在Play Slick中分配动态注入的数据库名称

时间:2017-11-26 04:25:19

标签: scala playframework slick slick-3.0 play-slick

我有以下Play Slick DAO课程。请注意,数据库配置是常量control0001。 DAO有一个函数readUser,它根据用户ID读取用户:

class UsersDAO @Inject()(@NamedDatabase("control0001") 
    protected val dbConfigProvider: DatabaseConfigProvider) 
                  extends HasDatabaseConfigProvider[JdbcProfile] {

   import driver.api._

   def readUser (userid: String) = {
      val users = TableQuery[UserDB]
      val action = users.filter(_.userid === userid).result
      val future = db.run(action.asTry)
      future.map{
        case Success(s) => 
          if (s.length>0)
            Some(s(0))
          else
            None
        case Failure(e) => throw new Exception ("Failure in readUser: " + e.getMessage)
      }
   }
}

我需要数据库变量,而不是@NamedDatabase("control0001")中的常量。在应用程序中,我在control0001中配置了多个数据库(control002application.conf等)。根据变量值,我需要确定要在DAO中使用的数据库。所有数据库都相似并且具有相同的表(每个数据库中的数据不同)。

以下Play类调用DAO函数,但首先需要确定要注入的数据库名称:

class TestSlick  @Inject()(dao: UsersDAO) extends Controller  {

  def test(someCode: Int, userId: String) = Action { request =>

    val databaseName = if (someCode == 1) "control0001" else "control0002"

    // Run the method in UsersDAO accessing the database set by databaseName

    val future = dao.readUser(userId) 
    future.map { result =>
      result match {
        case Some(user) => Ok(user.firstName)
        case _ => Ok("user not found")
      }
    }
  }        
}

如何在Play Slick中实现这一目标?

1 个答案:

答案 0 :(得分:1)

在这种情况下,不要尝试使用Play的运行时依赖项注入实用程序,而是直接在DAO中使用SlickApi并将数据源名称传递给dbConfig(DbName(name))方法。要获得alter system set open_cursors = 1000 ; ,请混合SlickComponents特征:

SlickApi

然后在你的控制器中:

class UsersDAO extends SlickComponents {

  def readUser(userid: String, dbName: String) = {
    val users = TableQuery[UserDB]
    val action = users.filter(_.userid === userid).result

    val dbConfig = slickApi.dbConfig(DbName(dbName))
    val future = dbConfig.db.run(action.asTry)
    ...
  }
}