Merge branch 'streaming' into unisea

Author: eivindr

internal/tmpl/ktiface.tmpl

@@ -18,6 +18,9 @@ interface Queries {
   {{- if eq .Cmd ":many"}}
   fun {{.MethodName}}({{.Arg.Args}}): List<{{.Ret.Type}}>
   {{- end}}
+  {{- if eq .Cmd ":iter"}}
+  fun {{.MethodName}}(iter: ({{.Ret.Name}}: {{.Ret.Type}}) -> Unit{{ if .Arg.Args }}, {{ .Arg.Args}}{{end}})
+  {{- end}}
   {{- if eq .Cmd ":exec"}}
   fun {{.MethodName}}({{.Arg.Args}})
   {{- end}}

internal/tmpl/ktsql.tmpl

@@ -64,6 +64,27 @@ class QueriesImpl(private val conn: Connection) : Queries {
   }
 {{end}}
 
+{{if eq .Cmd ":iter"}}
+{{range .Comments}}//{{.}}
+{{end}}
+  @Throws(SQLException::class)
+  override fun {{.MethodName}}(iter: ({{.Ret.Name}}: {{.Ret.Type}}) -> Unit{{ if .Arg.Args }}, {{ .Arg.Args}}{{end}}) {
+    return conn.prepareStatement({{.ConstantName}}).use { stmt ->
+      {{.Arg.Bindings}}
+
+      val results = stmt.executeQuery()
+      while (results.next()) {
+          val ret = {{.Ret.ResultSet}}
+          try {
+            iter(ret)
+          } catch (e: Exception) {
+            throw SQLException("error calling iter function at row %d".format(results.row), e)
+          }
+      }
+    }
+  }
+{{end}}
+
 {{if eq .Cmd ":exec"}}
 {{range .Comments}}//{{.}}
 {{end}}