首页 文章

Slick 2.0 Generic CRUD操作

提问于
浏览
6

我一直在寻找如何为公共CRUD和其他类型的操作实现通用特性,我看了thisthis并且指定的方法运行良好 .

我想要的是一个通用的插入方法,我的类目前看起来像这样(非泛型实现):

object CampaignModel {
  val campaigns = TableQuery[Campaign]

  def insert(campaign: CampaignRow)(implicit s: Session) = {
    campaigns.insert(campaign)
  }
}

到目前为止,我在第一个链接之后尝试了这个(通用实现):

trait PostgresGeneric[T <: Table[A], A]  {
  val tableReference = TableQuery[T]

  def insertGeneric(row: ? What type goes here ?)(implicit s: Session) = tableReference.insert(row)

}

当我检查 insert 方法时,看起来正确的类型应该是 T#TableElementType ,但我的知识非常基本,我无法绕过类型,我尝试 TA 并且编译器说classtype不符合特征那些 .

其他信息,这些表是使用光滑的表生成工具生成的

case class CampaignRow(id: Long, name: Option[String])

/** Table description of table campaign. Objects of this class serve as prototypes for rows in queries. */
class Campaign(tag: Tag) extends Table[CampaignRow](tag, "campaign") {
  def * = (id, name) <>(CampaignRow.tupled, CampaignRow.unapply)

  /** Maps whole row to an option. Useful for outer joins. */
  def ? = (id.?, name).shaped.<>({
    r => import r._; _1.map(_ => CampaignRow.tupled((_1.get, _2)))
  }, (_: Any) => throw new Exception("Inserting into ? projection not supported."))

  /** Database column id AutoInc, PrimaryKey */
  val id: Column[Long] = column[Long]("id", O.AutoInc, O.PrimaryKey)
  /** Database column name  */
  val name: Column[Option[String]] = column[Option[String]]("name")
}

1 回答

  • 11

    我设法让它工作,这是我的通用特性:

    import scala.slick.driver.PostgresDriver
    import scala.slick.driver.PostgresDriver.simple._
    import path.to.RichTable
    
    trait PostgresGeneric[T <: RichTable[A], A] {
    
      val tableReference: TableQuery[T]
    
      def insert(row: T#TableElementType)(implicit s: Session) = 
        tableReference.insert(row)
    
      def insertAndGetId(row: T#TableElementType)(implicit s: Session) = 
        (tableReference returning tableReference.map(_.id)) += row
    
      def deleteById(id: Long)(implicit s: Session): Boolean = 
        tableReference.filter(_.id === id).delete == 1
    
      def updateById(id: Long, row: T#TableElementType)(implicit s: Session): Boolean = 
        tableReference.filter(_.id === id).update(row) == 1
    
      def selectById(id: Long)(implicit s: Session): Option[T#TableElementType] = 
        tableReference.filter(_.id === id).firstOption
    
      def existsById(id: Long)(implicit s: Session): Boolean = {
        (for {
          row <- tableReference
          if row.id === id
        } yield row).firstOption.isDefined
      }
    }
    

    其中 RichTable 是一个带有id字段的抽象类,这个带有上限约束对于获取 T#TableElementType 的id字段很有用(有关详细信息,请参阅this):

    import scala.slick.driver.PostgresDriver.simple._
    import scala.slick.jdbc.{GetResult => GR}
    
    abstract class RichTable[T](tag: Tag, name: String) extends Table[T](tag, name) {
      val id: Column[Long] = column[Long]("id", O.PrimaryKey, O.AutoInc)
    }
    

    我的广告系列表现在看起来像这样:

    import scala.slick.driver.PostgresDriver.simple._
    import scala.slick.jdbc.{GetResult => GR}
    import scala.slick.lifted.TableQuery
    
    case class CampaignRow(id: Long, name: Option[String])
    
    class Campaign(tag: Tag) extends RichTable[CampaignRow](tag, "campaign") {
      def * = (id, name) <>(CampaignRow.tupled, CampaignRow.unapply)
    
      def ? = (id.?, name).shaped.<>({
        r => import r._; _1.map(_ => CampaignRow.tupled((_1.get, _2)))
      }, (_: Any) => throw new Exception("Inserting into ? projection not supported."))
    
      override val id: Column[Long] = column[Long]("id", O.AutoInc, O.PrimaryKey)
      val name: Column[Option[String]] = column[Option[String]]("name")
    }
    

    实现通用特征的模型如下所示:

    object CampaignModel extends PostgresGeneric[Campaign, CampaignRow] {
    
       override val tableReference: PostgresDriver.simple.TableQuery[Tables.Campaign] = 
         TableQuery[Campaign]
    
       def insertCampaign(row: CampaignRow) = {
         insert(CampaignRow(0, "test"))
       }
     }
    

相关问题