feat: add get conn

opengovern · Oct 7, 2024 · 12b2e77 · 12b2e77
1 parent fff8217
commit 12b2e77
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/steampipe/live-context.go b/pkg/steampipe/live-context.go
@@ -60,6 +60,10 @@ func NewSelfClient(ctx context.Context) (*SelfClient, error) {
 	return &SelfClient{conn: conn, createdAt: time.Now()}, nil
 }
 
+func (sc *SelfClient) GetConnection() *pgxpool.Pool {
+	return sc.conn
+}
+
 func (sc *SelfClient) GetConfigTableValueOrNil(ctx context.Context, key KaytuConfigKey) (*string, error) {
 	var value *string
 	// Create table if not exists