Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 18 additions & 0 deletions builder.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,10 @@ package ormshift

import (
"database/sql"
"errors"
"slices"

"github.com/ordershift/ormshift/errs"
"github.com/ordershift/ormshift/schema"
)

Expand All @@ -19,6 +21,22 @@ type DDLSQLBuilder interface {
// ColumnsValues represents a mapping between column names and their corresponding values.
type ColumnsValues map[string]any

// NewColumnsValues helps to initialize a pointer for ColumnsValues map​, with given columns and values slices params.
func NewColumnsValues(columns []string, values []any) (*ColumnsValues, error) {
if len(columns) != len(values) {
return nil, failedToGetColumnsValues(errors.New("columns len must be equal to values len"))
}
cv := ColumnsValues{}
for i, c := range columns {
cv[c] = values[i]
}
return &cv, nil
}

func failedToGetColumnsValues(err error) error {
return errs.FailedTo("get columns values", err)
}

// ToNamedArgs transforms ColumnsValues to a sql.NamedArg array ordered by name, e.g.:
//
// values := ColumnsValues{"id": 5, "sku": "ZTX-9000", "is_simple": true}
Expand Down
22 changes: 22 additions & 0 deletions builder_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -26,3 +26,25 @@ func TestColumnsValuesToColumns(t *testing.T) {
testutils.AssertEqualWithLabel(t, columns[0], "id", "ColumnsValues.ToColumns[0]")
testutils.AssertEqualWithLabel(t, columns[1], "sku", "ColumnsValues.ToColumns[1]")
}

func TestNewColumnsValues(t *testing.T) {
columns := []string{"id", "sku"}
values := []any{1, "ABC1234"}
cv, err := ormshift.NewColumnsValues(columns, values)
if !testutils.AssertNotNilResultAndNilError(t, cv, err, "ormshift.NewColumnsValues") {
return
}
testutils.AssertEqualWithLabel(t, 2, len(*cv), "len(ColumnsValues)")
testutils.AssertEqualWithLabel(t, (*cv)["id"], 1, "id")
testutils.AssertEqualWithLabel(t, (*cv)["sku"], "ABC1234", "sku")
}

func TestNewColumnsValuesFailsWhenIncompatibleLengths(t *testing.T) {
columns := []string{"id", "sku"}
values := []any{1}
cv, err := ormshift.NewColumnsValues(columns, values)
if !testutils.AssertNilResultAndNotNilError(t, cv, err, "ormshift.NewColumnsValues") {
return
}
testutils.AssertErrorMessage(t, "failed to get columns values: columns len must be equal to values len", err, "ormshift.NewColumnsValues")
}