Using pacakges.NeedName (#418)

packages.LoadFiles is deprecated. It is not future-proof to use it. In addition, I found all mockgen need is Package.PkgPath, so packages.NeedName is sufficient.
golang · Mar 28, 2020 · ccaa079 · ccaa079
1 parent 0b87a54
commit ccaa079
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mockgen/parse.go b/mockgen/parse.go
@@ -548,7 +548,11 @@ func packageNameOfDir(srcDir string) (string, error) {
 
 // parseImportPackage get package import path via source file
 func parsePackageImport(source, srcDir string) (string, error) {
-	cfg := &packages.Config{Mode: packages.LoadFiles, Tests: true, Dir: srcDir}
+	cfg := &packages.Config{
+		Mode:  packages.NeedName,
+		Tests: true,
+		Dir:   srcDir,
+	}
 	pkgs, err := packages.Load(cfg, "file="+source)
 	if err != nil {
 		return "", err