add an posture-check-id to public-id index

Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io>

management/internals/network_map_db/pgsql/posture.go

@@ -2,6 +2,7 @@ package networkmap_pgsql
 
 import (
 	"context"
+	"database/sql"
 	"encoding/json"
 	"reflect"
 
@@ -12,45 +13,48 @@ import (
 
 const (
 	GetPostureChecksQuery = `
-	select public_id as id, checks
+	select id, public_id, checks
 	from posture_checks
 	where account_id=$1
 	`
 )
 
-func (pg *PgStore) GetPostureChecks(ctx context.Context, accountId string) ([]nmdata.PostureChecks, error) {
+func (pg *PgStore) GetPostureChecks(ctx context.Context, accountId string) ([]nmdata.PostureChecks, map[string]string, error) {
 	c, err := pg.Pool.Acquire(ctx)
 	if err != nil {
-		return nil, err
+		return nil, nil, err
 	}
 	return GetPostureChecksViaPgxConnection(ctx, c.Conn(), accountId)
 }
 
-func GetPostureChecksViaPgxConnection(ctx context.Context, con *pgx.Conn, accountId string) ([]nmdata.PostureChecks, error) {
+func GetPostureChecksViaPgxConnection(ctx context.Context, con *pgx.Conn, accountId string) ([]nmdata.PostureChecks, map[string]string, error) {
 	rows, err := con.Query(ctx, GetPostureChecksQuery, accountId)
 	if err != nil {
-		return nil, err
+		return nil, nil, err
 	}
 
 	checks, err := pgx.CollectRows(rows, pgx.RowToStructByName[posturechecks])
 	if err != nil {
-		return nil, err
+		return nil, nil, err
 	}
 
 	toret := make([]nmdata.PostureChecks, 0, len(checks))
+	idToPublicIDIdx := make(map[string]string)
 	for _, c := range checks {
 		checks := nmdata.PostureChecks{}
 		err := networkmapdb.FromSqlTypesToSharedTypes(reflect.ValueOf(&c), reflect.ValueOf(&checks))
 		if err != nil {
-			return nil, err
+			return nil, nil, err
 		}
 		toret = append(toret, checks)
+		idToPublicIDIdx[checks.ID] = c.PublicID.String
 	}
 
-	return toret, nil
+	return toret, idToPublicIDIdx, nil
 }
 
 type posturechecks struct {
-	ID     string
-	Checks json.RawMessage
+	ID       string
+	PublicID sql.NullString `nmap:"skip"`
+	Checks   json.RawMessage
 }