update

pkg/opencatd/db/db.go

@@ -5,12 +5,6 @@ import (
 	"gorm.io/gorm"
 )
 
-type User struct {
-	gorm.Model
-	Name  string `gorm:"not null"`
-	Token string `gorm:"unique;not null"`
-}
-
 var db *gorm.DB
 
 func init() {
@@ -26,32 +20,3 @@ func init() {
 		panic(err)
 	}
 }
-
-// 添加用户
-func AddUser(name, token string) error {
-	user := &User{Name: name, Token: token}
-	result := db.Create(&user)
-	if result.Error != nil {
-		return result.Error
-	}
-	return nil
-}
-
-// 删除用户
-func DeleteUser(id uint) error {
-	result := db.Delete(&User{}, id)
-	if result.Error != nil {
-		return result.Error
-	}
-	return nil
-}
-
-// 修改用户
-func UpdateUser(id uint, name, token string) error {
-	user := &User{Name: name, Token: token}
-	result := db.Model(&User{}).Where("id = ?", id).Updates(user)
-	if result.Error != nil {
-		return result.Error
-	}
-	return nil
-}

pkg/opencatd/db/keydb.go

@@ -0,0 +1,41 @@
+package db
+
+import "gorm.io/gorm"
+
+type Key struct {
+	gorm.Model
+	ApiKey string
+	UserId string
+}
+
+// 添加记录
+func AddKey(apiKey string, userId string) error {
+	key := Key{
+		ApiKey: apiKey,
+		UserId: userId,
+	}
+	if err := db.Create(&key).Error; err != nil {
+		return err
+	}
+	return nil
+}
+
+// 删除记录
+func DeleteKey(id uint) error {
+	if err := db.Delete(&Key{}, id).Error; err != nil {
+		return err
+	}
+	return nil
+}
+
+// 更新记录
+func UpdateKey(id uint, apiKey string, userId string) error {
+	key := Key{
+		ApiKey: apiKey,
+		UserId: userId,
+	}
+	if err := db.Model(&Key{}).Where("id = ?", id).Updates(key).Error; err != nil {
+		return err
+	}
+	return nil
+}

pkg/opencatd/db/userdb.go

@@ -0,0 +1,38 @@
+package db
+
+import "gorm.io/gorm"
+
+type User struct {
+	gorm.Model
+	Name  string `gorm:"not null"`
+	Token string `gorm:"unique;not null"`
+}
+
+// 添加用户
+func AddUser(name, token string) error {
+	user := &User{Name: name, Token: token}
+	result := db.Create(&user)
+	if result.Error != nil {
+		return result.Error
+	}
+	return nil
+}
+
+// 删除用户
+func DeleteUser(id uint) error {
+	result := db.Delete(&User{}, id)
+	if result.Error != nil {
+		return result.Error
+	}
+	return nil
+}
+
+// 修改用户
+func UpdateUser(id uint, name, token string) error {
+	user := &User{Name: name, Token: token}
+	result := db.Model(&User{}).Where("id = ?", id).Updates(user)
+	if result.Error != nil {
+		return result.Error
+	}
+	return nil
+}