build: fix generation when vendoring dependencies (#22089)

.circleci/config.yml

@@ -382,6 +382,12 @@ jobs:
           name: Check flag generation
           command: ./scripts/ci/lint/flags.bash
       - run: make checkgenerate
+      - run:
+          name: Check generation when vendoring deps
+          command: |
+            go mod vendor
+            make checkgenerate
+            rm -rf vendor
       - run: make vet
       - run: make checkfmt
       - run: go install honnef.co/go/tools/cmd/staticcheck

pkg/tracing/wire/binary.go

@@ -1,4 +1,4 @@
 //Package wire is used to serialize a trace.
 package wire
 
-//go:generate sh -c "protoc -I$(go list -f '{{ .Dir }}' -m github.com/gogo/protobuf) -I. --gogofaster_out=Mgoogle/protobuf/timestamp.proto=github.com/gogo/protobuf/types:. binary.proto"
+//go:generate sh -c "protoc -I$(../../../scripts/gogo-path.sh) -I. --gogofaster_out=Mgoogle/protobuf/timestamp.proto=github.com/gogo/protobuf/types:. binary.proto"

scripts/gogo-path.sh

@@ -0,0 +1,13 @@
+#!/usr/bin/env bash
+set -e
+
+declare -r SCRIPT_DIR=$(cd $(dirname ${0}) >/dev/null 2>&1 && pwd)
+declare -r ROOT_DIR=$(dirname ${SCRIPT_DIR})
+
+declare -r GOGO_PATH=github.com/gogo/protobuf
+
+if [ -d "${ROOT_DIR}/vendor" ]; then
+  echo "${ROOT_DIR}/vendor/${GOGO_PATH}"
+else
+  go list -f '{{ .Dir }}' -m ${GOGO_PATH}
+fi

storage/reads/datatypes/gen.go

@@ -1,3 +1,3 @@
 package datatypes
 
-//go:generate sh -c "protoc -I$(go list -f '{{ .Dir }}' -m github.com/gogo/protobuf) -I. --plugin ../../../scripts/protoc-gen-gogofaster --gogofaster_out=Mgoogle/protobuf/empty.proto=github.com/gogo/protobuf/types,Mgoogle/protobuf/any.proto=github.com/gogo/protobuf/types,plugins=grpc:. storage_common.proto predicate.proto"
+//go:generate sh -c "protoc -I$(../../../scripts/gogo-path.sh) -I. --plugin ../../../scripts/protoc-gen-gogofaster --gogofaster_out=Mgoogle/protobuf/empty.proto=github.com/gogo/protobuf/types,Mgoogle/protobuf/any.proto=github.com/gogo/protobuf/types,plugins=grpc:. storage_common.proto predicate.proto"

v1/services/storage/gen.go

@@ -1,3 +1,3 @@
 package storage
 
-//go:generate sh -c "protoc -I$(go list -f '{{ .Dir }}' -m github.com/gogo/protobuf) -I. --gogofaster_out=. source.proto"
+//go:generate sh -c "protoc -I$(../../../scripts/gogo-path.sh) -I. --gogofaster_out=. source.proto"