aboutsummaryrefslogtreecommitdiffstats
path: root/zkfacade/src/main/java/com/yahoo/vespa/curator/MultiplePathsLock.java
blob: 65ae77c8ffa70fa9b4823dad87972126fe9bd209 (plain) (blame)
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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.curator;

import com.yahoo.concurrent.UncheckedTimeoutException;
import com.yahoo.path.Path;

import java.time.Duration;

/**
 * Class that holds two locks, originally used for transitioning from one lock to
 * another, where you need to hold both the old lock and the new lock in the
 * transition period.
 *
 * @author hmusum
 */
public class MultiplePathsLock extends Lock {

    private final Lock oldLock;
    private final Lock newLock;

    public MultiplePathsLock(Path newLockPath, Path oldLockPath, Duration timeout, Curator curator) {
        super(newLockPath.getAbsolute(), curator);
        this.newLock = curator.lock(newLockPath, timeout);
        this.oldLock = curator.lock(oldLockPath, timeout);;
    }

    @Override
    public void acquire(Duration timeout) throws UncheckedTimeoutException {
        oldLock.acquire(timeout);
        newLock.acquire(timeout);
    }

    @Override
    public void close() {
        oldLock.close();
        newLock.close();
    }

}