From 2226fb42630e931d577b01bba615d0ad5fffb4d0 Mon Sep 17 00:00:00 2001
From: Karol Kokoszka <karol.kokoszka@scylladb.com>
Date: Thu, 28 Mar 2024 18:12:26 +0100
Subject: [PATCH] fix(session): close client after session creation

---
 pkg/service/cluster/service.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/pkg/service/cluster/service.go b/pkg/service/cluster/service.go
index 4fcc05f5a4..888b3ed617 100644
--- a/pkg/service/cluster/service.go
+++ b/pkg/service/cluster/service.go
@@ -578,6 +578,11 @@ func (s *Service) GetSession(ctx context.Context, clusterID uuid.UUID) (session
 	if err != nil {
 		return session, errors.Wrap(err, "get client")
 	}
+	defer func() {
+		if err := client.Close(); err != nil {
+			s.logger.Error(ctx, "Couldn't close scylla client", "error", err)
+		}
+	}()
 
 	ni, err := client.AnyNodeInfo(ctx)
 	if err != nil {