Add um.Set() method that takes a slice of expressions

stephenafamo · Jan 2, 2024 · ed4d0c1 · ed4d0c1
1 parent 07da5c0
commit ed4d0c1
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/dialect/mysql/um/qm.go b/dialect/mysql/um/qm.go
@@ -4,6 +4,7 @@ import (
 	"github.com/stephenafamo/bob"
 	"github.com/stephenafamo/bob/clause"
 	"github.com/stephenafamo/bob/dialect/mysql/dialect"
+	"github.com/stephenafamo/bob/internal"
 	"github.com/stephenafamo/bob/mods"
 )
 
@@ -51,6 +52,12 @@ func StraightJoin(e any) bob.Mod[*dialect.UpdateQuery] {
 	return dialect.StraightJoin[*dialect.UpdateQuery](e)
 }
 
+func Set(sets ...bob.Expression) bob.Mod[*dialect.UpdateQuery] {
+	return mods.QueryModFunc[*dialect.UpdateQuery](func(q *dialect.UpdateQuery) {
+		q.Set.Set = append(q.Set.Set, internal.ToAnySlice(sets)...)
+	})
+}
+
 func SetCol(from ...string) mods.Set[*dialect.UpdateQuery] {
 	return mods.Set[*dialect.UpdateQuery](from)
 }

diff --git a/dialect/psql/um/qm.go b/dialect/psql/um/qm.go
@@ -4,6 +4,7 @@ import (
 	"github.com/stephenafamo/bob"
 	"github.com/stephenafamo/bob/clause"
 	"github.com/stephenafamo/bob/dialect/psql/dialect"
+	"github.com/stephenafamo/bob/internal"
 	"github.com/stephenafamo/bob/mods"
 )
 
@@ -38,6 +39,12 @@ func TableAs(name any, alias string) bob.Mod[*dialect.UpdateQuery] {
 	})
 }
 
+func Set(sets ...bob.Expression) bob.Mod[*dialect.UpdateQuery] {
+	return mods.QueryModFunc[*dialect.UpdateQuery](func(q *dialect.UpdateQuery) {
+		q.Set.Set = append(q.Set.Set, internal.ToAnySlice(sets)...)
+	})
+}
+
 func SetCol(from string) mods.Set[*dialect.UpdateQuery] {
 	return mods.Set[*dialect.UpdateQuery]([]string{from})
 }

diff --git a/dialect/sqlite/um/qm.go b/dialect/sqlite/um/qm.go
@@ -3,6 +3,7 @@ package um
 import (
 	"github.com/stephenafamo/bob"
 	"github.com/stephenafamo/bob/dialect/sqlite/dialect"
+	"github.com/stephenafamo/bob/internal"
 	"github.com/stephenafamo/bob/mods"
 )
 
@@ -60,6 +61,12 @@ func TableNotIndexed() bob.Mod[*dialect.UpdateQuery] {
 	})
 }
 
+func Set(sets ...bob.Expression) bob.Mod[*dialect.UpdateQuery] {
+	return mods.QueryModFunc[*dialect.UpdateQuery](func(q *dialect.UpdateQuery) {
+		q.Set.Set = append(q.Set.Set, internal.ToAnySlice(sets)...)
+	})
+}
+
 func SetCol(from string) mods.Set[*dialect.UpdateQuery] {
 	return mods.Set[*dialect.UpdateQuery]([]string{from})
 }