package lock

import (
	"context"

	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/errors/gerror"
)

// TryOptimisticLock 乐观锁通用方法（基于 version 字段）
// 参数说明：
// - ctx: 请求上下文
// - model: 对应表的 gdb.Model，如 dao.User.Ctx(ctx)
// - idKey: 主键名，通常是 "id"
// - idVal: 主键值
// - clientVersion: 前端传来的 version 值
// 返回：新的 version，如果冲突返回错误
func TryOptimisticLock(ctx context.Context, model *gdb.Model, idKey string, idVal any, clientVersion int) (newVersion int, err error) {
	if idVal == 0 || clientVersion == 0 {
		return 0, gerror.New("缺少 ID 或版本号参数")
	}

	// 尝试根据 version 进行原子更新
	result, err := model.
		Where(idKey, idVal).
		Where("version", clientVersion).
		Increment("version", 1)

	if err != nil {
		return 0, err
	}

	// 检查是否真的更新成功
	rows, _ := result.RowsAffected()
	if rows == 0 {
		return 0, gerror.New("数据版本冲突，请刷新后重试")
	}

	return clientVersion + 1, nil
}