修复并发安全

farseer-go · Jul 27, 2023 · 7cfada3 · 7cfada3
1 parent 232662f
commit 7cfada3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/modules/getDependModule.go b/modules/getDependModule.go
@@ -30,7 +30,9 @@ func GetDependModule(module ...FarseerModule) []FarseerModule {
 		moduleName := reflect.TypeOf(farseerModule).String()
 		flog.LogBuffer <- fmt.Sprint("Loading Module：" + flog.Colors[5](moduleName) + "")
 		modules = append(modules, farseerModule)
+		moduleMapLocker.Lock()
 		moduleMap[moduleName] = 0
+		moduleMapLocker.Unlock()
 	}
 	return modules
 }

diff --git a/modules/moduleMap.go b/modules/moduleMap.go
@@ -4,12 +4,17 @@ import (
 	"fmt"
 	"github.com/farseer-go/fs/flog"
 	"reflect"
+	"sync"
 )
 
 var moduleMap = make(map[string]int64)
+var moduleMapLocker sync.RWMutex
 
 // IsLoad 模块是否加载
 func IsLoad(module FarseerModule) bool {
+	moduleMapLocker.RLock()
+	defer moduleMapLocker.RUnlock()
+
 	moduleName := reflect.TypeOf(module).String()
 	_, isExists := moduleMap[moduleName]
 	return isExists