update curd & code logic

pkg/opencatd/db/keydb.go

@@ -1,18 +1,38 @@
 package db
 
-import "gorm.io/gorm"
+import "time"
 
 type Key struct {
-	gorm.Model
-	ApiKey string
-	UserId string
+	ID        uint      `gorm:"primarykey" json:"id,omitempty"`
+	Key       string    `gorm:"unique;not null" json:"key,omitempty"`
+	Name      string    `gorm:"unique;not null" json:"name,omitempty"`
+	UserId    string    `json:"-,omitempty"`
+	CreatedAt time.Time `json:"createdAt,omitempty"`
+	UpdatedAt time.Time `json:"updatedAt,omitempty"`
+}
+
+func GetKeyrByName(name string) (*Key, error) {
+	var key Key
+	result := db.First(&key, "name = ?", name)
+	if result.Error != nil {
+		return nil, result.Error
+	}
+	return &key, nil
+}
+
+func GetAllKeys() ([]Key, error) {
+	var keys []Key
+	if err := db.Find(&keys).Error; err != nil {
+		return nil, err
+	}
+	return keys, nil
 }
 
 // 添加记录
-func AddKey(apiKey string, userId string) error {
+func AddKey(apikey, name string) error {
 	key := Key{
-		ApiKey: apiKey,
-		UserId: userId,
+		Key:  apikey,
+		Name: name,
 	}
 	if err := db.Create(&key).Error; err != nil {
 		return err
@@ -29,9 +49,9 @@ func DeleteKey(id uint) error {
 }
 
 // 更新记录
-func UpdateKey(id uint, apiKey string, userId string) error {
+func UpdateKey(id uint, apikey string, userId string) error {
 	key := Key{
-		ApiKey: apiKey,
+		Key:    apikey,
 		UserId: userId,
 	}
 	if err := db.Model(&Key{}).Where("id = ?", id).Updates(key).Error; err != nil {