EvoBGP/internal/repository/module_snapshot.go

package repository

import (
	"context"
	"encoding/json"
	"errors"
	"strings"
	"time"

	"evobgp/internal/store"

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

func (p *Postgres) GetModulePrefixSnapshot(tenantID, moduleID string) (*store.ModulePrefixSnapshot, bool, error) {
	ctx := context.Background()
	var inputHash string
	var collectedAt time.Time
	var raw []byte
	err := p.pool.QueryRow(ctx, `
		SELECT input_hash, collected_at, prefixes_json
		FROM module_prefix_snapshot
		WHERE tenant_id = $1 AND module_id = $2`,
		tenantID, moduleID).Scan(&inputHash, &collectedAt, &raw)
	if err != nil {
		if errors.Is(err, pgx.ErrNoRows) {
			return nil, false, nil
		}
		return nil, false, err
	}
	var prefixes []store.PrefixRow
	if len(raw) > 0 {
		if err := json.Unmarshal(raw, &prefixes); err != nil {
			return nil, false, err
		}
	}
	return &store.ModulePrefixSnapshot{
		InputHash:   inputHash,
		CollectedAt: collectedAt.UTC(),
		Prefixes:    prefixes,
	}, true, nil
}

func (p *Postgres) SetModulePrefixSnapshot(tenantID, moduleID, inputHash string, prefixes []store.PrefixRow) error {
	if strings.TrimSpace(tenantID) == "" || strings.TrimSpace(moduleID) == "" || strings.TrimSpace(inputHash) == "" {
		return store.ErrInvalidInput
	}
	raw, err := json.Marshal(prefixes)
	if err != nil {
		return err
	}
	ctx := context.Background()
	_, err = p.pool.Exec(ctx, `
		INSERT INTO module_prefix_snapshot (tenant_id, module_id, input_hash, collected_at, prefixes_json)
		VALUES ($1::uuid, $2::uuid, $3, now(), $4::jsonb)
		ON CONFLICT (tenant_id, module_id) DO UPDATE SET
			input_hash = EXCLUDED.input_hash,
			collected_at = EXCLUDED.collected_at,
			prefixes_json = EXCLUDED.prefixes_json`,
		tenantID, moduleID, inputHash, string(raw))
	return err
}

func (p *Postgres) DeleteModulePrefixSnapshot(tenantID, moduleID string) error {
	ctx := context.Background()
	_, err := p.pool.Exec(ctx, `
		DELETE FROM module_prefix_snapshot WHERE tenant_id = $1 AND module_id = $2`,
		tenantID, moduleID)
	return err
}