add table escape

This commit is contained in:
Simon Vieille 2024-03-20 11:42:18 +01:00
commit b547166c83
Signed by untrusted user: deblan
GPG key ID: 579388D585F70417
3 changed files with 22 additions and 6 deletions

View file

@ -2,10 +2,19 @@ package database
import (
"database/sql"
"fmt"
"gitnet.fr/deblan/database-anonymizer/data"
"gitnet.fr/deblan/database-anonymizer/logger"
)
func EscapeTable(dbType, table string) string {
if dbType == "mysql" {
return fmt.Sprintf("`%s`", table)
}
return fmt.Sprintf("\"%s\"", table)
}
func GetRows(db *sql.DB, query string) map[int]map[string]data.Data {
rows, err := db.Query(query)
defer rows.Close()