update curd & code logic

pkg/opencatd/db/db.go

@@ -1,6 +1,9 @@
 package db
 
 import (
+	"log"
+	"os"
+
 	"gorm.io/driver/sqlite"
 	"gorm.io/gorm"
 )
@@ -8,6 +11,12 @@ import (
 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 {
@@ -15,7 +24,7 @@ func init() {
 	}
 
 	// 自动迁移 User 结构体
-	err = db.AutoMigrate(&User{})
+	err = db.AutoMigrate(&User{}, &Key{})
 	if err != nil {
 		panic(err)
 	}