分布式锁实现方式

方法一 基于数据实现分布式锁

select * from lock_table where xxx_code = 'xxx' for update;
对于该条数据加锁,可以实现分布式锁
加锁后,可以进行select,不可以进行再次加锁或者更改
释放时,运行 commit 即可  
在java项目中需要加事务,否则mapper查询后自动commit

优点
简单方便、易于理解、易于操作

缺点
并发量大时,对数据库压力较大

建议
作为锁的数据库与业务数据库分开

方法二 基于redis实现分布式锁

获取锁
set resource_name my_random_value NX PX 30000

resource_name:资源名称,可根据不同的业务区分不同的锁
my_random_value:随机值,每个线程的随机值都不同,用于释放锁时的校验
NX: key不存在时设置成功,key存在则设置不成功
PX: 自动失效时间,出现异常情况,锁可以过期失效

实现原理
利用NX的原子性,多个线程并发时,只有一个线程可以设置成功
设置成功即获得锁,可以执行后续的业务处理
如果出现异常,过了锁的有效期,锁自动释放

释放锁(需要校验 value 是否相等)
采用lua脚本
if redis.call("get",KEYS[1]) == ARGV[1] 
	then return redis.call("del",KEYS[1])
else
	return 0
end
Demo代码:
/**
 * redis分布式锁demo
 */
@RestController
public class RedisController {
    @Autowired
    private RedisTemplate redisTemplate;
    public String redisLock(){
        String key = "redisKey";
        String value = UUID.randomUUID().toString();
        RedisCallback<Boolean> redisCallback = redisConnection -> {
            // NX
            RedisStringCommands.SetOption setOption = RedisStringCommands.SetOption.ifAbsent();
            // 过期时间
            Expiration expiration = Expiration.seconds(30);
            byte[] redisKey = redisTemplate.getKeySerializer().serialize(key);
            byte[] redisValue = redisTemplate.getValueSerializer().serialize(value);
            // 设置锁
            Boolean result = redisConnection.set(redisKey, redisValue, expiration, setOption);
            return result;
        };
        // 获取锁
        Boolean result = (Boolean)redisTemplate.execute(redisCallback);
        if(result) System.out.println("获取成功");
        // 释放锁
        String script = "if redis.call(\"get\",KEYS[1]) == ARGV[1] \n" +
                "\tthen return redis.call(\"del\",KEYS[1])\n" +
                "else\n" +
                "\treturn 0\n" +
                "end";
        RedisScript<Boolean> redisScript = RedisScript.of(script,Boolean.class);
        List<String> keys = Arrays.asList(key);
        Boolean result1 = (Boolean)redisTemplate.execute(redisScript, keys, value);
        if(result1) System.out.println("释放锁成功");
        return "";
    }
}
Redis分布式锁抽象类:
/**
分布式锁抽象类
**/
package com.example.distributelock.lock;

import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.connection.RedisStringCommands;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.RedisScript;
import org.springframework.data.redis.core.types.Expiration;

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

@Slf4j
public class RedisLock implements AutoCloseable {

    private RedisTemplate redisTemplate;
    private String key;
    private String value;
    //单位:秒
    private int expireTime;

    public RedisLock(RedisTemplate redisTemplate,String key,int expireTime){
        this.redisTemplate = redisTemplate;
        this.key = key;
        this.expireTime=expireTime;
        this.value = UUID.randomUUID().toString();
    }

    /**
     * 获取分布式锁
     * @return
     */
    public boolean getLock(){
        RedisCallback<Boolean> redisCallback = connection -> {
            //设置NX
            RedisStringCommands.SetOption setOption = RedisStringCommands.SetOption.ifAbsent();
            //设置过期时间
            Expiration expiration = Expiration.seconds(expireTime);
            //序列化key
            byte[] redisKey = redisTemplate.getKeySerializer().serialize(key);
            //序列化value
            byte[] redisValue = redisTemplate.getValueSerializer().serialize(value);
            //执行setnx操作
            Boolean result = connection.set(redisKey, redisValue, expiration, setOption);
            return result;
        };

        //获取分布式锁
        Boolean lock = (Boolean)redisTemplate.execute(redisCallback);
        return lock;
    }

    public boolean unLock() {
        String script = "if redis.call(\"get\",KEYS[1]) == ARGV[1] then\n" +
                "    return redis.call(\"del\",KEYS[1])\n" +
                "else\n" +
                "    return 0\n" +
                "end";
        RedisScript<Boolean> redisScript = RedisScript.of(script,Boolean.class);
        List<String> keys = Arrays.asList(key);

        Boolean result = (Boolean)redisTemplate.execute(redisScript, keys, value);
        log.info("释放锁的结果:"+result);
        return result;
    }


    @Override
    public void close() throws Exception {
        unLock();
    }
}


{
    try(RedisLock redisLock = new RedisLock(XXX,XXX,XXX))
    这样写时,自动调用close()进行unlock
}

方法三 基于zk实现分布式锁

package com.springboot.test.controller;

import org.apache.zookeeper.*;
import org.apache.zookeeper.data.Stat;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Collections;
import java.util.List;

public class ZkLock implements AutoCloseable, Watcher {

    private ZooKeeper zooKeeper;

    private String znode;

    public ZkLock() throws IOException {
        this.zooKeeper = new ZooKeeper("localhost:2181",
                10000,this);
    }

    public Boolean getLock(String code){
        try {
            // 创建业务根节点
            Stat stat = zooKeeper.exists("/" + code, false);
            if(stat==null){
                zooKeeper.create("/"+code,code.getBytes(StandardCharsets.UTF_8),
                        ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
            }

            // 创建瞬时有序节点 /order/order_
            znode = zooKeeper.create("/" + code + "/" + code + "_", code.getBytes(StandardCharsets.UTF_8),
                    ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);
            // 获取业务节点下所有的子节点
            List<String> childrenNodes = zooKeeper.getChildren("/" + code, false);
            // 子节点排序
            Collections.sort(childrenNodes);
            String firstNode = childrenNodes.get(0);
            // 如果创建的节点是第一个子节点,则获得锁
            if(znode.endsWith(firstNode))return true;

            // 不是第一个子节点,则监听前一个节点
            String lastNode = firstNode;
            for(String node : childrenNodes){
                if(znode.endsWith(node)){
                    zooKeeper.exists("/"+code+"/"+lastNode,true);
                }else {
                    lastNode = node;
                }
            }

            synchronized (this){
                wait();
            }

            return true;

        }catch (Exception e){
            e.printStackTrace();
        }

        return false;
    }

    @Override
    public void close() throws Exception {
        zooKeeper.delete(znode,-1);
        zooKeeper.close();
    }

    @Override
    public void process(WatchedEvent watchedEvent) {
        if(watchedEvent.getType() == Event.EventType.NodeDeleted){
            synchronized (this){
                notify();
            }
        }
    }
}

方法四 基于curator实现分布式锁

 RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000, 3);
 CuratorFramework client = CuratorFrameworkFactory.newClient("localhost:2181", retryPolicy);
 client.start();
 InterProcessMutex lock = new InterProcessMutex(client, "/order");
 if (lock.acquire(30, TimeUnit.SECONDS)) {
   try {
      // do some work inside of the critical section here
   } finally {
      lock.release();
   }
 }
client.close();

方法五 基于redisson实现分布式锁

 Config config = new Config();
 config.useSingleServer().setAddress("redis://192.168.73.130:6379");
 RedissonClient redisson = Redisson.create(config);
 RLock rLock = redisson.getLock("order");
 try {
    rLock.lock(30, TimeUnit.SECONDS);
    log.info("我获得了锁!!!");
    Thread.sleep(10000);
  } catch (InterruptedException e) {
            e.printStackTrace();
   }finally {
     log.info("我释放了锁!!");
     rLock.unlock();
 }

{
	spring boot
	@Autowired
    private RedissonClient redisson;

    @RequestMapping("redissonLock")
    public String redissonLock() {
        RLock rLock = redisson.getLock("order");
        log.info("我进入了方法!!");
        try {
            rLock.lock(30, TimeUnit.SECONDS);
            log.info("我获得了锁!!!");
            Thread.sleep(10000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }finally {
            log.info("我释放了锁!!");
            rLock.unlock();
        }
        log.info("方法执行完成!!");
        return "方法执行完成!!";
      }
    }

对比

数据库       实现简单、易于理解   对数据库压力大
Redis       易于理解           自己实现、不支持阻塞
Zookeeper   支持阻塞           需理解Zookeeper、程序复杂
Curator     提供锁的方法        依赖Zookeeper,强一致
Redisson    提供锁的方法,可阻塞

不推荐自己编写的分布式锁
推荐Redisson和Curator实现的分布式锁
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值