storj/pkg/accounting/dbx/accounting.dbx.go
Bill Thorp b2a8a10b16
redo tally query (#742)
* redo tally query

* broken, sharing w/ Jen

* tally tests added, failiing

* fix txn issue in tally

* fix lint
2018-12-05 09:03:23 -05:00

2031 lines
51 KiB
Go

// AUTOGENERATED BY gopkg.in/spacemonkeygo/dbx.v1
// DO NOT EDIT.
package dbx
import (
"bytes"
"context"
"database/sql"
"errors"
"fmt"
"reflect"
"regexp"
"strconv"
"strings"
"sync"
"time"
"unicode"
"github.com/lib/pq"
"github.com/mattn/go-sqlite3"
)
// Prevent conditional imports from causing build failures
var _ = strconv.Itoa
var _ = strings.LastIndex
var _ = fmt.Sprint
var _ sync.Mutex
var (
WrapErr = func(err *Error) error { return err }
Logger func(format string, args ...interface{})
errTooManyRows = errors.New("too many rows")
errUnsupportedDriver = errors.New("unsupported driver")
errEmptyUpdate = errors.New("empty update")
)
func logError(format string, args ...interface{}) {
if Logger != nil {
Logger(format, args...)
}
}
type ErrorCode int
const (
ErrorCode_Unknown ErrorCode = iota
ErrorCode_UnsupportedDriver
ErrorCode_NoRows
ErrorCode_TxDone
ErrorCode_TooManyRows
ErrorCode_ConstraintViolation
ErrorCode_EmptyUpdate
)
type Error struct {
Err error
Code ErrorCode
Driver string
Constraint string
QuerySuffix string
}
func (e *Error) Error() string {
return e.Err.Error()
}
func wrapErr(e *Error) error {
if WrapErr == nil {
return e
}
return WrapErr(e)
}
func makeErr(err error) error {
if err == nil {
return nil
}
e := &Error{Err: err}
switch err {
case sql.ErrNoRows:
e.Code = ErrorCode_NoRows
case sql.ErrTxDone:
e.Code = ErrorCode_TxDone
}
return wrapErr(e)
}
func unsupportedDriver(driver string) error {
return wrapErr(&Error{
Err: errUnsupportedDriver,
Code: ErrorCode_UnsupportedDriver,
Driver: driver,
})
}
func emptyUpdate() error {
return wrapErr(&Error{
Err: errEmptyUpdate,
Code: ErrorCode_EmptyUpdate,
})
}
func tooManyRows(query_suffix string) error {
return wrapErr(&Error{
Err: errTooManyRows,
Code: ErrorCode_TooManyRows,
QuerySuffix: query_suffix,
})
}
func constraintViolation(err error, constraint string) error {
return wrapErr(&Error{
Err: err,
Code: ErrorCode_ConstraintViolation,
Constraint: constraint,
})
}
type driver interface {
Exec(query string, args ...interface{}) (sql.Result, error)
Query(query string, args ...interface{}) (*sql.Rows, error)
QueryRow(query string, args ...interface{}) *sql.Row
}
var (
notAPointer = errors.New("destination not a pointer")
lossyConversion = errors.New("lossy conversion")
)
type DB struct {
*sql.DB
dbMethods
Hooks struct {
Now func() time.Time
}
}
func Open(driver, source string) (db *DB, err error) {
var sql_db *sql.DB
switch driver {
case "postgres":
sql_db, err = openpostgres(source)
case "sqlite3":
sql_db, err = opensqlite3(source)
default:
return nil, unsupportedDriver(driver)
}
if err != nil {
return nil, makeErr(err)
}
defer func(sql_db *sql.DB) {
if err != nil {
sql_db.Close()
}
}(sql_db)
if err := sql_db.Ping(); err != nil {
return nil, makeErr(err)
}
db = &DB{
DB: sql_db,
}
db.Hooks.Now = time.Now
switch driver {
case "postgres":
db.dbMethods = newpostgres(db)
case "sqlite3":
db.dbMethods = newsqlite3(db)
default:
return nil, unsupportedDriver(driver)
}
return db, nil
}
func (obj *DB) Close() (err error) {
return obj.makeErr(obj.DB.Close())
}
func (obj *DB) Open(ctx context.Context) (*Tx, error) {
tx, err := obj.DB.Begin()
if err != nil {
return nil, obj.makeErr(err)
}
return &Tx{
Tx: tx,
txMethods: obj.wrapTx(tx),
}, nil
}
func (obj *DB) NewRx() *Rx {
return &Rx{db: obj}
}
func DeleteAll(ctx context.Context, db *DB) (int64, error) {
tx, err := db.Open(ctx)
if err != nil {
return 0, err
}
defer func() {
if err == nil {
err = db.makeErr(tx.Commit())
return
}
if err_rollback := tx.Rollback(); err_rollback != nil {
logError("delete-all: rollback failed: %v", db.makeErr(err_rollback))
}
}()
return tx.deleteAll(ctx)
}
type Tx struct {
Tx *sql.Tx
txMethods
}
type dialectTx struct {
tx *sql.Tx
}
func (tx *dialectTx) Commit() (err error) {
return makeErr(tx.tx.Commit())
}
func (tx *dialectTx) Rollback() (err error) {
return makeErr(tx.tx.Rollback())
}
type postgresImpl struct {
db *DB
dialect __sqlbundle_postgres
driver driver
}
func (obj *postgresImpl) Rebind(s string) string {
return obj.dialect.Rebind(s)
}
func (obj *postgresImpl) logStmt(stmt string, args ...interface{}) {
postgresLogStmt(stmt, args...)
}
func (obj *postgresImpl) makeErr(err error) error {
constraint, ok := obj.isConstraintError(err)
if ok {
return constraintViolation(err, constraint)
}
return makeErr(err)
}
type postgresDB struct {
db *DB
*postgresImpl
}
func newpostgres(db *DB) *postgresDB {
return &postgresDB{
db: db,
postgresImpl: &postgresImpl{
db: db,
driver: db.DB,
},
}
}
func (obj *postgresDB) Schema() string {
return `CREATE TABLE aggregates (
node_id text NOT NULL,
start_time timestamp with time zone NOT NULL,
interval bigint NOT NULL,
created_at timestamp with time zone NOT NULL,
updated_at timestamp with time zone NOT NULL,
PRIMARY KEY ( node_id )
);
CREATE TABLE granulars (
node_id text NOT NULL,
start_time timestamp with time zone NOT NULL,
end_time timestamp with time zone NOT NULL,
data_total bigint NOT NULL,
created_at timestamp with time zone NOT NULL,
updated_at timestamp with time zone NOT NULL,
PRIMARY KEY ( node_id )
);
CREATE TABLE timestamps (
name text NOT NULL,
value timestamp with time zone NOT NULL,
PRIMARY KEY ( name )
);`
}
func (obj *postgresDB) wrapTx(tx *sql.Tx) txMethods {
return &postgresTx{
dialectTx: dialectTx{tx: tx},
postgresImpl: &postgresImpl{
db: obj.db,
driver: tx,
},
}
}
type postgresTx struct {
dialectTx
*postgresImpl
}
func postgresLogStmt(stmt string, args ...interface{}) {
// TODO: render placeholders
if Logger != nil {
out := fmt.Sprintf("stmt: %s\nargs: %v\n", stmt, pretty(args))
Logger(out)
}
}
type sqlite3Impl struct {
db *DB
dialect __sqlbundle_sqlite3
driver driver
}
func (obj *sqlite3Impl) Rebind(s string) string {
return obj.dialect.Rebind(s)
}
func (obj *sqlite3Impl) logStmt(stmt string, args ...interface{}) {
sqlite3LogStmt(stmt, args...)
}
func (obj *sqlite3Impl) makeErr(err error) error {
constraint, ok := obj.isConstraintError(err)
if ok {
return constraintViolation(err, constraint)
}
return makeErr(err)
}
type sqlite3DB struct {
db *DB
*sqlite3Impl
}
func newsqlite3(db *DB) *sqlite3DB {
return &sqlite3DB{
db: db,
sqlite3Impl: &sqlite3Impl{
db: db,
driver: db.DB,
},
}
}
func (obj *sqlite3DB) Schema() string {
return `CREATE TABLE aggregates (
node_id TEXT NOT NULL,
start_time TIMESTAMP NOT NULL,
interval INTEGER NOT NULL,
created_at TIMESTAMP NOT NULL,
updated_at TIMESTAMP NOT NULL,
PRIMARY KEY ( node_id )
);
CREATE TABLE granulars (
node_id TEXT NOT NULL,
start_time TIMESTAMP NOT NULL,
end_time TIMESTAMP NOT NULL,
data_total INTEGER NOT NULL,
created_at TIMESTAMP NOT NULL,
updated_at TIMESTAMP NOT NULL,
PRIMARY KEY ( node_id )
);
CREATE TABLE timestamps (
name TEXT NOT NULL,
value TIMESTAMP NOT NULL,
PRIMARY KEY ( name )
);`
}
func (obj *sqlite3DB) wrapTx(tx *sql.Tx) txMethods {
return &sqlite3Tx{
dialectTx: dialectTx{tx: tx},
sqlite3Impl: &sqlite3Impl{
db: obj.db,
driver: tx,
},
}
}
type sqlite3Tx struct {
dialectTx
*sqlite3Impl
}
func sqlite3LogStmt(stmt string, args ...interface{}) {
// TODO: render placeholders
if Logger != nil {
out := fmt.Sprintf("stmt: %s\nargs: %v\n", stmt, pretty(args))
Logger(out)
}
}
type pretty []interface{}
func (p pretty) Format(f fmt.State, c rune) {
fmt.Fprint(f, "[")
nextval:
for i, val := range p {
if i > 0 {
fmt.Fprint(f, ", ")
}
rv := reflect.ValueOf(val)
if rv.Kind() == reflect.Ptr {
if rv.IsNil() {
fmt.Fprint(f, "NULL")
continue
}
val = rv.Elem().Interface()
}
switch v := val.(type) {
case string:
fmt.Fprintf(f, "%q", v)
case time.Time:
fmt.Fprintf(f, "%s", v.Format(time.RFC3339Nano))
case []byte:
for _, b := range v {
if !unicode.IsPrint(rune(b)) {
fmt.Fprintf(f, "%#x", v)
continue nextval
}
}
fmt.Fprintf(f, "%q", v)
default:
fmt.Fprintf(f, "%v", v)
}
}
fmt.Fprint(f, "]")
}
type Aggregate struct {
NodeId string
StartTime time.Time
Interval int64
CreatedAt time.Time
UpdatedAt time.Time
}
func (Aggregate) _Table() string { return "aggregates" }
type Aggregate_Update_Fields struct {
StartTime Aggregate_StartTime_Field
Interval Aggregate_Interval_Field
}
type Aggregate_NodeId_Field struct {
_set bool
_value string
}
func Aggregate_NodeId(v string) Aggregate_NodeId_Field {
return Aggregate_NodeId_Field{_set: true, _value: v}
}
func (f Aggregate_NodeId_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Aggregate_NodeId_Field) _Column() string { return "node_id" }
type Aggregate_StartTime_Field struct {
_set bool
_value time.Time
}
func Aggregate_StartTime(v time.Time) Aggregate_StartTime_Field {
return Aggregate_StartTime_Field{_set: true, _value: v}
}
func (f Aggregate_StartTime_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Aggregate_StartTime_Field) _Column() string { return "start_time" }
type Aggregate_Interval_Field struct {
_set bool
_value int64
}
func Aggregate_Interval(v int64) Aggregate_Interval_Field {
return Aggregate_Interval_Field{_set: true, _value: v}
}
func (f Aggregate_Interval_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Aggregate_Interval_Field) _Column() string { return "interval" }
type Aggregate_CreatedAt_Field struct {
_set bool
_value time.Time
}
func Aggregate_CreatedAt(v time.Time) Aggregate_CreatedAt_Field {
return Aggregate_CreatedAt_Field{_set: true, _value: v}
}
func (f Aggregate_CreatedAt_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Aggregate_CreatedAt_Field) _Column() string { return "created_at" }
type Aggregate_UpdatedAt_Field struct {
_set bool
_value time.Time
}
func Aggregate_UpdatedAt(v time.Time) Aggregate_UpdatedAt_Field {
return Aggregate_UpdatedAt_Field{_set: true, _value: v}
}
func (f Aggregate_UpdatedAt_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Aggregate_UpdatedAt_Field) _Column() string { return "updated_at" }
type Granular struct {
NodeId string
StartTime time.Time
EndTime time.Time
DataTotal int64
CreatedAt time.Time
UpdatedAt time.Time
}
func (Granular) _Table() string { return "granulars" }
type Granular_Update_Fields struct {
StartTime Granular_StartTime_Field
EndTime Granular_EndTime_Field
DataTotal Granular_DataTotal_Field
}
type Granular_NodeId_Field struct {
_set bool
_value string
}
func Granular_NodeId(v string) Granular_NodeId_Field {
return Granular_NodeId_Field{_set: true, _value: v}
}
func (f Granular_NodeId_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Granular_NodeId_Field) _Column() string { return "node_id" }
type Granular_StartTime_Field struct {
_set bool
_value time.Time
}
func Granular_StartTime(v time.Time) Granular_StartTime_Field {
return Granular_StartTime_Field{_set: true, _value: v}
}
func (f Granular_StartTime_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Granular_StartTime_Field) _Column() string { return "start_time" }
type Granular_EndTime_Field struct {
_set bool
_value time.Time
}
func Granular_EndTime(v time.Time) Granular_EndTime_Field {
return Granular_EndTime_Field{_set: true, _value: v}
}
func (f Granular_EndTime_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Granular_EndTime_Field) _Column() string { return "end_time" }
type Granular_DataTotal_Field struct {
_set bool
_value int64
}
func Granular_DataTotal(v int64) Granular_DataTotal_Field {
return Granular_DataTotal_Field{_set: true, _value: v}
}
func (f Granular_DataTotal_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Granular_DataTotal_Field) _Column() string { return "data_total" }
type Granular_CreatedAt_Field struct {
_set bool
_value time.Time
}
func Granular_CreatedAt(v time.Time) Granular_CreatedAt_Field {
return Granular_CreatedAt_Field{_set: true, _value: v}
}
func (f Granular_CreatedAt_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Granular_CreatedAt_Field) _Column() string { return "created_at" }
type Granular_UpdatedAt_Field struct {
_set bool
_value time.Time
}
func Granular_UpdatedAt(v time.Time) Granular_UpdatedAt_Field {
return Granular_UpdatedAt_Field{_set: true, _value: v}
}
func (f Granular_UpdatedAt_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Granular_UpdatedAt_Field) _Column() string { return "updated_at" }
type Timestamps struct {
Name string
Value time.Time
}
func (Timestamps) _Table() string { return "timestamps" }
type Timestamps_Update_Fields struct {
Value Timestamps_Value_Field
}
type Timestamps_Name_Field struct {
_set bool
_value string
}
func Timestamps_Name(v string) Timestamps_Name_Field {
return Timestamps_Name_Field{_set: true, _value: v}
}
func (f Timestamps_Name_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Timestamps_Name_Field) _Column() string { return "name" }
type Timestamps_Value_Field struct {
_set bool
_value time.Time
}
func Timestamps_Value(v time.Time) Timestamps_Value_Field {
return Timestamps_Value_Field{_set: true, _value: v}
}
func (f Timestamps_Value_Field) value() interface{} {
if !f._set {
return nil
}
return f._value
}
func (Timestamps_Value_Field) _Column() string { return "value" }
func toUTC(t time.Time) time.Time {
return t.UTC()
}
func toDate(t time.Time) time.Time {
// keep up the minute portion so that translations between timezones will
// continue to reflect properly.
return t.Truncate(time.Minute)
}
//
// runtime support for building sql statements
//
type __sqlbundle_SQL interface {
Render() string
private()
}
type __sqlbundle_Dialect interface {
Rebind(sql string) string
}
type __sqlbundle_RenderOp int
const (
__sqlbundle_NoFlatten __sqlbundle_RenderOp = iota
__sqlbundle_NoTerminate
)
func __sqlbundle_Render(dialect __sqlbundle_Dialect, sql __sqlbundle_SQL, ops ...__sqlbundle_RenderOp) string {
out := sql.Render()
flatten := true
terminate := true
for _, op := range ops {
switch op {
case __sqlbundle_NoFlatten:
flatten = false
case __sqlbundle_NoTerminate:
terminate = false
}
}
if flatten {
out = __sqlbundle_flattenSQL(out)
}
if terminate {
out += ";"
}
return dialect.Rebind(out)
}
var __sqlbundle_reSpace = regexp.MustCompile(`\s+`)
func __sqlbundle_flattenSQL(s string) string {
return strings.TrimSpace(__sqlbundle_reSpace.ReplaceAllString(s, " "))
}
// this type is specially named to match up with the name returned by the
// dialect impl in the sql package.
type __sqlbundle_postgres struct{}
func (p __sqlbundle_postgres) Rebind(sql string) string {
out := make([]byte, 0, len(sql)+10)
j := 1
for i := 0; i < len(sql); i++ {
ch := sql[i]
if ch != '?' {
out = append(out, ch)
continue
}
out = append(out, '$')
out = append(out, strconv.Itoa(j)...)
j++
}
return string(out)
}
// this type is specially named to match up with the name returned by the
// dialect impl in the sql package.
type __sqlbundle_sqlite3 struct{}
func (s __sqlbundle_sqlite3) Rebind(sql string) string {
return sql
}
type __sqlbundle_Literal string
func (__sqlbundle_Literal) private() {}
func (l __sqlbundle_Literal) Render() string { return string(l) }
type __sqlbundle_Literals struct {
Join string
SQLs []__sqlbundle_SQL
}
func (__sqlbundle_Literals) private() {}
func (l __sqlbundle_Literals) Render() string {
var out bytes.Buffer
first := true
for _, sql := range l.SQLs {
if sql == nil {
continue
}
if !first {
out.WriteString(l.Join)
}
first = false
out.WriteString(sql.Render())
}
return out.String()
}
type __sqlbundle_Condition struct {
// set at compile/embed time
Name string
Left string
Equal bool
Right string
// set at runtime
Null bool
}
func (*__sqlbundle_Condition) private() {}
func (c *__sqlbundle_Condition) Render() string {
switch {
case c.Equal && c.Null:
return c.Left + " is null"
case c.Equal && !c.Null:
return c.Left + " = " + c.Right
case !c.Equal && c.Null:
return c.Left + " is not null"
case !c.Equal && !c.Null:
return c.Left + " != " + c.Right
default:
panic("unhandled case")
}
}
type __sqlbundle_Hole struct {
// set at compiile/embed time
Name string
// set at runtime
SQL __sqlbundle_SQL
}
func (*__sqlbundle_Hole) private() {}
func (h *__sqlbundle_Hole) Render() string { return h.SQL.Render() }
//
// end runtime support for building sql statements
//
type Value_Row struct {
Value time.Time
}
func (obj *postgresImpl) Create_Timestamps(ctx context.Context,
timestamps_name Timestamps_Name_Field,
timestamps_value Timestamps_Value_Field) (
timestamps *Timestamps, err error) {
__name_val := timestamps_name.value()
__value_val := timestamps_value.value()
var __embed_stmt = __sqlbundle_Literal("INSERT INTO timestamps ( name, value ) VALUES ( ?, ? ) RETURNING timestamps.name, timestamps.value")
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __name_val, __value_val)
timestamps = &Timestamps{}
err = obj.driver.QueryRow(__stmt, __name_val, __value_val).Scan(&timestamps.Name, &timestamps.Value)
if err != nil {
return nil, obj.makeErr(err)
}
return timestamps, nil
}
func (obj *postgresImpl) Create_Aggregate(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field,
aggregate_start_time Aggregate_StartTime_Field,
aggregate_interval Aggregate_Interval_Field) (
aggregate *Aggregate, err error) {
__now := obj.db.Hooks.Now().UTC()
__node_id_val := aggregate_node_id.value()
__start_time_val := aggregate_start_time.value()
__interval_val := aggregate_interval.value()
__created_at_val := __now
__updated_at_val := __now
var __embed_stmt = __sqlbundle_Literal("INSERT INTO aggregates ( node_id, start_time, interval, created_at, updated_at ) VALUES ( ?, ?, ?, ?, ? ) RETURNING aggregates.node_id, aggregates.start_time, aggregates.interval, aggregates.created_at, aggregates.updated_at")
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __node_id_val, __start_time_val, __interval_val, __created_at_val, __updated_at_val)
aggregate = &Aggregate{}
err = obj.driver.QueryRow(__stmt, __node_id_val, __start_time_val, __interval_val, __created_at_val, __updated_at_val).Scan(&aggregate.NodeId, &aggregate.StartTime, &aggregate.Interval, &aggregate.CreatedAt, &aggregate.UpdatedAt)
if err != nil {
return nil, obj.makeErr(err)
}
return aggregate, nil
}
func (obj *postgresImpl) Create_Granular(ctx context.Context,
granular_node_id Granular_NodeId_Field,
granular_start_time Granular_StartTime_Field,
granular_end_time Granular_EndTime_Field,
granular_data_total Granular_DataTotal_Field) (
granular *Granular, err error) {
__now := obj.db.Hooks.Now().UTC()
__node_id_val := granular_node_id.value()
__start_time_val := granular_start_time.value()
__end_time_val := granular_end_time.value()
__data_total_val := granular_data_total.value()
__created_at_val := __now
__updated_at_val := __now
var __embed_stmt = __sqlbundle_Literal("INSERT INTO granulars ( node_id, start_time, end_time, data_total, created_at, updated_at ) VALUES ( ?, ?, ?, ?, ?, ? ) RETURNING granulars.node_id, granulars.start_time, granulars.end_time, granulars.data_total, granulars.created_at, granulars.updated_at")
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __node_id_val, __start_time_val, __end_time_val, __data_total_val, __created_at_val, __updated_at_val)
granular = &Granular{}
err = obj.driver.QueryRow(__stmt, __node_id_val, __start_time_val, __end_time_val, __data_total_val, __created_at_val, __updated_at_val).Scan(&granular.NodeId, &granular.StartTime, &granular.EndTime, &granular.DataTotal, &granular.CreatedAt, &granular.UpdatedAt)
if err != nil {
return nil, obj.makeErr(err)
}
return granular, nil
}
func (obj *postgresImpl) Find_Timestamps_Value_By_Name(ctx context.Context,
timestamps_name Timestamps_Name_Field) (
row *Value_Row, err error) {
var __embed_stmt = __sqlbundle_Literal("SELECT timestamps.value FROM timestamps WHERE timestamps.name = ?")
var __values []interface{}
__values = append(__values, timestamps_name.value())
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
row = &Value_Row{}
err = obj.driver.QueryRow(__stmt, __values...).Scan(&row.Value)
if err == sql.ErrNoRows {
return nil, nil
}
if err != nil {
return nil, obj.makeErr(err)
}
return row, nil
}
func (obj *postgresImpl) Get_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field) (
aggregate *Aggregate, err error) {
var __embed_stmt = __sqlbundle_Literal("SELECT aggregates.node_id, aggregates.start_time, aggregates.interval, aggregates.created_at, aggregates.updated_at FROM aggregates WHERE aggregates.node_id = ?")
var __values []interface{}
__values = append(__values, aggregate_node_id.value())
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
aggregate = &Aggregate{}
err = obj.driver.QueryRow(__stmt, __values...).Scan(&aggregate.NodeId, &aggregate.StartTime, &aggregate.Interval, &aggregate.CreatedAt, &aggregate.UpdatedAt)
if err != nil {
return nil, obj.makeErr(err)
}
return aggregate, nil
}
func (obj *postgresImpl) Get_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field) (
granular *Granular, err error) {
var __embed_stmt = __sqlbundle_Literal("SELECT granulars.node_id, granulars.start_time, granulars.end_time, granulars.data_total, granulars.created_at, granulars.updated_at FROM granulars WHERE granulars.node_id = ?")
var __values []interface{}
__values = append(__values, granular_node_id.value())
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
granular = &Granular{}
err = obj.driver.QueryRow(__stmt, __values...).Scan(&granular.NodeId, &granular.StartTime, &granular.EndTime, &granular.DataTotal, &granular.CreatedAt, &granular.UpdatedAt)
if err != nil {
return nil, obj.makeErr(err)
}
return granular, nil
}
func (obj *postgresImpl) Update_Timestamps_By_Name(ctx context.Context,
timestamps_name Timestamps_Name_Field,
update Timestamps_Update_Fields) (
timestamps *Timestamps, err error) {
var __sets = &__sqlbundle_Hole{}
var __embed_stmt = __sqlbundle_Literals{Join: "", SQLs: []__sqlbundle_SQL{__sqlbundle_Literal("UPDATE timestamps SET "), __sets, __sqlbundle_Literal(" WHERE timestamps.name = ? RETURNING timestamps.name, timestamps.value")}}
__sets_sql := __sqlbundle_Literals{Join: ", "}
var __values []interface{}
var __args []interface{}
if update.Value._set {
__values = append(__values, update.Value.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("value = ?"))
}
if len(__sets_sql.SQLs) == 0 {
return nil, emptyUpdate()
}
__args = append(__args, timestamps_name.value())
__values = append(__values, __args...)
__sets.SQL = __sets_sql
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
timestamps = &Timestamps{}
err = obj.driver.QueryRow(__stmt, __values...).Scan(&timestamps.Name, &timestamps.Value)
if err == sql.ErrNoRows {
return nil, nil
}
if err != nil {
return nil, obj.makeErr(err)
}
return timestamps, nil
}
func (obj *postgresImpl) Update_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field,
update Aggregate_Update_Fields) (
aggregate *Aggregate, err error) {
var __sets = &__sqlbundle_Hole{}
var __embed_stmt = __sqlbundle_Literals{Join: "", SQLs: []__sqlbundle_SQL{__sqlbundle_Literal("UPDATE aggregates SET "), __sets, __sqlbundle_Literal(" WHERE aggregates.node_id = ? RETURNING aggregates.node_id, aggregates.start_time, aggregates.interval, aggregates.created_at, aggregates.updated_at")}}
__sets_sql := __sqlbundle_Literals{Join: ", "}
var __values []interface{}
var __args []interface{}
if update.StartTime._set {
__values = append(__values, update.StartTime.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("start_time = ?"))
}
if update.Interval._set {
__values = append(__values, update.Interval.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("interval = ?"))
}
__now := obj.db.Hooks.Now().UTC()
__values = append(__values, __now)
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("updated_at = ?"))
__args = append(__args, aggregate_node_id.value())
__values = append(__values, __args...)
__sets.SQL = __sets_sql
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
aggregate = &Aggregate{}
err = obj.driver.QueryRow(__stmt, __values...).Scan(&aggregate.NodeId, &aggregate.StartTime, &aggregate.Interval, &aggregate.CreatedAt, &aggregate.UpdatedAt)
if err == sql.ErrNoRows {
return nil, nil
}
if err != nil {
return nil, obj.makeErr(err)
}
return aggregate, nil
}
func (obj *postgresImpl) Update_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field,
update Granular_Update_Fields) (
granular *Granular, err error) {
var __sets = &__sqlbundle_Hole{}
var __embed_stmt = __sqlbundle_Literals{Join: "", SQLs: []__sqlbundle_SQL{__sqlbundle_Literal("UPDATE granulars SET "), __sets, __sqlbundle_Literal(" WHERE granulars.node_id = ? RETURNING granulars.node_id, granulars.start_time, granulars.end_time, granulars.data_total, granulars.created_at, granulars.updated_at")}}
__sets_sql := __sqlbundle_Literals{Join: ", "}
var __values []interface{}
var __args []interface{}
if update.StartTime._set {
__values = append(__values, update.StartTime.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("start_time = ?"))
}
if update.EndTime._set {
__values = append(__values, update.EndTime.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("end_time = ?"))
}
if update.DataTotal._set {
__values = append(__values, update.DataTotal.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("data_total = ?"))
}
__now := obj.db.Hooks.Now().UTC()
__values = append(__values, __now)
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("updated_at = ?"))
__args = append(__args, granular_node_id.value())
__values = append(__values, __args...)
__sets.SQL = __sets_sql
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
granular = &Granular{}
err = obj.driver.QueryRow(__stmt, __values...).Scan(&granular.NodeId, &granular.StartTime, &granular.EndTime, &granular.DataTotal, &granular.CreatedAt, &granular.UpdatedAt)
if err == sql.ErrNoRows {
return nil, nil
}
if err != nil {
return nil, obj.makeErr(err)
}
return granular, nil
}
func (obj *postgresImpl) Delete_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field) (
deleted bool, err error) {
var __embed_stmt = __sqlbundle_Literal("DELETE FROM aggregates WHERE aggregates.node_id = ?")
var __values []interface{}
__values = append(__values, aggregate_node_id.value())
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
__res, err := obj.driver.Exec(__stmt, __values...)
if err != nil {
return false, obj.makeErr(err)
}
__count, err := __res.RowsAffected()
if err != nil {
return false, obj.makeErr(err)
}
return __count > 0, nil
}
func (obj *postgresImpl) Delete_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field) (
deleted bool, err error) {
var __embed_stmt = __sqlbundle_Literal("DELETE FROM granulars WHERE granulars.node_id = ?")
var __values []interface{}
__values = append(__values, granular_node_id.value())
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
__res, err := obj.driver.Exec(__stmt, __values...)
if err != nil {
return false, obj.makeErr(err)
}
__count, err := __res.RowsAffected()
if err != nil {
return false, obj.makeErr(err)
}
return __count > 0, nil
}
func (impl postgresImpl) isConstraintError(err error) (
constraint string, ok bool) {
if e, ok := err.(*pq.Error); ok {
if e.Code.Class() == "23" {
return e.Constraint, true
}
}
return "", false
}
func (obj *postgresImpl) deleteAll(ctx context.Context) (count int64, err error) {
var __res sql.Result
var __count int64
__res, err = obj.driver.Exec("DELETE FROM timestamps;")
if err != nil {
return 0, obj.makeErr(err)
}
__count, err = __res.RowsAffected()
if err != nil {
return 0, obj.makeErr(err)
}
count += __count
__res, err = obj.driver.Exec("DELETE FROM granulars;")
if err != nil {
return 0, obj.makeErr(err)
}
__count, err = __res.RowsAffected()
if err != nil {
return 0, obj.makeErr(err)
}
count += __count
__res, err = obj.driver.Exec("DELETE FROM aggregates;")
if err != nil {
return 0, obj.makeErr(err)
}
__count, err = __res.RowsAffected()
if err != nil {
return 0, obj.makeErr(err)
}
count += __count
return count, nil
}
func (obj *sqlite3Impl) Create_Timestamps(ctx context.Context,
timestamps_name Timestamps_Name_Field,
timestamps_value Timestamps_Value_Field) (
timestamps *Timestamps, err error) {
__name_val := timestamps_name.value()
__value_val := timestamps_value.value()
var __embed_stmt = __sqlbundle_Literal("INSERT INTO timestamps ( name, value ) VALUES ( ?, ? )")
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __name_val, __value_val)
__res, err := obj.driver.Exec(__stmt, __name_val, __value_val)
if err != nil {
return nil, obj.makeErr(err)
}
__pk, err := __res.LastInsertId()
if err != nil {
return nil, obj.makeErr(err)
}
return obj.getLastTimestamps(ctx, __pk)
}
func (obj *sqlite3Impl) Create_Aggregate(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field,
aggregate_start_time Aggregate_StartTime_Field,
aggregate_interval Aggregate_Interval_Field) (
aggregate *Aggregate, err error) {
__now := obj.db.Hooks.Now().UTC()
__node_id_val := aggregate_node_id.value()
__start_time_val := aggregate_start_time.value()
__interval_val := aggregate_interval.value()
__created_at_val := __now
__updated_at_val := __now
var __embed_stmt = __sqlbundle_Literal("INSERT INTO aggregates ( node_id, start_time, interval, created_at, updated_at ) VALUES ( ?, ?, ?, ?, ? )")
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __node_id_val, __start_time_val, __interval_val, __created_at_val, __updated_at_val)
__res, err := obj.driver.Exec(__stmt, __node_id_val, __start_time_val, __interval_val, __created_at_val, __updated_at_val)
if err != nil {
return nil, obj.makeErr(err)
}
__pk, err := __res.LastInsertId()
if err != nil {
return nil, obj.makeErr(err)
}
return obj.getLastAggregate(ctx, __pk)
}
func (obj *sqlite3Impl) Create_Granular(ctx context.Context,
granular_node_id Granular_NodeId_Field,
granular_start_time Granular_StartTime_Field,
granular_end_time Granular_EndTime_Field,
granular_data_total Granular_DataTotal_Field) (
granular *Granular, err error) {
__now := obj.db.Hooks.Now().UTC()
__node_id_val := granular_node_id.value()
__start_time_val := granular_start_time.value()
__end_time_val := granular_end_time.value()
__data_total_val := granular_data_total.value()
__created_at_val := __now
__updated_at_val := __now
var __embed_stmt = __sqlbundle_Literal("INSERT INTO granulars ( node_id, start_time, end_time, data_total, created_at, updated_at ) VALUES ( ?, ?, ?, ?, ?, ? )")
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __node_id_val, __start_time_val, __end_time_val, __data_total_val, __created_at_val, __updated_at_val)
__res, err := obj.driver.Exec(__stmt, __node_id_val, __start_time_val, __end_time_val, __data_total_val, __created_at_val, __updated_at_val)
if err != nil {
return nil, obj.makeErr(err)
}
__pk, err := __res.LastInsertId()
if err != nil {
return nil, obj.makeErr(err)
}
return obj.getLastGranular(ctx, __pk)
}
func (obj *sqlite3Impl) Find_Timestamps_Value_By_Name(ctx context.Context,
timestamps_name Timestamps_Name_Field) (
row *Value_Row, err error) {
var __embed_stmt = __sqlbundle_Literal("SELECT timestamps.value FROM timestamps WHERE timestamps.name = ?")
var __values []interface{}
__values = append(__values, timestamps_name.value())
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
row = &Value_Row{}
err = obj.driver.QueryRow(__stmt, __values...).Scan(&row.Value)
if err == sql.ErrNoRows {
return nil, nil
}
if err != nil {
return nil, obj.makeErr(err)
}
return row, nil
}
func (obj *sqlite3Impl) Get_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field) (
aggregate *Aggregate, err error) {
var __embed_stmt = __sqlbundle_Literal("SELECT aggregates.node_id, aggregates.start_time, aggregates.interval, aggregates.created_at, aggregates.updated_at FROM aggregates WHERE aggregates.node_id = ?")
var __values []interface{}
__values = append(__values, aggregate_node_id.value())
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
aggregate = &Aggregate{}
err = obj.driver.QueryRow(__stmt, __values...).Scan(&aggregate.NodeId, &aggregate.StartTime, &aggregate.Interval, &aggregate.CreatedAt, &aggregate.UpdatedAt)
if err != nil {
return nil, obj.makeErr(err)
}
return aggregate, nil
}
func (obj *sqlite3Impl) Get_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field) (
granular *Granular, err error) {
var __embed_stmt = __sqlbundle_Literal("SELECT granulars.node_id, granulars.start_time, granulars.end_time, granulars.data_total, granulars.created_at, granulars.updated_at FROM granulars WHERE granulars.node_id = ?")
var __values []interface{}
__values = append(__values, granular_node_id.value())
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
granular = &Granular{}
err = obj.driver.QueryRow(__stmt, __values...).Scan(&granular.NodeId, &granular.StartTime, &granular.EndTime, &granular.DataTotal, &granular.CreatedAt, &granular.UpdatedAt)
if err != nil {
return nil, obj.makeErr(err)
}
return granular, nil
}
func (obj *sqlite3Impl) Update_Timestamps_By_Name(ctx context.Context,
timestamps_name Timestamps_Name_Field,
update Timestamps_Update_Fields) (
timestamps *Timestamps, err error) {
var __sets = &__sqlbundle_Hole{}
var __embed_stmt = __sqlbundle_Literals{Join: "", SQLs: []__sqlbundle_SQL{__sqlbundle_Literal("UPDATE timestamps SET "), __sets, __sqlbundle_Literal(" WHERE timestamps.name = ?")}}
__sets_sql := __sqlbundle_Literals{Join: ", "}
var __values []interface{}
var __args []interface{}
if update.Value._set {
__values = append(__values, update.Value.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("value = ?"))
}
if len(__sets_sql.SQLs) == 0 {
return nil, emptyUpdate()
}
__args = append(__args, timestamps_name.value())
__values = append(__values, __args...)
__sets.SQL = __sets_sql
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
timestamps = &Timestamps{}
_, err = obj.driver.Exec(__stmt, __values...)
if err != nil {
return nil, obj.makeErr(err)
}
var __embed_stmt_get = __sqlbundle_Literal("SELECT timestamps.name, timestamps.value FROM timestamps WHERE timestamps.name = ?")
var __stmt_get = __sqlbundle_Render(obj.dialect, __embed_stmt_get)
obj.logStmt("(IMPLIED) "+__stmt_get, __args...)
err = obj.driver.QueryRow(__stmt_get, __args...).Scan(&timestamps.Name, &timestamps.Value)
if err == sql.ErrNoRows {
return nil, nil
}
if err != nil {
return nil, obj.makeErr(err)
}
return timestamps, nil
}
func (obj *sqlite3Impl) Update_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field,
update Aggregate_Update_Fields) (
aggregate *Aggregate, err error) {
var __sets = &__sqlbundle_Hole{}
var __embed_stmt = __sqlbundle_Literals{Join: "", SQLs: []__sqlbundle_SQL{__sqlbundle_Literal("UPDATE aggregates SET "), __sets, __sqlbundle_Literal(" WHERE aggregates.node_id = ?")}}
__sets_sql := __sqlbundle_Literals{Join: ", "}
var __values []interface{}
var __args []interface{}
if update.StartTime._set {
__values = append(__values, update.StartTime.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("start_time = ?"))
}
if update.Interval._set {
__values = append(__values, update.Interval.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("interval = ?"))
}
__now := obj.db.Hooks.Now().UTC()
__values = append(__values, __now)
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("updated_at = ?"))
__args = append(__args, aggregate_node_id.value())
__values = append(__values, __args...)
__sets.SQL = __sets_sql
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
aggregate = &Aggregate{}
_, err = obj.driver.Exec(__stmt, __values...)
if err != nil {
return nil, obj.makeErr(err)
}
var __embed_stmt_get = __sqlbundle_Literal("SELECT aggregates.node_id, aggregates.start_time, aggregates.interval, aggregates.created_at, aggregates.updated_at FROM aggregates WHERE aggregates.node_id = ?")
var __stmt_get = __sqlbundle_Render(obj.dialect, __embed_stmt_get)
obj.logStmt("(IMPLIED) "+__stmt_get, __args...)
err = obj.driver.QueryRow(__stmt_get, __args...).Scan(&aggregate.NodeId, &aggregate.StartTime, &aggregate.Interval, &aggregate.CreatedAt, &aggregate.UpdatedAt)
if err == sql.ErrNoRows {
return nil, nil
}
if err != nil {
return nil, obj.makeErr(err)
}
return aggregate, nil
}
func (obj *sqlite3Impl) Update_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field,
update Granular_Update_Fields) (
granular *Granular, err error) {
var __sets = &__sqlbundle_Hole{}
var __embed_stmt = __sqlbundle_Literals{Join: "", SQLs: []__sqlbundle_SQL{__sqlbundle_Literal("UPDATE granulars SET "), __sets, __sqlbundle_Literal(" WHERE granulars.node_id = ?")}}
__sets_sql := __sqlbundle_Literals{Join: ", "}
var __values []interface{}
var __args []interface{}
if update.StartTime._set {
__values = append(__values, update.StartTime.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("start_time = ?"))
}
if update.EndTime._set {
__values = append(__values, update.EndTime.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("end_time = ?"))
}
if update.DataTotal._set {
__values = append(__values, update.DataTotal.value())
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("data_total = ?"))
}
__now := obj.db.Hooks.Now().UTC()
__values = append(__values, __now)
__sets_sql.SQLs = append(__sets_sql.SQLs, __sqlbundle_Literal("updated_at = ?"))
__args = append(__args, granular_node_id.value())
__values = append(__values, __args...)
__sets.SQL = __sets_sql
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
granular = &Granular{}
_, err = obj.driver.Exec(__stmt, __values...)
if err != nil {
return nil, obj.makeErr(err)
}
var __embed_stmt_get = __sqlbundle_Literal("SELECT granulars.node_id, granulars.start_time, granulars.end_time, granulars.data_total, granulars.created_at, granulars.updated_at FROM granulars WHERE granulars.node_id = ?")
var __stmt_get = __sqlbundle_Render(obj.dialect, __embed_stmt_get)
obj.logStmt("(IMPLIED) "+__stmt_get, __args...)
err = obj.driver.QueryRow(__stmt_get, __args...).Scan(&granular.NodeId, &granular.StartTime, &granular.EndTime, &granular.DataTotal, &granular.CreatedAt, &granular.UpdatedAt)
if err == sql.ErrNoRows {
return nil, nil
}
if err != nil {
return nil, obj.makeErr(err)
}
return granular, nil
}
func (obj *sqlite3Impl) Delete_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field) (
deleted bool, err error) {
var __embed_stmt = __sqlbundle_Literal("DELETE FROM aggregates WHERE aggregates.node_id = ?")
var __values []interface{}
__values = append(__values, aggregate_node_id.value())
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
__res, err := obj.driver.Exec(__stmt, __values...)
if err != nil {
return false, obj.makeErr(err)
}
__count, err := __res.RowsAffected()
if err != nil {
return false, obj.makeErr(err)
}
return __count > 0, nil
}
func (obj *sqlite3Impl) Delete_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field) (
deleted bool, err error) {
var __embed_stmt = __sqlbundle_Literal("DELETE FROM granulars WHERE granulars.node_id = ?")
var __values []interface{}
__values = append(__values, granular_node_id.value())
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, __values...)
__res, err := obj.driver.Exec(__stmt, __values...)
if err != nil {
return false, obj.makeErr(err)
}
__count, err := __res.RowsAffected()
if err != nil {
return false, obj.makeErr(err)
}
return __count > 0, nil
}
func (obj *sqlite3Impl) getLastTimestamps(ctx context.Context,
pk int64) (
timestamps *Timestamps, err error) {
var __embed_stmt = __sqlbundle_Literal("SELECT timestamps.name, timestamps.value FROM timestamps WHERE _rowid_ = ?")
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, pk)
timestamps = &Timestamps{}
err = obj.driver.QueryRow(__stmt, pk).Scan(&timestamps.Name, &timestamps.Value)
if err != nil {
return nil, obj.makeErr(err)
}
return timestamps, nil
}
func (obj *sqlite3Impl) getLastAggregate(ctx context.Context,
pk int64) (
aggregate *Aggregate, err error) {
var __embed_stmt = __sqlbundle_Literal("SELECT aggregates.node_id, aggregates.start_time, aggregates.interval, aggregates.created_at, aggregates.updated_at FROM aggregates WHERE _rowid_ = ?")
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, pk)
aggregate = &Aggregate{}
err = obj.driver.QueryRow(__stmt, pk).Scan(&aggregate.NodeId, &aggregate.StartTime, &aggregate.Interval, &aggregate.CreatedAt, &aggregate.UpdatedAt)
if err != nil {
return nil, obj.makeErr(err)
}
return aggregate, nil
}
func (obj *sqlite3Impl) getLastGranular(ctx context.Context,
pk int64) (
granular *Granular, err error) {
var __embed_stmt = __sqlbundle_Literal("SELECT granulars.node_id, granulars.start_time, granulars.end_time, granulars.data_total, granulars.created_at, granulars.updated_at FROM granulars WHERE _rowid_ = ?")
var __stmt = __sqlbundle_Render(obj.dialect, __embed_stmt)
obj.logStmt(__stmt, pk)
granular = &Granular{}
err = obj.driver.QueryRow(__stmt, pk).Scan(&granular.NodeId, &granular.StartTime, &granular.EndTime, &granular.DataTotal, &granular.CreatedAt, &granular.UpdatedAt)
if err != nil {
return nil, obj.makeErr(err)
}
return granular, nil
}
func (impl sqlite3Impl) isConstraintError(err error) (
constraint string, ok bool) {
if e, ok := err.(sqlite3.Error); ok {
if e.Code == sqlite3.ErrConstraint {
msg := err.Error()
colon := strings.LastIndex(msg, ":")
if colon != -1 {
return strings.TrimSpace(msg[colon:]), true
}
return "", true
}
}
return "", false
}
func (obj *sqlite3Impl) deleteAll(ctx context.Context) (count int64, err error) {
var __res sql.Result
var __count int64
__res, err = obj.driver.Exec("DELETE FROM timestamps;")
if err != nil {
return 0, obj.makeErr(err)
}
__count, err = __res.RowsAffected()
if err != nil {
return 0, obj.makeErr(err)
}
count += __count
__res, err = obj.driver.Exec("DELETE FROM granulars;")
if err != nil {
return 0, obj.makeErr(err)
}
__count, err = __res.RowsAffected()
if err != nil {
return 0, obj.makeErr(err)
}
count += __count
__res, err = obj.driver.Exec("DELETE FROM aggregates;")
if err != nil {
return 0, obj.makeErr(err)
}
__count, err = __res.RowsAffected()
if err != nil {
return 0, obj.makeErr(err)
}
count += __count
return count, nil
}
type Rx struct {
db *DB
tx *Tx
}
func (rx *Rx) UnsafeTx(ctx context.Context) (unsafe_tx *sql.Tx, err error) {
tx, err := rx.getTx(ctx)
if err != nil {
return nil, err
}
return tx.Tx, nil
}
func (rx *Rx) getTx(ctx context.Context) (tx *Tx, err error) {
if rx.tx == nil {
if rx.tx, err = rx.db.Open(ctx); err != nil {
return nil, err
}
}
return rx.tx, nil
}
func (rx *Rx) Rebind(s string) string {
return rx.db.Rebind(s)
}
func (rx *Rx) Commit() (err error) {
if rx.tx != nil {
err = rx.tx.Commit()
rx.tx = nil
}
return err
}
func (rx *Rx) Rollback() (err error) {
if rx.tx != nil {
err = rx.tx.Rollback()
rx.tx = nil
}
return err
}
func (rx *Rx) Create_Aggregate(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field,
aggregate_start_time Aggregate_StartTime_Field,
aggregate_interval Aggregate_Interval_Field) (
aggregate *Aggregate, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Create_Aggregate(ctx, aggregate_node_id, aggregate_start_time, aggregate_interval)
}
func (rx *Rx) Create_Granular(ctx context.Context,
granular_node_id Granular_NodeId_Field,
granular_start_time Granular_StartTime_Field,
granular_end_time Granular_EndTime_Field,
granular_data_total Granular_DataTotal_Field) (
granular *Granular, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Create_Granular(ctx, granular_node_id, granular_start_time, granular_end_time, granular_data_total)
}
func (rx *Rx) Create_Timestamps(ctx context.Context,
timestamps_name Timestamps_Name_Field,
timestamps_value Timestamps_Value_Field) (
timestamps *Timestamps, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Create_Timestamps(ctx, timestamps_name, timestamps_value)
}
func (rx *Rx) Delete_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field) (
deleted bool, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Delete_Aggregate_By_NodeId(ctx, aggregate_node_id)
}
func (rx *Rx) Delete_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field) (
deleted bool, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Delete_Granular_By_NodeId(ctx, granular_node_id)
}
func (rx *Rx) Find_Timestamps_Value_By_Name(ctx context.Context,
timestamps_name Timestamps_Name_Field) (
row *Value_Row, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Find_Timestamps_Value_By_Name(ctx, timestamps_name)
}
func (rx *Rx) Get_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field) (
aggregate *Aggregate, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Get_Aggregate_By_NodeId(ctx, aggregate_node_id)
}
func (rx *Rx) Get_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field) (
granular *Granular, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Get_Granular_By_NodeId(ctx, granular_node_id)
}
func (rx *Rx) Update_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field,
update Aggregate_Update_Fields) (
aggregate *Aggregate, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Update_Aggregate_By_NodeId(ctx, aggregate_node_id, update)
}
func (rx *Rx) Update_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field,
update Granular_Update_Fields) (
granular *Granular, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Update_Granular_By_NodeId(ctx, granular_node_id, update)
}
func (rx *Rx) Update_Timestamps_By_Name(ctx context.Context,
timestamps_name Timestamps_Name_Field,
update Timestamps_Update_Fields) (
timestamps *Timestamps, err error) {
var tx *Tx
if tx, err = rx.getTx(ctx); err != nil {
return
}
return tx.Update_Timestamps_By_Name(ctx, timestamps_name, update)
}
type Methods interface {
Create_Aggregate(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field,
aggregate_start_time Aggregate_StartTime_Field,
aggregate_interval Aggregate_Interval_Field) (
aggregate *Aggregate, err error)
Create_Granular(ctx context.Context,
granular_node_id Granular_NodeId_Field,
granular_start_time Granular_StartTime_Field,
granular_end_time Granular_EndTime_Field,
granular_data_total Granular_DataTotal_Field) (
granular *Granular, err error)
Create_Timestamps(ctx context.Context,
timestamps_name Timestamps_Name_Field,
timestamps_value Timestamps_Value_Field) (
timestamps *Timestamps, err error)
Delete_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field) (
deleted bool, err error)
Delete_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field) (
deleted bool, err error)
Find_Timestamps_Value_By_Name(ctx context.Context,
timestamps_name Timestamps_Name_Field) (
row *Value_Row, err error)
Get_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field) (
aggregate *Aggregate, err error)
Get_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field) (
granular *Granular, err error)
Update_Aggregate_By_NodeId(ctx context.Context,
aggregate_node_id Aggregate_NodeId_Field,
update Aggregate_Update_Fields) (
aggregate *Aggregate, err error)
Update_Granular_By_NodeId(ctx context.Context,
granular_node_id Granular_NodeId_Field,
update Granular_Update_Fields) (
granular *Granular, err error)
Update_Timestamps_By_Name(ctx context.Context,
timestamps_name Timestamps_Name_Field,
update Timestamps_Update_Fields) (
timestamps *Timestamps, err error)
}
type TxMethods interface {
Methods
Rebind(s string) string
Commit() error
Rollback() error
}
type txMethods interface {
TxMethods
deleteAll(ctx context.Context) (int64, error)
makeErr(err error) error
}
type DBMethods interface {
Methods
Schema() string
Rebind(sql string) string
}
type dbMethods interface {
DBMethods
wrapTx(tx *sql.Tx) txMethods
makeErr(err error) error
}
func openpostgres(source string) (*sql.DB, error) {
return sql.Open("postgres", source)
}
var sqlite3DriverName = "sqlite3_" + fmt.Sprint(time.Now().UnixNano())
func init() {
sql.Register(sqlite3DriverName, &sqlite3.SQLiteDriver{
ConnectHook: sqlite3SetupConn,
})
}
// SQLite3JournalMode controls the journal_mode pragma for all new connections.
// Since it is read without a mutex, it must be changed to the value you want
// before any Open calls.
var SQLite3JournalMode = "WAL"
func sqlite3SetupConn(conn *sqlite3.SQLiteConn) (err error) {
_, err = conn.Exec("PRAGMA foreign_keys = ON", nil)
if err != nil {
return makeErr(err)
}
_, err = conn.Exec("PRAGMA journal_mode = "+SQLite3JournalMode, nil)
if err != nil {
return makeErr(err)
}
return nil
}
func opensqlite3(source string) (*sql.DB, error) {
return sql.Open(sqlite3DriverName, source)
}