de.comhix.database.config.DatabaseModule.kt Maven / Gradle / Ivy

Go to download
package de.comhix.database.config

import com.google.common.collect.Lists
import com.google.inject.AbstractModule
import com.mongodb.MongoClient
import com.mongodb.MongoCredential
import com.mongodb.ServerAddress
import org.mongodb.morphia.Datastore
import org.mongodb.morphia.Morphia

/**
 * @author Benjamin Beeker
 */
class DatabaseModule @JvmOverloads constructor(host: String,
                                               port: Int,
                                               dbName: String,
                                               user: String? = null,
                                               password: String? = null,
                                               authDatabase: String = DEFAULT_AUTH_DB) : AbstractModule() {
    private val host: String
    private val port: Int
    private val dbName: String
    private val user: String?
    private val password: String?
    private val authDatabase: String
    override fun configure() {
        val credential = MongoCredential.createCredential(user, authDatabase, password!!.toCharArray())
        val mongo = MongoClient(ServerAddress(host, port), Lists.newArrayList(credential))
        val datastore = Morphia().createDatastore(mongo, dbName)
        bind(Datastore::class.java).toInstance(datastore)
    }

    companion object {
        const val DEFAULT_AUTH_DB = "admin"
    }

    init {
        this.host = host
        this.port = port
        this.dbName = dbName
        this.user = user
        this.password = password
        this.authDatabase = authDatabase
    }
}