io.tracee.contextlogger.impl.ConfigBuilderImpl Maven / Gradle / Ivy

Go to download
package io.tracee.contextlogger.impl;

import io.tracee.contextlogger.api.ConfigBuilder;
import io.tracee.contextlogger.api.ToStringBuilder;
import io.tracee.contextlogger.api.internal.Configuration;
import io.tracee.contextlogger.api.internal.ContextLoggerBuilderAccessible;
import io.tracee.contextlogger.contextprovider.api.Profile;
import io.tracee.contextlogger.outputgenerator.TraceeContextStringRepresentationBuilderImpl;
import io.tracee.contextlogger.outputgenerator.writer.BasicOutputWriterConfiguration;
import io.tracee.contextlogger.outputgenerator.writer.OutputWriterConfiguration;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Implementation class to create a configuration by using the fluent api.
 */
public class ConfigBuilderImpl implements Configuration {

	private final ContextLoggerBuilderAccessible contextLogger;
	private final ContextLoggerConfiguration contextLoggerConfiguration;

	private Profile profile = null;

	private boolean enforceOrder = false;

	private Map manualContextOverrides = new HashMap();

	private OutputWriterConfiguration outputWriterConfiguration = BasicOutputWriterConfiguration.JSON_INTENDED;

	public ConfigBuilderImpl(ContextLoggerBuilderAccessible traceeContextLoggerBuilderAccessible) {

		this.contextLogger = traceeContextLoggerBuilderAccessible;
		this.contextLoggerConfiguration = traceeContextLoggerBuilderAccessible.getContextLoggerConfiguration();

	}

	@Override
	public final ConfigBuilderImpl enforceProfile(Profile profile) {
		this.profile = profile;
		return this;
	}

	@Override
	public ConfigBuilder enable(String... contexts) {
		fillManualContextOverrideMap(contexts, true);
		return this;
	}

	@Override
	public ConfigBuilder disableTypes(final Class... types) {
		List classNames = new ArrayList();
		if (types != null) {

			for (Class type : types) {
				if (type != null) {
					this.manualContextOverrides.put(type.getCanonicalName(), Boolean.FALSE);
				}
			}

		}
		return this;
	}

	@Override
	public ConfigBuilder disable(String... contexts) {
		fillManualContextOverrideMap(contexts, false);
		return this;
	}

	@Override
	public ConfigBuilder enforceOrder() {
		this.enforceOrder = true;
		return this;
	}

	@Override
	public ConfigBuilder enforceOutputWriterConfiguration(final BasicOutputWriterConfiguration outputWriterConfiguration) {
		this.outputWriterConfiguration = outputWriterConfiguration;
		return this;
	}

	@Override
	public T apply() {
		contextLogger.setStringRepresentationBuilder(createContextStringRepresentationLogBuilder());
		return (T) contextLogger;
	}

	public Map getManualContextOverrides() {
		return manualContextOverrides;
	}

	public Profile getProfile() {
		return profile;
	}

	public boolean getEnforceOrder() {
		return enforceOrder;
	}

	@Override
	public OutputWriterConfiguration getOutputWriterConfiguration() {
		return this.outputWriterConfiguration;
	}

	/**
	 * Adds passed contexts value pairs to manualContextOverrides.
	 *
	 * @param contexts The property name of the context data.
	 * @param value    the value which should be set.
	 */
	private void fillManualContextOverrideMap(final String[] contexts, final boolean value) {
		if (contexts != null) {

			for (String context : contexts) {

				if (!context.isEmpty()) {
					this.manualContextOverrides.put(context, value);
				}

			}

		}
	}

	/**
	 * Creates a TraceeGsonContextStringRepresentationBuilder instance which can be used for creating the createStringRepresentation message.
	 *
	 * @return An instance of TraceeGsonContextStringRepresentationBuilder
	 */
	private TraceeContextStringRepresentationBuilderImpl createContextStringRepresentationLogBuilder() {

		TraceeContextStringRepresentationBuilderImpl traceeContextStringRepresentationBuilderImpl = new TraceeContextStringRepresentationBuilderImpl();
		traceeContextStringRepresentationBuilderImpl.setManualContextOverrides(this.getManualContextOverrides());
		traceeContextStringRepresentationBuilderImpl.setProfile(this.getProfile());
		traceeContextStringRepresentationBuilderImpl.setEnforceOrder(this.getEnforceOrder());
		traceeContextStringRepresentationBuilderImpl.setOutputWriterConfiguration(this.getOutputWriterConfiguration());

		return traceeContextStringRepresentationBuilderImpl;
	}
}