adding support for bulding of network_map_data

Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io>
This commit is contained in:
Dmitri Dolguikh
2026-07-31 15:58:40 +02:00
parent ea1b4d56e8
commit c2f8360b00
12 changed files with 125 additions and 34 deletions
@@ -7,7 +7,6 @@ import (
"reflect"
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgxpool"
networkmapdb "github.com/netbirdio/netbird/management/internals/network_map_db"
"github.com/netbirdio/netbird/shared/management/networkmap/nmdata"
)
@@ -25,10 +24,10 @@ func (pg *PgStore) GetNetwork(ctx context.Context, accountId string) (nmdata.Net
if err != nil {
return nmdata.Network{}, err
}
return GetNetworkViaConnection(ctx, c, accountId)
return GetNetworkViaPgxConnection(ctx, c.Conn(), accountId)
}
func GetNetworkViaConnection(ctx context.Context, con *pgxpool.Conn, accountId string) (nmdata.Network, error) {
func GetNetworkViaPgxConnection(ctx context.Context, con *pgx.Conn, accountId string) (nmdata.Network, error) {
rows, err := con.Query(ctx, GetNetworkQuery, accountId)
if err != nil {
return nmdata.Network{}, err