From 85473656174b1b1d6221d3bb76cc12fa5f7f7e8d Mon Sep 17 00:00:00 2001 From: Laria Carolin Chabowski Date: Fri, 1 Jan 2021 14:46:54 +0100 Subject: Initial commit --- dbutils/dbutils.go | 60 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 60 insertions(+) create mode 100644 dbutils/dbutils.go (limited to 'dbutils') diff --git a/dbutils/dbutils.go b/dbutils/dbutils.go new file mode 100644 index 0000000..4082efd --- /dev/null +++ b/dbutils/dbutils.go @@ -0,0 +1,60 @@ +package dbutils + +import ( + "database/sql" + "fmt" + "strings" +) + +func TxCommitIfOk(tx *sql.Tx, err error) error { + if err != nil { + if rollbackErr := tx.Rollback(); rollbackErr != nil { + return fmt.Errorf("Failed rolling back transaction with error \"%s\" while handling error %w", rollbackErr, err) + } + + return err + } else { + if err = tx.Commit(); err != nil { + return err + } + + return nil + } +} + +func BuildInIdsSqlAndArgs(field string, ids []int) (string, []interface{}) { + if len(ids) == 0 { + return "0", []interface{}{} + } + + sb := new(strings.Builder) + sb.WriteString(field) + sb.WriteString(" IN (?") + + for i := 1; i < len(ids); i++ { + sb.WriteString(",?") + } + sb.WriteRune(')') + + args := make([]interface{}, 0, len(ids)) + for _, id := range ids { + args = append(args, interface{}(id)) + } + + return sb.String(), args +} + +// GetIndexedValues +// func GetIndexedValues(db *sql.DB, m interface{}, query, args ...interface{}) error { +// mv := reflect.ValueOf(m) +// mt := mv.Type() +// if mt.Kind() != reflect.Map { +// panic("GetIndexValue needs a map as argument m") +// } +// if mt. + +// rows, err := db.Query(query, args...) +// if err != nil { +// return err +// } +// } -- cgit v1.2.3-70-g09d2