netbird/management/internals/network_map_db/pgsql/dns_settings.go

package networkmap_pgsql

import (
	"context"
	"encoding/json"

	"github.com/jackc/pgx/v5"
	"github.com/netbirdio/netbird/shared/management/networkmap/nmdata"
)

const (
	GetDnsSettingsQuery = `
	select dns_settings_disabled_management_groups
	from accounts
	where id=$1
	`
)

func (pg *PgStore) GetDnsSettings(ctx context.Context, accountId string) (nmdata.DNSSettings, error) {
	c, err := pg.Pool.Acquire(ctx)
	if err != nil {
		return nmdata.DNSSettings{}, err
	}
	return GetDnsSettingsViaPgxConnection(ctx, c.Conn(), accountId)
}

func GetDnsSettingsViaPgxConnection(ctx context.Context, con *pgx.Conn, accountId string) (nmdata.DNSSettings, error) {
	rows, err := con.Query(ctx, GetDnsSettingsQuery, accountId)
	if err != nil {
		return nmdata.DNSSettings{}, err
	}

	return pgx.CollectOneRow(rows, rowToDnsSettings)
}

func rowToDnsSettings(row pgx.CollectableRow) (nmdata.DNSSettings, error) {
	var value nmdata.DNSSettings
	var settings json.RawMessage

	if err := row.Scan(&settings); err != nil {
		return value, err
	}

	if err := json.Unmarshal(settings, &value.DisabledManagementGroups); err != nil {
		return value, err
	}

	return value, nil
}