com.stratio.crossdata.connectors.config.ConnectConfig.scala Maven / Gradle / Ivy

Go to download
/*
 * Licensed to STRATIO (C) under one or more contributor license agreements.
 * See the NOTICE file distributed with this work for additional information
 * regarding copyright ownership.  The STRATIO (C) licenses this file
 * to you 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.
 */

package com.stratio.crossdata.connectors.config

import java.io.File

import com.typesafe.config.{Config, ConfigFactory}
import org.apache.log4j.Logger

object ConnectConfig {
  val CONNECTOR_BASIC_CONFIG = "connector-reference.conf"
  val PARENT_CONFIG_NAME = "crossdata-connector"

  val CONNECTOR_CLUSTER_NAME_KEY = "config.cluster.name"
  val CONNECTOR_ACTOR_NAME_KEY = "config.cluster.actor"
  val CONNECTOR_USER_CONFIG_FILE = "external.config.filename"
  val CONNECTOR_USER_CONFIG_RESOURCE = "external.config.resource"
  val CONNECTOR_ACTOR_NUM = "config.akka.number.connector-actor"

  val ConnectorAutoRestartKey = "config.auto-restart"
}

trait ConnectConfig {

  lazy val logger: Logger = ???
  lazy val clusterName = config.getString(ConnectConfig.CONNECTOR_CLUSTER_NAME_KEY)
  lazy val actorName = config.getString(ConnectConfig.CONNECTOR_ACTOR_NAME_KEY)
  lazy val num_connector_actor: Int = config.getString(ConnectConfig.CONNECTOR_ACTOR_NUM).toInt
  lazy val autoRestartConnector = config.getString(ConnectConfig.ConnectorAutoRestartKey).toBoolean

  val config: Config = {

    var defaultConfig = ConfigFactory.load(ConnectConfig.CONNECTOR_BASIC_CONFIG).getConfig(ConnectConfig.PARENT_CONFIG_NAME)
    val configFile = defaultConfig.getString(ConnectConfig.CONNECTOR_USER_CONFIG_FILE)
    val configResource = defaultConfig.getString(ConnectConfig.CONNECTOR_USER_CONFIG_RESOURCE)

    if (configResource != "") {
      val resource = ConnectConfig.getClass.getClassLoader.getResource(configResource)
      if (resource != null) {
        val userConfig = ConfigFactory.parseResources(configResource).getConfig(ConnectConfig.PARENT_CONFIG_NAME)
        defaultConfig = userConfig.withFallback(defaultConfig)
      } else {
        logger.warn("User resource (" + configResource + ") haven't been found")
        val file = new File(configResource)
        if (file.exists()) {
          val userConfig = ConfigFactory.parseFile(file).getConfig(ConnectConfig.PARENT_CONFIG_NAME)
          defaultConfig = userConfig.withFallback(defaultConfig)
        } else {
          logger.warn("User file (" + configResource + ") haven't been found in classpath")
        }
      }
    }
    if (configFile != "") {
      val file = new File(configFile)
      if (file.exists()) {
        val userConfig = ConfigFactory.parseFile(file).getConfig(ConnectConfig.PARENT_CONFIG_NAME)
        defaultConfig = userConfig.withFallback(defaultConfig)
      } else {
        logger.warn("User file (" + configFile + ") haven't been found")
      }
    }

    /*
    val modifiedConfig = ConfigFactory.parseString("akka.cluster.roles = [CassandraConnector]")
    val finalConfig = modifiedConfig.withFallback(defaultConfig)
    ConfigFactory.load(finalConfig)
    */
    ConfigFactory.load(defaultConfig)
  }

}