milvus/scripts/proto_gen_go.sh
SimFG a55f739608
Separate public proto files (#19782)
Signed-off-by: SimFG <bang.fu@zilliz.com>

Signed-off-by: SimFG <bang.fu@zilliz.com>
2022-10-16 20:49:27 +08:00

95 lines
4.1 KiB
Bash
Executable File

#!/usr/bin/env bash
# Licensed to the LF AI & Data foundation under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you 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.
SCRIPTS_DIR=$(dirname "$0")
PROTO_DIR=$SCRIPTS_DIR/../internal/proto/
GOOGLE_PROTO_DIR=$SCRIPTS_DIR/../cmake_build/thirdparty/protobuf/protobuf-src/src/
API_PROTO_DIR=$SCRIPTS_DIR/../cmake_build/thirdparty/milvus-proto/proto/
PROGRAM=$(basename "$0")
GOPATH=$(go env GOPATH)
if [ -z $GOPATH ]; then
printf "Error: the environment variable GOPATH is not set, please set it before running %s\n" $PROGRAM > /dev/stderr
exit 1
fi
export PATH=${GOPATH}/bin:$PATH
echo `which protoc-gen-go`
# official go code ship with the crate, so we need to generate it manually.
pushd ${PROTO_DIR}
mkdir -p etcdpb
mkdir -p indexcgopb
mkdir -p internalpb
mkdir -p rootcoordpb
mkdir -p segcorepb
mkdir -p proxypb
mkdir -p indexpb
mkdir -p datapb
mkdir -p querypb
mkdir -p planpb
mkdir -p ../../cmd/tools/migration/legacy/legacypb
#${protoc} --proto_path="${GOOGLE_PROTO_DIR}" --proto_path=. \
# --go_opt="Mmilvus.proto=github.com/milvus-io/milvus/api/milvuspb;milvuspb" \
# --go_opt=Mcommon.proto=github.com/milvus-io/milvus/api/commonpb \
# --go_opt=Mschema.proto=github.com/milvus-io/milvus/api/schemapb \
# --go_out=plugins=grpc,paths=source_relative:./../../api/milvuspb milvus.proto
#
#${protoc} --proto_path="${GOOGLE_PROTO_DIR}" --proto_path=. \
# --go_opt=Mmilvus.proto=github.com/milvus-io/milvus/api/milvuspb \
# --go_opt=Mcommon.proto=github.com/milvus-io/milvus/api/commonpb \
# --go_opt="Mschema.proto=github.com/milvus-io/milvus/api/schemapb;schemapb" \
# --go_out=plugins=grpc,paths=source_relative:./../../api/schemapb schema.proto
#
#${protoc} --proto_path="${GOOGLE_PROTO_DIR}" --proto_path=. \
# --go_opt=Mmilvus.proto=github.com/milvus-io/milvus/api/milvuspb \
# --go_opt="Mcommon.proto=github.com/milvus-io/milvus/api/commonpb;commonpb" \
# --go_opt=Mschema.proto=github.com/milvus-io/milvus/api/schemapb \
# --go_out=plugins=grpc,paths=source_relative:./../../api/commonpb common.proto
#protoc_opt="${protoc} --proto_path=${GOOGLE_PROTO_DIR} --proto_path=${API_PROTO_DIR} --proto_path=.
# --go_opt=Mmilvus.proto=github.com/milvus-io/milvus/api/milvuspb
# --go_opt=Mcommon.proto=github.com/milvus-io/milvus/api/commonpb
# --go_opt=Mschema.proto=github.com/milvus-io/milvus/api/schemapb"
protoc_opt="${protoc} --proto_path=${GOOGLE_PROTO_DIR} --proto_path=${API_PROTO_DIR} --proto_path=."
${protoc_opt} --go_out=plugins=grpc,paths=source_relative:./etcdpb etcd_meta.proto
${protoc_opt} --go_out=plugins=grpc,paths=source_relative:./indexcgopb index_cgo_msg.proto
${protoc_opt} --go_out=plugins=grpc,paths=source_relative:./rootcoordpb root_coord.proto
${protoc_opt} --go_out=plugins=grpc,paths=source_relative:./internalpb internal.proto
${protoc_opt} --go_out=plugins=grpc,paths=source_relative:./proxypb proxy.proto
${protoc_opt} --go_out=plugins=grpc,paths=source_relative:./indexpb index_coord.proto
${protoc_opt} --go_out=plugins=grpc,paths=source_relative:./datapb data_coord.proto
${protoc_opt} --go_out=plugins=grpc,paths=source_relative:./querypb query_coord.proto
${protoc_opt} --go_out=plugins=grpc,paths=source_relative:./planpb plan.proto
${protoc_opt} --go_out=plugins=grpc,paths=source_relative:./segcorepb segcore.proto
${protoc_opt} --proto_path=../../cmd/tools/migration/legacy/ \
--go_out=plugins=grpc,paths=source_relative:../../cmd/tools/migration/legacy/legacypb legacy.proto
popd