apoc.load.LoadDirectoryItem Maven / Gradle / Ivy

Go to download
package apoc.load;

import org.apache.commons.lang3.StringUtils;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.concurrent.atomic.AtomicReference;

public class LoadDirectoryItem {

    public static final String LISTEN_EVENT_TYPE = "listenEventType";
    public static final String INTERVAL = "interval";
    public static final Set DEFAULT_EVENT_TYPES = Set.of("CREATE", "DELETE", "MODIFY");

    public static class LoadDirectoryConfig {

        private final List listenEventType;
        private final Long interval;

        public LoadDirectoryConfig(Map config) {
            if (config == null) config = Collections.emptyMap();
            this.interval = (Long) config.getOrDefault(INTERVAL, 1000L);
            this.listenEventType = (List) config.getOrDefault(LISTEN_EVENT_TYPE, new ArrayList<>(DEFAULT_EVENT_TYPES));
        }

        public List getListenEventType() {
            return listenEventType;
        }

        public Long getInterval() {
            return interval;
        }
    }

    public static class LoadDirectoryResult {

        public final String name;
        public final String status;
        public final String pattern;
        public final String cypher;
        public final String urlDir;
        public final Map config;
        public final String error;

        public LoadDirectoryResult(String name, String status, String pattern, String cypher, String urlDir, LoadDirectoryConfig configClass, String error) {
            this.name = name;
            this.status = status;
            this.pattern = pattern;
            this.cypher = cypher;
            this.urlDir = urlDir;
            this.config = Map.of(LISTEN_EVENT_TYPE, configClass.getListenEventType(), INTERVAL, configClass.getInterval());
            this.error = error;
        }
    }

    enum Status { RUNNING, CREATED, ERROR }

    private final String pattern;
    private final String cypher;
    private final String name;
    private final String urlDir;
    private final LoadDirectoryConfig config;
    private final AtomicReference status;
    private AtomicReference error;

    public LoadDirectoryItem(String name) {
        this(name, null, null, null, null);
    }

    public LoadDirectoryItem(String name, String pattern, String cypher, String urlDir, LoadDirectoryConfig config) {
        this.name = name;
        this.cypher = cypher;
        this.pattern = pattern;
        this.urlDir = urlDir;
        this.config = config;
        this.status = new AtomicReference<>(Status.CREATED);
        this.error = new AtomicReference<>(StringUtils.EMPTY);
    }

    public String getPattern() {
        return pattern;
    }

    public String getCypher() {
        return cypher;
    }

    public String getName() {
        return name;
    }

    public String getUrlDir() {
        return urlDir;
    }

    public LoadDirectoryConfig getConfig() {
        return config;
    }

    public synchronized void setError(String errorMessage) {
        error.set(errorMessage);
        status.set(Status.ERROR);
    }

    public void setStatusRunning() {
        status.set(Status.RUNNING);
    }

    public synchronized LoadDirectoryResult toResult() {
        return new LoadDirectoryResult(name, status.get().name(), pattern, cypher, urlDir, config, error.get());
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        LoadDirectoryItem that = (LoadDirectoryItem) o;
        return Objects.equals(name, that.name);
    }

    @Override
    public int hashCode() {
        return Objects.hash(name);
    }
}