forked from deblan/database-anonymizer
fix query builder
This commit is contained in:
parent
93f79e898b
commit
db5dba17a1
1 changed files with 20 additions and 7 deletions
27
app/app.go
27
app/app.go
|
|
@ -5,6 +5,7 @@ import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
// "os"
|
// "os"
|
||||||
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
nq "github.com/Knetic/go-namedParameterQuery"
|
nq "github.com/Knetic/go-namedParameterQuery"
|
||||||
|
|
@ -91,18 +92,27 @@ func (a *App) ApplyRule(c config.SchemaConfigData, globalColumns map[string]stri
|
||||||
|
|
||||||
var scan any
|
var scan any
|
||||||
|
|
||||||
|
x := 1
|
||||||
|
t := len(rows)
|
||||||
|
|
||||||
for _, row := range rows {
|
for _, row := range rows {
|
||||||
|
fmt.Printf("%+v/%+v\n", x, t)
|
||||||
|
x = x + 1
|
||||||
|
|
||||||
updates := []string{}
|
updates := []string{}
|
||||||
pkeys := []string{}
|
pkeys := []string{}
|
||||||
|
pCounter := 1
|
||||||
|
|
||||||
for col, value := range row {
|
for col, value := range row {
|
||||||
if value.IsUpdated {
|
if value.IsUpdated {
|
||||||
updates = append(updates, fmt.Sprintf("%s=:%s", col, col))
|
updates = append(updates, fmt.Sprintf("%s=:p%s", col, strconv.Itoa(pCounter)))
|
||||||
|
pCounter = pCounter + 1
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, col := range c.PrimaryKey {
|
for _, col := range c.PrimaryKey {
|
||||||
pkeys = append(pkeys, fmt.Sprintf("%s=:%s", col, col))
|
pkeys = append(pkeys, fmt.Sprintf("%s=:p%s", col, strconv.Itoa(pCounter)))
|
||||||
|
pCounter = pCounter + 1
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(updates) > 0 {
|
if len(updates) > 0 {
|
||||||
|
|
@ -114,20 +124,23 @@ func (a *App) ApplyRule(c config.SchemaConfigData, globalColumns map[string]stri
|
||||||
)
|
)
|
||||||
|
|
||||||
stmt := nq.NewNamedParameterQuery(sql)
|
stmt := nq.NewNamedParameterQuery(sql)
|
||||||
|
pCounter = 1
|
||||||
|
|
||||||
for col, value := range row {
|
for _, value := range row {
|
||||||
if value.IsUpdated {
|
if value.IsUpdated {
|
||||||
stmt.SetValue(col, value.Value)
|
stmt.SetValue(fmt.Sprintf("p%s", strconv.Itoa(pCounter)), value.Value)
|
||||||
|
pCounter = pCounter + 1
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, col := range c.PrimaryKey {
|
for _, col := range c.PrimaryKey {
|
||||||
stmt.SetValue(col, row[col].Value)
|
stmt.SetValue(fmt.Sprintf("p%s", strconv.Itoa(pCounter)), row[col].Value)
|
||||||
|
pCounter = pCounter + 1
|
||||||
}
|
}
|
||||||
|
|
||||||
r := a.Db.QueryRow(stmt.GetParsedQuery(), (stmt.GetParsedParameters())...).Scan(&scan)
|
a.Db.QueryRow(stmt.GetParsedQuery(), (stmt.GetParsedParameters())...).Scan(&scan)
|
||||||
|
|
||||||
fmt.Printf("%+v\n", r)
|
// fmt.Printf("%+v\n", r)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue