Separate exec types into files, document code (#10)
Go Tests / test (push) Successful in 1m2s
Go Tests / test (push) Successful in 1m2s
Reviewed-on: #10
This commit was merged in pull request #10.
This commit is contained in:
-262
@@ -157,268 +157,6 @@ type Exec[T any] interface {
|
||||
execute(conn *simpleorm.DBConnection, tx *sql.Tx) error
|
||||
}
|
||||
|
||||
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
|
||||
}
|
||||
|
||||
type Insert[T any] struct {
|
||||
target schema.Table
|
||||
toInsert []T
|
||||
results []T
|
||||
}
|
||||
|
||||
func NewInsert[T any](orm *simpleorm.ORM, toInsert ...T) (Insert[T], error) {
|
||||
table, ok := orm.Cache().Get(reflect.TypeFor[T]().Name())
|
||||
if !ok {
|
||||
return Insert[T]{}, errors.New("Failed to get table from schema cache")
|
||||
}
|
||||
|
||||
return Insert[T]{target: *table, toInsert: toInsert}, nil
|
||||
}
|
||||
|
||||
func (ins *Insert[T]) Results() []T {
|
||||
return ins.results
|
||||
}
|
||||
|
||||
func (ins *Insert[T]) Execute(conn *simpleorm.DBConnection) error {
|
||||
return ins.execute(conn, nil)
|
||||
}
|
||||
|
||||
func (ins *Insert[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
|
||||
dml, err := ins.target.GetInsertDML(len(ins.toInsert))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var params []any
|
||||
for i := range len(ins.toInsert) {
|
||||
actualParams := prepareParams(ins.toInsert[i], ins.target)
|
||||
if len(params) == 0 {
|
||||
params = make([]any, len(ins.toInsert)*len(actualParams))
|
||||
}
|
||||
for j := range actualParams {
|
||||
params[(i*len(actualParams))+j] = actualParams[j]
|
||||
}
|
||||
}
|
||||
|
||||
log.LogDebug("%s [%s]", dml, params)
|
||||
|
||||
var stmt *sql.Stmt
|
||||
if tx != nil {
|
||||
stmt, err = tx.Prepare(dml)
|
||||
} else {
|
||||
stmt, err = conn.Prepare(dml)
|
||||
}
|
||||
defer stmt.Close()
|
||||
|
||||
// Flattent args to make sure it can be parsed correctly
|
||||
var flatParams []any
|
||||
for _, param := range params {
|
||||
if s, ok := param.([]any); ok {
|
||||
flatParams = append(flatParams, s...)
|
||||
} else {
|
||||
flatParams = append(flatParams, param)
|
||||
}
|
||||
}
|
||||
|
||||
rows, err := stmt.Query(flatParams...)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
ins.results, err = readRows[T](ins.target, rows)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
type Update[T any] struct {
|
||||
target schema.Table
|
||||
toUpdate T
|
||||
}
|
||||
|
||||
func NewUpdate[T any](orm *simpleorm.ORM, toUpdate T) (Update[T], error) {
|
||||
table, ok := orm.Cache().Get(reflect.TypeFor[T]().Name())
|
||||
if !ok {
|
||||
return Update[T]{}, errors.New("Failed to get table from schema cache")
|
||||
}
|
||||
|
||||
return Update[T]{target: *table, toUpdate: toUpdate}, nil
|
||||
}
|
||||
|
||||
func (u Update[T]) Execute(conn *simpleorm.DBConnection) error {
|
||||
return u.execute(conn, nil)
|
||||
}
|
||||
|
||||
func (u Update[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
|
||||
dml, err := u.target.GetUpdateDML()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
params := prepareParams(u.toUpdate, u.target)
|
||||
|
||||
pkCols, err := getPk(u.toUpdate, u.target)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// Put the pks back at the end
|
||||
for _, pk := range pkCols {
|
||||
params = append(params, pk)
|
||||
}
|
||||
|
||||
log.LogDebug("%s [%s]", dml, params)
|
||||
|
||||
var stmt *sql.Stmt
|
||||
if tx != nil {
|
||||
stmt, err = tx.Prepare(dml)
|
||||
} else {
|
||||
stmt, err = conn.Prepare(dml)
|
||||
}
|
||||
defer stmt.Close()
|
||||
|
||||
result, err := stmt.Exec(params...)
|
||||
if err != nil {
|
||||
return err
|
||||
} else {
|
||||
rowsAffected, _ := result.RowsAffected()
|
||||
log.LogInfo("Updated %s row", rowsAffected)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type Delete[T any] struct {
|
||||
target schema.Table
|
||||
toDelete []T
|
||||
}
|
||||
|
||||
func NewDelete[T any](orm *simpleorm.ORM, toDelete []T) (Delete[T], error) {
|
||||
table, ok := orm.Cache().Get(reflect.TypeFor[T]().Name())
|
||||
if !ok {
|
||||
return Delete[T]{}, errors.New("Failed to get table from schema cache")
|
||||
}
|
||||
|
||||
return Delete[T]{target: *table, toDelete: toDelete}, nil
|
||||
}
|
||||
|
||||
func (d *Delete[T]) Execute(conn *simpleorm.DBConnection) error {
|
||||
return d.execute(conn, nil)
|
||||
}
|
||||
|
||||
func (d *Delete[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
|
||||
count := len(d.toDelete)
|
||||
dml, err := d.target.GetDeleteDML(count)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var params []any
|
||||
|
||||
for _, del := range d.toDelete {
|
||||
pkCols, err := getPk(del, d.target)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, pk := range pkCols {
|
||||
params = append(params, pk)
|
||||
}
|
||||
}
|
||||
|
||||
log.LogDebug("%s [%s]", dml, params)
|
||||
|
||||
var stmt *sql.Stmt
|
||||
if tx != nil {
|
||||
stmt, err = tx.Prepare(dml)
|
||||
} else {
|
||||
stmt, err = conn.Prepare(dml)
|
||||
}
|
||||
defer stmt.Close()
|
||||
|
||||
result, err := stmt.Exec(params...)
|
||||
if err != nil {
|
||||
return err
|
||||
} else {
|
||||
rowsAffected, _ := result.RowsAffected()
|
||||
log.LogInfo("Deleted %s row", rowsAffected)
|
||||
}
|
||||
|
||||
return nil
|
||||
|
||||
}
|
||||
|
||||
func createSelectResultContainer(t schema.Table) []any {
|
||||
typ := t.Type()
|
||||
vals := make([]any, typ.NumField())
|
||||
|
||||
Reference in New Issue
Block a user