mirror of
https://github.com/agent-substrate/substrate.git
synced 2026-10-14 05:29:26 +08:00
Julian Gutierrez Oschmann
21b0b030bc
protoc's Python output puts the serialized descriptor on a single line which makes any two concurrent ateapi.proto changes conflict. This is unfortunate as both proto and go generated code usually merges with no conflict. Stop checking in the *_pb2*.py files. The locust and nighthawk-ingress images now generate them in a build stage via benchmarking/locust/codegen/generate.sh. Fixes #1824
106 lines
3.1 KiB
Bash
Executable File
106 lines
3.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# Copyright 2026 Google LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
set -o errexit -o nounset -o pipefail
|
|
|
|
ROOT="$(git rev-parse --show-toplevel)"
|
|
cd "${ROOT}"
|
|
|
|
# The boilerpolate header we use for generated Go code.
|
|
GO_BOILERPLATE=hack/boilerplate/go.txt
|
|
|
|
#
|
|
# The order of these matters - some of the codegen tools depend on the output
|
|
# of others.
|
|
#
|
|
|
|
# TODO: We should move the rest of these into this file.
|
|
function codegen::go_generate() {
|
|
echo "Running go generate"
|
|
go generate ./...
|
|
}
|
|
codegen::go_generate
|
|
|
|
function codegen::protobuf() {
|
|
local proto_dirs=()
|
|
# shellcheck disable=SC2207 # reading array
|
|
proto_dirs=(
|
|
$(git ls-files \
|
|
-cm \
|
|
--exclude-standard \
|
|
-- \
|
|
':(glob)**/*.proto' \
|
|
':!:vendor/*' \
|
|
':!:**/vendor/*' \
|
|
':!:third_party/*' \
|
|
':!:**/third_party/*' \
|
|
':!:_LICENSES/*' \
|
|
| while read -r FILE; do dirname "${FILE}"; done \
|
|
| sort \
|
|
| uniq)
|
|
)
|
|
|
|
echo "Generating protobuf code"
|
|
for dir in "${proto_dirs[@]}"; do
|
|
local protoc_gen_go
|
|
protoc_gen_go="$(./hack/run-tool.sh --print-bin-path protoc-gen-go)"
|
|
local protoc_gen_go_rpc
|
|
protoc_gen_go_rpc="$(./hack/run-tool.sh --print-bin-path protoc-gen-go-grpc)"
|
|
(
|
|
cd "${dir}" || exit 1
|
|
"${ROOT}"/hack/protoc.sh \
|
|
-I "${ROOT}" -I . \
|
|
--plugin=protoc-gen-go="${protoc_gen_go}" \
|
|
--plugin=protoc-gen-go-grpc="${protoc_gen_go_rpc}" \
|
|
--go_out=paths=source_relative:. \
|
|
--go-grpc_out=paths=source_relative:. \
|
|
./*.proto
|
|
)
|
|
done
|
|
}
|
|
codegen::protobuf
|
|
|
|
function codegen::validation() {
|
|
local validation_dirs=()
|
|
# shellcheck disable=SC2207 # reading array
|
|
validation_dirs=(
|
|
$(git grep -l \
|
|
'+k8s:validation-gen' \
|
|
-- \
|
|
':(glob)**/doc.go' \
|
|
':!:vendor/*' \
|
|
':!:**/vendor/*' \
|
|
':!:third_party/*' \
|
|
':!:**/third_party/*' \
|
|
':!:_LICENSES/*' \
|
|
| while read -r FILE; do dirname "${FILE}"; done \
|
|
| sort \
|
|
| uniq)
|
|
)
|
|
|
|
echo "Generating validation code"
|
|
for dir in "${validation_dirs[@]}"; do
|
|
./hack/run-tool.sh validation-gen \
|
|
--go-header-file="${GO_BOILERPLATE}" \
|
|
--output-file=zz_generated.validation.go \
|
|
--readonly-pkg=google.golang.org/protobuf/types/known/timestamppb \
|
|
--readonly-pkg=google.golang.org/protobuf/types/known/fieldmaskpb \
|
|
--readonly-pkg=google.golang.org/protobuf/types/known/emptypb \
|
|
"./${dir}"
|
|
done
|
|
}
|
|
codegen::validation
|