alpha
This commit is contained in:
parent
53e59ddf89
commit
079542e431
9 changed files with 191 additions and 57 deletions
|
@ -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
|
||||
}
|
||||
|
|
|
@ -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
|
||||
);
|
Loading…
Add table
Add a link
Reference in a new issue