dal/pkg/facade/SQLIteShared.go
Anton Nesterov 07a61af8e5
[feat] test nodejs bindings
Signed-off-by: Anton Nesterov <anton@demiurg.io>
2024-08-20 03:36:41 +02:00

80 lines
1.6 KiB
Go

package facade
import (
"log"
"reflect"
"l12.xyz/dal/adapter"
"l12.xyz/dal/proto"
)
var db adapter.DBAdapter
func InitSQLite(pragmas []string) {
if db.Type == "" {
adapter.RegisterDialect("sqlite3", adapter.CommonDialect{})
db = adapter.DBAdapter{
Type: "sqlite3",
}
db.AfterOpen("PRAGMA journal_mode=WAL")
}
for _, pragma := range pragmas {
if pragma == "" {
continue
}
db.AfterOpen(pragma)
}
}
func HandleQuery(input *[]byte, output *[]byte) int {
InitSQLite([]string{})
req := proto.Request{}
_, err := req.UnmarshalMsg(*input)
if err != nil {
log.Println(*input)
log.Printf("failed to unmarshal request: %v", err)
return 1
}
query, err := req.Parse(adapter.GetDialect(db.Type))
if err != nil {
log.Printf("failed to parse request: %v", err)
return 1
}
if query.Exec {
result, err := db.Exec(query)
if err != nil {
log.Printf("failed to exec query: %v", err)
return 1
}
ra, _ := result.RowsAffected()
la, _ := result.LastInsertId()
res := proto.Response{
Id: 0,
RowsAffected: ra,
LastInsertId: la,
}
*output, _ = res.MarshalMsg(nil)
return 0
}
rows, err := db.Query(query)
if err != nil {
log.Printf("failed to query: %v", err)
return 1
}
columns, _ := rows.Columns()
types, _ := rows.ColumnTypes()
cols, _ := proto.MarshalRow(columns)
*output = append(*output, cols...)
for rows.Next() {
data := make([]interface{}, len(columns))
for i := range data {
typ := reflect.New(types[i].ScanType()).Interface()
data[i] = &typ
}
rows.Scan(data...)
cols, _ := proto.MarshalRow(data)
*output = append(*output, cols...)
}
return 0
}