java如何使用redis加鎖

更新時(shí)間：2023年01月09日 16:54:26 作者：KLOCIC

這篇文章主要介紹了java如何使用redis加鎖問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

java使用redis加鎖

編寫(xiě)LockUtil工具類

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.connection.RedisStringCommands.SetOption;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.types.Expiration;
import org.springframework.stereotype.Service;
?
/**
?* LockUtil <br>
?*
?*/
@Service
public class LockUtil {
?
? ? @Autowired
? ? private RedisTemplate redisTemplate;
?
? ? @Autowired
? ? private StringRedisTemplate stringRedisTemplate;
?
? ? /**
? ? ?* @param lockKey 上鎖的key
? ? ?* @param lockSeconds 上鎖的秒數(shù)
? ? ?* @return
? ? ?*/
? ? public boolean lock(String lockKey, int lockSeconds) {
? ? ? ? return (Boolean) redisTemplate.execute((RedisCallback) connection -> {
? ? ? ? ? ? byte[] key = lockKey.getBytes();
? ? ? ? ? ? Boolean set = connection.set(key, key, Expiration.seconds(lockSeconds), SetOption.SET_IF_ABSENT);
? ? ? ? ? ? if (set == null) {
? ? ? ? ? ? ? ? return false;
? ? ? ? ? ? }
? ? ? ? ? ? return set;
? ? ? ? });
? ? }
?
? ? public boolean isLock(String lockKey) {
? ? ? ? return stringRedisTemplate.opsForValue().get(lockKey)!=null;
? ? }
?
?
? ? public boolean clearLock(String lockKey){
? ? ? ?return redisTemplate.delete(lockKey);
? ? }
}

使用鎖

public abstract class AbstractTask {
?
? ? @Autowired
? ? private LockUtil lockUtil;
?
? ? /**
? ? ?* 獲取redis鎖的key
? ? ?*
? ? ?* @return
? ? ?*/
? ? protected abstract String getLockKey();
?
? ? protected boolean lock() {
? ? ? ? return lockUtil.lock(getLockKey(), 120);
? ? }
?
? ? protected boolean lockManual() {
? ? ? ? return lockUtil.lock(getLockKey(), 299);
? ? }
?
? ? protected boolean clearLock() {
? ? ? ? return lockUtil.clearLock(getLockKey());
? ? }
}

@Component
@Slf4j
@RefreshScope
public class FileCapacityCountTask extends AbstractTask{
? ? @Autowired
? ? private FileCapacityCountService fileCapacityCountService;
? ?
?
? ? @Scheduled(cron = "${batch.verification.schedule.capacity}")
? ? public void task(){
? ? ? ? if (!lock()) {
? ? ? ? ? ? log.info("本實(shí)例無(wú)需執(zhí)行定時(shí)任務(wù)");
? ? ? ? ? ? return;
? ? ? ? }
? ? ? ? fileCapacityCountService.fileCapacityCountTask();
? ? }
?
? ? @Override
? ? protected String getLockKey() {
? ? ? ? String today = DateUtil.formatDate(new Date());
? ? ? ? return FileCapacityCountTask.class.getSimpleName() + CommonConstant.APPLICATION_NAME + today;
? ? }
}

redis鎖用法java代碼

由于redis是串行的，所以可以用redis實(shí)現(xiàn)鎖機(jī)制。

下方是java代碼

@Component
@Slf4j
public class RedisSingleLock {
    private final StringRedisTemplate redis;

    public SimpleDistributedLock(StringRedisTemplate redis) {
        this.redis = redis;
    }

    //這個(gè)方法，可以傳入key加鎖；多線程調(diào)用時(shí)，只有1個(gè)能獲取鎖成功，其它線程則會(huì)進(jìn)入循環(huán)，不停嘗試獲取鎖
    public void lock(String key) {
        do {
            Boolean lockSuccess = redis.opsForValue().setIfAbsent(key, "1", 1, TimeUnit.DAYS);
            if (lockSuccess == null) {
                throw new IllegalStateException();
            }
            if (!lockSuccess) {
                try {
                    TimeUnit.MILLISECONDS.sleep(100);
                } catch (InterruptedException e) {
                    log.error(e.getMessage(), e);
                }
            } else {
                break;
            }
        } while (true);
    }

    //這個(gè)方法，傳入key釋放鎖，當(dāng)持有鎖的線程執(zhí)行業(yè)務(wù)代碼完畢后調(diào)用，釋放這個(gè)鎖；上方某一個(gè)在lock方法中循環(huán)嘗試獲得鎖的線程可以獲得鎖，另外的線程則繼續(xù)循環(huán)等待
    public void releaseLock(String key) {
        redis.delete(key);
    }
	
	//這個(gè)方法只嘗試獲取一次鎖，返回獲取結(jié)果
    public boolean tryLock(String key) {
        Boolean lockSuccess = redis.opsForValue().setIfAbsent(key, "1", 1, TimeUnit.DAYS);
        if (lockSuccess == null) {
            throw new IllegalStateException();
        }

        return lockSuccess;
    }
}