com.talanlabs.processmanager.messages.injector.AbstractInjector Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of process-manager-messages-plugin Show documentation
The newest version!
package com.talanlabs.processmanager.messages.injector;

import com.talanlabs.processmanager.messages.flux.AbstractImportFlux;
import com.talanlabs.processmanager.messages.gate.Gate;
import com.talanlabs.processmanager.messages.trigger.event.FileTriggerEvent;
import com.talanlabs.processmanager.shared.logging.LogManager;
import com.talanlabs.processmanager.shared.logging.LogService;
import org.apache.commons.lang3.StringUtils;

import java.io.File;

/**
 * An abstract injector used to import flux

 */
public abstract class AbstractInjector implements IInjector {

    private final LogService logService;

    private final String name;

    private final String acceptedPath;
    private final String retryPath;
    private final String rejectedPath;
    private final String archivePath;

    private Gate gate;
    private final File workDir;

    /**
     * Builds an injector dedicated to managing flux of the given type
     *
     * @param name        the name of the flux (which is also the name of the final folder). Must be unique
     * @param baseWorkdir the base work directory (parent of where the injector folder will be located)
     */
    public AbstractInjector(String name, String baseWorkdir) {
        super();

        logService = LogManager.getLogService(getClass());

        this.name = name;

        String workDirPath = StringUtils.appendIfMissing(baseWorkdir, File.separator) + name + File.separator;
        this.acceptedPath = workDirPath + "accepted" + File.separator;
        this.retryPath = workDirPath + "retry" + File.separator;
        this.rejectedPath = workDirPath + "rejected" + File.separator;
        this.archivePath = workDirPath + "archive" + File.separator;
        this.workDir = new File(workDirPath);
    }

    @Override
    public final Gate getGate() {
        return gate;
    }

    @Override
    public final void setGate(Gate g) {
        this.gate = g;
    }

    @Override
    public final String getName() {
        return name;
    }

    @Override
    public final File getWorkDir() {
        return workDir;
    }

    /**
     * Called when a file is injected
     */
    @Override
    public final Object inject(FileTriggerEvent evt) {
        File file = evt.getAttachment();
        if (evt.isNewFileEvent() && file.getParentFile().equals(getWorkDir())) {
            try {
                injectMessage(null, file); // content will be fetched later (lazy loading)
            } catch (Exception e) {
                logService.error(() -> "Error when injecting file", e);
            }
        }
        return null;
    }

    @Override
    public final void injectMessage(String content, File file) {
        M flux = createFlux();
        flux.setContent(content);
        flux.setFile(file);
        handleFlux(flux);
    }

    /**
     * Implement this method to tell the injector what to do with the given flux
     */
    protected abstract void handleFlux(M flux);

    @Override
    public final String getAcceptedPath() {
        return acceptedPath;
    }

    @Override
    public final String getRejectedPath() {
        return rejectedPath;
    }

    @Override
    public final String getRetryPath() {
        return retryPath;
    }

    @Override
    public final String getArchivePath() {
        return archivePath;
    }

    @Override
    public long getDelay() {
        return 20000;
    }

    public abstract M createFlux();

}