io.microconfig.MicroconfigMain Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of microconfig-cli Show documentation
Powerful tool for microservice configuration management
There is a newer version: 4.9.3
package io.microconfig;

import io.microconfig.core.MicroconfigRunner;
import io.microconfig.core.properties.Properties;
import io.microconfig.core.properties.serializers.ConfigResult;
import lombok.RequiredArgsConstructor;
import lombok.val;

import java.io.File;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;

import static io.microconfig.core.properties.serializers.ConfigResult.toJson;
import static io.microconfig.core.properties.serializers.PropertySerializers.asConfigResult;
import static io.microconfig.utils.IoUtils.readClasspathResource;
import static io.microconfig.utils.Logger.*;
import static java.lang.System.exit;
import static java.lang.System.nanoTime;
import static java.util.concurrent.TimeUnit.NANOSECONDS;
import static java.util.stream.Collectors.toCollection;

/**
 * Command line params example: *
 * -r configs/repo -d build -e dev
 * 
 * VM speedup params:
 * -XX:TieredStopAtLevel=1
 */
@RequiredArgsConstructor
public class MicroconfigMain {
    private final File rootDir;
    private final String destinationDir;
    private final Set environments;
    private final List groups;
    private final List services;
    private final boolean stacktrace;
    private final boolean jsonOutput;
    private final boolean isSingleEnvBuild;

    public static void main(String... args) {
        val params = MicroconfigParams.parse(args);
        if (params.version()) {
            printVersion();
            return;
        }

        File rootDir = params.rootDir();
        String destinationDir = params.destinationDir();
        Set environments = params.environments();
        List groups = params.groups();
        List services = params.services();
        boolean stacktrace = params.stacktrace();
        boolean jsonOutput = params.jsonOutput();
        boolean isSingleEnvBuild = params.isSingleEnvBuild();

        new MicroconfigMain(rootDir, destinationDir, environments, groups, services,
                stacktrace, jsonOutput, isSingleEnvBuild)
                .build();
    }

    private void build() {
        try {
            enableLogger(!jsonOutput);
            doBuild();
        } catch (RuntimeException e) {
            if (stacktrace || e.getMessage() == null) throw e;
            error(e.getMessage());
            exit(-1);
        }
    }

    private void doBuild() {
        environmentsToBuild().forEach(env -> {
            long startTime = nanoTime();
            String resultDestinationDir = isSingleEnvBuild ? destinationDir : destinationDir + "/" + env;
            MicroconfigRunner runner = new MicroconfigRunner(rootDir, new File(resultDestinationDir));
            Properties properties = runner.buildProperties(env, groups, services);
            if (jsonOutput) {
                List results = properties.save(asConfigResult());
                System.out.println(toJson(results));
            } else {
                properties.save(runner.toFiles());
            }
            announce("\nGenerated [" + env + "] configs in " + (NANOSECONDS.toMillis(nanoTime() - startTime)) + "ms");
        });
    }

    private Set environmentsToBuild() {
        if (!environments.contains("*")) {
            return environments.stream()
                    .filter(e -> !e.startsWith("!"))
                    .collect(toCollection(LinkedHashSet::new));
        }

        return new MicroconfigRunner(rootDir, new File(destinationDir))
                .getMicroconfig()
                .environments()
                .environmentNames()
                .stream()
                .filter(e -> !environments.contains("!" + e))
                .collect(toCollection(LinkedHashSet::new));
    }

    private static void printVersion() {
        String version = readClasspathResource("version.properties")
                .split("\n")[0]
                .split("=")[1]
                .trim();
        info(version);
    }
}