Question

我想在Task表中插入10行以上的数据，因此我创建了这些记录的列表。除了我的表，我有一个存储库，其中包含create，update等功能。所以我可以使用create a function来添加一条记录，但我想用这个函数批量插入数据。

case class Task (
  idTask: Option[Long],
  ownerId: Long,
  name: String,
  finished: Boolean
)

class TaskTable(tag: Tag) extends Table[Task](tag, "tasks"){
  val idTask = column[Long]("id_task", O.PrimaryKey)
  val ownerId = column[Long]("owner")
  val name = column[String]("name")
  val finished = column[Boolean]("finished")

  val ownerFk = foreignKey("owner_id_fk", ownerId, TableQuery[UserTable])(_.idUser)

  def * = (idTask.?, ownerId, name, finished) <> (Task.apply _ tupled, Task.unapply)
}

object TaskTable{
  lazy val table = TableQuery[TaskTable]
}

class TaskRepository(db: Database) {
  val taskTableQuery = TableQuery[TaskTable]
  def create(task: Task): Future[Task] =
    db.run(taskTableQuery returning taskTableQuery += task)

 def createTasks(tasks: List[Task]): Future[Option[Task]] = 
    db.run(taskTableQuery ++= tasks)

  def update(task: Task): Future[Int] =
    db.run(taskTableQuery.filter(_.idTask === task.idTask).update(task))

  def delete(task: Task): Future[Int] =
    db.run(taskTableQuery.filter(_.idTask === task.idTask).delete)

  def getById(task: Task): Future[Option[Task]] =
    db.run(taskTableQuery.filter(_.idTask === task.idTask).result.headOption)
}

我试过这样的事情：

val tasks = List(
    Task(Some(1),  1,"Analyze logs with Spark", false),
    Task(Some(2),  1,"Clean and process data", false),
...
)
val createTasks = tasks.map(taskRepository.create(_))

但是这个createTasks值的类型为Unit，我无法使用db.run或Await.result运行它。由于返回类型不匹配，我的方法createTasks无法编译。那么如何为批量插入创建此方法或更改现有方法呢？我将非常感谢任何帮助！

Answer 1

如果您在编译时知道操作，请使用{{3}}链接数据库操作：

db.run(
  DBIO.seq( taskTableQuery += Task(....), 
            taskTableQuery += Task(....), 
            taskTableQuery += Task(....)...
  ).transactionally
)

否则，请使用DBIO.sequence：

val taskSeq:Seq[Task] = ... however you get the tasks
db.run( 
    DBIO.sequence( taskSeq.map(t=>taskTableQuery+=t) ).transactionally 
)

批量插入记录中的记录

1 个答案: