85 lines
1.6 KiB
Go
85 lines
1.6 KiB
Go
package exec
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"reflect"
|
|
|
|
"git.gdulai.com/gdulai/simpleorm"
|
|
"git.gdulai.com/gdulai/simpleorm/schema"
|
|
log "gitlab.com/gdulai/simpleloglvl"
|
|
)
|
|
|
|
type Count[T any] struct {
|
|
target schema.Table
|
|
whereStmt string
|
|
args []any
|
|
result int64
|
|
}
|
|
|
|
func CreateCount[T any](orm *simpleorm.ORM, whereStmt string, args ...any) (Count[T], error) {
|
|
table, ok := orm.Cache().Get(reflect.TypeFor[T]().Name())
|
|
if !ok {
|
|
return Count[T]{}, errors.New("Failed to get table from schema cache")
|
|
}
|
|
|
|
return Count[T]{target: *table, whereStmt: whereStmt, args: args}, nil
|
|
}
|
|
|
|
func (c *Count[T]) Result() int64 {
|
|
return c.result
|
|
}
|
|
|
|
func (c *Count[T]) Execute(conn *simpleorm.DBConnection) error {
|
|
return c.execute(conn, nil)
|
|
}
|
|
|
|
func (c *Count[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
|
|
// Reinit the results, new execution
|
|
c.result = -1
|
|
dml := c.target.GetCountDML()
|
|
|
|
if c.whereStmt != "" {
|
|
dml += " WHERE " + c.whereStmt
|
|
}
|
|
|
|
log.LogDebug("Preparing sql: %s, with args: %s", dml, c.args)
|
|
|
|
var stmt *sql.Stmt
|
|
var err error
|
|
if tx != nil {
|
|
stmt, err = tx.Prepare(dml)
|
|
} else {
|
|
stmt, err = conn.Prepare(dml)
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer stmt.Close()
|
|
|
|
log.LogDebug("Executing statement: %s", stmt)
|
|
|
|
if len(c.args) == 0 {
|
|
err = stmt.QueryRow().Scan(&c.result)
|
|
} else {
|
|
// Flattent args to make sure it can be parsed correctly
|
|
var flatArgs []any
|
|
for _, a := range c.args {
|
|
if s, ok := a.([]any); ok {
|
|
flatArgs = append(flatArgs, s...)
|
|
} else {
|
|
flatArgs = append(flatArgs, a)
|
|
}
|
|
}
|
|
|
|
err = stmt.QueryRow(flatArgs...).Scan(&c.result)
|
|
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|