only check Go source files for non-thread-locked MakeGitError calls

1 files changed, 3 insertions, 1 deletions

diff --git a/script/check-MakeGitError-thread-lock.go b/script/check-MakeGitError-thread-lock.go
index 10ab39f..f6b01b3 100644
--- a/script/check-MakeGitError-thread-lock.go
+++ b/script/check-MakeGitError-thread-lock.go
@@ -9,6 +9,8 @@ import (
 	"go/printer"
 	"go/token"
 	"log"
+	"os"
+	"path/filepath"
 	"strings"
 )
 
@@ -24,7 +26,7 @@ func main() {
 		log.Fatal(err)
 	}
 
-	pkgs, err := parser.ParseDir(fset, bpkg.Dir, nil, 0)
+	pkgs, err := parser.ParseDir(fset, bpkg.Dir, func(fi os.FileInfo) bool { return filepath.Ext(fi.Name()) == ".go" }, 0)
 	if err != nil {
 		log.Fatal(err)
 	}