gorm 使用更新数据优化建议

jetsung

gorm 使用更新数据优化建议

// ShortenUpdate 更新短链接
func (t *ShortenLogic) ShortenUpdate(code string, originalURL string, describe string) (int, types.ResShorten) {
    // 初始化返回结构，避免零值问题
    url := types.ResShorten{}
    
    // 查询现有记录
    var existingURL model.Urls
    if err := t.db.Where("short_code = ?", code).First(&existingURL).Error; err != nil {
        if errors.Is(err, gorm.ErrRecordNotFound) {
            return ecodes.ErrCodeNotFound, url
        }
        return ecodes.ErrCodeDatabaseError, url
    }

    // 准备更新字段
    updates := make(map[string]interface{})
    updates["updated_at"] = time.Now().Unix()
    
    if originalURL != "" {
        updates["original_url"] = originalURL
    }
    if describe != "" {
        updates["describe"] = describe
    }

    // 使用 Update 而不是 Save，只更新必要字段
    if err := t.db.Model(&existingURL).Where("short_code = ?", code).Updates(updates).Error; err != nil {
        return ecodes.ErrCodeDatabaseError, url
    }

    // 重新查询获取完整记录（可选，如果担心缓存问题）
    // if err := t.db.Where("short_code = ?", code).First(&existingURL).Error; err != nil {
    //     return ecodes.ErrCodeDatabaseError, url
    // }

    // 构建返回结果
    url = types.ResShorten{
        ID:          existingURL.ID,
        Code:        existingURL.ShortCode,
        ShortURL:    t.GetSiteURL(existingURL.ShortCode),
        OriginalURL: existingURL.OriginalURL,
        Describe:    existingURL.Describe,
        Status:      existingURL.Status,
        CreatedTime: t.GetTimeFormat(existingURL.CreatedAt),
        UpdatedTime: t.GetTimeFormat(existingURL.UpdatedAt),
    }

    return ecodes.ErrCodeSuccess, url
}

优化后的版本：

添加输入参数验证
添加事务支持
添加日志记录
考虑并发更新问题（乐观锁）

// ShortenUpdate 更新短链接
func (t *ShortenLogic) ShortenUpdate(code string, originalURL string, describe string) (int, types.ResShorten) {
    // 1. 输入参数验证
    if code == "" {
        t.logger.Warn("short code is empty")
        return ecodes.ErrCodeInvalidParam, types.ResShorten{}
    }
    
    if originalURL != "" {
        if _, err := url.ParseRequestURI(originalURL); err != nil {
            t.logger.Warn("invalid original URL", zap.String("url", originalURL), zap.Error(err))
            return ecodes.ErrCodeInvalidParam, types.ResShorten{}
        }
    }

    // 初始化返回结构
    result := types.ResShorten{}
    
    // 2. 开启事务
    tx := t.db.Begin()
    if tx.Error != nil {
        t.logger.Error("failed to begin transaction", zap.Error(tx.Error))
        return ecodes.ErrCodeDatabaseError, result
    }
    
    // 确保在函数退出时处理事务
    defer func() {
        if r := recover(); r != nil {
            tx.Rollback()
            t.logger.Error("panic occurred in ShortenUpdate", zap.Any("recover", r))
        }
    }()

    // 3. 查询现有记录（带乐观锁）
    var existingURL model.Urls
    if err := tx.Set("gorm:query_option", "FOR UPDATE").Where("short_code = ?", code).First(&existingURL).Error; err != nil {
        tx.Rollback()
        if errors.Is(err, gorm.ErrRecordNotFound) {
            t.logger.Info("short code not found", zap.String("code", code))
            return ecodes.ErrCodeNotFound, result
        }
        t.logger.Error("failed to query URL", zap.String("code", code), zap.Error(err))
        return ecodes.ErrCodeDatabaseError, result
    }

    t.logger.Info("updating short URL", 
        zap.String("code", code),
        zap.String("originalURL", originalURL),
        zap.String("describe", describe))

    // 4. 准备更新字段
    updates := make(map[string]interface{})
    updates["updated_at"] = time.Now().Unix()
    
    if originalURL != "" {
        updates["original_url"] = originalURL
    }
    if describe != "" {
        updates["describe"] = describe
    }

    // 5. 执行更新（带版本检查）
    updates["version"] = existingURL.Version + 1
    db := tx.Model(&existingURL).
        Where("short_code = ? AND version = ?", code, existingURL.Version).
        Updates(updates)

    if db.Error != nil {
        tx.Rollback()
        t.logger.Error("failed to update URL", zap.String("code", code), zap.Error(db.Error))
        return ecodes.ErrCodeDatabaseError, result
    }

    if db.RowsAffected == 0 {
        tx.Rollback()
        t.logger.Warn("optimistic lock failed, record may have been modified by others", zap.String("code", code))
        return ecodes.ErrCodeConcurrentUpdate, result
    }

    // 6. 提交事务
    if err := tx.Commit().Error; err != nil {
        t.logger.Error("failed to commit transaction", zap.Error(err))
        return ecodes.ErrCodeDatabaseError, result
    }

    // 7. 重新查询获取最新数据（可选）
    if err := t.db.Where("short_code = ?", code).First(&existingURL).Error; err != nil {
        t.logger.Error("failed to query updated URL", zap.String("code", code), zap.Error(err))
        return ecodes.ErrCodeDatabaseError, result
    }

    // 8. 构建返回结果
    result = types.ResShorten{
        ID:          existingURL.ID,
        Code:        existingURL.ShortCode,
        ShortURL:    t.GetSiteURL(existingURL.ShortCode),
        OriginalURL: existingURL.OriginalURL,
        Describe:    existingURL.Describe,
        Status:      existingURL.Status,
        CreatedTime: t.GetTimeFormat(existingURL.CreatedAt),
        UpdatedTime: t.GetTimeFormat(existingURL.UpdatedAt),
        Version:     existingURL.Version, // 返回版本号给客户端
    }

    t.logger.Info("successfully updated short URL", zap.String("code", code))
    return ecodes.ErrCodeSuccess, result
}