Files
substrate/hack/update/codegen.sh
T
Benjamin Elder 2c429a9906 multi-actor worker API (#1283)
Part of #1266 

This is a draft of the core API + data store changes.

It's still a large PR, apologies.

The "as rows" commit could be split out, but this takes it to ~all of
the breaking changes we can't hide behind updating internals.

Same for the claimlock, but in both cases it seems these are worth
understanding when considering the API shape.

They're loadbearing for performance once we actually have multi-actor
workers.
2026-09-03 21:19:21 -07:00

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