code.grnet.gr Git - aquarium/blob - src/main/scala/gr/grnet/aquarium/store/mongodb/MongoDBStoreProvider.scala

   1 /*
   2  * Copyright 2011-2012 GRNET S.A. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or
   5  * without modification, are permitted provided that the following
   6  * conditions are met:
   7  *
   8  *   1. Redistributions of source code must retain the above
   9  *      copyright notice, this list of conditions and the following
  10  *      disclaimer.
  11  *
  12  *   2. Redistributions in binary form must reproduce the above
  13  *      copyright notice, this list of conditions and the following
  14  *      disclaimer in the documentation and/or other materials
  15  *      provided with the distribution.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
  18  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  20  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
  21  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  22  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  23  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  24  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  25  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  27  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  28  * POSSIBILITY OF SUCH DAMAGE.
  29  *
  30  * The views and conclusions contained in the software and
  31  * documentation are those of the authors and should not be
  32  * interpreted as representing official policies, either expressed
  33  * or implied, of GRNET S.A.
  34  */
  35
  36 package gr.grnet.aquarium.store.mongodb
  37
  38 import com.ckkloverdos.props.Props
  39 import com.mongodb.{MongoException, Mongo, MongoOptions, ServerAddress}
  40 import gr.grnet.aquarium.store._
  41 import gr.grnet.aquarium.{AquariumException, Configurable}
  42
  43 /**
  44  *
  45  * @author Christos KK Loverdos <loverdos@gmail.com>
  46  */
  47
  48 class MongoDBStoreProvider extends StoreProvider with Configurable {
  49   private[this] var _mongo: Mongo = _
  50   private[this] var _database: String = _
  51   private[this] var _username: String = _
  52   private[this] var _password: String = _
  53
  54   private[this] var _mongoDBStore: MongoDBStore = _
  55
  56   def propertyPrefix = Some(MongoDBStoreProvider.MongoDBKeys.Prefix)
  57
  58   def configure(props: Props) = {
  59     import MongoDBStoreProvider.MongoDBKeys
  60
  61     this._database = props.getEx(MongoDBKeys.dbschema)
  62     this._username = props.getEx(MongoDBKeys.username)
  63     this._password = props.getEx(MongoDBKeys.password)
  64     val host = props.getEx(MongoDBKeys.host)
  65     val port = props.getEx(MongoDBKeys.port).toInt
  66
  67     try {
  68       val addr = new ServerAddress(host, port)
  69
  70       val opt = new MongoOptions()
  71       opt.connectionsPerHost = props.getEx(MongoDBKeys.connection_pool_size).toInt
  72       opt.threadsAllowedToBlockForConnectionMultiplier = 8
  73
  74       this._mongo = new Mongo(addr, opt)
  75       this._mongoDBStore = new MongoDBStore(this._mongo, this._database, this._username, this._password)
  76     } catch {
  77       case e: MongoException =>
  78         throw new AquariumException("Cannot connect to mongo at %s:%s".format(host, port), e)
  79     }
  80   }
  81
  82   def userStateStore = _mongoDBStore
  83   def resourceEventStore = _mongoDBStore
  84   def imEventStore = _mongoDBStore
  85   def policyStore = _mongoDBStore
  86 }
  87
  88 /**
  89  * Provides configuration keys.
  90  *
  91  * @author Christos KK Loverdos <loverdos@gmail.com>
  92  */
  93 object MongoDBStoreProvider {
  94
  95   /**
  96    * Note that these keys must be prefixed by `mongodb` in the configuration file
  97    *
  98    * @author Christos KK Loverdos <loverdos@gmail.com>
  99    */
 100   object MongoDBKeys {
 101     final val Prefix = "mongodb"
 102     final val PrefixAndDot = Prefix + "."
 103
 104     private[this] def p(name: String) = PrefixAndDot + name
 105
 106     /**
 107      * Hostname for the MongoDB
 108      */
 109     final val host = p("host")
 110
 111     /**
 112      * Username for connecting to the MongoDB
 113      */
 114     final val username = p("username")
 115
 116     /**
 117      *  Password for connecting to the MongoDB
 118      */
 119     final val password = p("password")
 120
 121     /**
 122      *  Password for connecting to the MongoDB
 123      */
 124     final val port = p("port")
 125
 126     /**
 127      *  The DB schema to use
 128      */
 129     final val dbschema = p("dbschema")
 130
 131     /**
 132      * Maximum number of open connections to MongoDB
 133      */
 134     final val connection_pool_size = p("connection.pool.size")
 135
 136   }
 137 }