enhance: [skip e2e] Improve update-milvus-api command for version consistency (#42074)

issue: #42073 
Enhanced the update-milvus-api Makefile command to update milvus-proto
version across all 4 go.mod files in the project:

- Update main go.mod (root module)
- Update client/go.mod (client module)
- Update pkg/go.mod (pkg module)
- Update tests/go_client/go.mod (test client module)

Improvements:
- Add intelligent version detection (git tag vs commit ID)
- Add comprehensive error handling and validation
- Add detailed progress feedback and success reporting
- Add proper directory existence checks
- Add UPDATE_MILVUS_API.md documentation with usage examples

This ensures all modules use the same milvus-proto version, improving
project consistency and maintainability.

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
This commit is contained in:
wei liu 2025-05-26 17:08:47 +08:00 committed by GitHub
parent bfa948c2d4
commit b6723296b2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 125 additions and 9 deletions

72
UPDATE_MILVUS_API.md Normal file
View File

@ -0,0 +1,72 @@
# Update Milvus API Version
This document explains how to use the `update-milvus-api` command to update the milvus-proto version across all go.mod files in the project.
## Overview
The `update-milvus-api` command has been enhanced to update the milvus-proto version in all 4 go.mod files:
1. `go.mod` (main module)
2. `client/go.mod` (client module)
3. `pkg/go.mod` (pkg module)
4. `tests/go_client/go.mod` (test client module)
## Usage
### Update to a specific tagged version
```bash
make update-milvus-api PROTO_API_VERSION=v2.3.0-dev.1
```
### Update to a specific commit ID
```bash
make update-milvus-api PROTO_API_VERSION=4080770055ad
```
## What the command does
1. **Validates the version**: Checks if the provided version is a valid git tag or treats it as a commit ID
2. **Updates all go.mod files**: Updates the milvus-proto dependency in all 4 go.mod files
3. **Runs go mod tidy**: Ensures all dependencies are properly resolved
4. **Updates the local milvus-proto repo**: Updates the third-party milvus-proto repository if it exists
5. **Provides feedback**: Shows which files were updated
## Example output
```
----------------------------
Update the milvus-proto/go-api/v2@v2.3.0-dev.1
Updating milvus-proto version in all go.mod files...
Updating main go.mod...
Updating client/go.mod...
Updating pkg/go.mod...
Updating tests/go_client/go.mod...
----------------------------
Update the milvus-proto repo
----------------------------
Successfully updated milvus-proto version to v2.3.0-dev.1 in all go.mod files:
- go.mod
- client/go.mod
- pkg/go.mod
- tests/go_client/go.mod
```
## Error handling
- If no version is provided, the command will show usage examples
- If the milvus-proto third-party directory doesn't exist, a warning will be displayed but the go.mod updates will still proceed
- If a git branch with the same name already exists, the command will checkout the commit directly
## Prerequisites
- Go must be installed and available in PATH
- Git must be installed and available in PATH
- Internet connection to fetch the milvus-proto repository
## Notes
- The command will automatically determine if the provided version is a git tag or commit ID
- All go.mod files will be updated to the same version
- The command includes proper error handling and user feedback

View File

@ -24,20 +24,64 @@ echo "Update the milvus-proto/go-api/v2@${version}"
if [[ $commitID == "" ]]; then if [[ $commitID == "" ]]; then
echo "${version} is not a valid tag, try to use it as commit ID" echo "${version} is not a valid tag, try to use it as commit ID"
commitID=$version commitID=$version
go get -u github.com/milvus-io/milvus-proto/go-api/v2@$commitID update_version=$commitID
else else
go get -u github.com/milvus-io/milvus-proto/go-api/v2@$version update_version=$version
fi fi
SCRIPTS_DIR=$(dirname "$0") SCRIPTS_DIR=$(dirname "$0")
EXAMPLE_DIR=$SCRIPTS_DIR/../cmake_build/thirdparty/protobuf/protobuf-src/examples PROJECT_ROOT=$(cd "$SCRIPTS_DIR/.." && pwd)
rm -rf $EXAMPLE_DIR
# Update all 4 go.mod files
echo "Updating milvus-proto version in all go.mod files..."
# 1. Update main go.mod
echo "Updating main go.mod..."
cd "$PROJECT_ROOT"
go get -u github.com/milvus-io/milvus-proto/go-api/v2@$update_version
go mod tidy go mod tidy
line echo "Update the milvus-proto repo" # 2. Update client/go.mod
echo "Updating client/go.mod..."
cd "$PROJECT_ROOT/client"
go get -u github.com/milvus-io/milvus-proto/go-api/v2@$update_version
go mod tidy
# 3. Update pkg/go.mod
echo "Updating pkg/go.mod..."
cd "$PROJECT_ROOT/pkg"
go get -u github.com/milvus-io/milvus-proto/go-api/v2@$update_version
go mod tidy
# 4. Update tests/go_client/go.mod
echo "Updating tests/go_client/go.mod..."
cd "$PROJECT_ROOT/tests/go_client"
go get -u github.com/milvus-io/milvus-proto/go-api/v2@$update_version
go mod tidy
# Return to project root
cd "$PROJECT_ROOT"
# Clean up protobuf examples directory
EXAMPLE_DIR=$SCRIPTS_DIR/../cmake_build/thirdparty/protobuf/protobuf-src/examples
rm -rf $EXAMPLE_DIR
line
echo "Update the milvus-proto repo"
THIRD_PARTY_DIR=$SCRIPTS_DIR/../cmake_build/thirdparty THIRD_PARTY_DIR=$SCRIPTS_DIR/../cmake_build/thirdparty
if [ -d "$THIRD_PARTY_DIR/milvus-proto" ]; then
pushd $THIRD_PARTY_DIR/milvus-proto pushd $THIRD_PARTY_DIR/milvus-proto
git fetch git fetch
git checkout -b $version $commitID git checkout -b $version $commitID 2>/dev/null || git checkout $commitID
popd popd
else
echo "Warning: milvus-proto directory not found at $THIRD_PARTY_DIR/milvus-proto"
fi
line
echo "Successfully updated milvus-proto version to $update_version in all go.mod files:"
echo " - go.mod"
echo " - client/go.mod"
echo " - pkg/go.mod"
echo " - tests/go_client/go.mod"