add sql query generation

This commit is contained in:
Simon Vieille 2024-03-19 06:51:07 +01:00
commit 79fd8195cd
Signed by untrusted user: deblan
GPG key ID: 579388D585F70417

View file

@ -4,8 +4,10 @@ import (
"database/sql" "database/sql"
"errors" "errors"
"fmt" "fmt"
"os" // "os"
"strings"
nq "github.com/Knetic/go-namedParameterQuery"
"gitnet.fr/deblan/database-anonymizer/config" "gitnet.fr/deblan/database-anonymizer/config"
"gitnet.fr/deblan/database-anonymizer/data" "gitnet.fr/deblan/database-anonymizer/data"
"gitnet.fr/deblan/database-anonymizer/database" "gitnet.fr/deblan/database-anonymizer/database"
@ -33,8 +35,6 @@ func (a *App) ApplyRule(c config.SchemaConfigData, globalColumns map[string]stri
c.PrimaryKey = []string{"id"} c.PrimaryKey = []string{"id"}
} }
fmt.Printf("%+v\n", c.PrimaryKey)
rows := database.GetRows(a.Db, query) rows := database.GetRows(a.Db, query)
for key, row := range rows { for key, row := range rows {
@ -86,14 +86,50 @@ func (a *App) ApplyRule(c config.SchemaConfigData, globalColumns map[string]stri
rows[key][col] = value rows[key][col] = value
} }
fmt.Printf("%+v\n", rows[key])
rows[key] = a.UpdateRow(rows[key]) rows[key] = a.UpdateRow(rows[key])
fmt.Printf("%+v\n", rows[key])
os.Exit(0)
} }
// fmt.Printf("%+v\n", rows) var scan any
for _, row := range rows {
updates := []string{}
pkeys := []string{}
for col, value := range row {
if value.IsUpdated {
updates = append(updates, fmt.Sprintf("%s=:%s", col, col))
}
}
for _, col := range c.PrimaryKey {
pkeys = append(pkeys, fmt.Sprintf("%s=:%s", col, col))
}
if len(updates) > 0 {
sql := fmt.Sprintf(
"UPDATE %s SET %s WHERE %s",
c.Table,
strings.Join(updates, ", "),
strings.Join(pkeys, " AND "),
)
stmt := nq.NewNamedParameterQuery(sql)
for col, value := range row {
if value.IsUpdated {
stmt.SetValue(col, value.Value)
}
}
for _, col := range c.PrimaryKey {
stmt.SetValue(col, row[col].Value)
}
r := a.Db.QueryRow(stmt.GetParsedQuery(), (stmt.GetParsedParameters())...).Scan(&scan)
fmt.Printf("%+v\n", r)
}
}
return nil return nil
} }