在項(xiàng)目中經(jīng)常會(huì)遇到并發(fā)安全問(wèn)題，這時(shí)我們可以使用鎖來(lái)進(jìn)行線程同步。于是我們可以根據(jù)具體的情況使用synchronized 關(guān)鍵字來(lái)修飾方法或者代碼塊。也可以使用 java 5以后的 Lock 來(lái)實(shí)現(xiàn)，與 synchronized 關(guān)鍵字相比，Lock 的使用更靈活，可以有加鎖超時(shí)時(shí)間、公平性等優(yōu)勢(shì)。但是synchronized關(guān)鍵字和Lock作用范圍也只是當(dāng)前應(yīng)用，如果分布式部署，那無(wú)法保證某個(gè)數(shù)據(jù)在同時(shí)間只有一個(gè)線程訪問(wèn)，這時(shí)我們可以考慮使用中間層

接下來(lái)簡(jiǎn)單介紹本人開(kāi)源的一個(gè)分布式鎖的用法

一. 導(dǎo)入依賴

<dependency>
    <groupId>cn.gjing</groupId>
    <artifactId>tools-redis</artifactId>
    <version>1.2.0</version>
</dependency>

二. 啟動(dòng)類標(biāo)注注解

/**
 * @author Gjing
 */
@SpringBootApplication
@EnableToolsLock
public class TestRedisApplication {
    public static void main(String[] args) {
        SpringApplication.run(TestRedisApplication.class, args);
    }
}

三. 具體使用

1、注解方式

在需要加鎖的方法上使用@Lock注解即可

/**
 * @author Gjing
 **/
@RestController
public class TestController {
    private static int num = 20;
    
    @GetMapping("/test1")
    @Lock(key = "test1")
    public void test1() {
        System.out.println("當(dāng)前線程:" + Thread.currentThread().getName());
        if (num == 0) {
            System.out.println("賣完了");
            return;
        }
        num--;
        System.out.println("還剩余：" + num);
    }
}

參數(shù)信息

ab壓測(cè)執(zhí)行結(jié)果

lock

2、手動(dòng)控制方式

在需要使用的方法的類中通過(guò)@Resource注入

a、lock(): 加鎖

獲取鎖成功后會(huì)返回一個(gè)用于解鎖的值，失敗返回null

abstractLock.lock(key, expire, timeout, retry)
參數(shù)說(shuō)明

b、release()：解鎖

釋放鎖成功返回當(dāng)前被解鎖的key，失敗返回null
abstractLock.release(key, value)
參數(shù)說(shuō)明

使用案例

/**
 * @author Gjing
 **/
@RestController
public class LockController {
    @Resource
    private AbstractLock abstractLock;
    
    private static int num = 10;
    
    @GetMapping("/test2")
    public void test2() {
        String lock = null;
        try {
            lock = this.abstractLock.lock("testLock", 20, 10000, 50);
            System.out.println("當(dāng)前線程:" + Thread.currentThread().getName());
            if (num == 0) {
                System.out.println("賣完了");
                return;
            }
            num--;
            System.out.println("還剩余：" + num);
        } finally {
            this.abstractLock.release("testLock", lock);
        }
    }
}

ab壓測(cè)結(jié)果

lock2

3. 重寫(xiě)異常處理

在獲取鎖時(shí)往往會(huì)出現(xiàn)長(zhǎng)時(shí)間未獲取鎖，達(dá)到我們加鎖設(shè)置的超時(shí)時(shí)間后會(huì)拋出超時(shí)異常，如果要走自己的邏輯，可以重寫(xiě)異常處理

/**
 * @author Gjing
 **/
@Component
public class TimeoutHandler extends AbstractLockTimeoutHandler {
    @Override
    public void getLockTimeoutFallback(String s, int i, int i1, int i2) {
        // TODO: 2019/8/19 自定義邏輯 
    }
}

4. 自定義實(shí)現(xiàn)鎖

本項(xiàng)目使用Redis和lua腳本結(jié)合使用實(shí)現(xiàn)鎖，如若想使用自己的鎖，可以繼承AbstracttLock類

/**
 * @author Gjing
 **/
@Component
public class DemoLock extends AbstractLock {
    @Override
    public String lock(String s, String s1, int i, int i1, int i2) {
        return null;
    }
    @Override
    public String release(String s, String s1) {
        return null;
    }
}

四. 使用建議

該鎖建議使用單獨(dú)的單機(jī)redis，如果是在redis sentinel集群中情況就有所不同在redis sentinel集群中，我們具有多臺(tái)redis，他們之間有著主從的關(guān)系，例如一主二從。我們的set命令對(duì)應(yīng)的數(shù)據(jù)寫(xiě)到主庫(kù)，然后同步到從庫(kù)。當(dāng)我們申請(qǐng)一個(gè)鎖的時(shí)候，對(duì)應(yīng)就是一條命令 setnx mykey myvalue ，在redis sentinel集群中，這條命令先是落到了主庫(kù)。假設(shè)這時(shí)主庫(kù)down了，而這條數(shù)據(jù)還沒(méi)來(lái)得及同步到從庫(kù)，sentinel將從庫(kù)中的一臺(tái)選舉為主庫(kù)了。這時(shí)，我們的新主庫(kù)中并沒(méi)有mykey這條數(shù)據(jù)，若此時(shí)另外一個(gè)client執(zhí)行 setnx mykey hisvalue , 也會(huì)成功，即也能得到鎖。這就意味著，此時(shí)有兩個(gè)client獲得了鎖

使用中如果有任何問(wèn)題，歡迎評(píng)論留言，我會(huì)及時(shí)回復(fù)以及更新，源代碼地址：tools-redis