io.tracee.contextlogger.contextprovider.ContextProviderServiceLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of contextlogger-core Show documentation
Please refer to https://github.com/tracee/contextlogger.
There is a newer version: 0.11.0
package io.tracee.contextlogger.contextprovider;

import io.tracee.contextlogger.contextprovider.api.Profile;
import io.tracee.contextlogger.contextprovider.api.TraceeContextProviderServiceProvider;
import io.tracee.contextlogger.profile.ProfileLookup;
import io.tracee.contextlogger.profile.ProfilePropertyNames;
import io.tracee.contextlogger.utility.GenericServiceLocator;

import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;

/**
 * Service loader for context providers.
 */
public class ContextProviderServiceLoader implements TraceeContextProviderServiceProvider {

	private static TraceeContextProviderServiceProvider instance = null;

	private final Class[] immplicitContextProviders;
	private final Class[] contextProviders;

	private static final Map profilePropertiesMap = new HashMap();

	private ContextProviderServiceLoader() {

		List serviceProviders = GenericServiceLocator.locateAll(TraceeContextProviderServiceProvider.class);

		List implicitContextProviders = new ArrayList();
		List contextProviders = new ArrayList();

		for (TraceeContextProviderServiceProvider serviceProvider : serviceProviders) {
			if (serviceProvider != null) {
				implicitContextProviders.addAll(Arrays.asList(serviceProvider.getImplicitContextProvider()));
				contextProviders.addAll(Arrays.asList(serviceProvider.getContextProvider()));
			}
		}

		this.contextProviders = contextProviders.toArray(new Class[contextProviders.size()]);
		this.immplicitContextProviders = implicitContextProviders.toArray(new Class[implicitContextProviders.size()]);

	}

	@Override
	public Class[] getImplicitContextProvider() {
		return this.immplicitContextProviders;
	}

	@Override
	public Class[] getContextProvider() {
		return this.contextProviders;
	}


	public static Properties getProfileSettings(final Profile profile) {

		if (profile == null || Profile.NONE.equals(profile)) {
			return new Properties();
		}

		Properties properties = profilePropertiesMap.get(profile);

		if (properties == null) {

			properties = new Properties();

			try {
				Enumeration resources = ContextProviderServiceLoader.class.getClassLoader().getResources(profile.getFilename());

				while (resources.hasMoreElements()) {

					URL url = resources.nextElement();

					InputStream inputStream = null;
					try {
						properties.load(url.openStream());
					} finally {
						if (inputStream != null) {
							inputStream.close();
						}
					}
				}

			} catch (IOException e) {
				// Ignore
			}

			// Add Custom Properties
			properties.putAll(getCustomProperties());


			profilePropertiesMap.put(profile, properties);
		}

		return properties;

	}

	public void clear() {
		profilePropertiesMap.clear();
	}

	private static Properties getCustomProperties() {

		Properties properties = null;

		try {
			properties = ProfileLookup.openProperties(getCustomPropertyFileName());
		} catch (IOException e) {
		}

		return properties != null ? properties : new Properties();
	}

	private static String getCustomPropertyFileName() {

		String filename = System.getProperty(ProfilePropertyNames.CUSTOM_PROFILE_FILENAME_SET_GLOBALLY_VIA_SYSTEM_PROPERTIES);


		if (filename == null) {
			// load custom profile filename vi file in classpath
			try {
				Properties selectorProperties = ProfileLookup.openProperties(ProfilePropertyNames.PROFILE_SET_BY_FILE_IN_CLASSPATH_FILENAME);
				if (selectorProperties != null) {
					filename = selectorProperties.getProperty(ProfilePropertyNames.CUSTOM_PROFILE_FILENAME_SET_BY_FILE_IN_CLASSPATH_PROPERTY);

				}
			} catch (IOException e) {
			}
		}

		if (filename == null) {
			// load custom profile filename via system property
			filename = ProfilePropertyNames.DEFAULT_CUSTOM_PROFILE_FILE;
		}

		return "/" + filename;
	}

	public static TraceeContextProviderServiceProvider getServiceLocator() {

		if (instance == null) {
			instance = new ContextProviderServiceLoader();
		}
		return instance;

	}

}