优化

internal/data/parameter.go

@@ -28,31 +28,34 @@ type parameterPO struct {
 func (parameterPO) TableName() string { return "sys_params" }
 
 func (r *parameterRepo) CreateParameter(ctx context.Context, v *biz.SystemParameter) error {
-	po := parameterPO{Name: v.Name, Key: v.Key, Value: v.Value, Desc: v.Desc}
+	po := parameterPO{ID: v.ID, CreatedAt: v.CreatedAt, UpdatedAt: v.UpdatedAt, Name: v.Name, Key: v.Key, Value: v.Value, Desc: v.Desc}
 	if err := r.data.gormDB.WithContext(ctx).Create(&po).Error; err != nil {
 		return err
 	}
-	v.ID = po.ID
+	v.ID, v.CreatedAt, v.UpdatedAt = po.ID, po.CreatedAt, po.UpdatedAt
 	return nil
 }
 func (r *parameterRepo) UpdateParameter(ctx context.Context, v *biz.SystemParameter) error {
 	// The compatible update uses a struct, so zero-value optional fields (notably desc)
 	// are intentionally ignored rather than clearing an existing value.
-	return r.data.gormDB.WithContext(ctx).Model(&parameterPO{}).Where("id = ?", v.ID).Updates(&parameterPO{Name: v.Name, Key: v.Key, Value: v.Value, Desc: v.Desc}).Error
+	return r.data.gormDB.WithContext(ctx).Model(&parameterPO{}).Where("id = ?", v.ID).Updates(&parameterPO{ID: v.ID, CreatedAt: v.CreatedAt, UpdatedAt: v.UpdatedAt, Name: v.Name, Key: v.Key, Value: v.Value, Desc: v.Desc}).Error
 }
-func (r *parameterRepo) DeleteParameters(ctx context.Context, ids []uint) error {
-	return r.data.gormDB.WithContext(ctx).Delete(&parameterPO{}, ids).Error
-}
-func (r *parameterRepo) FindParameter(ctx context.Context, id uint, key string) (*biz.SystemParameter, error) {
-	var po parameterPO
-	db := r.data.gormDB.WithContext(ctx)
-	var err error
-	if id != 0 {
-		err = db.First(&po, id).Error
-	} else {
-		err = db.Where(map[string]any{"key": key}).First(&po).Error
+func (r *parameterRepo) DeleteParameters(ctx context.Context, ids []string) error {
+	if len(ids) == 1 {
+		return r.data.gormDB.WithContext(ctx).Delete(&parameterPO{}, "id = ?", ids[0]).Error
 	}
-	if err != nil {
+	return r.data.gormDB.WithContext(ctx).Delete(&[]parameterPO{}, "id in ?", ids).Error
+}
+func (r *parameterRepo) FindParameterByID(ctx context.Context, id string) (*biz.SystemParameter, error) {
+	var po parameterPO
+	if err := r.data.gormDB.WithContext(ctx).Where("id = ?", id).First(&po).Error; err != nil {
+		return nil, err
+	}
+	return parameterFromPO(po), nil
+}
+func (r *parameterRepo) FindParameterByKey(ctx context.Context, key string) (*biz.SystemParameter, error) {
+	var po parameterPO
+	if err := r.data.gormDB.WithContext(ctx).Where(parameterPO{Key: key}).First(&po).Error; err != nil {
 		return nil, err
 	}
 	return parameterFromPO(po), nil