fix(wire): imports mutex

go-zing · Dec 1, 2023 · 794fa9e · 794fa9e
1 parent a37ac55
commit 794fa9e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/internal/plugins/wire.go b/internal/plugins/wire.go
@@ -417,8 +417,9 @@ func (w Wire) parseInterfaceMethods(name string, dir string, imports zcore.Impor
 
 func (w Wire) generateSet(dir string, sets map[string]*wireDeclSet) (err error) {
 	var (
-		dstImports    = zcore.Imports{"github.com/google/wire": "wire"}
-		dstImportPath = zcore.GetImportPath(dir)
+		dstImports      = zcore.Imports{"github.com/google/wire": "wire"}
+		dstImportPath   = zcore.GetImportPath(dir)
+		dstImportsMutex sync.Mutex
 
 		aopImportsMutex sync.Mutex
 		aopImports      = make(zcore.Imports)
@@ -436,6 +437,8 @@ func (w Wire) generateSet(dir string, sets map[string]*wireDeclSet) (err error)
 
 		// fix name import package selector
 		fp := func(name string) string {
+			dstImportsMutex.Lock()
+			defer dstImportsMutex.Unlock()
 			return zcore.FixPackage(name, el.Path, dstImportPath, srcImports, dstImports)
 		}