mirror of
https://github.com/anand34577/ferrum.git
synced 2026-09-18 08:35:15 +00:00
Anand
d9b2519b75
- New PBS Backups page (/pbs): datastore usage, group browser, per-group prune, GC with live status, sync/verify jobs with admin-gated run - Shared single-connection SSE store (lib/sse.ts) feeding NotificationBell and the telemetry pill; global offline banner with reconnect invalidation - Correctness: CIFS storage credentials sent in JSON body; per-row alert silencing + unsilence endpoint; graceful stop-and-delete for running guests; accent/shadow token fixes for all look presets - Safer operations: confirms for SDN apply and host upgrade-all; Refresh control on every page; 30s polling added to previously-static pages - Wayfinding: guests deep-linkable via /inventory?focusGuest, scoped inventory views, topology guest click-through, Overview links - Forms/consistency: node storage/bridge pick-lists in CreateGuestDialog, backup schedule validation, BulkOperationsPage on shared DataTable, progressive pagination on cluster access lists, shared Timestamp and chartToneFor helpers, editable alert rules and webhooks - Security/tests: CSRF, authz-matrix, redaction and PBS test coverage; webhook outbox + connection TLS fingerprint migrations (00032/00033, sqlite + postgres) - Comment audit: removed stale, duplicated and orphaned comments; corrected inaccurate doc comments (UpdateGuestConfig, InvalidateAll, breakpoint references); no behavior changes
112 lines
2.6 KiB
Go
112 lines
2.6 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// DB wraps *sql.DB so callers can always write queries using "?" placeholders
|
|
// (SQLite's native style) while Postgres gets them rebound to "$1, $2, ..."
|
|
// automatically. This is what lets internal/store/seed.go and future query
|
|
// code stay driver-agnostic without a full query builder.
|
|
type DB struct {
|
|
*sql.DB
|
|
driver string
|
|
}
|
|
|
|
func newDB(driver string, sqlDB *sql.DB) *DB {
|
|
return &DB{DB: sqlDB, driver: driver}
|
|
}
|
|
|
|
func (d *DB) rebind(query string) string {
|
|
if d.driver != "postgres" {
|
|
return query
|
|
}
|
|
var b strings.Builder
|
|
n := 0
|
|
for _, r := range query {
|
|
if r == '?' {
|
|
n++
|
|
b.WriteByte('$')
|
|
b.WriteString(strconv.Itoa(n))
|
|
continue
|
|
}
|
|
b.WriteRune(r)
|
|
}
|
|
return b.String()
|
|
}
|
|
|
|
func (d *DB) Exec(query string, args ...any) (sql.Result, error) {
|
|
return d.DB.Exec(d.rebind(query), args...)
|
|
}
|
|
|
|
func (d *DB) ExecContext(ctx context.Context, query string, args ...any) (sql.Result, error) {
|
|
return d.DB.ExecContext(ctx, d.rebind(query), args...)
|
|
}
|
|
|
|
func (d *DB) Query(query string, args ...any) (*sql.Rows, error) {
|
|
return d.DB.Query(d.rebind(query), args...)
|
|
}
|
|
|
|
func (d *DB) QueryContext(ctx context.Context, query string, args ...any) (*sql.Rows, error) {
|
|
return d.DB.QueryContext(ctx, d.rebind(query), args...)
|
|
}
|
|
|
|
func (d *DB) QueryRow(query string, args ...any) *sql.Row {
|
|
return d.DB.QueryRow(d.rebind(query), args...)
|
|
}
|
|
|
|
func (d *DB) QueryRowContext(ctx context.Context, query string, args ...any) *sql.Row {
|
|
return d.DB.QueryRowContext(ctx, d.rebind(query), args...)
|
|
}
|
|
|
|
// Begin starts a transaction whose Exec/Query/QueryRow calls are also rebound.
|
|
func (d *DB) Begin() (*Tx, error) {
|
|
tx, err := d.DB.Begin()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Tx{Tx: tx, driver: d.driver}, nil
|
|
}
|
|
|
|
type Tx struct {
|
|
*sql.Tx
|
|
driver string
|
|
}
|
|
|
|
func (t *Tx) rebind(query string) string {
|
|
if t.driver != "postgres" {
|
|
return query
|
|
}
|
|
var b strings.Builder
|
|
n := 0
|
|
for _, r := range query {
|
|
if r == '?' {
|
|
n++
|
|
b.WriteByte('$')
|
|
b.WriteString(strconv.Itoa(n))
|
|
continue
|
|
}
|
|
b.WriteRune(r)
|
|
}
|
|
return b.String()
|
|
}
|
|
|
|
func (t *Tx) Exec(query string, args ...any) (sql.Result, error) {
|
|
return t.Tx.Exec(t.rebind(query), args...)
|
|
}
|
|
|
|
func (t *Tx) ExecContext(ctx context.Context, query string, args ...any) (sql.Result, error) {
|
|
return t.Tx.ExecContext(ctx, t.rebind(query), args...)
|
|
}
|
|
|
|
func (t *Tx) Query(query string, args ...any) (*sql.Rows, error) {
|
|
return t.Tx.Query(t.rebind(query), args...)
|
|
}
|
|
|
|
func (t *Tx) QueryRow(query string, args ...any) *sql.Row {
|
|
return t.Tx.QueryRow(t.rebind(query), args...)
|
|
}
|