enable session active when calling close()

snowflakedb · Mar 11, 2021 · 6d4a132 · 6d4a132
1 parent 8dbd9b2
commit 6d4a132
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/connection.go b/connection.go
@@ -203,9 +203,11 @@ func (sc *snowflakeConn) Close() (err error) {
 	logger.WithContext(sc.ctx).Infoln("Close")
 	sc.stopHeartBeat()
 
-	err = sc.rest.FuncCloseSession(sc.ctx, sc.rest, sc.rest.RequestTimeout)
-	if err != nil {
-		logger.Error(err)
+	if !sc.cfg.KeepSessionAlive {
+		err = sc.rest.FuncCloseSession(context.TODO(), sc.rest, sc.rest.RequestTimeout)
+		if err != nil {
+			logger.Error(err)
+		}
 	}
 	sc.cleanup()
 	return nil

diff --git a/dsn.go b/dsn.go
@@ -71,8 +71,9 @@ type Config struct {
 	InsecureMode bool             // driver doesn't check certificate revocation status
 	OCSPFailOpen OCSPFailOpenMode // OCSP Fail Open
 
-	Token         string        // Token to use for OAuth other forms of token based auth
-	TokenAccessor TokenAccessor // Optional token accessor to use
+	Token            string        // Token to use for OAuth other forms of token based auth
+	TokenAccessor    TokenAccessor // Optional token accessor to use
+	KeepSessionAlive bool          // Enables the session to persist even after the connection is closed
 
 	PrivateKey *rsa.PrivateKey // Private key used to sign JWT