blob: a3459ff227f86cf71771e4b17620a23af41dc43a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
package lesci
import (
"context"
"database/sql"
"database/sql/driver"
"github.com/duckdb/duckdb-go/v2"
)
type AppendFunc func([]driver.Value) error
type AppendRowsFunc func(appendFunc AppendFunc) error
func AppendRows(db *sql.DB, table string, appendRowsFunc AppendRowsFunc) error {
var conn *sql.Conn
if c, err := db.Conn(context.Background()); err != nil {
return err
} else {
conn = c
}
defer conn.Close()
return conn.Raw(func(driverConn any) error {
duckConn := driverConn.(*duckdb.Conn)
var appender *duckdb.Appender
if a, err := duckdb.NewAppenderFromConn(duckConn, "", table); err != nil {
return err
} else {
appender = a
}
defer appender.Close()
if err := appendRowsFunc(func(row []driver.Value) error {
return appender.AppendRow(row...)
}); err != nil {
return err
}
return appender.Flush()
})
}
|
