fix concurrent map write

Phuong39 · Jul 28, 2022 · 45953fc · 45953fc
1 parent 43388e8
commit 45953fc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/internal/db/cache.go b/internal/db/cache.go
@@ -1,11 +1,15 @@
 package db
 
-import "github.com/zu1k/nali/pkg/dbif"
+import (
+	"sync"
+
+	"github.com/zu1k/nali/pkg/dbif"
+)
 
 var (
 	dbNameCache = make(map[string]dbif.DB)
 	dbTypeCache = make(map[dbif.QueryType]dbif.DB)
-	queryCache  = make(map[string]string)
+	queryCache  = sync.Map{}
 )
 
 var (

diff --git a/internal/db/db.go b/internal/db/db.go
@@ -70,14 +70,14 @@ func GetDB(typ dbif.QueryType) (db dbif.DB) {
 }
 
 func Find(typ dbif.QueryType, query string) string {
-	if result, found := queryCache[query]; found {
-		return result
+	if result, found := queryCache.Load(query); found {
+		return result.(string)
 	}
 	result, err := GetDB(typ).Find(query)
 	if err != nil {
 		return ""
 	}
 	r := strings.Trim(result.String(), " ")
-	queryCache[query] = r
+	queryCache.Store(query, r)
 	return r
 }