+7
-15
@@ -2,30 +2,22 @@ package db
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"honoka-chan/pkg/utils"
|
||||
"os"
|
||||
|
||||
_ "modernc.org/sqlite"
|
||||
"xorm.io/xorm"
|
||||
)
|
||||
|
||||
var (
|
||||
DB *Instance
|
||||
Ldb *LdbInstance
|
||||
|
||||
ExampleDb = "assets/data.example.db"
|
||||
MainDb = "assets/main.db"
|
||||
UserDb = "assets/data.db"
|
||||
MainEng *xorm.Engine
|
||||
UserEng *xorm.Engine
|
||||
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))
|
||||
}
|
||||
Ldb = GetLdbInstance()
|
||||
|
||||
eng, err := xorm.NewEngine("sqlite", MainDb)
|
||||
if err != nil {
|
||||
@@ -53,7 +45,7 @@ func init() {
|
||||
}
|
||||
|
||||
func MatchTokenUid(token, uid string) bool {
|
||||
res, err := DB.Get([]byte(uid))
|
||||
res, err := Ldb.Get([]byte(uid))
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return false
|
||||
|
||||
Reference in New Issue
Block a user