12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 |
- package data
- import (
- "database/sql"
- log "github.com/sirupsen/logrus"
- )
- func createAccessTokenTable(db *sql.DB) {
- createTableSQL := `CREATE TABLE IF NOT EXISTS user_access_tokens (
- "token" TEXT NOT NULL PRIMARY KEY,
- "user_id" TEXT NOT NULL,
- "timestamp" DATE DEFAULT CURRENT_TIMESTAMP NOT NULL,
- FOREIGN KEY(user_id) REFERENCES users(id)
- );`
- stmt, err := db.Prepare(createTableSQL)
- if err != nil {
- log.Fatal(err)
- }
- defer stmt.Close()
- _, err = stmt.Exec()
- if err != nil {
- log.Warnln(err)
- }
- }
- func createUsersTable(db *sql.DB) {
- log.Traceln("Creating users table...")
- createTableSQL := `CREATE TABLE IF NOT EXISTS users (
- "id" TEXT,
- "display_name" TEXT NOT NULL,
- "display_color" NUMBER NOT NULL,
- "created_at" TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
- "disabled_at" TIMESTAMP,
- "previous_names" TEXT DEFAULT '',
- "namechanged_at" TIMESTAMP,
- "authenticated_at" TIMESTAMP,
- "scopes" TEXT,
- "type" TEXT DEFAULT 'STANDARD',
- "last_used" DATETIME DEFAULT CURRENT_TIMESTAMP,
- PRIMARY KEY (id)
- );`
- MustExec(createTableSQL, db)
- MustExec(`CREATE INDEX IF NOT EXISTS idx_user_id ON users (id);`, db)
- MustExec(`CREATE INDEX IF NOT EXISTS idx_user_id_disabled ON users (id, disabled_at);`, db)
- MustExec(`CREATE INDEX IF NOT EXISTS idx_user_disabled_at ON users (disabled_at);`, db)
- }
- // GetUsersCount will return the number of users in the database.
- func GetUsersCount() int64 {
- query := `SELECT COUNT(*) FROM users`
- rows, err := _db.Query(query)
- if err != nil || rows.Err() != nil {
- return 0
- }
- defer rows.Close()
- var count int64
- for rows.Next() {
- if err := rows.Scan(&count); err != nil {
- return 0
- }
- }
- return count
- }
|