consul.KeyValue Maven / Gradle / Ivy

Go to download
package consul;

import com.fasterxml.jackson.databind.JsonNode;
import org.apache.commons.codec.binary.Base64;

import java.io.IOException;

public class KeyValue extends ConsulChain {

    public KeyValue(Consul consul) {
        super(consul);
    }

    public boolean set(String key, String value) throws ConsulException {
        // Give garbage, get garbage
        if (key == null || key.trim().length() == 0) {
            return false;
        }
        try {
            final HttpResp resp = Http.put(consul().getUrl() + EndpointCategory.KV.getUri() + key, value);
            return resp.getStatus() == Http.OK;
        } catch (IOException e) {
            throw new ConsulException(e);
        }
    }

    public String get(String key) throws ConsulException {
        // Give garbage, get garbage
        if (key == null || key.trim().length() == 0)
            return null;

        final KV keyValue = getDetails(key);
        byte[] valueDecoded = Base64.decodeBase64(keyValue.getValue());

        return new String(valueDecoded);
    }

    public KV getDetails(String key) throws ConsulException {
        try {
            // Give garbage, get garbage
            if (key == null || key.trim().length() == 0) {
                return null;
            }
            final HttpResp resp = Http.get(consul().getUrl() + EndpointCategory.KV.getUri() + key);
            JsonNode node = checkResponse(resp);
            if (node.isArray()) {
                node = node.get(0);
            }
            return new KV(node);
        } catch (IOException e) {
            throw new ConsulException(e);
        }
    }

    public boolean delete(String key) throws ConsulException {
        // Give garbage, get garbage
        if (key == null || key.trim().length() == 0) {
            return false;
        }
        try {
            final HttpResp resp = Http.delete(consul().getUrl() + EndpointCategory.KV.getUri() + key);
            return resp.getStatus() == Http.OK;
        } catch (IOException e) {
            throw new ConsulException(e);
        }
    }

    public boolean acquireLock(String key, String value, String sessionId) throws ConsulException {
        return lock(key, value, sessionId, "acquire");
    }

    public boolean releaseLock(String key, String value, String sessionId) throws ConsulException {
        return lock(key, value, sessionId, "release");
    }

    private boolean lock(String key, String value, String sessionId, String type) throws ConsulException {
        // Give garbage, get garbage
        if (key == null || key.trim().length() == 0 || value == null || value.trim().length() == 0 ||
            sessionId == null || sessionId.trim().length() == 0) {
            return false;
        }
        try {
            // Allow the lock to be acquired multiple times.
            final KV kv = getDetails(key);
            if (type.equals("acquire") && kv.getSessionId() != null && !kv.getSessionId().equals("")) {
                return kv.getSessionId().equals(sessionId);
            }
            final HttpResp resp = Http.put(consul().getUrl() + EndpointCategory.KV.getUri() + key + "?" + type + "=" + sessionId, value);
            if (resp.getStatus() != Http.OK) {
                return false;
            }
           return mapper.readValue(resp.getBody(), Boolean.class);
        } catch (IOException e) {
            throw new ConsulException(e);
        }
    }
}