titan.scala

object TitanCon extends InMemory with InCassandra{

  lazy var Graph: TitanGraph

  def connect(backendType: String): Unit = {
    Graph = backendType match {

      case "inMemory" => memoryConnect()
      case "inCassandra" => cassandraConnect()

    }

  }
}

trait InMemory {

  def memoryConnect(): TitanGraph = {
        val conf = new BaseConfiguration()
        conf.setProperty("storage.backend","inmemory")
        TitanFactory.open(conf)
      }
}

trait InCassandra {
  def cassandraConnect(): TitanGraph = {
    TitanFactory.open("titan-cassandra.properties")

  }
}