fx: mutex

0xPolygonID · Oct 2, 2024 · f07b643 · f07b643
1 parent 94a4e82
commit f07b643
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/internal/loader/w3c_loader.go b/internal/loader/w3c_loader.go
@@ -18,13 +18,13 @@ type W3CDocumentLoader struct {
 }
 
 type w3CDocumentLoaderLocalCache struct {
-	mutex sync.Mutex
+	mutex sync.RWMutex
 	data  map[string]*ld.RemoteDocument
 }
 
 func newLocalCache() *w3CDocumentLoaderLocalCache {
 	return &w3CDocumentLoaderLocalCache{
-		mutex: sync.Mutex{},
+		mutex: sync.RWMutex{},
 		data:  make(map[string]*ld.RemoteDocument),
 	}
 }
@@ -39,8 +39,8 @@ func (c *w3CDocumentLoaderLocalCache) Get(key string) (doc *ld.RemoteDocument, e
 
 func (c *w3CDocumentLoaderLocalCache) Set(key string, doc *ld.RemoteDocument, expireTime time.Time) error {
 	c.mutex.Lock()
+	defer c.mutex.Unlock()
 	c.data[key] = doc
-	c.mutex.Unlock()
 	return nil
 }