This commit is contained in:
“xHuPo” 2025-05-22 12:06:34 +08:00
parent 53e59ddf89
commit 079542e431
9 changed files with 191 additions and 57 deletions

View file

@ -2,6 +2,7 @@ package database
import (
_ "embed"
"fmt"
"log"
_ "github.com/go-sql-driver/mysql"
@ -24,11 +25,11 @@ func InitDB() (*sqlx.DB, error) {
db, err := sqlx.Open(driver, dsn)
if err != nil {
return nil, err
return nil, fmt.Errorf("failed to connect to database: %w", err)
}
if err := db.Ping(); err != nil {
return nil, err
return nil, fmt.Errorf("failed to ping database: %w", err)
}
log.Println("Connected to database!")
@ -36,14 +37,12 @@ func InitDB() (*sqlx.DB, error) {
}
func MigrateDB(db *sqlx.DB) error {
_, err := db.Exec(userTable)
if err != nil {
return err
if _, err := db.Exec(userTable); err != nil {
return fmt.Errorf("failed to create user migration: %w", err)
}
_, err = db.Exec(otpTable)
if err != nil {
return err
if _, err := db.Exec(otpTable); err != nil {
return fmt.Errorf("failed to create otp migration: %w", err)
}
return nil
}

View file

@ -2,5 +2,6 @@ CREATE TABLE IF NOT EXISTS otp (
id SERIAL PRIMARY KEY,
openid VARCHAR(255),
num INTEGER,
token VARCHAR(255)
token VARCHAR(255),
createtime TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);