From c3322c570012c4ddf93999d5eb46a017f135d933 Mon Sep 17 00:00:00 2001
From: kai <kai@turbot.com>
Date: Wed, 8 May 2024 10:50:06 +0100
Subject: [PATCH] Ensure QueryData passed to ConnectionKeyColumns value
 callback is populated with ConnectionManager. Closes #797

(cherry picked from commit 95898beee5345a7b212b972cee02dd0d7cc135d2)
---
 plugin/plugin_connection_key_columns.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/plugin/plugin_connection_key_columns.go b/plugin/plugin_connection_key_columns.go
index e05ab055..7c5b33f6 100644
--- a/plugin/plugin_connection_key_columns.go
+++ b/plugin/plugin_connection_key_columns.go
@@ -3,6 +3,7 @@ package plugin
 import (
 	"context"
 	typeHelpers "github.com/turbot/go-kit/types"
+	connection_manager "github.com/turbot/steampipe-plugin-sdk/v5/connection"
 	"github.com/turbot/steampipe-plugin-sdk/v5/grpc"
 	"github.com/turbot/steampipe-plugin-sdk/v5/grpc/proto"
 	"github.com/turbot/steampipe-plugin-sdk/v5/plugin/quals"
@@ -164,9 +165,11 @@ func (p *Plugin) getConnectionKeyColumnValue(ctx context.Context, connectionName
 		return nil, err
 	}
 	d := &QueryData{
-		Connection:      p.ConnectionMap[connectionName].Connection,
-		ConnectionCache: connectionCache,
+		Connection:        p.ConnectionMap[connectionName].Connection,
+		ConnectionCache:   connectionCache,
+		ConnectionManager: connection_manager.NewManager(connectionCache),
 	}
+
 	h := &HydrateData{}
 	val, err := valueFunc(ctx, d, h)
 	if err != nil {