backport of commit 1fe69f2 (#147)

Co-authored-by: Robert <17119716+robmonte@users.noreply.github.com>

Author: hc-github-team-secure-vault-ecosystem

.changes/unreleased/BUG FIXES-20250909-165037.yaml

@@ -0,0 +1,5 @@
+kind: BUG FIXES
+body: Refresh the connection upon expiration
+time: 2025-09-09T16:50:37.281682-05:00
+custom:
+    Issue: "134"

connection_producer.go

@@ -10,15 +10,16 @@ import (
 	"database/sql"
 	"encoding/pem"
 	"fmt"
-	"github.com/hashicorp/errwrap"
-	log "github.com/hashicorp/go-hclog"
-	"github.com/hashicorp/go-secure-stdlib/parseutil"
-	"github.com/hashicorp/vault/sdk/database/helper/dbutil"
 	"net/url"
 	"regexp"
 	"sync"
 	"time"
 
+	"github.com/hashicorp/errwrap"
+	log "github.com/hashicorp/go-hclog"
+	"github.com/hashicorp/go-secure-stdlib/parseutil"
+	"github.com/hashicorp/vault/sdk/database/helper/dbutil"
+
 	"github.com/hashicorp/vault/sdk/database/helper/connutil"
 	"github.com/mitchellh/mapstructure"
 	"github.com/snowflakedb/gosnowflake"
@@ -149,8 +150,14 @@ func (c *snowflakeConnectionProducer) Connection(ctx context.Context) (interface
 		return nil, connutil.ErrNotInitialized
 	}
 
+	// If we already have a DB, test it and return
 	if c.snowflakeDB != nil {
-		return c.snowflakeDB, nil
+		if err := c.snowflakeDB.PingContext(ctx); err == nil {
+			return c.snowflakeDB, nil
+		}
+		// If the ping was unsuccessful, close it and ignore errors as we'll be
+		// reestablishing anyways
+		c.snowflakeDB.Close()
 	}
 
 	var db *sql.DB