1
2
3
4
// RedissonLock의 tryLock 메소드 : 타임아웃 시간을 명시
public boolean tryLock(long waitTime, long leaseTime, TimeUnit unit) throws InterruptedException
// waitTime: 락을 획득하기 위한 대기 시간, leaseTime: 락이 만료되는 시간, unit: 시간단
// 락 획득에 성공하면 true 반환

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
@Override
public boolean tryLock(long waitTime, long leaseTime, TimeUnit unit) throws InterruptedException {
    /*
    1. 메서드가 호출된 시점의 현재 시간과 스레드 ID를 가져오기
    */
    long time = unit.toMillis(waitTime);
    long current = System.currentTimeMillis();
    long threadId = Thread.currentThread().getId();

    /*
    2. tryAcquire 메서드를 호출하여 락을 획득하려고 시도 
      락을 획득하면 null을 반환하고, 그렇지 않으면 락의 남은 TTL(Time To Live)을 반환
    */
    Long ttl = tryAcquire(waitTime, leaseTime, unit, threadId);
    // lock acquired
    if (ttl == null) {
        return true;  락 획득 
    }

    /*
    3. 락 획득에 실패한 경우, 남은 대기 시간을 계산
    */
    time -= System.currentTimeMillis() - current;
    if (time <= 0) {
        acquireFailed(waitTime, unit, threadId);
        return false; // 대기시간 초과되어 락 획득 실패
    }
    
    current = System.currentTimeMillis();

    /*
    4. 락 해제 알림을 받기 위해 Pub/Sub 채널을 구독
    */
    CompletableFuture<RedissonLockEntry> subscribeFuture = subscribe(threadId);
    try {
        subscribeFuture.get(time, TimeUnit.MILLISECONDS);
    } catch (TimeoutException e) {
        if (!subscribeFuture.completeExceptionally(new RedisTimeoutException(
                "Unable to acquire subscription lock after " + time + "ms. " +
                        "Try to increase 'subscriptionsPerConnection' and/or 'subscriptionConnectionPoolSize' parameters."))) {
            subscribeFuture.whenComplete((res, ex) -> {
                if (ex == null) {
                    unsubscribe(res, threadId);
                }
            });
        }
        acquireFailed(waitTime, unit, threadId); // 구독 중 오류 발생 시 락 획득 실패
        return false;
    } catch (ExecutionException e) {
        LOGGER.error(e.getMessage(), e);
        acquireFailed(waitTime, unit, threadId);
        return false; // 구독 중 오류 발생 시 락 획득 실패
    }

    /*
    5. 남은 시간 동안 주기적으로 락 획득 재시도
      락을 획득하면 true를 반환하고, 대기 시간이 초과되면 false를 반환
    */
    try {
        time -= System.currentTimeMillis() - current;
        if (time <= 0) {
            acquireFailed(waitTime, unit, threadId);
            return false; // 대기 시간이 초과되어 락 획득 실패
        }
    
        while (true) {
            long currentTime = System.currentTimeMillis();
            ttl = tryAcquire(waitTime, leaseTime, unit, threadId);
            // lock acquired
            if (ttl == null) {
                return true; // 락 획득 성공
            }

            time -= System.currentTimeMillis() - currentTime;
            if (time <= 0) {
                acquireFailed(waitTime, unit, threadId);
                return false; // 대기 시간이 초과되어 락 획득 실패
            }

            // waiting for message
            currentTime = System.currentTimeMillis();
            if (ttl >= 0 && ttl < time) {
                commandExecutor.getNow(subscribeFuture).getLatch().tryAcquire(ttl, TimeUnit.MILLISECONDS);
            } else {
                commandExecutor.getNow(subscribeFuture).getLatch().tryAcquire(time, TimeUnit.MILLISECONDS);
            }

            time -= System.currentTimeMillis() - currentTime;
            if (time <= 0) {
                acquireFailed(waitTime, unit, threadId);
                return false; // 대기 시간이 초과되어 락 획득 실패
            }
        }
    } finally {
        unsubscribe(commandExecutor.getNow(subscribeFuture), threadId);
    }
}