Question

以下代码使用scala anorm打印所有行：

import anorm._

val query = s"select col_str, col_num from mytable"

val rowParser: RowParser[~[String, Int]] = SqlParser.str(1) ~ SqlParser.int(2)

def f(row: (String, Int)) {
  println(row._1 + "\t" + row._2)
}

val rows: List[(String, Int)] = SQL(query).as(rowParser.*).map(SqlParser.flatten)

rows.foreach(f)

然而，它需要将所有数据加载到内存中。避免加载所有数据的解决方案是使用fold，如下所示：

SQL(query).fold(Unit, ColumnAliaser.empty) { (_, r: Row) =>
  println(r[String](1) + "\t" + r[Int](2))
  Unit
}

但是，我在这里没有使用rowParser和flatten。如何修改以前的代码以使用rowParser和flatten，而无需将所有内容加载到内存中？这样的事情（注意：这段代码不起作用）：

SQL(query).as(rowParser.*).map(SqlParser.flatten).fold(Unit, ColumnAliaser.empty) { (_, row: (String, Int)) =>
  f(row)
  Unit
}

并且更难以制作隐式forEach2函数，以便我可以按如下方式运行它：

SQL(query).as(rowParser.*).map(SqlParser.flatten).forEach2(f)

OLD

我之前尝试过的一些代码：

def foreach[T, A, B](sqlQuery: SqlQuery, rowParser: RowParser[~[A, B]], f: (~[A, B]) => T) {
  val result: Either[List[Throwable], Unit.type] = sqlQuery.fold(Unit, ColumnAliaser.empty) { (_, row: Row) =>
    rowParser(row) match {
      case Success(r: ~[A, B]) =>
        f(r)
        Unit
      case Error(err) =>
        throw AnormException(err.toString)
    }
  }

  result.left.foreach { t: Seq[Throwable] =>
    t.foreach(_.printStackTrace)
    t.headOption.foreach { tt => throw tt}
  }
}

def f(row: ~[String, Int]) {
  println(row._1 + "\t" + row._2)
}

foreach(SQL(query), rowParser, f)

这很有效。但是，我需要将def f(row: (String, Int))转换为def f(row: ~[String, String])。如何在~函数中删除此f？此foreach函数还需要一个包含两列的行。如何将此概括为n列？。

Answer 1

foreach的改善是否足够好？或者你还需要更多东西吗？

object AnormForEachOps {

  import anorm._
  import java.sql.Connection

  implicit class ForEachOps(val query: SqlQuery) extends AnyVal {
    def foreach[T1, T2, R](rowParser: RowParser[~[T1, T2]], f: R => Unit)(implicit connection: Connection, fl: TupleFlattener[(T1 ~ T2) => R]): Unit = {
      query.fold(Unit, ColumnAliaser.empty) { (_, r: Row) =>
        rowParser(r).map(SqlParser.flatten) match {
          case Success(t) =>
            f(t)
            Unit
          case Error(err) =>
            throw AnormException(err.toString)
        }
        Unit
      }
    }
  }

}

所以你的例子会变成这样：

import AnormForEachOps._
val query = s"select col_str, col_num from mytable"
val rowParser = SqlParser.str(1) ~ SqlParser.int(2)
SQL(query).foreach(rowParser, (r: (String, Int)) => {
    Logger.info(r._1 + "\t" + r._2)
})

此修改的主要技巧是隐式TupleFlattener参数，它从~转换为常规元组（通过SqlParser.flatten调用）

scala anorm foreach没有将所有内容加载到内存中

1 个答案: