java 防重复提交_厉害了,一文教你如何避免重复提交问题的8种解决方案!

package com.battcn.utils;

import org.springframework.boot.autoconfigure.AutoConfigureAfter;

import org.springframework.boot.autoconfigure.data.redis.RedisAutoConfiguration;

import org.springframework.context.annotation.Configuration;

import org.springframework.data.redis.connection.RedisStringCommands;

import org.springframework.data.redis.core.RedisCallback;

import org.springframework.data.redis.core.StringRedisTemplate;

import org.springframework.data.redis.core.types.Expiration;

import org.springframework.util.StringUtils;

import java.util.concurrent.Executors;

import java.util.concurrent.ScheduledExecutorService;

import java.util.concurrent.TimeUnit;

import java.util.regex.Pattern;

/**

* 需要定义成 Bean

*

* @author Levin

* @since 2018/6/15 0015

*/

@Configuration

@AutoConfigureAfter(RedisAutoConfiguration.class)

public class RedisLockHelper {

private static final String DELIMITER = "|";

/**

* 如果要求比较高可以通过注入的方式分配

*/

private static final ScheduledExecutorService EXECUTOR_SERVICE = Executors.newScheduledThreadPool(10);

private final StringRedisTemplate stringRedisTemplate;

public RedisLockHelper(StringRedisTemplate stringRedisTemplate) {

this.stringRedisTemplate = stringRedisTemplate;

}

/**

* 获取锁(存在死锁风险)

*

* @param lockKey lockKey

* @param value  value

* @param time    超时时间

* @param unit    过期单位

* @return true or false

*/

public boolean tryLock(final String lockKey, final String value, final long time, final TimeUnit unit) {

return stringRedisTemplate.execute((RedisCallback) connection -> connection.set(lockKey.getBytes(), value.getBytes(), Expiration.from(time, unit), RedisStringCommands.SetOption.SET_IF_ABSENT));

}

/**

* 获取锁

*

* @param lockKey lockKey

* @param uuid    UUID

* @param timeout 超时时间

* @param unit    过期单位

* @return true or false

*/

public boolean lock(String lockKey, final String uuid, long timeout, final TimeUnit unit) {

final long milliseconds = Expiration.from(timeout, unit).getExpirationTimeInMilliseconds();

boolean success = stringRedisTemplate.opsForValue().setIfAbsent(lockKey, (System.currentTimeMillis() + milliseconds) + DELIMITER + uuid);

if (success) {

stringRedisTemplate.expire(lockKey, timeout, TimeUnit.SECONDS);

} else {

String oldVal = stringRedisTemplate.opsForValue().getAndSet(lockKey, (System.currentTimeMillis() + milliseconds) + DELIMITER + uuid);

final String[] oldValues = oldVal.split(Pattern.quote(DELIMITER));

if (Long.parseLong(oldValues[0]) + 1 <= System.currentTimeMillis()) {

return true;

}

}

return success;

}

/**

* @see Redis Documentation: SET

*/

public void unlock(String lockKey, String value) {

unlock(lockKey, value, 0, TimeUnit.MILLISECONDS);

}

/**

* 延迟unlock

*

* @param lockKey  key

* @param uuid      client(最好是唯一键的)

* @param delayTime 延迟时间

* @param unit      时间单位

*/

public void unlock(final String lockKey, final String uuid, long delayTime, TimeUnit unit) {

if (StringUtils.isEmpty(lockKey)) {

return;

}

if (delayTime <= 0) {

doUnlock(lockKey, uuid);

} else {

EXECUTOR_SERVICE.schedule(() -> doUnlock(lockKey, uuid), delayTime, unit);

}

}

/**

* @param lockKey key

* @param uuid    client(最好是唯一键的)

*/

private void doUnlock(final String lockKey, final String uuid) {

String val = stringRedisTemplate.opsForValue().get(lockKey);

final String[] values = val.split(Pattern.quote(DELIMITER));

if (values.length <= 0) {

return;

}

if (uuid.equals(values[1])) {

stringRedisTemplate.delete(lockKey);

}

}

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值