2018-07-27 22:11:44 +01:00
// AUTOGENERATED BY gopkg.in/spacemonkeygo/dbx.v1
// DO NOT EDIT.
package statdb
import (
"bytes"
"context"
"database/sql"
"errors"
"fmt"
"reflect"
"regexp"
"strconv"
"strings"
"sync"
"time"
"unicode"
2018-11-09 21:01:51 +00:00
"github.com/lib/pq"
2018-07-27 22:11:44 +01:00
"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 {
2018-11-09 21:01:51 +00:00
case "postgres" :
sql_db , err = openpostgres ( source )
2018-07-27 22:11:44 +01:00
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 {
2018-11-09 21:01:51 +00:00
case "postgres" :
db . dbMethods = newpostgres ( db )
2018-07-27 22:11:44 +01:00
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 ( ) )
}
2018-11-09 21:01:51 +00:00
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 nodes (
id bytea NOT NULL ,
audit_success_count bigint NOT NULL ,
total_audit_count bigint NOT NULL ,
audit_success_ratio double precision NOT NULL ,
uptime_success_count bigint NOT NULL ,
total_uptime_count bigint NOT NULL ,
uptime_ratio double precision NOT NULL ,
created_at timestamp with time zone NOT NULL ,
updated_at timestamp with time zone NOT NULL ,
PRIMARY KEY ( id )
) ; `
}
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 )
}
}
2018-07-27 22:11:44 +01:00
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 nodes (
2018-10-31 16:18:51 +00:00
id BLOB NOT NULL ,
2018-07-27 22:11:44 +01:00
audit_success_count INTEGER NOT NULL ,
total_audit_count INTEGER NOT NULL ,
audit_success_ratio REAL NOT NULL ,
uptime_success_count INTEGER NOT NULL ,
total_uptime_count INTEGER NOT NULL ,
uptime_ratio REAL NOT NULL ,
created_at TIMESTAMP NOT NULL ,
updated_at TIMESTAMP NOT NULL ,
PRIMARY KEY ( id )
) ; `
}
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 Node struct {
2018-10-31 16:18:51 +00:00
Id [ ] byte
2018-07-27 22:11:44 +01:00
AuditSuccessCount int64
TotalAuditCount int64
AuditSuccessRatio float64
UptimeSuccessCount int64
TotalUptimeCount int64
UptimeRatio float64
CreatedAt time . Time
UpdatedAt time . Time
}
func ( Node ) _Table ( ) string { return "nodes" }
type Node_Update_Fields struct {
AuditSuccessCount Node_AuditSuccessCount_Field
TotalAuditCount Node_TotalAuditCount_Field
AuditSuccessRatio Node_AuditSuccessRatio_Field
UptimeSuccessCount Node_UptimeSuccessCount_Field
TotalUptimeCount Node_TotalUptimeCount_Field
UptimeRatio Node_UptimeRatio_Field
}
type Node_Id_Field struct {
_set bool
2018-10-31 16:18:51 +00:00
_value [ ] byte
2018-07-27 22:11:44 +01:00
}
2018-10-31 16:18:51 +00:00
func Node_Id ( v [ ] byte ) Node_Id_Field {
2018-07-27 22:11:44 +01:00
return Node_Id_Field { _set : true , _value : v }
}
func ( f Node_Id_Field ) value ( ) interface { } {
if ! f . _set {
return nil
}
return f . _value
}
func ( Node_Id_Field ) _Column ( ) string { return "id" }
type Node_AuditSuccessCount_Field struct {
_set bool
_value int64
}
func Node_AuditSuccessCount ( v int64 ) Node_AuditSuccessCount_Field {
return Node_AuditSuccessCount_Field { _set : true , _value : v }
}
func ( f Node_AuditSuccessCount_Field ) value ( ) interface { } {
if ! f . _set {
return nil
}
return f . _value
}
func ( Node_AuditSuccessCount_Field ) _Column ( ) string { return "audit_success_count" }
type Node_TotalAuditCount_Field struct {
_set bool
_value int64
}
func Node_TotalAuditCount ( v int64 ) Node_TotalAuditCount_Field {
return Node_TotalAuditCount_Field { _set : true , _value : v }
}
func ( f Node_TotalAuditCount_Field ) value ( ) interface { } {
if ! f . _set {
return nil
}
return f . _value
}
func ( Node_TotalAuditCount_Field ) _Column ( ) string { return "total_audit_count" }
type Node_AuditSuccessRatio_Field struct {
_set bool
_value float64
}
func Node_AuditSuccessRatio ( v float64 ) Node_AuditSuccessRatio_Field {
return Node_AuditSuccessRatio_Field { _set : true , _value : v }
}
func ( f Node_AuditSuccessRatio_Field ) value ( ) interface { } {
if ! f . _set {
return nil
}
return f . _value
}
func ( Node_AuditSuccessRatio_Field ) _Column ( ) string { return "audit_success_ratio" }
type Node_UptimeSuccessCount_Field struct {
_set bool
_value int64
}
func Node_UptimeSuccessCount ( v int64 ) Node_UptimeSuccessCount_Field {
return Node_UptimeSuccessCount_Field { _set : true , _value : v }
}
func ( f Node_UptimeSuccessCount_Field ) value ( ) interface { } {
if ! f . _set {
return nil
}
return f . _value
}
func ( Node_UptimeSuccessCount_Field ) _Column ( ) string { return "uptime_success_count" }
type Node_TotalUptimeCount_Field struct {
_set bool
_value int64
}
func Node_TotalUptimeCount ( v int64 ) Node_TotalUptimeCount_Field {
return Node_TotalUptimeCount_Field { _set : true , _value : v }
}
func ( f Node_TotalUptimeCount_Field ) value ( ) interface { } {
if ! f . _set {
return nil
}
return f . _value
}
func ( Node_TotalUptimeCount_Field ) _Column ( ) string { return "total_uptime_count" }
type Node_UptimeRatio_Field struct {
_set bool
_value float64
}
func Node_UptimeRatio ( v float64 ) Node_UptimeRatio_Field {
return Node_UptimeRatio_Field { _set : true , _value : v }
}
func ( f Node_UptimeRatio_Field ) value ( ) interface { } {
if ! f . _set {
return nil
}
return f . _value
}
func ( Node_UptimeRatio_Field ) _Column ( ) string { return "uptime_ratio" }
type Node_CreatedAt_Field struct {
_set bool
_value time . Time
}
func Node_CreatedAt ( v time . Time ) Node_CreatedAt_Field {
return Node_CreatedAt_Field { _set : true , _value : v }
}
func ( f Node_CreatedAt_Field ) value ( ) interface { } {
if ! f . _set {
return nil
}
return f . _value
}
func ( Node_CreatedAt_Field ) _Column ( ) string { return "created_at" }
type Node_UpdatedAt_Field struct {
_set bool
_value time . Time
}
func Node_UpdatedAt ( v time . Time ) Node_UpdatedAt_Field {
return Node_UpdatedAt_Field { _set : true , _value : v }
}
func ( f Node_UpdatedAt_Field ) value ( ) interface { } {
if ! f . _set {
return nil
}
return f . _value
}
func ( Node_UpdatedAt_Field ) _Column ( ) string { return "updated_at" }
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
//
2018-11-09 21:01:51 +00:00
func ( obj * postgresImpl ) Create_Node ( ctx context . Context ,
node_id Node_Id_Field ,
node_audit_success_count Node_AuditSuccessCount_Field ,
node_total_audit_count Node_TotalAuditCount_Field ,
node_audit_success_ratio Node_AuditSuccessRatio_Field ,
node_uptime_success_count Node_UptimeSuccessCount_Field ,
node_total_uptime_count Node_TotalUptimeCount_Field ,
node_uptime_ratio Node_UptimeRatio_Field ) (
node * Node , err error ) {
__now := obj . db . Hooks . Now ( ) . UTC ( )
__id_val := node_id . value ( )
__audit_success_count_val := node_audit_success_count . value ( )
__total_audit_count_val := node_total_audit_count . value ( )
__audit_success_ratio_val := node_audit_success_ratio . value ( )
__uptime_success_count_val := node_uptime_success_count . value ( )
__total_uptime_count_val := node_total_uptime_count . value ( )
__uptime_ratio_val := node_uptime_ratio . value ( )
__created_at_val := __now
__updated_at_val := __now
var __embed_stmt = __sqlbundle_Literal ( "INSERT INTO nodes ( id, audit_success_count, total_audit_count, audit_success_ratio, uptime_success_count, total_uptime_count, uptime_ratio, created_at, updated_at ) VALUES ( ?, ?, ?, ?, ?, ?, ?, ?, ? ) RETURNING nodes.id, nodes.audit_success_count, nodes.total_audit_count, nodes.audit_success_ratio, nodes.uptime_success_count, nodes.total_uptime_count, nodes.uptime_ratio, nodes.created_at, nodes.updated_at" )
var __stmt = __sqlbundle_Render ( obj . dialect , __embed_stmt )
obj . logStmt ( __stmt , __id_val , __audit_success_count_val , __total_audit_count_val , __audit_success_ratio_val , __uptime_success_count_val , __total_uptime_count_val , __uptime_ratio_val , __created_at_val , __updated_at_val )
node = & Node { }
err = obj . driver . QueryRow ( __stmt , __id_val , __audit_success_count_val , __total_audit_count_val , __audit_success_ratio_val , __uptime_success_count_val , __total_uptime_count_val , __uptime_ratio_val , __created_at_val , __updated_at_val ) . Scan ( & node . Id , & node . AuditSuccessCount , & node . TotalAuditCount , & node . AuditSuccessRatio , & node . UptimeSuccessCount , & node . TotalUptimeCount , & node . UptimeRatio , & node . CreatedAt , & node . UpdatedAt )
if err != nil {
return nil , obj . makeErr ( err )
}
return node , nil
}
func ( obj * postgresImpl ) Get_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ) (
node * Node , err error ) {
var __embed_stmt = __sqlbundle_Literal ( "SELECT nodes.id, nodes.audit_success_count, nodes.total_audit_count, nodes.audit_success_ratio, nodes.uptime_success_count, nodes.total_uptime_count, nodes.uptime_ratio, nodes.created_at, nodes.updated_at FROM nodes WHERE nodes.id = ?" )
var __values [ ] interface { }
__values = append ( __values , node_id . value ( ) )
var __stmt = __sqlbundle_Render ( obj . dialect , __embed_stmt )
obj . logStmt ( __stmt , __values ... )
node = & Node { }
err = obj . driver . QueryRow ( __stmt , __values ... ) . Scan ( & node . Id , & node . AuditSuccessCount , & node . TotalAuditCount , & node . AuditSuccessRatio , & node . UptimeSuccessCount , & node . TotalUptimeCount , & node . UptimeRatio , & node . CreatedAt , & node . UpdatedAt )
if err != nil {
return nil , obj . makeErr ( err )
}
return node , nil
}
func ( obj * postgresImpl ) Update_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ,
update Node_Update_Fields ) (
node * Node , err error ) {
var __sets = & __sqlbundle_Hole { }
var __embed_stmt = __sqlbundle_Literals { Join : "" , SQLs : [ ] __sqlbundle_SQL { __sqlbundle_Literal ( "UPDATE nodes SET " ) , __sets , __sqlbundle_Literal ( " WHERE nodes.id = ? RETURNING nodes.id, nodes.audit_success_count, nodes.total_audit_count, nodes.audit_success_ratio, nodes.uptime_success_count, nodes.total_uptime_count, nodes.uptime_ratio, nodes.created_at, nodes.updated_at" ) } }
__sets_sql := __sqlbundle_Literals { Join : ", " }
var __values [ ] interface { }
var __args [ ] interface { }
if update . AuditSuccessCount . _set {
__values = append ( __values , update . AuditSuccessCount . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "audit_success_count = ?" ) )
}
if update . TotalAuditCount . _set {
__values = append ( __values , update . TotalAuditCount . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "total_audit_count = ?" ) )
}
if update . AuditSuccessRatio . _set {
__values = append ( __values , update . AuditSuccessRatio . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "audit_success_ratio = ?" ) )
}
if update . UptimeSuccessCount . _set {
__values = append ( __values , update . UptimeSuccessCount . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "uptime_success_count = ?" ) )
}
if update . TotalUptimeCount . _set {
__values = append ( __values , update . TotalUptimeCount . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "total_uptime_count = ?" ) )
}
if update . UptimeRatio . _set {
__values = append ( __values , update . UptimeRatio . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "uptime_ratio = ?" ) )
}
__now := obj . db . Hooks . Now ( ) . UTC ( )
__values = append ( __values , __now )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "updated_at = ?" ) )
__args = append ( __args , node_id . value ( ) )
__values = append ( __values , __args ... )
__sets . SQL = __sets_sql
var __stmt = __sqlbundle_Render ( obj . dialect , __embed_stmt )
obj . logStmt ( __stmt , __values ... )
node = & Node { }
err = obj . driver . QueryRow ( __stmt , __values ... ) . Scan ( & node . Id , & node . AuditSuccessCount , & node . TotalAuditCount , & node . AuditSuccessRatio , & node . UptimeSuccessCount , & node . TotalUptimeCount , & node . UptimeRatio , & node . CreatedAt , & node . UpdatedAt )
if err == sql . ErrNoRows {
return nil , nil
}
if err != nil {
return nil , obj . makeErr ( err )
}
return node , nil
}
func ( obj * postgresImpl ) Delete_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ) (
deleted bool , err error ) {
var __embed_stmt = __sqlbundle_Literal ( "DELETE FROM nodes WHERE nodes.id = ?" )
var __values [ ] interface { }
__values = append ( __values , 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 nodes;" )
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
}
2018-07-27 22:11:44 +01:00
func ( obj * sqlite3Impl ) Create_Node ( ctx context . Context ,
node_id Node_Id_Field ,
node_audit_success_count Node_AuditSuccessCount_Field ,
node_total_audit_count Node_TotalAuditCount_Field ,
node_audit_success_ratio Node_AuditSuccessRatio_Field ,
node_uptime_success_count Node_UptimeSuccessCount_Field ,
node_total_uptime_count Node_TotalUptimeCount_Field ,
node_uptime_ratio Node_UptimeRatio_Field ) (
node * Node , err error ) {
__now := obj . db . Hooks . Now ( ) . UTC ( )
__id_val := node_id . value ( )
__audit_success_count_val := node_audit_success_count . value ( )
__total_audit_count_val := node_total_audit_count . value ( )
__audit_success_ratio_val := node_audit_success_ratio . value ( )
__uptime_success_count_val := node_uptime_success_count . value ( )
__total_uptime_count_val := node_total_uptime_count . value ( )
__uptime_ratio_val := node_uptime_ratio . value ( )
__created_at_val := __now
__updated_at_val := __now
var __embed_stmt = __sqlbundle_Literal ( "INSERT INTO nodes ( id, audit_success_count, total_audit_count, audit_success_ratio, uptime_success_count, total_uptime_count, uptime_ratio, created_at, updated_at ) VALUES ( ?, ?, ?, ?, ?, ?, ?, ?, ? )" )
var __stmt = __sqlbundle_Render ( obj . dialect , __embed_stmt )
obj . logStmt ( __stmt , __id_val , __audit_success_count_val , __total_audit_count_val , __audit_success_ratio_val , __uptime_success_count_val , __total_uptime_count_val , __uptime_ratio_val , __created_at_val , __updated_at_val )
__res , err := obj . driver . Exec ( __stmt , __id_val , __audit_success_count_val , __total_audit_count_val , __audit_success_ratio_val , __uptime_success_count_val , __total_uptime_count_val , __uptime_ratio_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 . getLastNode ( ctx , __pk )
}
func ( obj * sqlite3Impl ) Get_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ) (
node * Node , err error ) {
var __embed_stmt = __sqlbundle_Literal ( "SELECT nodes.id, nodes.audit_success_count, nodes.total_audit_count, nodes.audit_success_ratio, nodes.uptime_success_count, nodes.total_uptime_count, nodes.uptime_ratio, nodes.created_at, nodes.updated_at FROM nodes WHERE nodes.id = ?" )
var __values [ ] interface { }
__values = append ( __values , node_id . value ( ) )
var __stmt = __sqlbundle_Render ( obj . dialect , __embed_stmt )
obj . logStmt ( __stmt , __values ... )
node = & Node { }
err = obj . driver . QueryRow ( __stmt , __values ... ) . Scan ( & node . Id , & node . AuditSuccessCount , & node . TotalAuditCount , & node . AuditSuccessRatio , & node . UptimeSuccessCount , & node . TotalUptimeCount , & node . UptimeRatio , & node . CreatedAt , & node . UpdatedAt )
if err != nil {
return nil , obj . makeErr ( err )
}
return node , nil
}
func ( obj * sqlite3Impl ) Update_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ,
update Node_Update_Fields ) (
node * Node , err error ) {
var __sets = & __sqlbundle_Hole { }
var __embed_stmt = __sqlbundle_Literals { Join : "" , SQLs : [ ] __sqlbundle_SQL { __sqlbundle_Literal ( "UPDATE nodes SET " ) , __sets , __sqlbundle_Literal ( " WHERE nodes.id = ?" ) } }
__sets_sql := __sqlbundle_Literals { Join : ", " }
var __values [ ] interface { }
var __args [ ] interface { }
if update . AuditSuccessCount . _set {
__values = append ( __values , update . AuditSuccessCount . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "audit_success_count = ?" ) )
}
if update . TotalAuditCount . _set {
__values = append ( __values , update . TotalAuditCount . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "total_audit_count = ?" ) )
}
if update . AuditSuccessRatio . _set {
__values = append ( __values , update . AuditSuccessRatio . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "audit_success_ratio = ?" ) )
}
if update . UptimeSuccessCount . _set {
__values = append ( __values , update . UptimeSuccessCount . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "uptime_success_count = ?" ) )
}
if update . TotalUptimeCount . _set {
__values = append ( __values , update . TotalUptimeCount . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "total_uptime_count = ?" ) )
}
if update . UptimeRatio . _set {
__values = append ( __values , update . UptimeRatio . value ( ) )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "uptime_ratio = ?" ) )
}
__now := obj . db . Hooks . Now ( ) . UTC ( )
__values = append ( __values , __now )
__sets_sql . SQLs = append ( __sets_sql . SQLs , __sqlbundle_Literal ( "updated_at = ?" ) )
__args = append ( __args , node_id . value ( ) )
__values = append ( __values , __args ... )
__sets . SQL = __sets_sql
var __stmt = __sqlbundle_Render ( obj . dialect , __embed_stmt )
obj . logStmt ( __stmt , __values ... )
node = & Node { }
_ , err = obj . driver . Exec ( __stmt , __values ... )
if err != nil {
return nil , obj . makeErr ( err )
}
var __embed_stmt_get = __sqlbundle_Literal ( "SELECT nodes.id, nodes.audit_success_count, nodes.total_audit_count, nodes.audit_success_ratio, nodes.uptime_success_count, nodes.total_uptime_count, nodes.uptime_ratio, nodes.created_at, nodes.updated_at FROM nodes WHERE nodes.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 ( & node . Id , & node . AuditSuccessCount , & node . TotalAuditCount , & node . AuditSuccessRatio , & node . UptimeSuccessCount , & node . TotalUptimeCount , & node . UptimeRatio , & node . CreatedAt , & node . UpdatedAt )
if err == sql . ErrNoRows {
return nil , nil
}
if err != nil {
return nil , obj . makeErr ( err )
}
return node , nil
}
func ( obj * sqlite3Impl ) Delete_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ) (
deleted bool , err error ) {
var __embed_stmt = __sqlbundle_Literal ( "DELETE FROM nodes WHERE nodes.id = ?" )
var __values [ ] interface { }
__values = append ( __values , 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 ) getLastNode ( ctx context . Context ,
pk int64 ) (
node * Node , err error ) {
var __embed_stmt = __sqlbundle_Literal ( "SELECT nodes.id, nodes.audit_success_count, nodes.total_audit_count, nodes.audit_success_ratio, nodes.uptime_success_count, nodes.total_uptime_count, nodes.uptime_ratio, nodes.created_at, nodes.updated_at FROM nodes WHERE _rowid_ = ?" )
var __stmt = __sqlbundle_Render ( obj . dialect , __embed_stmt )
obj . logStmt ( __stmt , pk )
node = & Node { }
err = obj . driver . QueryRow ( __stmt , pk ) . Scan ( & node . Id , & node . AuditSuccessCount , & node . TotalAuditCount , & node . AuditSuccessRatio , & node . UptimeSuccessCount , & node . TotalUptimeCount , & node . UptimeRatio , & node . CreatedAt , & node . UpdatedAt )
if err != nil {
return nil , obj . makeErr ( err )
}
return node , 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 nodes;" )
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_Node ( ctx context . Context ,
node_id Node_Id_Field ,
node_audit_success_count Node_AuditSuccessCount_Field ,
node_total_audit_count Node_TotalAuditCount_Field ,
node_audit_success_ratio Node_AuditSuccessRatio_Field ,
node_uptime_success_count Node_UptimeSuccessCount_Field ,
node_total_uptime_count Node_TotalUptimeCount_Field ,
node_uptime_ratio Node_UptimeRatio_Field ) (
node * Node , err error ) {
var tx * Tx
if tx , err = rx . getTx ( ctx ) ; err != nil {
return
}
return tx . Create_Node ( ctx , node_id , node_audit_success_count , node_total_audit_count , node_audit_success_ratio , node_uptime_success_count , node_total_uptime_count , node_uptime_ratio )
}
func ( rx * Rx ) Delete_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ) (
deleted bool , err error ) {
var tx * Tx
if tx , err = rx . getTx ( ctx ) ; err != nil {
return
}
return tx . Delete_Node_By_Id ( ctx , node_id )
}
func ( rx * Rx ) Get_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ) (
node * Node , err error ) {
var tx * Tx
if tx , err = rx . getTx ( ctx ) ; err != nil {
return
}
return tx . Get_Node_By_Id ( ctx , node_id )
}
func ( rx * Rx ) Update_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ,
update Node_Update_Fields ) (
node * Node , err error ) {
var tx * Tx
if tx , err = rx . getTx ( ctx ) ; err != nil {
return
}
return tx . Update_Node_By_Id ( ctx , node_id , update )
}
type Methods interface {
Create_Node ( ctx context . Context ,
node_id Node_Id_Field ,
node_audit_success_count Node_AuditSuccessCount_Field ,
node_total_audit_count Node_TotalAuditCount_Field ,
node_audit_success_ratio Node_AuditSuccessRatio_Field ,
node_uptime_success_count Node_UptimeSuccessCount_Field ,
node_total_uptime_count Node_TotalUptimeCount_Field ,
node_uptime_ratio Node_UptimeRatio_Field ) (
node * Node , err error )
Delete_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ) (
deleted bool , err error )
Get_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ) (
node * Node , err error )
Update_Node_By_Id ( ctx context . Context ,
node_id Node_Id_Field ,
update Node_Update_Fields ) (
node * Node , 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
}
2018-11-09 21:01:51 +00:00
func openpostgres ( source string ) ( * sql . DB , error ) {
return sql . Open ( "postgres" , source )
}
2018-07-27 22:11:44 +01:00
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 )
}