flashcode/database-anonymizer
1
0
Fork 0
forked from deblan/database-anonymizer
Code Pull requests Activity

refactoring of app: separation of truncate and update

Browse source
This commit is contained in:
Simon Vieille 2024-03-20 10:59:50 +01:00
commit 4bdc6a04d3
Signed by untrusted user: deblan
GPG key ID: 579388D585F70417
2 changed files with 139 additions and 128 deletions
Download patch file Download diff file Expand all files Collapse all files

89
app/app.go
View file

@ -22,35 +22,32 @@ type App struct {
FakeManager faker.FakeManager
}
func (a *App) DoAction(c config.SchemaConfigAction, globalColumns map[string]string, generators map[string][]string) error {
var query string
func (a *App) Run(db *sql.DB, c config.SchemaConfig, fakeManager faker.FakeManager) error {
a.Db = db
a.FakeManager = fakeManager
if c.Table == "" {
return errors.New("Table must be defined")
for _, data := range c.Rules.Actions {
err := a.DoAction(data, c.Rules.Columns, c.Rules.Generators)
logger.LogFatalExitIf(err)
}
if c.Truncate {
if c.Query != "" {
query = c.Query
} else {
return a.TruncateTable(c.Table)
}
} else {
if c.Query != "" {
query = c.Query
} else {
query = fmt.Sprintf("SELECT * FROM %s", c.Table)
}
_ = db
return nil
}
if len(c.PrimaryKey) == 0 {
c.PrimaryKey = []string{"id"}
func (a *App) TruncateTable(c config.SchemaConfigAction) error {
if c.Query == "" {
_, err := a.Db.Exec(fmt.Sprintf("TRUNCATE %s", c.Table))
return err
}
query := a.CreateSelectQuery(c)
rows := database.GetRows(a.Db, query)
var scan any
if c.Truncate {
for _, row := range rows {
pkeys := []string{}
pCounter := 1
@ -76,7 +73,15 @@ func (a *App) DoAction(c config.SchemaConfigAction, globalColumns map[string]str
a.Db.QueryRow(stmt.GetParsedQuery(), (stmt.GetParsedParameters())...).Scan(&scan)
}
} else {
return nil
}
func (a *App) UpdateRows(c config.SchemaConfigAction, globalColumns map[string]string, generators map[string][]string) error {
query := a.CreateSelectQuery(c)
rows := database.GetRows(a.Db, query)
var scan any
for key, row := range rows {
if len(c.VirtualColumns) > 0 {
for col, faker := range c.VirtualColumns {
@ -171,11 +176,32 @@ func (a *App) DoAction(c config.SchemaConfigAction, globalColumns map[string]str
a.Db.QueryRow(stmt.GetParsedQuery(), (stmt.GetParsedParameters())...).Scan(&scan)
}
}
}
return nil
}
func (a *App) CreateSelectQuery(c config.SchemaConfigAction) string {
if c.Query != "" {
return c.Query
}
return fmt.Sprintf("SELECT * FROM %s", c.Table)
}
func (a *App) DoAction(c config.SchemaConfigAction, globalColumns map[string]string, generators map[string][]string) error {
if c.Table == "" {
return errors.New("Table must be defined")
}
c.InitPrimaryKey()
if c.Truncate {
return a.TruncateTable(c)
}
return a.UpdateRows(c, globalColumns, generators)
}
func (a *App) UpdateRow(row map[string]data.Data) (map[string]data.Data, error) {
for key, value := range row {
if value.IsVirtual && !value.IsTwigExpression() {
@ -215,24 +241,3 @@ func (a *App) UpdateRow(row map[string]data.Data) (map[string]data.Data, error)
return row, nil
}
func (a *App) TruncateTable(table string) error {
_, err := a.Db.Exec(fmt.Sprintf("TRUNCATE %s", table))
return err
}
func (a *App) Run(db *sql.DB, c config.SchemaConfig, fakeManager faker.FakeManager) error {
a.Db = db
a.FakeManager = fakeManager
for _, data := range c.Rules.Actions {
err := a.DoAction(data, c.Rules.Columns, c.Rules.Generators)
logger.LogFatalExitIf(err)
}
_ = db
return nil
}

6
config/schema_config.go
View file

@ -37,3 +37,9 @@ func LoadSchemaConfigFromFile(file string) (SchemaConfig, error) {
return value, nil
}
func (c *SchemaConfigAction) InitPrimaryKey() {
if len(c.PrimaryKey) == 0 {
c.PrimaryKey = []string{"id"}
}
}