src/com/ibm/matos/Config.java

/**
 * Copyright 2015 IBM
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
/**
 * Licensed Materials - Property of IBM
 * (c) Copyright IBM Corp. 2015
 */
package com.ibm.matos;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.Map;

import org.apache.log4j.Level;
import org.apache.log4j.Logger;

import com.google.gson.JsonObject;
import com.google.gson.JsonParser;

public class Config {

	private static Logger logger = Logger.getLogger(Config.class);
	
	// static Kafka properties
	final static String KAFKA_BROKER_PROP = "kafkaBroker";
	final static String KAFKA_REST_PROP = "kafkaRest";
	final static String KAFKA_API_KEY_PROP = "kafkaApiKey";
	final static String KAFKA_TOPIC_PROP = "kafkaTopic";
	final static String KAFKA_PARTITION_PROP = "kafkaPartition";
	final static String KAFKA_CONSUMER_ID_PROP = "kafkaConsumerId";
	// variable Kafka properties
	final static String KAFKA_START_OFFSET_PROP = "kafkaStartOffset";
	final static String KAFKA_END_OFFSET_PROP = "kafkaEndOffset";
	final static String KAFKA_NUM_RECORDS_PROP = "kafkaNumRecords";
	// static Swift properties
	final static String SWIFT_AUTH_URL_PROP = "swiftAuthUrl";
	final static String SWIFT_REGION_PROP = "swiftRegion";
	final static String SWIFT_TENANT_ID_PROP = "swiftTenantId";
	final static String SWIFT_USER_ID_PROP = "swiftUserId";
	final static String SWIFT_PASSWORD_PROP = "swiftPassword";
	final static String SWIFT_CONTAINER_PROP = "swiftContainer";
	final static String[] ALL_PROPS = {
			KAFKA_BROKER_PROP,
			KAFKA_REST_PROP,
			KAFKA_API_KEY_PROP,
			KAFKA_TOPIC_PROP,
			KAFKA_PARTITION_PROP,
			KAFKA_CONSUMER_ID_PROP,
			KAFKA_START_OFFSET_PROP,
			KAFKA_END_OFFSET_PROP,
			KAFKA_NUM_RECORDS_PROP,
			SWIFT_AUTH_URL_PROP,
			SWIFT_REGION_PROP,
			SWIFT_TENANT_ID_PROP,
			SWIFT_USER_ID_PROP,
			SWIFT_PASSWORD_PROP,
			SWIFT_CONTAINER_PROP
	};
	
	private final static String PROP_FILE_NAME = "matos.json";
	private JsonObject matos;
	
	public Config() throws IOException {
		this(PROP_FILE_NAME);
	}
	
	public Config(String filename) throws IOException {
		matos = loadProperties(filename); 
	}

	private JsonObject loadProperties(String filename) throws IOException {
		JsonObject matos = null;
		InputStream is = new FileInputStream(Utils.resourceDir + File.separator + filename);
		matos = new JsonParser().parse(new InputStreamReader(is)).getAsJsonObject();
		logger.log(Level.INFO, "Matos Properties:" + matos);
		return matos;
	}


	public void overrideProperties(JsonObject props) {
		// override config properties with those passed in params
		for (String prop : Config.ALL_PROPS) {
			if (props.has(prop))
				matos.addProperty(prop, props.getAsJsonPrimitive(prop).getAsString());
		}
	}

	public void overrideProperties(Map<String,String> props) {
		// override config properties with those passed in params
		for (String prop : Config.ALL_PROPS) {
			if (props.containsKey(prop))
				matos.addProperty(prop, props.get(prop));
		}
	}
	
	public String get(String key) {
		return matos.getAsJsonPrimitive(key).getAsString();
	}
	
	public String toString() {
		return matos.toString();
	}

}