milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make rootcoord client compatible with older version (#21651) (#21670) 

Signed-off-by: longjiquan <jiquan.long@zilliz.com>
pull/21695/head
Jiquan Long 2023-01-13 15:37:50 +08:00 committed by GitHub
parent 0a2dfd871f
commit 850bf9d0ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
internal/distributed/rootcoord/client/client.go
View File

@ -35,6 +35,8 @@ import (
clientv3 "go.etcd.io/etcd/client/v3"
"go.uber.org/zap"
"google.golang.org/grpc"
grpcCodes "google.golang.org/grpc/codes"
grpcStatus "google.golang.org/grpc/status"
)
var ClientParams paramtable.GrpcClientConfig
@ -240,8 +242,8 @@ func (c *Client) DescribeCollection(ctx context.Context, in *milvuspb.DescribeCo
return ret.(*milvuspb.DescribeCollectionResponse), err
}
// DescribeCollectionInternal return collection info
func (c *Client) DescribeCollectionInternal(ctx context.Context, in *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) {
// describeCollectionInternal return collection info
func (c *Client) describeCollectionInternal(ctx context.Context, in *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) {
in = typeutil.Clone(in)
commonpbutil.UpdateMsgBase(
in.GetBase(),
@ -259,6 +261,15 @@ func (c *Client) DescribeCollectionInternal(ctx context.Context, in *milvuspb.De
return ret.(*milvuspb.DescribeCollectionResponse), err
}
func (c *Client) DescribeCollectionInternal(ctx context.Context, in *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) {
resp, err := c.describeCollectionInternal(ctx, in)
status, ok := grpcStatus.FromError(err)
if ok && status.Code() == grpcCodes.Unimplemented {
return c.DescribeCollection(ctx, in)
}
return resp, err
}
// ShowCollections list all collection names
func (c *Client) ShowCollections(ctx context.Context, in *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error) {
in = typeutil.Clone(in)
@ -372,8 +383,8 @@ func (c *Client) ShowPartitions(ctx context.Context, in *milvuspb.ShowPartitions
return ret.(*milvuspb.ShowPartitionsResponse), err
}
// ShowPartitionsInternal list all partitions in collection
func (c *Client) ShowPartitionsInternal(ctx context.Context, in *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) {
// showPartitionsInternal list all partitions in collection
func (c *Client) showPartitionsInternal(ctx context.Context, in *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) {
in = typeutil.Clone(in)
commonpbutil.UpdateMsgBase(
in.GetBase(),
@ -391,6 +402,15 @@ func (c *Client) ShowPartitionsInternal(ctx context.Context, in *milvuspb.ShowPa
return ret.(*milvuspb.ShowPartitionsResponse), err
}
func (c *Client) ShowPartitionsInternal(ctx context.Context, in *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) {
resp, err := c.showPartitionsInternal(ctx, in)
status, ok := grpcStatus.FromError(err)
if ok && status.Code() == grpcCodes.Unimplemented {
return c.ShowPartitions(ctx, in)
}
return resp, err
}
// AllocTimestamp global timestamp allocator
func (c *Client) AllocTimestamp(ctx context.Context, in *rootcoordpb.AllocTimestampRequest) (*rootcoordpb.AllocTimestampResponse, error) {
in = typeutil.Clone(in)