91 lines
2 KiB
Go
91 lines
2 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 (
|
|
"encoding/json"
|
|
"time"
|
|
|
|
lib "git.sr.ht/~cco/go-scopes"
|
|
"git.sr.ht/~cco/go-scopes/storage"
|
|
sql "git.sr.ht/~cco/go-scopes/storage"
|
|
)
|
|
|
|
type track struct {
|
|
trackId uint
|
|
head map[string]string
|
|
timeStamp *time.Time
|
|
data lib.Data
|
|
container *container
|
|
}
|
|
|
|
func (tr *track) TrackId() uint {
|
|
return tr.trackId
|
|
}
|
|
|
|
type container struct {
|
|
tableName string
|
|
headFields []string
|
|
indexes [][]string
|
|
storage *sql.Storage
|
|
}
|
|
|
|
func SimpleTrack(taskId, userName string, data lib.Map) track {
|
|
return track{
|
|
head: map[string]string{"taskId": taskId, "userName": userName},
|
|
data: data,
|
|
}
|
|
}
|
|
|
|
func SimpleContainer(db *sql.Storage) *container {
|
|
return &container{
|
|
tableName: "tracks",
|
|
headFields: []string{"taskId", "userName"},
|
|
indexes: [][]string{[]string{"taskId", "userName"}, []string{"userName"}},
|
|
storage: db,
|
|
}
|
|
}
|
|
|
|
func (cont *container) Insert(tr *track) uint {
|
|
var columns []string
|
|
var values []any
|
|
for _, k := range cont.headFields {
|
|
columns = append(columns, k)
|
|
values = append(values, tr.head[k])
|
|
}
|
|
columns = append(columns, "Data")
|
|
b, _ := json.Marshal(tr.data)
|
|
values = append(values, b)
|
|
db := cont.storage
|
|
data := map[string]interface{}{
|
|
"schema": db.Schema,
|
|
"tablename": cont.tableName,
|
|
"columns": columns,
|
|
}
|
|
sql := storage.BuildSql(SqlInsert, data)
|
|
if res := storage.QueryCol[uint](db, sql, values...); res != nil {
|
|
trid := res[0]
|
|
tr.trackId = trid
|
|
tr.container = cont
|
|
return trid
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func (cont *container) CreateTable() *container {
|
|
db := cont.storage
|
|
data := map[string]interface{}{
|
|
"schema": db.Schema,
|
|
"tablename": cont.tableName,
|
|
"headFields": cont.headFields,
|
|
"indexes": cont.indexes,
|
|
"params": db.Params,
|
|
}
|
|
sql := storage.BuildSql(SqlCreate, data)
|
|
//println(sql)
|
|
_, err := db.Exec(sql)
|
|
if err == nil {
|
|
return cont
|
|
}
|
|
return nil
|
|
}
|