forked from deblan/database-anonymizer
manager all number types
add 'null' faker update readme
This commit is contained in:
parent
b8f3f59211
commit
628308e2fc
6 changed files with 297 additions and 257 deletions
|
|
@ -23,6 +23,23 @@ func GetNamedParameter(dbType, col string, number int) string {
|
|||
return fmt.Sprintf("%s=$%d", col, number)
|
||||
}
|
||||
|
||||
func IsPgNumberType(value string) bool {
|
||||
switch value {
|
||||
case
|
||||
"smallint",
|
||||
"integer",
|
||||
"bigint",
|
||||
"decimal",
|
||||
"numeric",
|
||||
"real",
|
||||
"double precision":
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func GetRows(db *sql.DB, query, table, dbType string) map[int]map[string]data.Data {
|
||||
rows, err := db.Query(query)
|
||||
defer rows.Close()
|
||||
|
|
@ -69,9 +86,9 @@ func GetRows(db *sql.DB, query, table, dbType string) map[int]map[string]data.Da
|
|||
|
||||
dataType := columnsTypes[col]
|
||||
|
||||
d.IsInteger = dataType == "integer"
|
||||
d.IsNumber = IsPgNumberType(dataType)
|
||||
d.IsBoolean = dataType == "boolean"
|
||||
d.IsString = !d.IsBoolean && !d.IsInteger
|
||||
d.IsString = !d.IsBoolean && !d.IsNumber
|
||||
} else {
|
||||
d.IsString = true
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue