Netpulse_SasS/server/internal/store/identify.go

package store

import (
	"context"

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

// IdentifyDecision — відповідь на «розпізнати зараз».
//
// Відмова несе готовий для людини текст, а не код: причин кілька, і
// кожна лікується по-своєму — одна вимагає завести доступ, інша
// почекати, доки зонд повернеться. Змушувати інтерфейс перекладати
// коди назад у речення означало б тримати той самий перелік у двох
// місцях і забути оновити одне з них.
type IdentifyDecision struct {
	Queued bool   `json:"queued"`
	Reason string `json:"reason,omitempty"`
}

// RequestIdentify просить опитати хост по SNMP просто зараз.
//
// Сам запит не надсилається: API не має сесій зондів — вони живуть у
// колекторі, іншому процесі. Тут лишається намір, який забере
// ClaimIdentifyRequests.
func (s *Store) RequestIdentify(ctx context.Context, tenantID, deviceID string) (IdentifyDecision, error) {
	var d IdentifyDecision

	err := s.InTenantTx(ctx, tenantID, func(tx pgx.Tx) error {
		var (
			agentID     string
			agentOnline bool
			hasSNMP     bool
		)
		err := tx.QueryRow(ctx, `
			SELECT COALESCE(d.agent_id::text, ''),
			       COALESCE(a.status::text = 'online', false),
			       EXISTS (
				SELECT 1 FROM inv.device_credentials dc
				JOIN inv.credentials c ON c.id = dc.credential_id
				WHERE dc.device_id = d.id
				  AND c.tenant_id = d.tenant_id
				  AND c.proto IN ('snmp_v2c','snmp_v3')
			       )
			FROM inv.devices d
			LEFT JOIN core.agents a ON a.id = d.agent_id
			WHERE d.id = $1 AND d.tenant_id = $2 AND d.deleted_at IS NULL
		`, deviceID, tenantID).Scan(&agentID, &agentOnline, &hasSNMP)
		if err != nil {
			return err
		}

		switch {
		case !hasSNMP:
			d.Reason = "у хоста немає SNMP-доступу — додайте його на вкладці «Доступи», " +
				"і хост розпізнається сам"
			return nil
		case agentID == "":
			d.Reason = "хост не прив'язаний до зонда — нікому його опитати"
			return nil
		case !agentOnline:
			d.Reason = "зонд не на зв'язку — прохання виконається, щойно він повернеться"
			// Намір усе одно лишаємо: коли зонд повернеться, диспетчер
			// його забере. Відмова тут описує затримку, а не поразку.
		}

		// Чек має існувати до того, як просити його запуску: зонд
		// зрушує задачу зі свого розкладу, а не виконує все, що
		// прилетіло. Без цього рядка прохання нікуди б не влучило.
		if _, err := tx.Exec(ctx, `
			INSERT INTO core.checks
				(tenant_id, device_id, check_type, params, interval_sec, timeout_ms, retries)
			SELECT $2, $1, $3, '{}'::jsonb, $4, 10000, 1
			WHERE NOT EXISTS (
				SELECT 1 FROM core.checks ch
				WHERE ch.device_id = $1 AND ch.tenant_id = $2
				  AND ch.check_type IN ($3, 'topology.discover')
			)
		`, deviceID, tenantID, IdentifyCheckType,
			int32(IdentifyProbeInterval.Seconds())); err != nil {
			return err
		}

		if _, err := tx.Exec(ctx, `
			UPDATE inv.devices SET identify_requested_at = now(), updated_at = now()
			WHERE id = $1 AND tenant_id = $2
		`, deviceID, tenantID); err != nil {
			return err
		}

		d.Queued = d.Reason == ""
		return nil
	})
	return d, err
}

// IdentifyRequest — прохання, готове до надсилання.
type IdentifyRequest struct {
	TenantID string
	AgentID  string
	DeviceID string
}

// ClaimIdentifyRequests забирає прохання для зондів, які зараз на
// зв'язку.
//
// Забирає назовсім: повторне надсилання гірше за втрачене прохання —
// людина натисне ще раз і побачить результат, а зациклене опитування
// вона не побачить узагалі й дізнається про нього з трафіку.
func (s *Store) ClaimIdentifyRequests(ctx context.Context, onlineAgents []string, limit int) ([]IdentifyRequest, error) {
	if len(onlineAgents) == 0 {
		return nil, nil
	}
	if limit <= 0 {
		limit = 64
	}

	rows, err := s.bg.Query(ctx, `
		UPDATE inv.devices d
		SET identify_requested_at = NULL
		WHERE d.id IN (
			SELECT id FROM inv.devices
			WHERE identify_requested_at IS NOT NULL
			  AND agent_id = ANY($1::uuid[])
			  AND deleted_at IS NULL
			ORDER BY identify_requested_at
			FOR UPDATE SKIP LOCKED
			LIMIT $2
		)
		RETURNING d.tenant_id::text, d.agent_id::text, d.id::text
	`, onlineAgents, limit)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var out []IdentifyRequest
	for rows.Next() {
		var r IdentifyRequest
		if err := rows.Scan(&r.TenantID, &r.AgentID, &r.DeviceID); err != nil {
			return nil, err
		}
		out = append(out, r)
	}
	return out, rows.Err()
}