com.marklogic.hub.deploy.util.ModuleWatchingConsumer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of marklogic-data-hub Show documentation

Library for Creating an Operational Data Hub on MarkLogic

There is a newer version: 6.1.1

package com.marklogic.hub.deploy.util;

import com.marklogic.appdeployer.command.Command;
import com.marklogic.appdeployer.command.CommandContext;
import com.marklogic.client.ext.helper.LoggingObject;
import org.springframework.core.io.Resource;

import java.util.Set;
import java.util.function.Consumer;

/**
 * Consumer implementation to be used with mlWatch. Expected to be invoked after one or more new/modified modules have
 * been detected and loaded.
 */
public class ModuleWatchingConsumer extends LoggingObject implements Consumer> {

    private final CommandContext commandContext;
    private final Command generateFunctionMetadataCommand;

    public ModuleWatchingConsumer(CommandContext commandContext, Command generateFunctionMetadataCommand) {
        this.commandContext = commandContext;
        this.generateFunctionMetadataCommand = generateFunctionMetadataCommand;
    }

    @Override
    public void accept(Set resources) {
        if (shouldFunctionMetadataBeGenerated(resources)) {
            if (generateFunctionMetadataCommand == null || commandContext == null) {
                logger.warn("Unable to generate function metadata for modules containing mapping functions; no command or command context found");
            } else {
                try {
                    logger.info("Generating function metadata for modules containing mapping functions");
                    generateFunctionMetadataCommand.execute(commandContext);
                } catch (Exception ex) {
                    logger.error("Unable to generate function metadata, cause: " + ex.getMessage());
                }

            }
        }
    }

    protected boolean shouldFunctionMetadataBeGenerated(Set resources) {
        if (resources != null && !resources.isEmpty()) {
            for (Resource r : resources) {
                try {
                    if (r.getFile().getAbsolutePath().contains("mapping-functions")) {
                        return true;
                    }
                } catch (Exception ex) {
                    // This is not expected to ever happen, and it shouldn't cause a failure, but a log message is
                    // still warranted in case there's a real problem, as it may result in function metadata never being
                    // generated while mlWatch is running
                    logger.warn("Unable to read resource: " + r + "; cause: " + ex.getMessage());
                }
            }
        }
        return false;
    }
}