70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
// Package `tracking` defines a generic track (sort of record) type
|
|
// and a container type that allows storing of tracks in a SQL database.
|
|
package tracking
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"text/template"
|
|
"time"
|
|
|
|
lib "git.sr.ht/~cco/go-scopes"
|
|
sql "git.sr.ht/~cco/go-scopes/storage"
|
|
)
|
|
|
|
type track struct {
|
|
TrackId uint
|
|
Head map[string]string
|
|
TimeStamp *time.Time
|
|
Data lib.Payload
|
|
Container *container
|
|
}
|
|
|
|
type container struct {
|
|
headFields []string
|
|
indexes [][]string
|
|
storage *sql.Storage
|
|
}
|
|
|
|
func Container(db *sql.Storage) *container {
|
|
return &container{
|
|
headFields: []string{"taskId", "userName"},
|
|
indexes: [][]string{[]string{"taskId", "userName"}, []string{"userName"}},
|
|
storage: db,
|
|
}
|
|
}
|
|
|
|
func Create(cont *container) *container {
|
|
db := cont.storage
|
|
t := template.New("create_table")
|
|
t = t.Funcs(map[string]any{
|
|
"ToLower": strings.ToLower,
|
|
"Add1": func(i int) int { return i + 1 },
|
|
})
|
|
t, err := t.Parse(sql_table)
|
|
if err != nil {
|
|
db.LogErr(err, "storage.tracking.Create", fmt.Sprintf("%+v", t))
|
|
return nil
|
|
}
|
|
var out strings.Builder
|
|
data := map[string]interface{}{
|
|
"schema": db.Schema,
|
|
"tablename": "tracks",
|
|
"headFields": cont.headFields,
|
|
"indexes": cont.indexes,
|
|
"idType": db.Params["idType"],
|
|
"jsonType": db.Params["jsonType"],
|
|
}
|
|
err = t.Execute(&out, data)
|
|
if err != nil {
|
|
db.LogErr(err, "storage.tracking.Create", fmt.Sprintf("%+v", t))
|
|
return nil
|
|
}
|
|
sql := out.String()
|
|
println(sql)
|
|
_, err = db.Exec(sql)
|
|
if err == nil {
|
|
return cont
|
|
}
|
|
return nil
|
|
}
|