Java封装Redis非阻塞分布式锁彻底解决表单重复提交主键冲突问题

@Slf4j
@Component
@SuppressWarnings(value = {"unchecked", "rawtypes"})
public class RedisService {
    @Autowired
    public RedisTemplate redisTemplate;
    /**
     * 非阻塞获取分布式锁（原子操作）
     * @param lockKey 锁键
     * @param requestId 唯一标识，防止误删锁
     * @param expireTime 过期时间(秒)
     * @return 获取结果
     */
    public boolean tryLock(String lockKey, String requestId, long expireTime) {
        try {
            Boolean result = redisTemplate.opsForValue().setIfAbsent(
                    lockKey,
                    requestId,
                    expireTime,
                    TimeUnit.SECONDS
            );
            return Boolean.TRUE.equals(result);
        } catch (Exception e) {
            log.error("获取分布式锁失败, lockKey:{}", lockKey, e);
            return false;
        }
    }
    /**
     * 安全释放锁（Lua脚本保证原子性）
     * @param lockKey 锁键
     * @param requestId 唯一标识
     * @return 释放结果
     */
    public boolean releaseLock(String lockKey, String requestId) {
        String luaScript = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
        try {
            DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
            redisScript.setScriptText(luaScript);
            redisScript.setResultType(Long.class);
            Long result = (Long) redisTemplate.execute(
                    redisScript,
                    Collections.singletonList(lockKey),
                    requestId
            );
            return result != null && result > 0;
        } catch (Exception e) {
            log.error("释放分布式锁失败, lockKey:{}", lockKey, e);
            return false;
        }
    }
    // 省略其他缓存方法...
}

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import java.util.UUID;
import java.util.function.Supplier;
/**
 * 通用Redis非阻塞分布式锁工具类
 * 适配：表单防重、接口防重复提交、并发插入控制
 */
@Slf4j
@Component
public class RedisLockUtil {
    @Autowired
    private RedisService redisService;
    /**
     * 非阻塞锁执行通用方法
     * @param lockKey 锁唯一键（建议：lock:业务模块:业务单号）
     * @param expireSeconds 锁过期时间（秒，兜底防死锁）
     * @param businessLogic 业务逻辑代码块
     * @param <T> 返回值泛型
     * @return 业务执行结果
     */
    public <T> T executeWithLock(String lockKey, int expireSeconds, Supplier<T> businessLogic) {
        // 生成唯一请求ID，防止误删其他线程的锁
        String requestId = UUID.randomUUID().toString();
        // 1. 非阻塞获取锁
        boolean isLocked = redisService.tryLock(lockKey, requestId, expireSeconds);
        if (!isLocked) {
            log.warn("[分布式锁] 获取失败，锁键:{}，操作进行中", lockKey);
            throw new RuntimeException("操作进行中，请勿重复提交！");
        }
        try {
            // 2. 执行核心业务逻辑
            return businessLogic.get();
        } catch (Exception e) {
            log.error("[分布式锁] 业务执行失败，锁键:{}，允许重试", lockKey, e);
            throw new RuntimeException("提交失败：" + e.getMessage() + "，可重试！");
        } finally {
            // 3. 无论成功/失败，必须释放锁
            boolean isReleased = redisService.releaseLock(lockKey, requestId);
            if (!isReleased) {
                log.warn("[分布式锁] 释放失败，锁键:{}，requestId:{}（可能已过期自动释放）", lockKey, requestId);
            }
        }
    }
}

@Service
@Slf4j
public class FormSubmitService {
    @Autowired
    private RedisLockUtil redisLockUtil;
    // 业务Mapper，隐藏具体表信息
    @Autowired
    private BizFormMapper bizFormMapper;
    /**
     * 表单提交核心方法
     * @param djbh 业务单号（唯一标识）
     * @param formDTO 表单参数
     * @return 接口返回结果
     */
    public Result submitForm(String djbh, BizFormDTO formDTO) {
        try {
            // 构造锁键：隐藏具体业务模块，规范命名
            String lockKey = "lock:biz_form_submit:" + djbh;
            // 调用锁工具类，60秒过期兜底，执行业务逻辑
            return redisLockUtil.executeWithLock(lockKey, 60, () -> {
                // ========== 核心业务逻辑 ==========
                // 1. 查询是否已存在（成功后禁止重复提交）
                Object existData = bizFormMapper.selectByDjbh(djbh);
                if (existData != null) {
                    return Result.fail("该单据已提交，请勿重复操作！");
                }
                // 2. 执行插入操作
                bizFormMapper.insertForm(formDTO);
                return Result.success("提交成功！");
                // ================================
            });
        } catch (RuntimeException e) {
            // 捕获锁异常、业务异常，统一返回
            return Result.fail(e.getMessage());
        }
    }
    // 省略Result成功/失败封装方法...
}