fix(gpd): don't panic on invalid root

bazelbuild · Dec 17, 2023 · 56e63e8 · 56e63e8
1 parent d5bc6ad
commit 56e63e8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/go/tools/gopackagesdriver/packageregistry.go b/go/tools/gopackagesdriver/packageregistry.go
@@ -16,6 +16,7 @@ package main
 
 import (
 	"fmt"
+	"os"
 	"strconv"
 	"strings"
 )
@@ -83,6 +84,11 @@ func (pr *PackageRegistry) ResolveImports() error {
 func (pr *PackageRegistry) walk(acc map[string]*FlatPackage, root string) {
 	pkg := pr.packagesByID[root]
 
+	if pkg == nil {
+		fmt.Fprintf(os.Stderr, "Error: package ID not found %v\n", root)
+		return
+	}
+
 	acc[pkg.ID] = pkg
 	for _, pkgID := range pkg.Imports {
 		if _, ok := acc[pkgID]; !ok {