Reorganize directory structures
Signed-off-by: Sean Du <do4suki@gmail.com>
This commit is contained in:
@@ -0,0 +1,63 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"honoka-chan/pkg/utils"
|
||||
"os"
|
||||
|
||||
_ "modernc.org/sqlite"
|
||||
"xorm.io/xorm"
|
||||
)
|
||||
|
||||
var (
|
||||
DB *Instance
|
||||
|
||||
ExampleDb = "assets/data.example.db"
|
||||
MainDb = "assets/main.db"
|
||||
UserDb = "assets/data.db"
|
||||
MainEng *xorm.Engine
|
||||
UserEng *xorm.Engine
|
||||
)
|
||||
|
||||
func init() {
|
||||
DB = GetInstance()
|
||||
|
||||
_, err := os.Stat(UserDb)
|
||||
if err != nil {
|
||||
utils.WriteAllText(UserDb, utils.ReadAllText(ExampleDb))
|
||||
}
|
||||
|
||||
eng, err := xorm.NewEngine("sqlite", MainDb)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = eng.Ping()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
eng.SetMaxOpenConns(10)
|
||||
eng.SetMaxIdleConns(5)
|
||||
MainEng = eng
|
||||
|
||||
eng, err = xorm.NewEngine("sqlite", UserDb)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = eng.Ping()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
eng.SetMaxOpenConns(10)
|
||||
eng.SetMaxIdleConns(5)
|
||||
UserEng = eng
|
||||
}
|
||||
|
||||
func MatchTokenUid(token, uid string) bool {
|
||||
res, err := DB.Get([]byte(uid))
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return false
|
||||
}
|
||||
|
||||
return string(res) == token
|
||||
}
|
||||
Reference in New Issue
Block a user