github.com/turbot/steampipe@v1.7.0-rc.0.0.20240517123944-7cef272d4458/pkg/serversettings/load.go (about)

     1  package serversettings
     2  
     3  import (
     4  	"context"
     5  	"fmt"
     6  
     7  	"github.com/jackc/pgx/v5"
     8  	"github.com/jackc/pgx/v5/pgxpool"
     9  	"github.com/turbot/steampipe-plugin-sdk/v5/sperr"
    10  	"github.com/turbot/steampipe/pkg/constants"
    11  	"github.com/turbot/steampipe/pkg/db/db_common"
    12  )
    13  
    14  func Load(ctx context.Context, pool *pgxpool.Pool) (serverSettings *db_common.ServerSettings, e error) {
    15  	conn, err := pool.Acquire(ctx)
    16  	if err != nil {
    17  		return nil, err
    18  	}
    19  	defer conn.Release()
    20  	defer func() {
    21  		// this function uses reflection to extract and convert values
    22  		// we need to be able to recover from panics while using reflection
    23  		if r := recover(); r != nil {
    24  			e = sperr.ToError(r, sperr.WithMessage("error loading server settings"))
    25  		}
    26  	}()
    27  	rows, err := conn.Query(ctx, fmt.Sprintf("SELECT * FROM %s.%s", constants.InternalSchema, constants.ServerSettingsTable))
    28  	if err != nil {
    29  		return nil, err
    30  	}
    31  	defer rows.Close()
    32  
    33  	serverSettings, e = pgx.CollectOneRow(rows, pgx.RowToAddrOfStructByName[db_common.ServerSettings])
    34  	return
    35  }