pixeldrain/gocqlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

dbutil: RewriteTable add possibility to skip rows

Browse Source
This commit is contained in:
Michał Matczuk
2021-12-20 15:39:46 +01:00
committed by Michal Jan Matczuk
parent 9a6aae51ed
commit 2b885ac61b
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dbutil/rewrite.go
View File

@@ -11,6 +11,7 @@ import (
// RewriteTable rewrites src table to dst table. // RewriteTable rewrites src table to dst table.
// Rows can be transformed using the transform function. // Rows can be transformed using the transform function.
// If row map is empty after transformation the row is skipped.
// Additional options can be passed to modify the insert query. // Additional options can be passed to modify the insert query.
func RewriteTable(session gocqlx.Session, dst, src *table.Table, transform func(map[string]interface{}), options ...func(q *gocqlx.Queryx)) error { func RewriteTable(session gocqlx.Session, dst, src *table.Table, transform func(map[string]interface{}), options ...func(q *gocqlx.Queryx)) error {
insert := dst.InsertQuery(session) insert := dst.InsertQuery(session)
@@ -28,6 +29,9 @@ func RewriteTable(session gocqlx.Session, dst, src *table.Table, transform func(
if transform != nil { if transform != nil {
transform(m) transform(m)
} }
if len(m) == 0 {
continue // map is empty - no need to clean
}
if err := insert.BindMap(m).Exec(); err != nil { if err := insert.BindMap(m).Exec(); err != nil {
return err return err
} }