com.github.aqiu202.lock.base.AbstractReentrantTtlLock Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of qlock-core Show documentation
There is a newer version: 1.2.0
package com.github.aqiu202.lock.base;

import com.github.aqiu202.id.IdGenerator;
import com.github.aqiu202.id.IdGeneratorFactory;
import com.github.aqiu202.lock.centralize.LocaleTtlLock;
import java.util.Objects;
import java.util.concurrent.TimeUnit;
import org.springframework.lang.Nullable;
import org.springframework.util.Assert;

/**
 * AbstractReentrantLock
 *
 * @author aqiu 2020/12/2 15:57
 **/
public abstract class AbstractReentrantTtlLock extends LocaleTtlLock {

    protected IdGenerator idGenerator;

    protected IdGeneratorFactory idGeneratorFactory;

    public IdGenerator getIdGenerator() {
        return idGenerator;
    }

    public void setIdGenerator(IdGenerator idGenerator) {
        this.idGenerator = idGenerator;
    }

    public IdGeneratorFactory getIdGeneratorFactory() {
        return idGeneratorFactory;
    }

    public void setIdGeneratorFactory(IdGeneratorFactory idGeneratorFactory) {
        this.idGeneratorFactory = idGeneratorFactory;
    }

    public void setLockValueHolderStrategy(LockValueHolderStrategy strategy) {
        LockValueHolder.strategy = strategy;
    }

    @Override
    public Boolean release(String key, long expired, TimeUnit timeUnit) {
        final String value = this.cache.get(key);
        if (Objects.equals(value, LockValueHolder.getValue())) {
            return this.doRelease(key, expired, timeUnit);
        }
        return false;
    }

    @Override
    public Boolean release(String key) {
        final String value = this.cache.get(key);
        if (Objects.equals(value, LockValueHolder.getValue())) {
            return this.doRelease(key);
        }
        return false;
    }

    @Override
    public Boolean acquire(String key, long expired, TimeUnit timeUnit) {
        String value = String.valueOf(this.idGenerator.nextId());
        value = LockValueHolder.setIfAbsent(value);
        final Boolean result = this.doAcquire(key, value, expired, timeUnit);
        if (result != null && !result) {
            return Objects.equals(this.cache.get(key), value);
        }
        return result;
    }

    @Override
    public Boolean acquire(String key) {
        String value = String.valueOf(this.idGenerator.nextId());
        value = LockValueHolder.setIfAbsent(value);
        final Boolean result = this.doAcquire(key, value);
        if (result != null && !result) {
            return Objects.equals(this.cache.get(key), value);
        }
        return result;
    }

    @Nullable
    public Boolean doAcquire(String key, String value) {
        return this.cache.setIfAbsent(key, value);
    }

    @Nullable
    public Boolean doAcquire(String key, String value, long expired, TimeUnit timeUnit) {
        return this.cache.setIfAbsent(key, value, expired, timeUnit);
    }

    @Nullable
    public Boolean doRelease(String key) {
        return super.release(key);
    }

    @Nullable
    public Boolean doRelease(String key, long expired, TimeUnit timeUnit) {
        return super.release(key, expired, timeUnit);
    }

    @Override
    public void afterPropertiesSet() throws Exception {
        this.initIdGenerator();
        Assert.notNull(this.idGenerator, "ID Generator无法构建");
        super.afterPropertiesSet();
    }

    protected void initIdGenerator() {
        if (this.idGenerator == null) {
            this.idGenerator = this.idGeneratorFactory.getIdGenerator();
        }
    }
}