add data struct
This commit is contained in:
parent
778ad1d10b
commit
1002d8f1d6
62
data/data.go
Normal file
62
data/data.go
Normal file
|
@ -0,0 +1,62 @@
|
|||
package data
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"github.com/tyler-sommer/stick"
|
||||
"github.com/tyler-sommer/stick/twig"
|
||||
"gitnet.fr/deblan/database-anonymizer/logger"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Data struct {
|
||||
Value string
|
||||
Faker string
|
||||
IsVirtual bool
|
||||
IsPrimaryKey bool
|
||||
}
|
||||
|
||||
func (d *Data) FromByte(v []byte) *Data {
|
||||
d.Value = string(v)
|
||||
|
||||
return d
|
||||
}
|
||||
|
||||
func (d *Data) FromInt64(v int64) *Data {
|
||||
d.Value = strconv.FormatInt(v, 10)
|
||||
|
||||
return d
|
||||
}
|
||||
|
||||
func (d *Data) FromString(v string) *Data {
|
||||
d.Value = v
|
||||
|
||||
return d
|
||||
}
|
||||
|
||||
func (d *Data) IsTwigExpression() bool {
|
||||
return strings.Contains(d.Faker, "{{") || strings.Contains(d.Faker, "}}")
|
||||
}
|
||||
|
||||
func (d *Data) Update(row map[string]Data) {
|
||||
if !d.IsPrimaryKey && d.Faker != "" {
|
||||
if d.IsTwigExpression() {
|
||||
env := twig.New(nil)
|
||||
params := map[string]stick.Value{}
|
||||
|
||||
for key, value := range row {
|
||||
params[key] = value.Value
|
||||
}
|
||||
|
||||
var buf bytes.Buffer
|
||||
err := env.Execute(d.Faker, &buf, params)
|
||||
|
||||
logger.LogFatalExitIf(err)
|
||||
|
||||
d.Value = buf.String()
|
||||
} else {
|
||||
d.Value = fmt.Sprintf("%s__UPDATE", d.Value)
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue