You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.3 KiB
57 lines
1.3 KiB
package data |
|
|
|
import ( |
|
"database/sql" |
|
|
|
log "github.com/sirupsen/logrus" |
|
) |
|
|
|
func createUsersTable(db *sql.DB) { |
|
log.Traceln("Creating users table...") |
|
|
|
createTableSQL := `CREATE TABLE IF NOT EXISTS users ( |
|
"id" TEXT, |
|
"access_token" string NOT NULL, |
|
"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, |
|
"scopes" TEXT, |
|
"type" TEXT DEFAULT 'STANDARD', |
|
"last_used" DATETIME DEFAULT CURRENT_TIMESTAMP, |
|
PRIMARY KEY (id) |
|
);CREATE INDEX index ON users (id, access_token, disabled_at); |
|
CREATE INDEX id ON users (id); |
|
CREATE INDEX id_disabled ON users (id, disabled_at); |
|
CREATE INDEX access_token ON users (access_token); |
|
CREATE INDEX disabled_at ON USERS (disabled_at);` |
|
|
|
stmt, err := db.Prepare(createTableSQL) |
|
if err != nil { |
|
log.Fatal(err) |
|
} |
|
defer stmt.Close() |
|
_, err = stmt.Exec() |
|
if err != nil { |
|
log.Warnln(err) |
|
} |
|
} |
|
|
|
// 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 |
|
}
|
|
|