add mutex lock in getQueryContextCacheSize

snowflakedb · Oct 12, 2023 · b0ac257 · b0ac257
1 parent c2f3b53
commit b0ac257
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/htap.go b/htap.go
@@ -79,7 +79,11 @@ func (qcc *queryContextCache) prune(size int) {
 }
 
 func (qcc *queryContextCache) getQueryContextCacheSize(sc *snowflakeConn) int {
-	if sizeStr, ok := sc.cfg.Params[queryContextCacheSizeParamName]; ok {
+	paramsMutex.Lock()
+	sizeStr, ok := sc.cfg.Params[queryContextCacheSizeParamName]
+	paramsMutex.Unlock()
+
+	if ok {
 		size, err := strconv.Atoi(*sizeStr)
 		if err != nil {
 			logger.Warnf("cannot parse %v as int as query context cache size: %v", sizeStr, err)