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

   1 /*
   2  * Copyright 2011 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
  37 package mongodb
  38
  39 import confmodel.MongoDBConfigurationModel
  40 import com.mongodb.casbah.MongoConnection
  41 import com.mongodb.{WriteConcern, ServerAddress}
  42 import gr.grnet.aquarium.util.Loggable
  43
  44 /**
  45  *
  46  * @author Christos KK Loverdos <loverdos@gmail.com>.
  47  */
  48 class MongoDBConnection(val confModel: MongoDBConfigurationModel) extends Loggable {
  49
  50   private[mongodb] lazy val _mongoConnection = {
  51     val hosts = confModel.hosts
  52     val serverAddresses = hosts.map(sacm => new ServerAddress(sacm.host, sacm.port))
  53
  54     val mongo = MongoConnection(serverAddresses)
  55     logger.info("Created MongoDB connection %s for hosts %s".format(mongo, confModel.hosts.map(h => "%s:%s".format(h.host, h.port)).mkString(", ")))
  56
  57     if(confModel.slaveOK) {
  58       mongo.slaveOk()
  59       logger.info("Set slaveOK for MongoDB connection %s".format(mongo))
  60     }
  61     val writeConcern = WriteConcern.valueOf(confModel.writeConcern)
  62     mongo.setWriteConcern(writeConcern)
  63     logger.info("Set WriteConcern %s for MongoDB connection %s".format(confModel.writeConcern, mongo))
  64
  65     mongo
  66   }
  67 }
  68
  69 object MongoDBConnection {
  70   object RCFolders {
  71     val mongodb = "mongodb"
  72   }
  73
  74   object PropFiles {
  75     val local_message_store = "local-message-store.xml"
  76     val aquarium_message_store = "aquarium-message-store.xml"
  77   }
  78
  79   object DBNames {
  80     val test = "test"
  81   }
  82
  83   object CollectionNames {
  84       val test = "test"
  85     }
  86 }