first up

store/db.go

@@ -0,0 +1,33 @@
+package store
+
+import (
+	"log"
+	"os"
+
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+)
+
+var db *gorm.DB
+
+func init() {
+	if _, err := os.Stat("db"); os.IsNotExist(err) {
+		errDir := os.MkdirAll("db", 0755)
+		if errDir != nil {
+			log.Fatalln("Error creating directory:", err)
+		}
+	}
+	var err error
+	db, err = gorm.Open(sqlite.Open("./db/cat.db"), &gorm.Config{})
+	if err != nil {
+		panic("failed to connect database")
+	}
+
+	// 自动迁移 User 结构体
+	err = db.AutoMigrate(&User{}, &Key{})
+	if err != nil {
+		panic(err)
+	}
+	LoadKeysCache()
+	LoadAuthCache()
+}