com.es.plus.adapter.lock.EsReadLock Maven / Gradle / Ivy

Go to download
package com.es.plus.adapter.lock;

import org.elasticsearch.action.DocWriteResponse;
import org.elasticsearch.action.update.UpdateResponse;
import org.elasticsearch.script.Script;
import org.elasticsearch.script.ScriptType;
import java.util.HashMap;
import java.util.Map;

import static com.es.plus.constant.EsConstant.*;

public class EsReadLock extends ELock {

    public EsReadLock(ELockClient esPlusLockClient, String key) {
        super(esPlusLockClient, key);
    }

    @Override
    public boolean tryLock0() {
        Map data = new HashMap<>();
        data.put("read_lock_count", 1);
        data.put("lock_type", "s");
        data.put(GLOBAL_LOCK_EXPIRETIME, System.currentTimeMillis() + GLOBAL_LOCK_TIMEOUT * 1000);

        Map params = new HashMap<>();
        params.put("currentTime", System.currentTimeMillis());
        params.put("expireTime", (System.currentTimeMillis() + GLOBAL_LOCK_TIMEOUT * 1000));
        String script = "if(ctx._source." + GLOBAL_LOCK_EXPIRETIME + " params = new HashMap<>();
        params.put("param", 0);
        //构建scipt语句
        String script = "if(ctx._source.lock_type == \"s\" && ctx._source.read_lock_count>=1){\n" +
                "            ctx._source.read_lock_count -= 1;\n" +
                "      }\n" +
                "if(ctx._source.read_lock_count>params.param) {" +
                "return;" +
                " }" +
                "      ctx.op = 'delete';";
        Script painless = new Script(ScriptType.INLINE, PAINLESS, script, params);
        UpdateResponse update = esPlusLockClient.updateByScript(lockIndexName(), key, painless);
    }

    @Override
    public String lockIndexName() {
        return "ep_read_write_lock";
    }
}