Netpulse_SasS/server/internal/grpcapi/ncm_retention.go

package grpcapi

import (
	"context"
	"time"
)

// retentionLockKey — окреме advisory-блокування під прибиральника.
//
// Своє, а не спільне з планувальником бекапів: обидва тікають незалежно,
// і спільний ключ означав би, що довгий перший прохід очистки заодно
// зупиняє розклад збору.
const retentionLockKey = 0x6e70_7263 // "nprc"

// RetentionInterval — як часто прибирати старі версії.
//
// Година. Найдрібніша одиниця політики — доба, тож частіше немає чого
// шукати; рідше — означало б, що на інсталяції, яка вперше вмикає
// чистку, звільнення місця починається невідомо коли.
const RetentionInterval = time.Hour

// SweepRetention прибирає старі версії конфігів за політикою.
//
// Живе в колекторі, а не в REST-процесі, і це не байдуже.
//
// По-перше, тут уже стоїть той, хто версії СТВОРЮЄ: планувальник
// бекапів і диспетчер збору. Механізм, що прибирає, доречно тримати
// поруч із тим, що накопичує, — інакше два процеси зі спільною
// відповідальністю за одну таблицю живуть у різних релізних циклах.
//
// По-друге, REST-процес — шлях запитів людини, і його масштабують
// копіями. Довге видалення в тому самому процесі конкурувало б за
// з'єднання з тим, хто зараз відкриває сторінку конфігів.
//
// По-третє, ключ шифрування тут не потрібен зовсім: видалення не читає
// тіл. Тому прохід безпечний і на інсталяції, де REST запущено без DEK.
//
// Кілька екземплярів колектора безпечні: тік бере advisory-блокування,
// тож прибирає рівно один.
func (s *Service) SweepRetention(ctx context.Context) {
	t := time.NewTicker(RetentionInterval)
	defer t.Stop()

	s.log.Info("прибиральник старих конфігів запущено", "інтервал", RetentionInterval)

	for {
		select {
		case <-ctx.Done():
			return
		case <-t.C:
			s.retentionTick(ctx)
		}
	}
}

func (s *Service) retentionTick(ctx context.Context) {
	conn, err := s.store.WorkerPool().Acquire(ctx)
	if err != nil {
		s.log.Error("прибиральник: з'єднання", "err", err)
		return
	}
	defer conn.Release()

	var got bool
	if err := conn.QueryRow(ctx, `SELECT pg_try_advisory_lock($1)`,
		int64(retentionLockKey)).Scan(&got); err != nil {
		s.log.Error("прибиральник: блокування", "err", err)
		return
	}
	if !got {
		// Прибирає інший інстанс — штатний стан.
		return
	}
	defer func() {
		_, _ = conn.Exec(context.WithoutCancel(ctx),
			`SELECT pg_advisory_unlock($1)`, int64(retentionLockKey))
	}()

	tenants, err := s.store.TenantsWithRetention(ctx)
	if err != nil {
		s.log.Error("прибиральник: перелік тенантів", "err", err)
		return
	}

	for _, tenantID := range tenants {
		stat, err := s.store.SweepRetention(ctx, tenantID)
		if err != nil {
			// Один тенант не має зупиняти решту: політика в кожного
			// своя, і збій на чужих даних тут нічого не означає.
			s.log.Error("прибиральник: прохід", "tenant", tenantID, "err", err)
			continue
		}
		if stat.Deleted > 0 {
			s.log.Info("прибрано старі версії конфігів",
				"tenant", tenantID, "версій", stat.Deleted, "байтів", stat.FreedBytes)
		}
	}
}