nexarch/dms/internal/batchsplit/store.go

package batchsplit

import (
	"context"
	"fmt"

	"github.com/jackc/pgx/v5/pgxpool"
)

// Store merkt sich die Pruefsummen bereits aufgeteilter Stapel.
type Store struct{ pool *pgxpool.Pool }

func NewStore(pool *pgxpool.Pool) *Store { return &Store{pool: pool} }

func (s *Store) Seen(ctx context.Context, checksum string) (bool, error) {
	var seen bool
	if err := s.pool.QueryRow(ctx, `SELECT EXISTS (SELECT 1 FROM batch_deliveries WHERE checksum_sha256 = $1)`, checksum).Scan(&seen); err != nil {
		return false, fmt.Errorf("batchsplit: lieferung pruefen: %w", err)
	}
	return seen, nil
}

func (s *Store) Record(ctx context.Context, checksum, sourceName string, parts int) error {
	if _, err := s.pool.Exec(ctx, `
		INSERT INTO batch_deliveries (checksum_sha256, source_name, part_count) VALUES ($1, $2, $3)
		ON CONFLICT (checksum_sha256) DO NOTHING`, checksum, sourceName, parts); err != nil {
		return fmt.Errorf("batchsplit: lieferung vermerken: %w", err)
	}
	return nil
}