12 Commits
Author SHA1 Message Date
gdulai 90d7bf7226 Fix order by and limi/offset incorrect order
Go Tests / test (push) Successful in 1m5s
2026-06-05 17:46:20 +02:00
gdulai 238361b066 Refactor parsing & select order by impl (#6)
Go Tests / test (push) Successful in 1m3s
Reviewed-on: #6
2026-06-05 15:22:41 +00:00
gdulai d10d0f8414 Count exec & Select limit/offset
Go Tests / test (push) Successful in 1m16s
2026-06-04 11:08:14 +02:00
gdulai 0cf4e63ac3 Correct build step
Go Tests / test (push) Successful in 1m4s
2026-06-02 22:30:23 +02:00
gdulai 357b765eb7 Add build step
Go Tests / test (push) Failing after 1m2s
2026-06-02 22:28:21 +02:00
gdulai 2c51bba6e4 Add go install to ci/cd
Go Tests / test (push) Successful in 1m4s
2026-06-02 22:23:33 +02:00
gdulai 82dd07ab59 Add go install to ci/cd
Go Tests / test (push) Failing after 2s
2026-06-02 22:21:54 +02:00
gdulai f3c5fb5fe6 Fix failing ci/cd setup
Go Tests / test (push) Failing after 2s
2026-06-02 22:14:44 +02:00
gdulai f19c82a2ad Fix failing ci/cd setup
Debug / test (push) Successful in 2s
2026-06-02 22:10:03 +02:00
gdulai 3e706bada5 Fix failing ci/cd setup
Go Tests / test (push) Failing after 2s
2026-06-02 22:06:53 +02:00
gdulai 91966fcf5a Actionless ci/cd setup
Go Tests / test (push) Failing after 1s
2026-06-02 22:04:53 +02:00
gdulai 7fff1f4d3d Correct insert/returning ddl
Go Tests / test (push) Failing after 1m33s
2026-05-24 21:36:56 +02:00
12 changed files with 983 additions and 432 deletions
+27 -15
View File
@@ -10,24 +10,36 @@ jobs:
runs-on: ubuntu-latest runs-on: ubuntu-latest
steps: steps:
- name: Checkout code - name: Checkout repository
uses: actions/checkout@v4 run: |
git clone \
"${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY}.git" \
.
- name: Setup Go git checkout "${GITHUB_SHA}"
uses: actions/setup-go@v5
with:
go-version: "1.22"
- name: Cache Go modules - name: Verify checkout
uses: actions/cache@v4 run: |
with: pwd
path: | ls -la
~/.cache/go-build test -f go.mod
~/go/pkg/mod
key: ${{ runner.os }}-go-${{ hashFiles('**/go.sum') }} - name: Install Go
run: |
rm -rf /usr/local/go
curl -LO https://go.dev/dl/go1.26.2.linux-amd64.tar.gz
tar -C /usr/local -xzf go1.26.2.linux-amd64.tar.gz
export PATH=/usr/local/go/bin:$PATH
go version
- name: Download dependencies - name: Download dependencies
run: go mod download run: |
go mod download
- name: Run tests - name: Run tests
run: go test ./... -v run: |
go test ./... -v
- name: Build binary
run: |
go build -o app .
+1 -1
View File
@@ -25,7 +25,7 @@ func NewSchemaCache(tables []*schema.Table) *SchemaCache {
func (o *SchemaCache) add(table *schema.Table) { func (o *SchemaCache) add(table *schema.Table) {
o.mu.Lock() o.mu.Lock()
defer o.mu.Unlock() defer o.mu.Unlock()
o.data[table.Type.Name()] = table o.data[table.Type().Name()] = table
} }
func (o *SchemaCache) Get(typeName string) (*schema.Table, bool) { func (o *SchemaCache) Get(typeName string) (*schema.Table, bool) {
+37 -41
View File
@@ -157,45 +157,43 @@ type Exec[T any] interface {
execute(conn *simpleorm.DBConnection, tx *sql.Tx) error execute(conn *simpleorm.DBConnection, tx *sql.Tx) error
} }
type Select[T any] struct { type Count[T any] struct {
target schema.Table target schema.Table
whereStmt string whereStmt string
args []any args []any
results []T result int64
} }
func CreateSelect[T any](orm *simpleorm.ORM, whereStmt string, args ...any) (Select[T], error) { func CreateCount[T any](orm *simpleorm.ORM, whereStmt string, args ...any) (Count[T], error) {
table, ok := orm.Cache().Get(reflect.TypeFor[T]().Name()) table, ok := orm.Cache().Get(reflect.TypeFor[T]().Name())
if !ok { if !ok {
return Select[T]{}, errors.New("Failed to get table from schema cache") return Count[T]{}, errors.New("Failed to get table from schema cache")
} }
return Select[T]{target: *table, whereStmt: whereStmt, args: args}, nil return Count[T]{target: *table, whereStmt: whereStmt, args: args}, nil
} }
func (s *Select[T]) Results() []T { func (c *Count[T]) Result() int64 {
return s.results return c.result
} }
func (s *Select[T]) Execute(conn *simpleorm.DBConnection) error { func (c *Count[T]) Execute(conn *simpleorm.DBConnection) error {
return s.execute(conn, nil) return c.execute(conn, nil)
} }
func (s *Select[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error { func (c *Count[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
// Reinit the results, new execution // Reinit the results, new execution
s.results = []T{} c.result = -1
dml, err := s.target.GetSelectDML() dml := c.target.GetCountDML()
if err != nil {
return err if c.whereStmt != "" {
dml += " WHERE " + c.whereStmt
} }
if s.whereStmt != "" { log.LogDebug("Preparing sql: %s, with args: %s", dml, c.args)
dml += " WHERE " + s.whereStmt
}
log.LogDebug("Preparing sql: %s, with args: %s", dml, s.args)
var stmt *sql.Stmt var stmt *sql.Stmt
var err error
if tx != nil { if tx != nil {
stmt, err = tx.Prepare(dml) stmt, err = tx.Prepare(dml)
} else { } else {
@@ -209,28 +207,22 @@ func (s *Select[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
log.LogDebug("Executing statement: %s", stmt) log.LogDebug("Executing statement: %s", stmt)
var rows *sql.Rows if len(c.args) == 0 {
if len(s.args) == 0 { err = stmt.QueryRow().Scan(&c.result)
rows, err = stmt.Query()
} else { } else {
// Flattent args to make sure it can be parsed correctly // Flattent args to make sure it can be parsed correctly
var flatArgs []any var flatArgs []any
for _, a := range s.args { for _, a := range c.args {
if s, ok := a.([]any); ok { if s, ok := a.([]any); ok {
flatArgs = append(flatArgs, s...) flatArgs = append(flatArgs, s...)
} else { } else {
flatArgs = append(flatArgs, a) flatArgs = append(flatArgs, a)
} }
} }
rows, err = stmt.Query(flatArgs...)
err = stmt.QueryRow(flatArgs...).Scan(&c.result)
} }
if err != nil {
return err
}
s.results, err = readRows[T](s.target, rows)
if err != nil { if err != nil {
return err return err
} }
@@ -278,7 +270,7 @@ func (ins *Insert[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
} }
} }
log.LogInfo("%s [%s]", dml, params) log.LogDebug("%s [%s]", dml, params)
var stmt *sql.Stmt var stmt *sql.Stmt
if tx != nil { if tx != nil {
@@ -347,7 +339,7 @@ func (u Update[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
params = append(params, pk) params = append(params, pk)
} }
log.LogInfo("%s [%s]", dml, params) log.LogDebug("%s [%s]", dml, params)
var stmt *sql.Stmt var stmt *sql.Stmt
if tx != nil { if tx != nil {
@@ -405,7 +397,7 @@ func (d *Delete[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
} }
} }
log.LogInfo("%s [%s]", dml, params) log.LogDebug("%s [%s]", dml, params)
var stmt *sql.Stmt var stmt *sql.Stmt
if tx != nil { if tx != nil {
@@ -428,9 +420,10 @@ func (d *Delete[T]) execute(conn *simpleorm.DBConnection, tx *sql.Tx) error {
} }
func createSelectResultContainer(t schema.Table) []any { func createSelectResultContainer(t schema.Table) []any {
vals := make([]any, t.Type.NumField()) typ := t.Type()
vals := make([]any, typ.NumField())
for i := range vals { for i := range vals {
switch t.Type.Field(i).Type.Kind().String() { switch typ.Field(i).Type.Kind().String() {
case "string": case "string":
var fieldContainer string var fieldContainer string
vals[i] = &fieldContainer vals[i] = &fieldContainer
@@ -446,13 +439,14 @@ func createSelectResultContainer(t schema.Table) []any {
} }
func prepareParams(src any, t schema.Table) []any { func prepareParams(src any, t schema.Table) []any {
typ := t.Type()
var params []any var params []any
for _, col := range t.Columns { for _, col := range t.Columns() {
_, ok := col.Modifiers["pk"] _, ok := col.Modifiers["pk"]
if ok && t.IsPkAuto() { if ok && t.IsPkAuto() {
continue continue
} }
field, ok := t.Type.FieldByName(col.FieldName) field, ok := typ.FieldByName(col.FieldName)
if !ok { if !ok {
continue continue
} }
@@ -464,11 +458,12 @@ func prepareParams(src any, t schema.Table) []any {
} }
func getPk(src any, t schema.Table) ([]any, error) { func getPk(src any, t schema.Table) ([]any, error) {
typ := t.Type()
var values []any var values []any
for _, constraint := range t.Constraints { for _, constraint := range t.Constraints() {
if constraint.Type == "pk" { if constraint.Type == "pk" {
for _, col := range constraint.Columns { for _, col := range constraint.Columns {
field, ok := t.Type.FieldByName(col.FieldName) field, ok := typ.FieldByName(col.FieldName)
if !ok { if !ok {
continue continue
@@ -496,10 +491,11 @@ func readRows[T any](table schema.Table, rows *sql.Rows) ([]T, error) {
return nil, err return nil, err
} }
targetType := table.Type targetType := table.Type()
cols := table.Columns()
parsedResult := reflect.New(targetType) parsedResult := reflect.New(targetType)
for i, fieldVal := range rowContainer { for i, fieldVal := range rowContainer {
col := table.Columns[i] col := cols[i]
targetField := parsedResult.Elem().Field(i) targetField := parsedResult.Elem().Field(i)
rawValue := reflect.Indirect(reflect.ValueOf(fieldVal)) rawValue := reflect.Indirect(reflect.ValueOf(fieldVal))
+153
View File
@@ -0,0 +1,153 @@
package exec
import (
"database/sql"
"errors"
"reflect"
"git.gdulai.com/gdulai/simpleorm"
"git.gdulai.com/gdulai/simpleorm/schema"
log "gitlab.com/gdulai/simpleloglvl"
)
// Select execution which is translated to a select sql query.
type Select[T any] struct {
target schema.Table
whereStmt string
args []any
limit int64
offset int64
ordering []schema.OrderBy
results []T
}
// Creates the select query builder
func CreateSelect[T any](orm *simpleorm.ORM) (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, limit: -1, offset: -1}, nil
}
// Sets the where part of the select query and the arguments
// Returns the pointer of the Select instance
func (s *Select[T]) Where(whereStmt string, args ...any) *Select[T] {
s.whereStmt = whereStmt
s.args = args
return s
}
// Sets the limit part of the select query
// Returns the pointer of the Select instance
func (s *Select[T]) Limit(limit int64) *Select[T] {
s.limit = limit
return s
}
// Sets the offset part of the select query
// Returns the pointer of the Select instance
func (s *Select[T]) Offset(offset int64) *Select[T] {
s.offset = offset
return s
}
// Sets the order by part of the select query
// Returns the pointer of the Select instance
func (s *Select[T]) OrderBy(ordering ...schema.OrderBy) *Select[T] {
s.ordering = ordering
return s
}
// Returns the result of the select
func (s *Select[T]) Results() []T {
return s.results
}
// Executes the select query based on the Select exec instance
// Returns an error if theres any
func (s *Select[T]) Execute(conn *simpleorm.DBConnection) error {
return s.execute(conn, nil)
}
// Executes the select query based on the Select exec instance
// If tx is given, the transaction is used instead of the conn
// Returns an error if theres any
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
}
if len(s.ordering) > 0 {
orderBy, err := s.target.GetOrderByDML(s.ordering...)
if err != nil {
log.LogError("Failed to create ORDER BY part: %s", err)
} else {
dml += orderBy
}
}
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.LogInfo("Preparing sql: %s", dml)
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
}
+26 -16
View File
@@ -7,7 +7,6 @@ import (
"strings" "strings"
"git.gdulai.com/gdulai/simpleorm/cache" "git.gdulai.com/gdulai/simpleorm/cache"
"git.gdulai.com/gdulai/simpleorm/parser"
"git.gdulai.com/gdulai/simpleorm/schema" "git.gdulai.com/gdulai/simpleorm/schema"
log "gitlab.com/gdulai/simpleloglvl" log "gitlab.com/gdulai/simpleloglvl"
@@ -21,28 +20,39 @@ type ORM struct {
// Inits the ORM library. // Inits the ORM library.
// Param objs is an array which should be an array of the types which describe the tables. // Param objs is an array which should be an array of the types which describe the tables.
func NewORM(objs ...any) *ORM { func NewORM(objs ...any) *ORM {
var parsers []*parser.Parser typeParsers := make(map[string]*schema.Parser)
var tables []*schema.Table
log.LogDebug("[ORM] Parsing entities to tables...")
log.LogDebug("[ORM] Step 1: Parsing table columns")
for _, obj := range objs { for _, obj := range objs {
log.LogDebug("[ORM] Mapping type for: %s", reflect.TypeOf(obj).Name()) typ := reflect.TypeOf(obj)
parser := parser.NewParser(obj) log.LogDebug("[ORM] Mapping type for: %s", typ.Name())
parsers = append(parsers, parser)
tables = append(tables, parser.ParseColumns()) parser := schema.NewParser(obj)
parser.ParseColumns()
typeParsers[typ.Name()] = parser
} }
log.LogDebug("[ORM] Tables initiated, creating cache.") log.LogDebug("[ORM] Step 1: Finished parsing columns!")
log.LogDebug("[ORM] Step 2: Parsing constraints...")
for _, parser := range typeParsers {
parser.ParseConstraints(typeParsers)
}
log.LogDebug("[ORM] Step 2: Finished parsing contraints!")
log.LogDebug("[ORM] Step 3: Creating and caching schema...")
var tables []*schema.Table
for _, parser := range typeParsers {
tables = append(tables, parser.ParseTable())
}
// Create cache with the initialized tables // Create cache with the initialized tables
cache := cache.NewSchemaCache(tables) cache := cache.NewSchemaCache(tables)
log.LogDebug("[ORM] Cache created.") log.LogDebug("[ORM] Step 3: Cache created.")
// Finish the parsing with the constraints and add the to the tables
for _, parser := range parsers {
log.LogDebug("[ORM] Parsing constraing for: %s", parser.Table.Name)
parser.ParseConstraints(cache)
log.LogDebug("[ORM] Parsed constraints for: %s", parser.Table.Name)
}
return &ORM{cache: cache} return &ORM{cache: cache}
} }
-101
View File
@@ -1,101 +0,0 @@
package parser
import (
"reflect"
"strings"
cache "git.gdulai.com/gdulai/simpleorm/cache"
"git.gdulai.com/gdulai/simpleorm/schema"
"git.gdulai.com/gdulai/simpleorm/util"
log "gitlab.com/gdulai/simpleloglvl"
)
type Parser struct {
typ reflect.Type
Table *schema.Table
}
func NewParser[T any](obj T) *Parser {
objType := reflect.TypeOf(obj)
return &Parser{typ: objType}
}
// This is step 1 of the parsing, it creates the table instance and
func (p *Parser) ParseColumns() *schema.Table {
table := schema.Table{Name: util.CamelToSnake(p.typ.Name()), Type: p.typ}
var columns []schema.Column
for field := range p.typ.Fields() {
field := field
col := schema.NewColumn(util.CamelToSnake((field.Name)), field.Name, determineType(field.Type), field.Tag)
columns = append(columns, col)
}
table.Columns = columns
p.Table = &table
return p.Table
}
func (p *Parser) ParseConstraints(cache *cache.SchemaCache) {
pkConstraint := schema.Constraint{Name: "pk_" + strings.ToLower(p.Table.Name), Type: "pk"}
fkConstraints := make(map[string]schema.Constraint)
for _, col := range p.Table.Columns {
_, ok := col.Modifiers["pk"]
if ok {
pkConstraint.Columns = append(pkConstraint.Columns, col)
continue
}
fkMod, ok := col.Modifiers["fk"]
if !ok {
continue
}
fkModParts := strings.Split(fkMod, ".")
refTable, ok := cache.Get(fkModParts[0])
if !ok {
log.LogError("[ORM] Table %s not found in OrmCache!", fkModParts[0])
return
}
fkId, ok := col.Modifiers["fk_id"]
if fkId == "" {
fkId = "fk_" + strings.ToLower(refTable.Name)
}
fkConstraint, ok := fkConstraints[fkId]
if !ok {
fkConstraint = schema.Constraint{Name: fkId, Type: "fk", RefTable: refTable}
}
fkConstraint.Columns = append(fkConstraint.Columns, col)
refField := fkModParts[1]
for _, refC := range refTable.Columns {
if refC.FieldName == refField {
fkConstraint.RefColumns = append(fkConstraint.RefColumns, refC)
}
}
fkConstraints[fkId] = fkConstraint
}
p.Table.Constraints = append(p.Table.Constraints, pkConstraint)
for _, fkConstraint := range fkConstraints {
p.Table.Constraints = append(p.Table.Constraints, fkConstraint)
}
}
func determineType(typ reflect.Type) string {
typStr := typ.String()
switch typStr {
case "string":
return "TEXT"
case "int", "bool":
return "INTEGER"
case "time.Time", "int64":
return "BIGINT"
}
return "VARCHAR(255)"
}
+5 -3
View File
@@ -23,7 +23,7 @@ func NewRepository[T HasPK](conn *simpleorm.DBConnection, orm *simpleorm.ORM) *R
} }
func (r *Repository[T]) SelectAll() []*T { func (r *Repository[T]) SelectAll() []*T {
selectExec, err := exec.CreateSelect[T](r.orm, "") selectExec, err := exec.CreateSelect[T](r.orm)
if err != nil { if err != nil {
log.LogError("Failed to create select execution: %s", err) log.LogError("Failed to create select execution: %s", err)
return []*T{} return []*T{}
@@ -53,7 +53,7 @@ func (r *Repository[T]) SelectByPk(pks ...any) *T {
var whereStmtBuilder strings.Builder var whereStmtBuilder strings.Builder
for _, constr := range table.Constraints { for _, constr := range table.Constraints() {
if constr.Type != "pk" { if constr.Type != "pk" {
continue continue
} }
@@ -68,12 +68,14 @@ func (r *Repository[T]) SelectByPk(pks ...any) *T {
} }
selectExec, err := exec.CreateSelect[T](r.orm, whereStmtBuilder.String(), pks) selectExec, err := exec.CreateSelect[T](r.orm)
if err != nil { if err != nil {
log.LogError("Failed to create select execution: %s", err) log.LogError("Failed to create select execution: %s", err)
return nil return nil
} }
selectExec.Where(whereStmtBuilder.String(), pks)
err = selectExec.Execute(r.conn) err = selectExec.Execute(r.conn)
if err != nil { if err != nil {
log.LogError("Failed to load entity: %s", err) log.LogError("Failed to load entity: %s", err)
+17 -8
View File
@@ -3,14 +3,16 @@ package schema
import ( import (
"errors" "errors"
"strings" "strings"
"git.gdulai.com/gdulai/simpleorm/util"
) )
type Constraint struct { type Constraint struct {
Name string Name string
Type string Type string
Columns []Column Columns []Column
RefTable *Table RefTypeName string
RefColumns []Column RefColumns []Column
} }
func (c *Constraint) GetDDL() (string, error) { func (c *Constraint) GetDDL() (string, error) {
@@ -26,7 +28,9 @@ func (c *Constraint) GetDDL() (string, error) {
func (c *Constraint) getPkDDL() string { func (c *Constraint) getPkDDL() string {
var ddl strings.Builder var ddl strings.Builder
ddl.WriteString("CONSTRAINT " + c.Name + " PRIMARY KEY(") ddl.WriteString("CONSTRAINT ")
ddl.WriteString(c.Name)
ddl.WriteString(" PRIMARY KEY(")
for i, col := range c.Columns { for i, col := range c.Columns {
if i != 0 { if i != 0 {
ddl.WriteString(", ") ddl.WriteString(", ")
@@ -39,7 +43,9 @@ func (c *Constraint) getPkDDL() string {
func (c *Constraint) getFkDDL() string { func (c *Constraint) getFkDDL() string {
var ddl strings.Builder var ddl strings.Builder
ddl.WriteString("CONSTRAINT " + c.Name + " FOREIGN KEY(") ddl.WriteString("CONSTRAINT ")
ddl.WriteString(c.Name)
ddl.WriteString(" FOREIGN KEY(")
for i, col := range c.Columns { for i, col := range c.Columns {
if i != 0 { if i != 0 {
@@ -47,7 +53,10 @@ func (c *Constraint) getFkDDL() string {
} }
ddl.WriteString(col.Name) ddl.WriteString(col.Name)
} }
ddl.WriteString(") REFERENCES " + c.RefTable.Name + "(")
ddl.WriteString(") REFERENCES ")
ddl.WriteString(util.CamelToSnake(c.RefTypeName))
ddl.WriteString("(")
for i, col := range c.RefColumns { for i, col := range c.RefColumns {
if i != 0 { if i != 0 {
+103
View File
@@ -0,0 +1,103 @@
package schema
import (
"reflect"
"strings"
"git.gdulai.com/gdulai/simpleorm/util"
log "gitlab.com/gdulai/simpleloglvl"
)
type Parser struct {
typ reflect.Type
columns []Column
constraints []Constraint
}
func NewParser[T any](obj T) *Parser {
objType := reflect.TypeOf(obj)
return &Parser{typ: objType}
}
// Step 1 of the parsing, it creates the table instance and
func (p *Parser) ParseColumns() {
var columns []Column
for field := range p.typ.Fields() {
field := field
col := NewColumn(util.CamelToSnake((field.Name)), field.Name, determineType(field.Type), field.Tag)
columns = append(columns, col)
}
p.columns = columns
}
// Step 2 of the parsing, it creates the constrains with the table references
func (p *Parser) ParseConstraints(tableParsers map[string]*Parser) {
pkConstraint := Constraint{Name: "pk_" + strings.ToLower(util.CamelToSnake((p.typ.Name()))), Type: "pk"}
fkConstraints := make(map[string]Constraint)
for _, col := range p.columns {
_, ok := col.Modifiers["pk"]
if ok {
pkConstraint.Columns = append(pkConstraint.Columns, col)
continue
}
fkMod, ok := col.Modifiers["fk"]
if !ok {
continue
}
fkModParts := strings.Split(fkMod, ".")
refTypeName := fkModParts[0]
refTableName := util.CamelToSnake(refTypeName)
parser, ok := tableParsers[refTypeName]
if !ok {
log.LogError("[ORM] Reference table not found", refTypeName)
}
fkId, ok := col.Modifiers["fk_id"]
if fkId == "" {
fkId = "fk_" + strings.ToLower(refTableName)
}
fkConstraint, ok := fkConstraints[fkId]
if !ok {
fkConstraint = Constraint{Name: fkId, Type: "fk", RefTypeName: refTypeName}
}
fkConstraint.Columns = append(fkConstraint.Columns, col)
refField := fkModParts[1]
for _, refC := range parser.columns {
if refC.FieldName == refField {
fkConstraint.RefColumns = append(fkConstraint.RefColumns, refC)
}
}
fkConstraints[fkId] = fkConstraint
}
p.constraints = append(p.constraints, pkConstraint)
for _, fkConstraint := range fkConstraints {
p.constraints = append(p.constraints, fkConstraint)
}
}
// Step 3 of the paring, create the schema.Table instance
// Returns the schema.Table pointer
func (p Parser) ParseTable() *Table {
return &Table{name: util.CamelToSnake(p.typ.Name()), typ: p.typ, columns: p.columns, constraints: p.constraints}
}
func determineType(typ reflect.Type) string {
typStr := typ.String()
switch typStr {
case "string":
return "TEXT"
case "int", "bool":
return "INTEGER"
case "time.Time", "int64":
return "BIGINT"
}
return "VARCHAR(255)"
}
+91 -25
View File
@@ -1,23 +1,47 @@
package schema package schema
import ( import (
"errors"
"reflect" "reflect"
"strings" "strings"
"git.gdulai.com/gdulai/simpleorm/util" "git.gdulai.com/gdulai/simpleorm/util"
) )
type OrderBy struct {
Field string
Direction string
}
type Table struct { type Table struct {
Name string name string
Type reflect.Type typ reflect.Type
Columns []Column columns []Column
Constraints []Constraint constraints []Constraint
}
func (t Table) Name() string {
return t.name
}
func (t Table) Type() reflect.Type {
return t.typ
}
func (t Table) Columns() []Column {
return t.columns
}
func (t Table) Constraints() []Constraint {
return t.constraints
} }
func (t Table) GetDDL() (string, error) { func (t Table) GetDDL() (string, error) {
var ddl strings.Builder var ddl strings.Builder
ddl.WriteString("CREATE TABLE IF NOT EXISTS " + t.Name + " (") ddl.WriteString("CREATE TABLE IF NOT EXISTS ")
for i, col := range t.Columns { ddl.WriteString(t.name)
ddl.WriteString(" (")
for i, col := range t.columns {
if i != 0 { if i != 0 {
ddl.WriteString(", ") ddl.WriteString(", ")
} }
@@ -27,12 +51,13 @@ func (t Table) GetDDL() (string, error) {
} }
ddl.WriteString(colDdl) ddl.WriteString(colDdl)
} }
for _, constr := range t.Constraints { for _, constr := range t.constraints {
constrDdl, err := constr.GetDDL() constrDdl, err := constr.GetDDL()
if err != nil { if err != nil {
return "", err return "", err
} }
ddl.WriteString(", " + constrDdl) ddl.WriteString(", ")
ddl.WriteString(constrDdl)
} }
ddl.WriteString(");") ddl.WriteString(");")
@@ -43,7 +68,7 @@ func (t Table) GetSelectDML() (string, error) {
var dml strings.Builder var dml strings.Builder
dml.WriteString("SELECT ") dml.WriteString("SELECT ")
for i, col := range t.Columns { for i, col := range t.columns {
if i != 0 { if i != 0 {
dml.WriteString(", ") dml.WriteString(", ")
} }
@@ -54,17 +79,24 @@ func (t Table) GetSelectDML() (string, error) {
dml.WriteString(colDml) dml.WriteString(colDml)
} }
dml.WriteString(" FROM " + t.Name) dml.WriteString(" FROM ")
dml.WriteString(t.name)
return dml.String(), nil return dml.String(), nil
} }
func (t Table) GetCountDML() string {
return "SELECT COUNT(*) FROM " + t.name
}
func (t Table) GetInsertDML(count int) (string, error) { func (t Table) GetInsertDML(count int) (string, error) {
var dml strings.Builder var dml strings.Builder
dml.WriteString("INSERT INTO " + util.CamelToSnake(t.Type.Name()) + " (") dml.WriteString("INSERT INTO ")
dml.WriteString(util.CamelToSnake(t.typ.Name()))
dml.WriteString(" (")
columnsLen := len(t.Columns) columnsLen := len(t.columns)
effectiveColumnsLen := 0 effectiveColumnsLen := 0
for i, col := range t.Columns { for i, col := range t.columns {
_, ok := col.Modifiers["pk"] _, ok := col.Modifiers["pk"]
if ok && t.IsPkAuto() { if ok && t.IsPkAuto() {
continue continue
@@ -100,19 +132,21 @@ func (t Table) GetInsertDML(count int) (string, error) {
} }
} }
dml.WriteString("RETURNING *") dml.WriteString(" RETURNING *")
return dml.String(), nil return dml.String(), nil
} }
func (t Table) GetUpdateDML() (string, error) { func (t Table) GetUpdateDML() (string, error) {
var dml strings.Builder var dml strings.Builder
dml.WriteString("UPDATE " + util.CamelToSnake(t.Type.Name()) + " SET ") dml.WriteString("UPDATE ")
dml.WriteString(util.CamelToSnake(t.typ.Name()))
dml.WriteString(" SET ")
colLen := len(t.Columns) colLen := len(t.columns)
var pkColumns []string var pkColumns []string
// Colum names // Colum names
for i, col := range t.Columns { for i, col := range t.columns {
modifiers := col.Modifiers modifiers := col.Modifiers
_, ok := modifiers["pk"] _, ok := modifiers["pk"]
@@ -129,9 +163,11 @@ func (t Table) GetUpdateDML() (string, error) {
} }
if i != colLen-1 { if i != colLen-1 {
dml.WriteString(colDml + ", ") dml.WriteString(colDml)
dml.WriteString(", ")
} else { } else {
dml.WriteString(colDml + " ") dml.WriteString(colDml)
dml.WriteString(" ")
} }
} }
@@ -141,7 +177,8 @@ func (t Table) GetUpdateDML() (string, error) {
if i != 0 { if i != 0 {
dml.WriteString(" AND ") dml.WriteString(" AND ")
} }
dml.WriteString(pkCol + " = ?") dml.WriteString(pkCol)
dml.WriteString(" = ?")
} }
return dml.String(), nil return dml.String(), nil
@@ -149,17 +186,21 @@ func (t Table) GetUpdateDML() (string, error) {
func (t Table) GetDeleteDML(count int) (string, error) { func (t Table) GetDeleteDML(count int) (string, error) {
var dml strings.Builder var dml strings.Builder
dml.WriteString("DELETE FROM " + t.Name + " WHERE ") dml.WriteString("DELETE FROM ")
dml.WriteString(t.name)
dml.WriteString(" WHERE ")
for _, col := range t.Columns { for _, col := range t.columns {
modifiers := col.Modifiers modifiers := col.Modifiers
_, ok := modifiers["pk"] _, ok := modifiers["pk"]
if ok { if ok {
if count > 1 { if count > 1 {
dml.WriteString(col.Name + " IN (") dml.WriteString(col.Name)
dml.WriteString(" IN (")
} else { } else {
dml.WriteString(col.Name + " = ") dml.WriteString(col.Name)
dml.WriteString(" = ")
} }
} }
} }
@@ -178,8 +219,33 @@ func (t Table) GetDeleteDML(count int) (string, error) {
return dml.String(), nil return dml.String(), nil
} }
func (t Table) GetOrderByDML(ordering ...OrderBy) (string, error) {
var dml strings.Builder
dml.WriteString(" ORDER BY")
for _, orderBy := range ordering {
col, err := t.getColumnByField(orderBy.Field)
if err != nil {
return "", err
}
dml.WriteString(" ")
dml.WriteString(col.Name)
dml.WriteString(" ")
dml.WriteString(orderBy.Direction)
}
return dml.String(), nil
}
func (t Table) getColumnByField(fieldName string) (Column, error) {
for _, col := range t.columns {
if col.FieldName == fieldName {
return col, nil
}
}
return Column{}, errors.New("No column for field " + fieldName + " in table " + t.name)
}
func (t Table) IsPkAuto() bool { func (t Table) IsPkAuto() bool {
for _, constr := range t.Constraints { for _, constr := range t.constraints {
if constr.Type != "pk" { if constr.Type != "pk" {
continue continue
} }
+104 -222
View File
@@ -1,6 +1,7 @@
package simpleorm_test package simpleorm_test
import ( import (
"strconv"
"testing" "testing"
"time" "time"
@@ -8,36 +9,6 @@ import (
log "gitlab.com/gdulai/simpleloglvl" log "gitlab.com/gdulai/simpleloglvl"
) )
func TestSelectEmpty(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
// WHEN
selectExec, err := exec.CreateSelect[Test](orm, "")
// THEN
if err != nil {
log.LogError("Failed to create select. %s", err)
t.Fail()
return
}
err = selectExec.Execute(conn)
if err != nil {
log.LogError("Select failure. %s", err)
t.Fail()
return
}
res := selectExec.Results()
if len(res) > 0 {
log.LogError("Expected empty result.")
t.Fail()
return
}
}
func TestInsert(t *testing.T) { func TestInsert(t *testing.T) {
// GIVEN // GIVEN
orm, conn := testSetup() orm, conn := testSetup()
@@ -132,193 +103,6 @@ func TestInsertMultiple(t *testing.T) {
} }
func TestInsertAndSelectSingle(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
// WHEN
insertExec, err := exec.NewInsert(orm, testObj)
if err != nil {
log.LogError("TestInsertAndSelectSingle setup failed: %s", err)
t.Fail()
return
}
selectExec, err := exec.CreateSelect[Test](orm, "")
if err != nil {
log.LogError("TestInsertAndSelectSingle setup failed: %s", err)
t.Fail()
return
}
insertErr := insertExec.Execute(conn)
selectErr := selectExec.Execute(conn)
// THEN
if insertErr != nil {
log.LogError("TestInsertAndSelectSingle insert failed: %s", err)
t.Fail()
return
}
if selectErr != nil {
log.LogError("TestInsertAndSelectSingle failure. %s", err)
t.Fail()
return
}
insertRes := insertExec.Results()
if len(insertRes) != 1 {
log.LogError("TestInsertAndSelectSingle test failed. Expected: 1, Actual: %s", len(insertRes))
t.Fail()
return
}
selectRes := selectExec.Results()
if len(selectRes) != 1 {
log.LogError("TestInsertAndSelectSingle test failed. Expected: 1, Actual: %s", len(selectRes))
t.Fail()
return
}
singleRes := insertRes[0]
if singleRes.ID != 1 || singleRes.Int64Field != 54 || singleRes.IntField != 12 || singleRes.StringField != "asdf" {
log.LogError("TestInsertAndSelectSingle invalid result. Expected: %s, Actual: %s", testObj, singleRes)
t.Fail()
}
singleRes = selectRes[0]
if singleRes.ID != 1 || singleRes.Int64Field != 54 || singleRes.IntField != 12 || singleRes.StringField != "asdf" {
log.LogError("TestInsertAndSelectSingle invalid result. Expected: %s, Actual: %s", testObj, singleRes)
t.Fail()
}
}
func TestInsertSelectWithCompositePk(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
testObj := TestWithCompositePk{ID: 12, Name: "Test"}
// WHEN
insertExec, err := exec.NewInsert(orm, testObj)
if err != nil {
log.LogError("TestInsertSelectWithCompositePk setup failed: %s", err)
t.Fail()
return
}
selectExec, err := exec.CreateSelect[TestWithCompositePk](orm, "")
if err != nil {
log.LogError("TestInsertSelectWithCompositePk setup failed: %s", err)
t.Fail()
return
}
err = insertExec.Execute(conn)
if err != nil {
log.LogError("TestInsertSelectWithCompositePk insert failed: %s", err)
t.Fail()
return
}
err = selectExec.Execute(conn)
// THEN
if err != nil {
log.LogError("TestInsertSelectWithCompositePk failure. %s", err)
t.Fail()
return
}
res := selectExec.Results()
if len(res) != 1 {
log.LogError("TestInsertSelectWithCompositePk test failed. Expected: 1, Actual: %s", len(res))
t.Fail()
return
}
singleRes := res[0]
if singleRes.ID != 12 || singleRes.Name != "Test" {
log.LogError("TestInsertSelectWithCompositePk invalid result. Expected: %s, Actual: %s", testObj, singleRes)
t.Fail()
}
}
func TestInsertMultipleAndSelectWithParam(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
// WHEN
testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
testObjSecond := Test{Int64Field: 12, IntField: 54, StringField: "fdsa"}
insertExec, err := exec.NewInsert(orm, testObj)
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
t.Fail()
return
}
insertSecond, err := exec.NewInsert(orm, testObjSecond)
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
t.Fail()
return
}
selectExec, err := exec.CreateSelect[Test](orm, "string_field = ?", "fdsa")
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
t.Fail()
return
}
err = insertExec.Execute(conn)
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam first insert failed: %s", err)
t.Fail()
return
}
err = insertSecond.Execute(conn)
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam second insert failed: %s", err)
t.Fail()
return
}
err = selectExec.Execute(conn)
// THEN
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam failure. %s", err)
t.Fail()
return
}
res := selectExec.Results()
if len(res) != 1 {
log.LogError(" TestInsertMultipleAndSelectWithParam test failed. Expected: 1, Actual: %s", len(res))
t.Fail()
return
}
singleRes := res[0]
if singleRes.ID != 2 || singleRes.Int64Field != 12 || singleRes.IntField != 54 || singleRes.StringField != "fdsa" {
log.LogError("TestInsertMultipleAndSelectWithParam invalid result. Expected: %s, Actual: %s", testObjSecond, singleRes)
t.Fail()
return
}
}
func TestInsertAndUpdate(t *testing.T) { func TestInsertAndUpdate(t *testing.T) {
// GIVEN // GIVEN
orm, conn := testSetup() orm, conn := testSetup()
@@ -334,13 +118,15 @@ func TestInsertAndUpdate(t *testing.T) {
return return
} }
selectExec, err := exec.CreateSelect[Test](orm, "string_field = ?", "asdf") selectExec, err := exec.CreateSelect[Test](orm)
if err != nil { if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err) log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
t.Fail() t.Fail()
return return
} }
selectExec.Where("string_field = ?", "asdf")
err = insertExec.Execute(conn) err = insertExec.Execute(conn)
if err != nil { if err != nil {
log.LogError("TestInsertAndUpdate insert failed: %s", err) log.LogError("TestInsertAndUpdate insert failed: %s", err)
@@ -421,7 +207,7 @@ func TestInsertAndDelete(t *testing.T) {
return return
} }
selectExec, err := exec.CreateSelect[Test](orm, "") selectExec, err := exec.CreateSelect[Test](orm)
if err != nil { if err != nil {
log.LogError("TestInsertAndDelete setup failed: %s", err) log.LogError("TestInsertAndDelete setup failed: %s", err)
t.Fail() t.Fail()
@@ -477,7 +263,7 @@ func TestBoolInsertAndSelectSingle(t *testing.T) {
return return
} }
selectExec, err := exec.CreateSelect[TestWithBool](orm, "") selectExec, err := exec.CreateSelect[TestWithBool](orm)
if err != nil { if err != nil {
log.LogError("TestBoolInsertAndSelectSingle setup failed: %s", err) log.LogError("TestBoolInsertAndSelectSingle setup failed: %s", err)
t.Fail() t.Fail()
@@ -531,7 +317,7 @@ func TestTimeInsertAndSelectSingle(t *testing.T) {
return return
} }
selectExec, err := exec.CreateSelect[TestWithTime](orm, "") selectExec, err := exec.CreateSelect[TestWithTime](orm)
if err != nil { if err != nil {
log.LogError("TestTimeInsertAndSelectSingle setup failed: %s", err) log.LogError("TestTimeInsertAndSelectSingle setup failed: %s", err)
t.Fail() t.Fail()
@@ -637,13 +423,15 @@ func TestTransactionRollback(t *testing.T) {
return return
} }
selectExec, err := exec.CreateSelect[Test](orm, "id = ?", insertExec.Results()[0].ID) selectExec, err := exec.CreateSelect[Test](orm)
if err != nil { if err != nil {
log.LogError("TestTransactionRollback WHEN failed: %s", err) log.LogError("TestTransactionRollback WHEN failed: %s", err)
t.Fail() t.Fail()
return return
} }
selectExec.Where("id = ?", insertExec.Results()[0].ID)
transErr = transaction.Execute(&selectExec) transErr = transaction.Execute(&selectExec)
if transErr != nil { if transErr != nil {
log.LogError("TestTransactionRollback WHEN failed: %s", err) log.LogError("TestTransactionRollback WHEN failed: %s", err)
@@ -681,3 +469,97 @@ func TestTransactionRollback(t *testing.T) {
return return
} }
} }
func TestCount(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
testEntites := []Test{}
for i := 0; i < 1000; i++ {
testEntites = append(testEntites, Test{Int64Field: -1, IntField: i, StringField: "Entity " + strconv.Itoa(i)})
}
insertExec, err := exec.NewInsert[Test](orm, testEntites...)
if err != nil {
log.LogError("TestSelectWithLimitAndOffset setup failed: %s", err)
t.Fail()
return
}
err = insertExec.Execute(conn)
if err != nil {
log.LogError("TestSelectWithLimitAndOffset setup failed: %s", err)
t.Fail()
return
}
// WHEN
countExec, err := exec.CreateCount[Test](orm, "")
if err != nil {
log.LogError("TestSelectWithLimitAndOffset failed: %s", err)
t.Fail()
return
}
err = countExec.Execute(conn)
if err != nil {
log.LogError("TestSelectWithLimitAndOffset failed: %s", err)
t.Fail()
return
}
// THEN
if countExec.Result() != 1000 {
log.LogError("TestSelectWithLimitAndOffset expected result size 1000, actual: %s", countExec.Result())
t.Fail()
}
}
func TestCountWithCondition(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
testEntites := []Test{}
for i := 0; i < 1000; i++ {
testEntites = append(testEntites, Test{Int64Field: -1, IntField: i, StringField: "Entity " + strconv.Itoa(i)})
}
insertExec, err := exec.NewInsert[Test](orm, testEntites...)
if err != nil {
log.LogError("TestSelectWithLimitAndOffset setup failed: %s", err)
t.Fail()
return
}
err = insertExec.Execute(conn)
if err != nil {
log.LogError("TestSelectWithLimitAndOffset setup failed: %s", err)
t.Fail()
return
}
// WHEN
countExec, err := exec.CreateCount[Test](orm, "id % 2 = 0")
if err != nil {
log.LogError("TestSelectWithLimitAndOffset failed: %s", err)
t.Fail()
return
}
err = countExec.Execute(conn)
if err != nil {
log.LogError("TestSelectWithLimitAndOffset failed: %s", err)
t.Fail()
return
}
// THEN
if countExec.Result() != 500 {
log.LogError("TestSelectWithLimitAndOffset expected result size 500, actual: %s", countExec.Result())
t.Fail()
}
}
+419
View File
@@ -0,0 +1,419 @@
package simpleorm_test
import (
"strconv"
"testing"
simpleorm "git.gdulai.com/gdulai/simpleorm"
"git.gdulai.com/gdulai/simpleorm/exec"
"git.gdulai.com/gdulai/simpleorm/schema"
log "gitlab.com/gdulai/simpleloglvl"
)
func TestSelectEmpty(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
// WHEN
selectExec, err := exec.CreateSelect[Test](orm)
// THEN
if err != nil {
log.LogError("Failed to create select. %s", err)
t.Fail()
return
}
err = selectExec.Execute(conn)
if err != nil {
log.LogError("Select failure. %s", err)
t.Fail()
return
}
res := selectExec.Results()
if len(res) > 0 {
log.LogError("Expected empty result.")
t.Fail()
return
}
}
func TestSelectSingle(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
// WHEN
insertExec, err := exec.NewInsert(orm, testObj)
if err != nil {
log.LogError("TestInsertAndSelectSingle setup failed: %s", err)
t.Fail()
return
}
selectExec, err := exec.CreateSelect[Test](orm)
if err != nil {
log.LogError("TestInsertAndSelectSingle setup failed: %s", err)
t.Fail()
return
}
insertErr := insertExec.Execute(conn)
selectErr := selectExec.Execute(conn)
// THEN
if insertErr != nil {
log.LogError("TestInsertAndSelectSingle insert failed: %s", err)
t.Fail()
return
}
if selectErr != nil {
log.LogError("TestInsertAndSelectSingle failure. %s", err)
t.Fail()
return
}
insertRes := insertExec.Results()
if len(insertRes) != 1 {
log.LogError("TestInsertAndSelectSingle test failed. Expected: 1, Actual: %s", len(insertRes))
t.Fail()
return
}
selectRes := selectExec.Results()
if len(selectRes) != 1 {
log.LogError("TestInsertAndSelectSingle test failed. Expected: 1, Actual: %s", len(selectRes))
t.Fail()
return
}
singleRes := insertRes[0]
if singleRes.ID != 1 || singleRes.Int64Field != 54 || singleRes.IntField != 12 || singleRes.StringField != "asdf" {
log.LogError("TestInsertAndSelectSingle invalid result. Expected: %s, Actual: %s", testObj, singleRes)
t.Fail()
}
singleRes = selectRes[0]
if singleRes.ID != 1 || singleRes.Int64Field != 54 || singleRes.IntField != 12 || singleRes.StringField != "asdf" {
log.LogError("TestInsertAndSelectSingle invalid result. Expected: %s, Actual: %s", testObj, singleRes)
t.Fail()
}
}
func TestSelectWithCompositePk(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
testObj := TestWithCompositePk{ID: 12, Name: "Test"}
// WHEN
insertExec, err := exec.NewInsert(orm, testObj)
if err != nil {
log.LogError("TestInsertSelectWithCompositePk setup failed: %s", err)
t.Fail()
return
}
selectExec, err := exec.CreateSelect[TestWithCompositePk](orm)
if err != nil {
log.LogError("TestInsertSelectWithCompositePk setup failed: %s", err)
t.Fail()
return
}
err = insertExec.Execute(conn)
if err != nil {
log.LogError("TestInsertSelectWithCompositePk insert failed: %s", err)
t.Fail()
return
}
err = selectExec.Execute(conn)
// THEN
if err != nil {
log.LogError("TestInsertSelectWithCompositePk failure. %s", err)
t.Fail()
return
}
res := selectExec.Results()
if len(res) != 1 {
log.LogError("TestInsertSelectWithCompositePk test failed. Expected: 1, Actual: %s", len(res))
t.Fail()
return
}
singleRes := res[0]
if singleRes.ID != 12 || singleRes.Name != "Test" {
log.LogError("TestInsertSelectWithCompositePk invalid result. Expected: %s, Actual: %s", testObj, singleRes)
t.Fail()
}
}
func TestSelectWithParam(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
// WHEN
testObj := Test{Int64Field: 54, IntField: 12, StringField: "asdf"}
testObjSecond := Test{Int64Field: 12, IntField: 54, StringField: "fdsa"}
insertExec, err := exec.NewInsert(orm, testObj)
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
t.Fail()
return
}
insertSecond, err := exec.NewInsert(orm, testObjSecond)
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
t.Fail()
return
}
selectExec, err := exec.CreateSelect[Test](orm)
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam setup failed: %s", err)
t.Fail()
return
}
selectExec.Where("string_field = ?", "fdsa")
err = insertExec.Execute(conn)
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam first insert failed: %s", err)
t.Fail()
return
}
err = insertSecond.Execute(conn)
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam second insert failed: %s", err)
t.Fail()
return
}
err = selectExec.Execute(conn)
// THEN
if err != nil {
log.LogError("TestInsertMultipleAndSelectWithParam failure. %s", err)
t.Fail()
return
}
res := selectExec.Results()
if len(res) != 1 {
log.LogError(" TestInsertMultipleAndSelectWithParam test failed. Expected: 1, Actual: %s", len(res))
t.Fail()
return
}
singleRes := res[0]
if singleRes.ID != 2 || singleRes.Int64Field != 12 || singleRes.IntField != 54 || singleRes.StringField != "fdsa" {
log.LogError("TestInsertMultipleAndSelectWithParam invalid result. Expected: %s, Actual: %s", testObjSecond, singleRes)
t.Fail()
return
}
}
func TestSelectWithLimitAndOffset(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
testEntites := []Test{}
for i := 0; i < 1000; i++ {
testEntites = append(testEntites, Test{Int64Field: -1, IntField: i, StringField: "Entity " + strconv.Itoa(i)})
}
insertExec, err := exec.NewInsert[Test](orm, testEntites...)
if err != nil {
log.LogError("TestSelectWithLimitAndOffset setup failed: %s", err)
t.Fail()
return
}
err = insertExec.Execute(conn)
if err != nil {
log.LogError("TestSelectWithLimitAndOffset setup failed: %s", err)
t.Fail()
return
}
// WHEN
selectExec, err := exec.CreateSelect[Test](orm)
if err != nil {
log.LogError("TestSelectWithLimitAndOffset failed: %s", err)
t.Fail()
return
}
selectExec.Limit(100).Offset(0)
err = selectExec.Execute(conn)
if err != nil {
log.LogError("TestSelectWithLimitAndOffset failed: %s", err)
t.Fail()
return
}
// THEN
if len(selectExec.Results()) != 100 {
log.LogError("TestSelectWithLimitAndOffset expected result size 100, actual: %s", len(selectExec.Results()))
t.Fail()
}
}
func TestSelectOrderByDesc(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
err := orderBySetup(conn, orm)
if err != nil {
log.LogError("TestSelectOrderByAsc setup failed: %s", err)
t.Fail()
return
}
// WHEN
selectExec, err := exec.CreateSelect[Test](orm)
if err != nil {
log.LogError("TestSelectOrderByDesc failed: %s", err)
t.Fail()
return
}
selectExec.OrderBy(schema.OrderBy{Field: "StringField", Direction: "DESC"})
err = selectExec.Execute(conn)
if err != nil {
log.LogError("TestSelectOrderByDesc failed: %s", err)
t.Fail()
return
}
// THEN
var resultStr string
for _, obj := range selectExec.Results() {
resultStr += obj.StringField
}
if resultStr != "EDCBA" {
log.LogError("TestSelectOrderByDesc failed, epxected: EDCBA actual: %s", resultStr)
t.Fail()
}
}
func TestSelectOrderByAsc(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
err := orderBySetup(conn, orm)
if err != nil {
log.LogError("TestSelectOrderByAsc setup failed: %s", err)
t.Fail()
return
}
// WHEN
selectExec, err := exec.CreateSelect[Test](orm)
if err != nil {
log.LogError("TestSelectOrderByAsc failed: %s", err)
t.Fail()
return
}
selectExec.OrderBy(schema.OrderBy{Field: "StringField", Direction: "ASC"})
err = selectExec.Execute(conn)
if err != nil {
log.LogError("TestSelectOrderByAsc failed: %s", err)
t.Fail()
return
}
// THEN
var resultStr string
for _, obj := range selectExec.Results() {
resultStr += obj.StringField
}
if resultStr != "ABCDE" {
log.LogError("TestSelectOrderByAsc failed, epxected: ABCDE actual: %s", resultStr)
t.Fail()
}
}
func TestSelectOrderByAscWithLimitAndOffset(t *testing.T) {
// GIVEN
orm, conn := testSetup()
defer cleanUp("test.db", conn)
err := orderBySetup(conn, orm)
if err != nil {
log.LogError("TestSelectOrderByAsc setup failed: %s", err)
t.Fail()
return
}
// WHEN
selectExec, err := exec.CreateSelect[Test](orm)
if err != nil {
log.LogError("TestSelectOrderByAsc failed: %s", err)
t.Fail()
return
}
selectExec.OrderBy(schema.OrderBy{Field: "StringField", Direction: "ASC"}).Limit(3).Offset(0)
err = selectExec.Execute(conn)
if err != nil {
log.LogError("TestSelectOrderByAsc failed: %s", err)
t.Fail()
return
}
// THEN
var resultStr string
for _, obj := range selectExec.Results() {
resultStr += obj.StringField
}
if resultStr != "ABC" {
log.LogError("TestSelectOrderByAsc failed, epxected: ABC actual: %s", resultStr)
t.Fail()
}
}
func orderBySetup(conn *simpleorm.DBConnection, orm *simpleorm.ORM) error {
insertExec, err := exec.NewInsert[Test](orm,
Test{Int64Field: -1, IntField: 1, StringField: "C"},
Test{Int64Field: -1, IntField: 1, StringField: "D"},
Test{Int64Field: -1, IntField: 1, StringField: "E"},
Test{Int64Field: -1, IntField: 1, StringField: "A"},
Test{Int64Field: -1, IntField: 1, StringField: "B"},
)
if err != nil {
return err
}
err = insertExec.Execute(conn)
if err != nil {
return err
}
return nil
}