@@ -43,8 +43,14 @@ func Run(dir, op, modName string, tag int, buildFlags []string) error {
4343 if err != nil {
4444 return errors .Wrap (err , "could not load package" )
4545 }
46+ ids := map [string ]struct {}{}
47+ files := map [string ]struct {}{}
4648 for _ , p := range pkgs {
47- err = updateImportPath (p , modName , newModPath )
49+ if _ , ok := ids [p .ID ]; ok {
50+ continue
51+ }
52+ ids [p .ID ] = struct {}{}
53+ err = updateImportPath (p , modName , newModPath , files )
4854 if err != nil {
4955 return err
5056 }
@@ -127,8 +133,13 @@ func getPrevious(s string) string {
127133 return strings .Join (ss [:len (ss )- 1 ], "/" ) + "/v" + strconv .Itoa (newV )
128134}
129135
130- func updateImportPath (p * packages.Package , old , new string ) error {
136+ func updateImportPath (p * packages.Package , old , new string , files map [ string ] struct {} ) error {
131137 for _ , syn := range p .Syntax {
138+ goFileName := p .Fset .File (syn .Pos ()).Name ()
139+ if _ , ok := files [goFileName ]; ok {
140+ continue
141+ }
142+ files [goFileName ] = struct {}{}
132143 var rewritten bool
133144 for _ , i := range syn .Imports {
134145 imp := strings .Replace (i .Path .Value , `"` , `` , 2 )
@@ -144,7 +155,6 @@ func updateImportPath(p *packages.Package, old, new string) error {
144155 continue
145156 }
146157
147- goFileName := p .Fset .File (syn .Pos ()).Name ()
148158 f , err := os .Create (goFileName )
149159 if err != nil {
150160 return errors .Wrapf (err , "could not create go file %v" , goFileName )
0 commit comments