WIP: Refactor Exec implementations to be chainable
This commit is contained in:
+3
-92
@@ -157,98 +157,9 @@ type Exec[T any] interface {
|
||||
execute(conn *simpleorm.DBConnection, tx *sql.Tx) error
|
||||
}
|
||||
|
||||
type Select[T any] struct {
|
||||
target schema.Table
|
||||
whereStmt string
|
||||
args []any
|
||||
results []T
|
||||
Limit int64
|
||||
Offset int64
|
||||
}
|
||||
|
||||
func CreateSelect[T any](orm *simpleorm.ORM, whereStmt string, args ...any) (Select[T], error) {
|
||||
table, ok := orm.Cache().Get(reflect.TypeFor[T]().Name())
|
||||
if !ok {
|
||||
return Select[T]{}, errors.New("Failed to get table from schema cache")
|
||||
}
|
||||
|
||||
return Select[T]{target: *table, whereStmt: whereStmt, args: args, Limit: -1, Offset: -1}, nil
|
||||
}
|
||||
|
||||
func (s *Select[T]) Results() []T {
|
||||
return s.results
|
||||
}
|
||||
|
||||
func (s *Select[T]) Execute(conn *simpleorm.DBConnection) error {
|
||||
return s.execute(conn, nil)
|
||||
}
|
||||
|
||||
func (s *Select[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
|
||||
// Reinit the results, new execution
|
||||
s.results = []T{}
|
||||
dml, err := s.target.GetSelectDML()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if s.whereStmt != "" {
|
||||
dml += " WHERE " + s.whereStmt
|
||||
}
|
||||
|
||||
var effectiveArgs []any = s.args
|
||||
if s.Limit != -1 {
|
||||
dml += " LIMIT ?"
|
||||
effectiveArgs = append(effectiveArgs, s.Limit)
|
||||
}
|
||||
|
||||
if s.Offset != -1 {
|
||||
dml += " OFFSET ?"
|
||||
effectiveArgs = append(effectiveArgs, s.Offset)
|
||||
}
|
||||
|
||||
log.LogDebug("Preparing sql: %s, with args: %s", dml, effectiveArgs)
|
||||
|
||||
var stmt *sql.Stmt
|
||||
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)
|
||||
|
||||
var rows *sql.Rows
|
||||
if len(effectiveArgs) == 0 {
|
||||
rows, err = stmt.Query()
|
||||
} else {
|
||||
// Flattent args to make sure it can be parsed correctly
|
||||
var flatArgs []any
|
||||
for _, a := range effectiveArgs {
|
||||
if s, ok := a.([]any); ok {
|
||||
flatArgs = append(flatArgs, s...)
|
||||
} else {
|
||||
flatArgs = append(flatArgs, a)
|
||||
}
|
||||
}
|
||||
rows, err = stmt.Query(flatArgs...)
|
||||
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
s.results, err = readRows[T](s.target, rows)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
type OrderBy struct {
|
||||
Column string
|
||||
Direction string
|
||||
}
|
||||
|
||||
type Count[T any] struct {
|
||||
|
||||
Reference in New Issue
Block a user