atmost working example

internal/store/pg/store.go

@@ -5,9 +5,10 @@ import (
 	"fmt"
 	"time"
 
-	"git.loyso.art/frx/devsim/internal/entities"
-
 	"github.com/jackc/pgx/v5/pgxpool"
+
+	"git.loyso.art/frx/devsim/internal/entities"
+	"git.loyso.art/frx/devsim/internal/store/pg/queries"
 )
 
 func Dial(ctx context.Context, addr string) (*repository, error) {
@@ -66,93 +67,38 @@ func (s deviceStatsDB) asDomain() entities.DeviceStatistics {
 }
 
 func (r statsRepository) Upsert(ctx context.Context, stats entities.DeviceStatistics) error {
-	const query = `INSERT INTO public.stats (
-			device_id,
-			inc_traffic,
-			out_traffic,
-			inc_rps, 
-			read_rps,
-			write_rps,
-			updated_at
-		) VALUES (
-			$1,
-			$2,
-			$3,
-			$4,
-			$5,
-			$6,
-			$7,
-		) ON CONFLICT(device_id) DO UPDATE SET
-			inc_traffic = EXCLUDED.inc_traffic,
-			out_traffic = EXCLUDED.out_traffic,
-			inc_rps = EXCLUDED.inc_rps,
-			read_rps = EXCLUDED.read_rps,
-			write_rps = EXCLUDED.write_rps,
-			updated_at = EXCLUDED.updated_at
-		RETURNING id;
-	`
-
-	_, err := r.db.Exec(
-		ctx, query,
-		stats.ID,
-		stats.IncomingTrafficBytes,
-		stats.OutgoingTrafficBytes,
-		stats.IncomingRPS,
-		stats.ReadRPS,
-		stats.WriteRPS,
-	)
+	err := queries.New(r.db).UpsertDeviceMetrics(ctx, queries.UpsertDeviceMetricsParams{
+		DeviceID:   string(stats.ID),
+		IncTraffic: int32(stats.IncomingTrafficBytes),
+		OutTraffic: int32(stats.OutgoingTrafficBytes),
+		IncRps:     int32(stats.IncomingRPS),
+		WriteRps:   int32(stats.WriteRPS),
+		ReadRps:    int32(stats.ReadRPS),
+	})
 	if err != nil {
-		return fmt.Errorf("executing query: %w", err)
+		return fmt.Errorf("upserting device metrics: %w", err)
 	}
 
 	return nil
 }
 
 func (r statsRepository) List(ctx context.Context) (out []entities.DeviceStatistics, err error) {
-	var count int
-	err = r.db.QueryRow(ctx, `SELECT COUNT(device_id) FROM public.stats`).Scan(&count)
+	stats, err := queries.New(r.db).ListDeviceStats(ctx)
 	if err != nil {
-		return nil, fmt.Errorf("getting count: %w", err)
+		return nil, fmt.Errorf("listing device stats: %w", err)
 	}
 
-	out = make([]entities.DeviceStatistics, 0, count)
+	out = make([]entities.DeviceStatistics, len(stats))
 
-	const query = `SELECT
-		device_id,
-		inc_traffic,
-		out_traffic,
-		inc_rps, 
-		read_rps,
-		write_rps,
-		updated_at
-	FROM public.stats;
-	`
-
-	rows, err := r.db.Query(ctx, query)
-	if err != nil {
-		return nil, fmt.Errorf("querying: %w", err)
-	}
-	defer rows.Close()
-
-	for rows.Next() {
-		var stat deviceStatsDB
-		err = rows.Scan(
-			&stat.DeviceID,
-			&stat.IncomingTraffic,
-			&stat.OutgoingTraffic,
-			&stat.IncomingRPS,
-			&stat.ReadRPS,
-			&stat.WriteRPS,
-			&stat.UpdatedAt,
-		)
-		if err != nil {
-			return nil, fmt.Errorf("scanning row: %w", err)
+	for i, stat := range stats {
+		out[i] = entities.DeviceStatistics{
+			IncomingTrafficBytes: int(stat.IncTraffic),
+			OutgoingTrafficBytes: int(stat.OutTraffic),
+			IncomingRPS:          int(stat.IncRps),
+			WriteRPS:             int(stat.WriteRps),
+			ReadRPS:              int(stat.ReadRps),
+			UpdatedAt:            stat.UpdatedAt.Time,
 		}
-
-		out = append(out, stat.asDomain())
-	}
-	if err = rows.Err(); err != nil {
-		return nil, fmt.Errorf("checking rows err: %w", err)
 	}
 
 	return out, nil