feat: add DbHandler and helpers for database operations with logging and utility functions

This commit is contained in:
Marco Pedone 2025-08-21 13:54:15 +02:00
parent 0837f447fc
commit 080002f4ae
3 changed files with 8 additions and 16 deletions

View file

@ -1,8 +1,8 @@
package main
package db
import (
"backend/dbhelpers"
"backend/typedefs"
"backend/utils"
"context"
"fmt"
"log/slog"
@ -35,7 +35,7 @@ func NewDbHandler() *DbHandler {
if err != nil {
panic(err)
}
config.ConnConfig.Tracer = dbhelpers.NewMultiQueryTracer(dbhelpers.NewLoggingQueryTracer(slog.Default()))
config.ConnConfig.Tracer = NewMultiQueryTracer(NewLoggingQueryTracer(slog.Default()))
config.AfterConnect = func(ctx context.Context, conn *pgx.Conn) error {
pgxuuid.Register(conn.TypeMap())
@ -88,7 +88,7 @@ func (h *DbHandler) InsertAnnunciInDB(annunci typedefs.AnnunciParsed) {
batch := &pgx.Batch{}
for _, annuncio := range annunci.Annuncio {
if contains(codici, annuncio.Codice) {
if utils.Contains(codici, annuncio.Codice) {
batch.Queue(`
UPDATE public.annunci
SET locatore = $1,
@ -225,15 +225,6 @@ func (h *DbHandler) InsertAnnunciInDB(annunci typedefs.AnnunciParsed) {
}
func contains(s []string, str string) bool {
for _, v := range s {
if v == str {
return true
}
}
return false
}
func (h *DbHandler) ClearRecords() {
//SET ALL WEB TO FALSE

View file

@ -1,4 +1,4 @@
package dbhelpers
package db
import (
"context"

View file

@ -1,6 +1,7 @@
package main
import (
"backend/db"
storagehandlers "backend/storage_handlers"
"fmt"
"io"
@ -23,7 +24,7 @@ var (
imageOption bool
concurrentImages int = 1
storagepath string
Db *DbHandler
Db *db.DbHandler
Miogest *MiogestHandler
ImageManager *ImageRefManager
Storage storagehandlers.StorageHandler
@ -63,7 +64,7 @@ func main() {
}
// SETUP DATABASE HANDLER
Db = NewDbHandler()
Db = db.NewDbHandler()
if Db == nil {
log.Fatal("Error connecting to db")
}