From 0aab478bd80381ed2f53c075762e4b2d18630e5f Mon Sep 17 00:00:00 2001
From: Joseph Watson <jtwatson@linux-consulting.us>
Date: Sun, 2 Apr 2023 02:00:26 -0400
Subject: [PATCH] feat(spanner): Add option to disable closing spanner clients

---
 database/spanner/spanner.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/database/spanner/spanner.go b/database/spanner/spanner.go
index b733302d5..12672b060 100644
--- a/database/spanner/spanner.go
+++ b/database/spanner/spanner.go
@@ -56,6 +56,9 @@ type Config struct {
 	// Parsing outputs clean DDL statements such as reformatted
 	// and void of comments.
 	CleanStatements bool
+	// DoNotCloseSpannerClients turns Close() into a no-op so the
+	// provided spanner clients are not closed
+	DoNotCloseSpannerClients bool
 }
 
 // Spanner implements database.Driver for Google Cloud Spanner
@@ -146,6 +149,9 @@ func (s *Spanner) Open(url string) (database.Driver, error) {
 
 // Close implements database.Driver
 func (s *Spanner) Close() error {
+	if s.config.DoNotCloseSpannerClients {
+		return nil
+	}
 	s.db.data.Close()
 	return s.db.admin.Close()
 }