mirror of https://github.com/aldy505/bob.git
feat: upsert (test not working yet)
This commit is contained in:
parent
50c8d98f3a
commit
4a75bedea3
19
bob.go
19
bob.go
|
|
@ -10,6 +10,15 @@ import (
|
|||
var ErrEmptyTable = errors.New("sql: no rows in result set")
|
||||
// ErrEmptyTable is a common pgx error if a table is empty or no rows is returned by the query.
|
||||
var ErrEmptyTablePgx = errors.New("no rows in result set")
|
||||
// ErrDialectNotSupported tells you whether the dialect is supported or not.
|
||||
var ErrDialectNotSupported = errors.New("provided database dialect is not supported")
|
||||
|
||||
const (
|
||||
Mysql int = iota
|
||||
Postgresql
|
||||
Sqlite
|
||||
MSSql
|
||||
)
|
||||
|
||||
// BobBuilderType is the type for BobBuilder
|
||||
type BobBuilderType builder.Builder
|
||||
|
|
@ -59,6 +68,10 @@ func (b BobBuilderType) Truncate(table string) TruncateBuilder {
|
|||
return TruncateBuilder(b).Truncate(table)
|
||||
}
|
||||
|
||||
func (b BobBuilderType) Upsert(table string, dialect int) UpsertBuilder {
|
||||
return UpsertBuilder(b).dialect(dialect).Into(table)
|
||||
}
|
||||
|
||||
// BobStmtBuilder is the parent builder for BobBuilderType
|
||||
var BobStmtBuilder = BobBuilderType(builder.EmptyBuilder)
|
||||
|
||||
|
|
@ -100,4 +113,10 @@ func RenameTable(from, to string) RenameBuilder {
|
|||
// Truncate performs TRUNCATE function. It deletes all contents from a table but not deleting the table.
|
||||
func Truncate(table string) TruncateBuilder {
|
||||
return BobStmtBuilder.Truncate(table)
|
||||
}
|
||||
|
||||
// Upsert performs a UPSERT query with specified database dialect.
|
||||
// Supported database includes MySQL, PostgreSQL, SQLite and MSSQL.
|
||||
func Upsert(table string, dialect int) UpsertBuilder {
|
||||
return BobStmtBuilder.Upsert(table, dialect)
|
||||
}
|
||||
|
|
@ -0,0 +1,172 @@
|
|||
package bob
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
"strings"
|
||||
|
||||
"github.com/lann/builder"
|
||||
)
|
||||
|
||||
type UpsertBuilder builder.Builder
|
||||
|
||||
type upsertData struct {
|
||||
Dialect int
|
||||
Into string
|
||||
Columns []string
|
||||
Values [][]interface{}
|
||||
Key []interface{}
|
||||
Replace [][]interface{}
|
||||
Placeholder string
|
||||
}
|
||||
|
||||
func init() {
|
||||
builder.Register(UpsertBuilder{}, upsertData{})
|
||||
}
|
||||
|
||||
func (b UpsertBuilder) dialect(db int) UpsertBuilder {
|
||||
return builder.Set(b, "Dialect", db).(UpsertBuilder)
|
||||
}
|
||||
|
||||
// Table sets which table to be dropped
|
||||
func (b UpsertBuilder) Into(name string) UpsertBuilder {
|
||||
return builder.Set(b, "Into", name).(UpsertBuilder)
|
||||
}
|
||||
|
||||
func (b UpsertBuilder) Columns(columns ...string) UpsertBuilder {
|
||||
return builder.Extend(b, "Columns", columns).(UpsertBuilder)
|
||||
}
|
||||
|
||||
// Values sets the values in relation with the columns.
|
||||
// Please not that only string, int, and bool type are supported.
|
||||
// Inputting other types other than those might result in your SQL not working properly.
|
||||
func (b UpsertBuilder) Values(values ...interface{}) UpsertBuilder {
|
||||
return builder.Append(b, "Values", values).(UpsertBuilder)
|
||||
}
|
||||
|
||||
func (b UpsertBuilder) Key(key ...interface{}) UpsertBuilder {
|
||||
return builder.Extend(b, "Key", []interface{}{key[0], key[1]}).(UpsertBuilder)
|
||||
}
|
||||
|
||||
func (b UpsertBuilder) Replace(column interface{}, value interface{}) UpsertBuilder {
|
||||
return builder.Extend(b, "Replace", []interface{}{column, value}).(UpsertBuilder)
|
||||
}
|
||||
|
||||
// PlaceholderFormat changes the default placeholder (?) to desired placeholder.
|
||||
func (b UpsertBuilder) PlaceholderFormat(f string) UpsertBuilder {
|
||||
return builder.Set(b, "Placeholder", f).(UpsertBuilder)
|
||||
}
|
||||
|
||||
// ToSql returns 3 variables filled out with the correct values based on bindings, etc.
|
||||
func (b UpsertBuilder) ToSql() (string, []interface{}, error) {
|
||||
data := builder.GetStruct(b).(upsertData)
|
||||
return data.ToSql()
|
||||
}
|
||||
|
||||
// ToSql returns 3 variables filled out with the correct values based on bindings, etc.
|
||||
func (d *upsertData) ToSql() (sqlStr string, args []interface{}, err error) {
|
||||
if len(d.Into) == 0 {
|
||||
err = errors.New("upsert statements must specify a table")
|
||||
return
|
||||
}
|
||||
|
||||
if len(d.Columns) == 0 {
|
||||
err = errors.New("upsert statement must have at least one column")
|
||||
return
|
||||
}
|
||||
|
||||
if len(d.Values) == 0 {
|
||||
err = errors.New("upsert statements must have at least one set of values")
|
||||
return
|
||||
}
|
||||
|
||||
if len(d.Replace) == 0 {
|
||||
err = errors.New("upsert statement must have at least one key value pair to be replaced")
|
||||
return
|
||||
}
|
||||
|
||||
sql := &bytes.Buffer{}
|
||||
|
||||
if d.Dialect == MSSql {
|
||||
if len(d.Key) == 0 {
|
||||
err = errors.New("unique key and value must be provided for MS SQL")
|
||||
}
|
||||
sql.WriteString("IF NOT EXISTS (SELECT * FROM \""+d.Into+"\" WHERE \""+d.Key[0].(string)+"\" = ?) ")
|
||||
args = append(args, d.Key[1])
|
||||
}
|
||||
|
||||
sql.WriteString("INSERT INTO ")
|
||||
sql.WriteString("\""+d.Into+"\"")
|
||||
sql.WriteString(" ")
|
||||
|
||||
var columns []string
|
||||
for _, v := range d.Columns {
|
||||
columns = append(columns, "\""+v+"\"")
|
||||
}
|
||||
|
||||
sql.WriteString("(")
|
||||
sql.WriteString(strings.Join(columns, ", "))
|
||||
sql.WriteString(") ")
|
||||
|
||||
sql.WriteString("VALUES ")
|
||||
|
||||
var values []string
|
||||
for i := 0; i < len(d.Values); i++ {
|
||||
var tempValues []string
|
||||
for _, v := range d.Values[i] {
|
||||
args = append(args, v)
|
||||
tempValues = append(tempValues, "?")
|
||||
}
|
||||
values = append(values, "("+strings.Join(tempValues, ", ")+")")
|
||||
}
|
||||
|
||||
sql.WriteString(strings.Join(values, ", "))
|
||||
|
||||
var replaces []string
|
||||
for i := 0; i < len(d.Replace); i++ {
|
||||
args = append(args, d.Replace[i][1])
|
||||
replace := "\"" + d.Replace[i][0].(string) + "\" = ?"
|
||||
replaces = append(replaces, replace)
|
||||
}
|
||||
|
||||
if d.Dialect == Mysql {
|
||||
// INSERT INTO table (col) VALUES (values) ON DUPLICATE KEY UPDATE col = value
|
||||
|
||||
sql.WriteString("ON DUPLICATE KEY UPDATE ")
|
||||
sql.WriteString(strings.Join(replaces, ", "))
|
||||
} else if d.Dialect == Postgresql || d.Dialect == Sqlite {
|
||||
// INSERT INTO players (user_name, age) VALUES('steven', 32) ON CONFLICT(user_name) DO UPDATE SET age=excluded.age;
|
||||
|
||||
sql.WriteString("ON CONFLICT ")
|
||||
sql.WriteString("(\""+d.Key[0].(string)+"\") ")
|
||||
sql.WriteString("DO UPDATE SET ")
|
||||
sql.WriteString(strings.Join(replaces, ", "))
|
||||
|
||||
} else if d.Dialect == MSSql {
|
||||
// IF NOT EXISTS (SELECT * FROM dbo.Table1 WHERE ID = @ID)
|
||||
// INSERT INTO dbo.Table1(ID, Name, ItemName, ItemCatName, ItemQty)
|
||||
// VALUES(@ID, @Name, @ItemName, @ItemCatName, @ItemQty)
|
||||
// ELSE
|
||||
// UPDATE dbo.Table1
|
||||
// SET Name = @Name,
|
||||
// ItemName = @ItemName,
|
||||
// ItemCatName = @ItemCatName,
|
||||
// ItemQty = @ItemQty
|
||||
// WHERE ID = @ID
|
||||
|
||||
sql.WriteString("ELSE ")
|
||||
sql.WriteString("UPDATE \""+d.Into+"\" SET ")
|
||||
sql.WriteString(strings.Join(replaces, ", "))
|
||||
sql.WriteString(" WHERE \""+d.Key[0].(string)+"\" = ?")
|
||||
args = append(args, d.Key[1])
|
||||
|
||||
} else {
|
||||
err = ErrDialectNotSupported
|
||||
return
|
||||
}
|
||||
|
||||
sql.WriteString(";")
|
||||
|
||||
sqlStr = ReplacePlaceholder(sql.String(), d.Placeholder)
|
||||
return
|
||||
}
|
||||
|
|
@ -0,0 +1,104 @@
|
|||
package bob_test
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"testing"
|
||||
|
||||
"github.com/aldy505/bob"
|
||||
)
|
||||
|
||||
func TestUpsert(t *testing.T) {
|
||||
t.Run("should be able to generate upsert query for postgres", func(t *testing.T) {
|
||||
sql, args, err := bob.
|
||||
Upsert("users", bob.Postgresql).
|
||||
Columns("name", "email").
|
||||
Values("John Doe", "john@doe.com").
|
||||
Key("email", nil).
|
||||
Replace("name", "John Does").
|
||||
PlaceholderFormat(bob.Dollar).
|
||||
ToSql()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
desiredSql := "INSERT INTO \"users\" (\"name\", \"email\") VALUES ($1, $2) ON CONFLICT (\"email\") DO UPDATE SET \"name\" = $3;"
|
||||
desiredArgs := []interface{}{"John Doe", "john@doe.com", "John Does"}
|
||||
|
||||
if sql != desiredSql {
|
||||
t.Error("sql is not the same as result: ", sql)
|
||||
}
|
||||
if reflect.DeepEqual(args, desiredArgs) {
|
||||
t.Error("args is not the same as result: ", args)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("should be able to generate upsert query for mysql", func(t *testing.T) {
|
||||
sql, args, err := bob.
|
||||
Upsert("users", bob.Mysql).
|
||||
Columns("name", "email").
|
||||
Values("John Doe", "john@doe.com").
|
||||
Replace("name", "John Does").
|
||||
ToSql()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
desiredSql := "INSERT INTO \"users\" (\"name\", \"email\") VALUES (?, ?) ON DUPLICATE KEY UPDATE \"name\" = ?;"
|
||||
desiredArgs := []interface{}{"John Doe", "john@doe.com", "John Does"}
|
||||
|
||||
if sql != desiredSql {
|
||||
t.Error("sql is not the same as result: ", sql)
|
||||
}
|
||||
if reflect.DeepEqual(args, desiredArgs) {
|
||||
t.Error("args is not the same as result: ", args)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("should be able to generate upsert query for sqlite", func(t *testing.T) {
|
||||
sql, args, err := bob.
|
||||
Upsert("users", bob.Sqlite).
|
||||
Columns("name", "email").
|
||||
Values("John Doe", "john@doe.com").
|
||||
Key("email", nil).
|
||||
Replace("name", "John Does").
|
||||
PlaceholderFormat(bob.Question).
|
||||
ToSql()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
desiredSql := "INSERT INTO \"users\" (\"name\", \"email\") VALUES ($1, $2) ON CONFLICT (\"email\") DO UPDATE SET \"name\" = $3;"
|
||||
desiredArgs := []interface{}{"John Doe", "john@doe.com", "John Does"}
|
||||
|
||||
if sql != desiredSql {
|
||||
t.Error("sql is not the same as result: ", sql)
|
||||
}
|
||||
if reflect.DeepEqual(args, desiredArgs) {
|
||||
t.Error("args is not the same as result: ", args)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("should be able to generate upsert query for mssql", func(t *testing.T) {
|
||||
sql, args, err := bob.
|
||||
Upsert("users", bob.MSSql).
|
||||
Columns("name", "email").
|
||||
Values("John Doe", "john@doe.com").
|
||||
Key("email", "john@doe.com").
|
||||
Replace("name", "John Does").
|
||||
PlaceholderFormat(bob.AtP).
|
||||
ToSql()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
desiredSql := "IF NOT EXISTS (SELECT * FROM \"users\" WHERE \"email\" = @p1) INSERT INTO \"users\" (\"name\", \"email\") VALUES (@p2, @p3) ELSE UPDATE SET \"name\" = @p4 WHERE \"users\" = @p5;"
|
||||
desiredArgs := []interface{}{"john@doe.com", "John Doe", "john@doe.com", "John Does"}
|
||||
|
||||
if sql != desiredSql {
|
||||
t.Error("sql is not the same as result: ", sql)
|
||||
}
|
||||
if reflect.DeepEqual(args, desiredArgs) {
|
||||
t.Error("args is not the same as result: ", args)
|
||||
}
|
||||
})
|
||||
}
|
||||
Loading…
Reference in New Issue