Redis锁相关

Redis锁相关

 

　　　　君不见，高堂明镜悲白发，朝如青丝暮成雪。

背景：面试的时候被问到有哪些锁，很快脱口而出Volatile、Synchronized和ReentrantLock，也能讲出他们之间的一些区别；当问到如在同一服务下同步锁可以起作用，但是在不同的服务器上部署同一个微服务工程，然后用nginx作代理，很明显，现在的线程锁不管用了。分布式情况下如何保证线程同步？当时就答不上来了，分布式环境下我们需要换一把锁，这把锁必须和两套甚至多套系统没有任何的耦合度。可以使用Redis锁实现分布式场景下的线程同步，使用Redies的API，如果key不存在，则设置一个key，这个key就是我们现在使用的一把锁。每个线程到此处，先设置锁，如果设置锁失败，则表明当前有线程获取到了锁，就返回。

一、单一服务器下的锁

例如将商品的数量存到Redis中。每个用户抢购前都需要到Redis中查询商品数量（代替mysql数据库。不考虑事务），如果商品数量大于0，则证明商品有库存；然后我们在进行库存扣减和接下来的操作；因为多线程并发问题，我们还需要在get()方法内部使用同步代码块，保证查询库存和减库存操作的原子性。

import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestHeader;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@AllArgsConstructor
@RequestMapping("/redis")
@Slf4j
public class TryRedisLock {

    private RedisTemplate<String, String> redisTemplate;

    @GetMapping(value = "/try/buy")
    public String get(@RequestHeader(required = false) String userId) {
        synchronized (this) {  // 单机同步
            String bird = redisTemplate.opsForValue().get("bird");
            Integer count = Integer.valueOf(bird);
            if (count > 0) {
                // 减库存
                redisTemplate.opsForValue().set("bird", String.valueOf(count - 1));
                log.info("用户{}, 抢到了, {} 号商品!", userId, bird);
            }
            return "零库存";
        }
    }
}

二、分布式场景redis锁

分布式场景下使用redis锁需要注意如下几个问题：

• 一台服务器宕机，导致无法释放锁；可以在try-catch的finally中释放锁或者给每一把锁加过期时间。
• 任务线程未执行完毕但锁已失效；可以延长锁的过期时间，使用定时器防止key过期。
• 使用Redisson框架简化代码；getLock()方法代替了Redis的setIfAbsent()，lock()设置过期时间，最终我们在交易结束后释放锁；延长锁的操作则由Redisson框架替我们完成，它会使用轮询去查看key是否过期，在交易没有完成时，自动重设Redis的key过期时间。

import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestHeader;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@AllArgsConstructor
@RequestMapping("/redis")
@Slf4j
public class TryRedisLock {

    private RedisTemplate<String, String> redisTemplate;


    @GetMapping(value = "/try/again/buy")
    public String getJudge(@RequestHeader(required = false) String userId) {
        // 每个线程到此处，先设置锁
        /**
         * 使用Redies的API如果key不存在，则设置一个key。这个key就是我们现在使用的一把锁
         * 每个线程到此处，先设置锁
         * 如果设置锁失败，则表明当前有线程获取到了锁，就返回。
         */
        Boolean birdLock = redisTemplate.opsForValue().setIfAbsent("birdLock", "");
        if (!birdLock) {
            return "";
        }
        try {
            String bird = redisTemplate.opsForValue().get("bird");
            Integer count = Integer.valueOf(bird);
            if (count > 0) {
                redisTemplate.opsForValue().set("bird", String.valueOf(count - 1));
                log.info("用户{}, 抢到了, {} 号商品!", userId, bird);
            }
        } finally {
            redisTemplate.delete("birdLock");  // 删除锁
        }
        return "";
    }

    @GetMapping(value = "/try/buy")
    public String get(@RequestHeader(required = false) String userId) {
        synchronized (this) {  // 单机同步
            String bird = redisTemplate.opsForValue().get("bird");
            Integer count = Integer.valueOf(bird);
            if (count > 0) {
                // 减库存
                redisTemplate.opsForValue().set("bird", String.valueOf(count - 1));
                log.info("用户{}, 抢到了, {} 号商品!", userId, bird);
            }
            return "零库存";
        }
    }
}

 

 

君不见

　　　　　　   高堂明镜悲白发

　　　　　　　　　　朝如青丝暮成雪