add database helper

database/database.go

@@ -0,0 +1,57 @@
+package database
+
+import (
+	"database/sql"
+	"gitnet.fr/deblan/database-anonymizer/data"
+	"gitnet.fr/deblan/database-anonymizer/logger"
+)
+
+func GetRows(db *sql.DB, query string) map[int]map[string]data.Data {
+	rows, err := db.Query(query)
+	defer rows.Close()
+	logger.LogFatalExitIf(err)
+
+	columns, err := rows.Columns()
+	logger.LogFatalExitIf(err)
+
+	values := make([]any, len(columns))
+	valuePointers := make([]any, len(columns))
+	datas := make(map[int]map[string]data.Data)
+
+	key := 0
+
+	for rows.Next() {
+		row := make(map[string]data.Data)
+
+		for i := range columns {
+			valuePointers[i] = &values[i]
+		}
+
+		if err := rows.Scan(valuePointers...); err != nil {
+			logger.LogFatalExitIf(err)
+		}
+
+		for i, col := range columns {
+			value := values[i]
+			d := data.Data{IsVirtual: false}
+
+			if value != nil {
+				switch v := value.(type) {
+				case []byte:
+					d.FromByte(v)
+				case string:
+					d.FromString(v)
+				case int64:
+					d.FromInt64(v)
+				}
+			}
+
+			row[col] = d
+		}
+
+		datas[key] = row
+		key = key + 1
+	}
+
+	return datas
+}