org.apache.activemq.apollo.broker.ConfigStore.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of apollo-broker Show documentation
A reliable messaging server.
The newest version!
/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF 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 org.apache.activemq.apollo.broker

import org.fusesource.hawtbuf.ByteArrayInputStream
import security.EncryptionSupport
import org.apache.activemq.apollo.util._
import FileSupport._
import java.util.Properties
import java.io.{InputStream, FileInputStream, File}
import javax.xml.bind.{ValidationEvent, ValidationEventHandler}
import org.apache.activemq.apollo.dto.{NullStoreDTO, XmlCodec, BrokerDTO}

/**
 * 
 * 
 *
 * @author Hiram Chirino
 */
object ConfigStore {

  def load(file:File, func: (String)=>Unit):BrokerDTO = {
    load(new FileInputStream(file), config_properties(file), func)
  }

  def load_xml(in:Array[Byte], func: (String)=>Unit):BrokerDTO = {
    load(new ByteArrayInputStream(in), config_properties(null), func)
  }

  def load(is: => InputStream, prop:Properties, func: (String)=>Unit):BrokerDTO = {
    val rc = XmlCodec.decode(classOf[BrokerDTO], is, prop, new ValidationEventHandler(){
        def handleEvent(event: ValidationEvent): Boolean = {
          val level = event.getSeverity match {
            case 0=> "warning"
            case 1=> "error"
            case 2=> "fatal error"
          }
          func("%s at (%d:%d): %s ".format(level, event.getLocator().getLineNumber(), event.getLocator().getColumnNumber(), event.getMessage()))
          true
        }
    })
    import collection.JavaConversions._
    for( host <- rc.virtual_hosts ) {
      if( host.store == null ) {
        func("error: virtual host '%s' does not have valid store configured".format(host.id))
      }
    }
    rc
  }

  def config_properties(file:File): Properties = {
    import collection.JavaConversions._
    val props = new Properties()
    for( entry <- System.getenv().entrySet() ) {
      props.put("env."+entry.getKey, entry.getValue)
    }
    props.putAll(System.getProperties)
    if( file!=null ) {
      val prop_file = file.getParentFile / (file.getName + ".properties")
      if (prop_file.exists()) {
        FileSupport.using(new FileInputStream(prop_file)) {
          is =>
            val p = new Properties
            p.load(new FileInputStream(prop_file))
            props.putAll(EncryptionSupport.decrypt(p))
        }
      }
    }
    props
  }
}