itsscb
6b1f30f0f7
TODO: Add support for multi table requests - currently the existing fields are being merged.
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"log/slog"
|
|
)
|
|
|
|
var (
|
|
restrictedFields = map[string]bool{
|
|
"passwordhash": true,
|
|
}
|
|
)
|
|
|
|
func (store *SQLStore) Query(ctx context.Context, statement string, args ...interface{}) (result []map[string]interface{}, err error) {
|
|
var rows *sql.Rows
|
|
// var err error
|
|
|
|
if len(args) > 0 {
|
|
rows, err = store.db.QueryContext(ctx, statement, args)
|
|
} else {
|
|
rows, err = store.db.QueryContext(ctx, statement)
|
|
}
|
|
if err != nil {
|
|
slog.Error("db_query", slog.String("statement", statement), slog.String("error", err.Error()))
|
|
return nil, err
|
|
}
|
|
|
|
cols, err := rows.Columns()
|
|
if err != nil {
|
|
slog.Error("db_query: getting columns", slog.String("error", err.Error()))
|
|
return nil, err
|
|
}
|
|
|
|
for rows.Next() {
|
|
values := make([]interface{}, len(cols))
|
|
res := make(map[string]interface{})
|
|
for i := range values {
|
|
values[i] = new(interface{})
|
|
}
|
|
err = rows.Scan(values...)
|
|
if err != nil {
|
|
slog.Error("db_query: scanning rows", slog.String("error", err.Error()))
|
|
return nil, err
|
|
}
|
|
for i, col := range cols {
|
|
if restrictedFields[col] {
|
|
continue
|
|
}
|
|
res[col] = *(values[i].(*interface{}))
|
|
}
|
|
|
|
result = append(result, res)
|
|
}
|
|
|
|
return result, rows.Err()
|
|
}
|