[Feature] add a cache to the FetchResult codepath to avoid API calls (#…

…45)

lint: fix `ConnectionId`

Fix CI for current master (#47)

* lint: fix `ConnectionId`

* Test fixes for previous change (`41c90a09`)

* nit: tweaks to dsn-test

Co-authored-by: Agam Brahma <agam@sigmacomputing.com>

fix var inits (#46)

connection.go

@@ -69,6 +69,7 @@ type snowflakeConn struct {
 	SQLState        string
 	telemetry       *snowflakeTelemetry
 	internal        InternalClient
+	execRespCache   *execRespCache
 }
 
 var (
@@ -223,6 +224,9 @@ func (sc *snowflakeConn) cleanup() {
 	// must flush log buffer while the process is running.
 	sc.rest = nil
 	sc.cfg = nil
+
+	releaseExecRespCache(sc.execRespCache)
+	sc.execRespCache = nil
 }
 
 func (sc *snowflakeConn) Close() (err error) {
@@ -502,6 +506,9 @@ func buildSnowflakeConn(ctx context.Context, config Config) (*snowflakeConn, err
 	if sc.cfg.DisableTelemetry {
 		sc.telemetry = &snowflakeTelemetry{enabled: false}
 	}
+	if sc.cfg.ConnectionID != "" {
+		sc.execRespCache = acquireExecRespCache(sc.cfg.ConnectionID)
+	}
 
 	// authenticate
 	sc.rest = &snowflakeRestful{

dsn.go

@@ -12,6 +12,8 @@ import (
 	"strconv"
 	"strings"
 	"time"
+
+	"github.com/google/uuid"
 )
 
 const (
@@ -80,6 +82,10 @@ type Config struct {
 	Transporter http.RoundTripper // RoundTripper to intercept HTTP requests and responses
 
 	DisableTelemetry bool // indicates whether to disable telemetry
+
+	// An identifier for this Config. Used to associate multiple connection instances with
+	// a single logical sql.DB connection.
+	ConnectionID string
 }
 
 // ocspMode returns the OCSP mode in string INSECURE, FAIL_OPEN, FAIL_CLOSED
@@ -195,6 +201,10 @@ func DSN(cfg *Config) (dsn string, err error) {
 
 	params.Add("validateDefaultParameters", strconv.FormatBool(cfg.ValidateDefaultParameters != ConfigBoolFalse))
 
+	if cfg.ConnectionID != "" {
+		params.Add("connectionId", cfg.ConnectionID)
+	}
+
 	dsn = fmt.Sprintf("%v:%v@%v:%v", url.QueryEscape(cfg.User), url.QueryEscape(cfg.Password), cfg.Host, cfg.Port)
 	if params.Encode() != "" {
 		dsn += "?" + params.Encode()
@@ -419,6 +429,10 @@ func fillMissingConfigParameters(cfg *Config) error {
 		cfg.ValidateDefaultParameters = ConfigBoolTrue
 	}
 
+	if cfg.ConnectionID == "" {
+		cfg.ConnectionID = uuid.New().String()
+	}
+
 	if strings.HasSuffix(cfg.Host, defaultDomain) && len(cfg.Host) == len(defaultDomain) {
 		return &SnowflakeError{
 			Number:      ErrCodeFailedToParseHost,