分布式锁与实现(一)——基于Redis实现

　　目前几乎很多大型网站及应用都是分布式部署的，分布式场景中的数据一致性问题一直是一个比较重要的话题。分布式CAP理论告诉我们“任何一个分布式系统都无法同时满足一致性（Consistency）、可用性（Availability）和分区容错性（Partition  tolerance），最多只能同时满足两项。”所以，很多系统在设计之初就要对这三者做出取舍。在互联网领域的绝大多数的场景中，都需要牺牲强一致性来换取系统的高可用性，系统往往只需要保证“最终一致性”，只要这个最终时间是在用户可以接受的范围内即可。

　　在很多场景中，我们为了保证数据的最终一致性，需要很多的技术方案来支持，比如分布式事物、分布式锁、分布式任务调度等。

为什么选用Redis实现分布式锁的原因

• Redis有很高的性能
• Redis命令对此支持较好，实现起来比较方便

在此将不介绍Redis安装，具体在Linux和Windows中的安装可以查看我的博客。

附上链接

主要使用的命令介绍

SETNX

SETNX key val
当且仅当key不存在时，set一个key为val的字符串，返回1；若key存在，则什么都不做，返回0。

expire

expire key timeout
为key设置一个超时时间，单位为second，超过这个时间锁会自动释放，避免死锁。

delete

delete key
删除key

在使用Redis实现分布式锁的时候，主要就会使用到这三个命令

其他命令请参照官网API

附上链接

实现

使用Jedis 来连接操作Redis

引入maven

        <dependency>
            <groupId>redis.clients</groupId>
            <artifactId>jedis</artifactId>
            <type>jar</type>
            <scope>compile</scope>
        </dependency>

实现思想/思路

• 获取锁的时候，使用setnx加锁，并使用expire命令为锁添加一个超时时间，超过该时间则自动释放锁，锁的value值为一个随机生成的UUID，通过此在释放锁的时候进行判断。
• 获取锁的时候还设置一个获取的超时时间，若超过这个时间则放弃获取锁。
• 释放锁的时候，通过UUID判断是不是该锁，若是该锁，则执行delete进行锁释放。

分布式锁的核心代码如下：

package com.example.RedisTest;

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;
import redis.clients.jedis.Transaction;
import redis.clients.jedis.exceptions.JedisException;

import java.util.List;
import java.util.UUID;

/**
 * @ClassName RedisUtil
 * @Description TODO
 * @Author 放牛男孩
 * @Date 2019/4/17 21:31
 * @Version 1.0
 **/
public class JedisUtil {
    private static final JedisUtil jedisUtil = new JedisUtil();
    private static JedisPool jedisPool = null;

    static {
        init();
    }

    public static void init() {
        JedisPoolConfig config = new JedisPoolConfig();
        // 设置最大连接数
        config.setMaxTotal(200);
        // 设置最大空闲数
        config.setMaxIdle(8);
        // 设置最大等待时间(毫秒)
        config.setMaxWaitMillis(1000 * 100);
        // 在borrow一个jedis实例时，是否需要验证，若为true，则所有jedis实例均是可用的
        config.setTestOnBorrow(true);
        jedisPool = new JedisPool(config, "127.0.0.1", 6379, 3000, "123456");
    }

    /**
     * 获取JedisUtil实例
     *
     * @return JedisUtil
     */
    public static JedisUtil getInstance() {
        return jedisUtil;
    }

    /**
     * 从jedis连接池中获取获取jedis对象
     *
     * @return jedisPool
     */
    public static Jedis getJedis() {
        return jedisPool.getResource();
    }

    /**
     * 加锁
     *
     * @param locaName       锁的key
     * @param acquireTimeout 获取超时时间
     * @param timeout        锁的超时时间
     * @return 锁标识
     */
    public String lockWithTimeout(String locaName, long acquireTimeout, long timeout) {
        Jedis jedis = null;
        String retIdentifier = null;
        try {
            // 获取连接
            jedis = getJedis();
            // 随机生成一个value
            String identifier = UUID.randomUUID().toString();
            // 锁名，即key值
            String lockKey = "lock:" + locaName;
            // 超时时间，上锁后超过此时间则自动释放锁
            int lockExpire = (int) (timeout / 1000);

            // 获取锁的超时时间，超过这个时间则放弃获取锁
            long end = System.currentTimeMillis() + acquireTimeout;
            while (System.currentTimeMillis() < end) {
                if (jedis.setnx(lockKey, identifier) == 1) {
                    jedis.expire(lockKey, lockExpire);
                    // 返回value值，用于释放锁时间确认
                    retIdentifier = identifier;
                    return retIdentifier;
                }
                // 返回-1代表key没有设置超时时间，为key设置一个超时时间
                if (jedis.ttl(lockKey) == -1) {
                    jedis.expire(lockKey, lockExpire);
                }

                try {
                    Thread.sleep(10);
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                }
            }
        } catch (JedisException e) {
            e.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (jedis != null) {
                jedis.close();
            }
        }
        return retIdentifier;
    }

    /**
     * 释放锁
     *
     * @param lockName   锁的key
     * @param identifier 释放锁的标识
     * @return
     */
    public static boolean releaseLock(String lockName, String identifier) {
        Jedis jedis = null;
        String lockKey = "lock:" + lockName;
        boolean retFlag = false;
        try {
            jedis = getJedis();
            jedis = jedisPool.getResource();
            while (true) {
                // 监视lock，准备开始事务
                jedis.watch(lockKey);
                // 通过前面返回的value值判断是不是该锁，若是该锁，则删除，释放锁
                if (identifier.equals(jedis.get(lockKey))) {
                    Transaction transaction = jedis.multi();
                    transaction.del(lockKey);
                    List<Object> results = transaction.exec();
                    if (results == null) {
                        continue;
                    }
                    retFlag = true;
                }
                jedis.unwatch();
                break;
            }
        } catch (JedisException e) {
            e.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (jedis != null) {
                jedis.close();
            }
        }
        return retFlag;
    }


    //秒杀业务
    int n = 500;

    public void seckill() {
//        返回锁的value值，供释放锁时候进行判断
        String indentifier = lockWithTimeout("resource", 5000, 1000);
        if(indentifier!=null){
            System.out.println(Thread.currentThread().getName() + "获得了锁");
            System.out.println(--n);
            releaseLock("resource", indentifier);
        }
    }
}

测试

下面就用一个简单的例子测试刚才实现的分布式锁。
例子中使用50个线程模拟秒杀一个商品，使用--运算符来实现商品减少，从结果有序性就可以看出是否为加锁状态。

模拟秒杀服务，在其中配置了jedis线程池，在初始化的时候传给分布式锁，供其使用。

public class ThreadTest extends Thread {
    @Override
    public void run() {
        JedisUtil.getInstance().seckill();
    }
}

// 模拟线程进行秒杀服务

public class Test {
    public static void main(String[] args) {
        for (int i = 0; i < 50; i++) {
            new ThreadTest().start();
        }
    }
}

执行结果-有序顺序

Thread-37获得了锁
499
Thread-26获得了锁
498
Thread-42获得了锁
497
Thread-0获得了锁
496
Thread-20获得了锁
495
Thread-46获得了锁
494
Thread-30获得了锁
493
Thread-14获得了锁
492
Thread-36获得了锁
491
Thread-41获得了锁
490
Thread-16获得了锁
489
... ...
... ...

试试不加锁的效果，注释掉使用锁的部分

    public void seckill() {
//        返回锁的value值，供释放锁时候进行判断
//        String indentifier = lockWithTimeout("resource", 5000, 1000);
//        if(indentifier!=null){
            System.out.println(Thread.currentThread().getName() + "获得了锁");
            System.out.println(--n);
//            releaseLock("resource", indentifier);
//        }
    }

从结果得出，有一些是异步进行的。

Thread-0获得了锁
Thread-2获得了锁
498
499
Thread-1获得了锁
497
Thread-4获得了锁
496
Thread-6获得了锁
495
Thread-5获得了锁
494
Thread-13获得了锁
493
Thread-16获得了锁
492
Thread-21获得了锁
491
Thread-12获得了锁
490
Thread-8获得了锁
489
... ...
... ...

在分布式环境中，对资源进行上锁有时候是很重要的，比如抢购某一资源，这时候使用分布式锁就可以很好地控制资源。
当然，在具体业务使用中，还需要考虑很多因素，比如超时时间的选取，获取锁时间的选取对并发量都有很大的影响。

上述实现的分布式锁也只是一种简单的实现，主要是一种思想。