From 760a2d9aa73f30be85b831abccd8aabaec464dea Mon Sep 17 00:00:00 2001 From: "cai.zhang" Date: Tue, 25 Jul 2023 10:05:00 +0800 Subject: [PATCH] Support AllocTimestamp api for Milvus (#25784) Signed-off-by: cai.zhang --- Makefile | 2 +- go.mod | 2 +- go.sum | 2 + internal/core/src/pb/common.pb.cc | 207 ++-- internal/core/src/pb/common.pb.h | 3 + internal/datacoord/index_service.go | 24 +- internal/datacoord/index_service_test.go | 1 + internal/distributed/proxy/service.go | 4 + internal/distributed/proxy/service_test.go | 9 + internal/mocks/mock_proxy.go | 1298 +++++--------------- internal/proto/index_coord.proto | 1 + internal/proto/indexpb/index_coord.pb.go | 293 ++--- internal/proxy/impl.go | 25 + internal/proxy/impl_test.go | 49 + internal/proxy/task_index.go | 2 +- internal/types/types.go | 2 + 16 files changed, 678 insertions(+), 1246 deletions(-) diff --git a/Makefile b/Makefile index 006115dfe8..3749302de3 100644 --- a/Makefile +++ b/Makefile @@ -314,7 +314,7 @@ rpm: install @QA_RPATHS="$$[ 0x001|0x0002|0x0020 ]" rpmbuild -ba ./build/rpm/milvus.spec mock-proxy: - mockery --name=ProxyComponent --dir=$(PWD)/internal/types --output=$(PWD)/internal/mocks --filename=mock_proxy.go --structname=Proxy --with-expecter + mockery --name=ProxyComponent --dir=$(PWD)/internal/types --output=$(PWD)/internal/mocks --filename=mock_proxy.go --structname=MockProxy --with-expecter mock-datanode: mockery --name=DataNode --dir=$(PWD)/internal/types --output=$(PWD)/internal/mocks --filename=mock_datanode.go --with-expecter diff --git a/go.mod b/go.mod index 0a76f55d73..82505dd1d2 100644 --- a/go.mod +++ b/go.mod @@ -23,7 +23,7 @@ require ( github.com/klauspost/compress v1.16.5 github.com/lingdor/stackerror v0.0.0-20191119040541-976d8885ed76 github.com/mgutz/ansi v0.0.0-20200706080929-d51e80ef957d - github.com/milvus-io/milvus-proto/go-api/v2 v2.3.0-dev.1 + github.com/milvus-io/milvus-proto/go-api/v2 v2.3.0-dev.1.0.20230716112827-c3fe148f5e1d github.com/minio/minio-go/v7 v7.0.17 github.com/nats-io/nats-server/v2 v2.9.17 github.com/nats-io/nats.go v1.24.0 diff --git a/go.sum b/go.sum index 743cfb6217..c391f3d02a 100644 --- a/go.sum +++ b/go.sum @@ -586,6 +586,8 @@ github.com/milvus-io/gorocksdb v0.0.0-20220624081344-8c5f4212846b h1:TfeY0NxYxZz github.com/milvus-io/gorocksdb v0.0.0-20220624081344-8c5f4212846b/go.mod h1:iwW+9cWfIzzDseEBCCeDSN5SD16Tidvy8cwQ7ZY8Qj4= github.com/milvus-io/milvus-proto/go-api/v2 v2.3.0-dev.1 h1:x6vhrVyK3wEuXIDHt0uk2l/UFPa/RRGWk1nkjgN5jkI= github.com/milvus-io/milvus-proto/go-api/v2 v2.3.0-dev.1/go.mod h1:1OIl0v5PQeNxIJhCvY+K55CBUOYDZevw9g9380u1Wek= +github.com/milvus-io/milvus-proto/go-api/v2 v2.3.0-dev.1.0.20230716112827-c3fe148f5e1d h1:XsQQ/MigebXEE2VXPKKmA3K7OHC+mkEUiErWvaWMikI= +github.com/milvus-io/milvus-proto/go-api/v2 v2.3.0-dev.1.0.20230716112827-c3fe148f5e1d/go.mod h1:1OIl0v5PQeNxIJhCvY+K55CBUOYDZevw9g9380u1Wek= github.com/milvus-io/pulsar-client-go v0.6.10 h1:eqpJjU+/QX0iIhEo3nhOqMNXL+TyInAs1IAHZCrCM/A= github.com/milvus-io/pulsar-client-go v0.6.10/go.mod h1:lQqCkgwDF8YFYjKA+zOheTk1tev2B+bKj5j7+nm8M1w= github.com/minio/asm2plan9s v0.0.0-20200509001527-cdd76441f9d8 h1:AMFGa4R4MiIpspGNG7Z948v4n35fFGB3RR3G/ry4FWs= diff --git a/internal/core/src/pb/common.pb.cc b/internal/core/src/pb/common.pb.cc index 9b1367f984..d683af868b 100644 --- a/internal/core/src/pb/common.pb.cc +++ b/internal/core/src/pb/common.pb.cc @@ -520,114 +520,116 @@ const char descriptor_table_protodef_common_2eproto[] PROTOBUF_SECTION_VARIABLE( "\020\004\022\t\n\005Retry\020\005*\202\001\n\014SegmentState\022\024\n\020Segmen" "tStateNone\020\000\022\014\n\010NotExist\020\001\022\013\n\007Growing\020\002\022" "\n\n\006Sealed\020\003\022\013\n\007Flushed\020\004\022\014\n\010Flushing\020\005\022\013" - "\n\007Dropped\020\006\022\r\n\tImporting\020\007*>\n\017Placeholde" + "\n\007Dropped\020\006\022\r\n\tImporting\020\007*V\n\017Placeholde" "rType\022\010\n\004None\020\000\022\020\n\014BinaryVector\020d\022\017\n\013Flo" - "atVector\020e*\237\020\n\007MsgType\022\r\n\tUndefined\020\000\022\024\n" - "\020CreateCollection\020d\022\022\n\016DropCollection\020e\022" - "\021\n\rHasCollection\020f\022\026\n\022DescribeCollection" - "\020g\022\023\n\017ShowCollections\020h\022\024\n\020GetSystemConf" - "igs\020i\022\022\n\016LoadCollection\020j\022\025\n\021ReleaseColl" - "ection\020k\022\017\n\013CreateAlias\020l\022\r\n\tDropAlias\020m" - "\022\016\n\nAlterAlias\020n\022\023\n\017AlterCollection\020o\022\024\n" - "\020RenameCollection\020p\022\021\n\rDescribeAlias\020q\022\017" - "\n\013ListAliases\020r\022\024\n\017CreatePartition\020\310\001\022\022\n" - "\rDropPartition\020\311\001\022\021\n\014HasPartition\020\312\001\022\026\n\021" - "DescribePartition\020\313\001\022\023\n\016ShowPartitions\020\314" - "\001\022\023\n\016LoadPartitions\020\315\001\022\026\n\021ReleasePartiti" - "ons\020\316\001\022\021\n\014ShowSegments\020\372\001\022\024\n\017DescribeSeg" - "ment\020\373\001\022\021\n\014LoadSegments\020\374\001\022\024\n\017ReleaseSeg" - "ments\020\375\001\022\024\n\017HandoffSegments\020\376\001\022\030\n\023LoadBa" - "lanceSegments\020\377\001\022\025\n\020DescribeSegments\020\200\002\022" - "\034\n\027FederListIndexedSegment\020\201\002\022\"\n\035FederDe" - "scribeSegmentIndexData\020\202\002\022\020\n\013CreateIndex" - "\020\254\002\022\022\n\rDescribeIndex\020\255\002\022\016\n\tDropIndex\020\256\002\022" - "\027\n\022GetIndexStatistics\020\257\002\022\013\n\006Insert\020\220\003\022\013\n" - "\006Delete\020\221\003\022\n\n\005Flush\020\222\003\022\027\n\022ResendSegmentS" - "tats\020\223\003\022\013\n\006Upsert\020\224\003\022\013\n\006Search\020\364\003\022\021\n\014Sea" - "rchResult\020\365\003\022\022\n\rGetIndexState\020\366\003\022\032\n\025GetI" - "ndexBuildProgress\020\367\003\022\034\n\027GetCollectionSta" - "tistics\020\370\003\022\033\n\026GetPartitionStatistics\020\371\003\022" - "\r\n\010Retrieve\020\372\003\022\023\n\016RetrieveResult\020\373\003\022\024\n\017W" - "atchDmChannels\020\374\003\022\025\n\020RemoveDmChannels\020\375\003" - "\022\027\n\022WatchQueryChannels\020\376\003\022\030\n\023RemoveQuery" - "Channels\020\377\003\022\035\n\030SealedSegmentsChangeInfo\020" - "\200\004\022\027\n\022WatchDeltaChannels\020\201\004\022\024\n\017GetShardL" - "eaders\020\202\004\022\020\n\013GetReplicas\020\203\004\022\023\n\016UnsubDmCh" - "annel\020\204\004\022\024\n\017GetDistribution\020\205\004\022\025\n\020SyncDi" - "stribution\020\206\004\022\020\n\013SegmentInfo\020\330\004\022\017\n\nSyste" - "mInfo\020\331\004\022\024\n\017GetRecoveryInfo\020\332\004\022\024\n\017GetSeg" - "mentState\020\333\004\022\r\n\010TimeTick\020\260\t\022\023\n\016QueryNode" - "Stats\020\261\t\022\016\n\tLoadIndex\020\262\t\022\016\n\tRequestID\020\263\t" - "\022\017\n\nRequestTSO\020\264\t\022\024\n\017AllocateSegment\020\265\t\022" - "\026\n\021SegmentStatistics\020\266\t\022\025\n\020SegmentFlushD" - "one\020\267\t\022\017\n\nDataNodeTt\020\270\t\022\014\n\007Connect\020\271\t\022\024\n" - "\017ListClientInfos\020\272\t\022\025\n\020CreateCredential\020" - "\334\013\022\022\n\rGetCredential\020\335\013\022\025\n\020DeleteCredenti" - "al\020\336\013\022\025\n\020UpdateCredential\020\337\013\022\026\n\021ListCred" - "Usernames\020\340\013\022\017\n\nCreateRole\020\300\014\022\r\n\010DropRol" - "e\020\301\014\022\024\n\017OperateUserRole\020\302\014\022\017\n\nSelectRole" - "\020\303\014\022\017\n\nSelectUser\020\304\014\022\023\n\016SelectResource\020\305" - "\014\022\025\n\020OperatePrivilege\020\306\014\022\020\n\013SelectGrant\020" - "\307\014\022\033\n\026RefreshPolicyInfoCache\020\310\014\022\017\n\nListP" - "olicy\020\311\014\022\030\n\023CreateResourceGroup\020\244\r\022\026\n\021Dr" - "opResourceGroup\020\245\r\022\027\n\022ListResourceGroups" - "\020\246\r\022\032\n\025DescribeResourceGroup\020\247\r\022\021\n\014Trans" - "ferNode\020\250\r\022\024\n\017TransferReplica\020\251\r\022\023\n\016Crea" - "teDatabase\020\211\016\022\021\n\014DropDatabase\020\212\016\022\022\n\rList" - "Databases\020\213\016*\"\n\007DslType\022\007\n\003Dsl\020\000\022\016\n\nBool" - "ExprV1\020\001*B\n\017CompactionState\022\021\n\rUndefiedS" - "tate\020\000\022\r\n\tExecuting\020\001\022\r\n\tCompleted\020\002*X\n\020" - "ConsistencyLevel\022\n\n\006Strong\020\000\022\013\n\007Session\020" - "\001\022\013\n\007Bounded\020\002\022\016\n\nEventually\020\003\022\016\n\nCustom" - "ized\020\004*\236\001\n\013ImportState\022\021\n\rImportPending\020" - "\000\022\020\n\014ImportFailed\020\001\022\021\n\rImportStarted\020\002\022\023" - "\n\017ImportPersisted\020\005\022\021\n\rImportFlushed\020\010\022\023" - "\n\017ImportCompleted\020\006\022\032\n\026ImportFailedAndCl" - "eaned\020\007*2\n\nObjectType\022\016\n\nCollection\020\000\022\n\n" - "\006Global\020\001\022\010\n\004User\020\002*\241\010\n\017ObjectPrivilege\022" - "\020\n\014PrivilegeAll\020\000\022\035\n\031PrivilegeCreateColl" - "ection\020\001\022\033\n\027PrivilegeDropCollection\020\002\022\037\n" - "\033PrivilegeDescribeCollection\020\003\022\034\n\030Privil" - "egeShowCollections\020\004\022\021\n\rPrivilegeLoad\020\005\022" - "\024\n\020PrivilegeRelease\020\006\022\027\n\023PrivilegeCompac" - "tion\020\007\022\023\n\017PrivilegeInsert\020\010\022\023\n\017Privilege" - "Delete\020\t\022\032\n\026PrivilegeGetStatistics\020\n\022\030\n\024" - "PrivilegeCreateIndex\020\013\022\030\n\024PrivilegeIndex" - "Detail\020\014\022\026\n\022PrivilegeDropIndex\020\r\022\023\n\017Priv" - "ilegeSearch\020\016\022\022\n\016PrivilegeFlush\020\017\022\022\n\016Pri" - "vilegeQuery\020\020\022\030\n\024PrivilegeLoadBalance\020\021\022" - "\023\n\017PrivilegeImport\020\022\022\034\n\030PrivilegeCreateO" - "wnership\020\023\022\027\n\023PrivilegeUpdateUser\020\024\022\032\n\026P" - "rivilegeDropOwnership\020\025\022\034\n\030PrivilegeSele" - "ctOwnership\020\026\022\034\n\030PrivilegeManageOwnershi" - "p\020\027\022\027\n\023PrivilegeSelectUser\020\030\022\023\n\017Privileg" - "eUpsert\020\031\022 \n\034PrivilegeCreateResourceGrou" - "p\020\032\022\036\n\032PrivilegeDropResourceGroup\020\033\022\"\n\036P" - "rivilegeDescribeResourceGroup\020\034\022\037\n\033Privi" - "legeListResourceGroups\020\035\022\031\n\025PrivilegeTra" - "nsferNode\020\036\022\034\n\030PrivilegeTransferReplica\020" - "\037\022\037\n\033PrivilegeGetLoadingProgress\020 \022\031\n\025Pr" - "ivilegeGetLoadState\020!\022\035\n\031PrivilegeRename" - "Collection\020\"\022\033\n\027PrivilegeCreateDatabase\020" - "#\022\031\n\025PrivilegeDropDatabase\020$\022\032\n\026Privileg" - "eListDatabases\020%\022\025\n\021PrivilegeFlushAll\020&*" - "S\n\tStateCode\022\020\n\014Initializing\020\000\022\013\n\007Health" - "y\020\001\022\014\n\010Abnormal\020\002\022\013\n\007StandBy\020\003\022\014\n\010Stoppi" - "ng\020\004*c\n\tLoadState\022\025\n\021LoadStateNotExist\020\000" - "\022\024\n\020LoadStateNotLoad\020\001\022\024\n\020LoadStateLoadi" - "ng\020\002\022\023\n\017LoadStateLoaded\020\003:^\n\021privilege_e" - "xt_obj\022\037.google.protobuf.MessageOptions\030" - "\351\007 \001(\0132!.milvus.proto.common.PrivilegeEx" - "tBi\n\016io.milvus.grpcB\013CommonProtoP\001Z4gith" - "ub.com/milvus-io/milvus-proto/go-api/v2/" - "commonpb\240\001\001\252\002\016IO.Milvus.Grpcb\006proto3" + "atVector\020e\022\t\n\005Int64\020\005\022\013\n\007VarChar\020\025*\264\020\n\007M" + "sgType\022\r\n\tUndefined\020\000\022\024\n\020CreateCollectio" + "n\020d\022\022\n\016DropCollection\020e\022\021\n\rHasCollection" + "\020f\022\026\n\022DescribeCollection\020g\022\023\n\017ShowCollec" + "tions\020h\022\024\n\020GetSystemConfigs\020i\022\022\n\016LoadCol" + "lection\020j\022\025\n\021ReleaseCollection\020k\022\017\n\013Crea" + "teAlias\020l\022\r\n\tDropAlias\020m\022\016\n\nAlterAlias\020n" + "\022\023\n\017AlterCollection\020o\022\024\n\020RenameCollectio" + "n\020p\022\021\n\rDescribeAlias\020q\022\017\n\013ListAliases\020r\022" + "\024\n\017CreatePartition\020\310\001\022\022\n\rDropPartition\020\311" + "\001\022\021\n\014HasPartition\020\312\001\022\026\n\021DescribePartitio" + "n\020\313\001\022\023\n\016ShowPartitions\020\314\001\022\023\n\016LoadPartiti" + "ons\020\315\001\022\026\n\021ReleasePartitions\020\316\001\022\021\n\014ShowSe" + "gments\020\372\001\022\024\n\017DescribeSegment\020\373\001\022\021\n\014LoadS" + "egments\020\374\001\022\024\n\017ReleaseSegments\020\375\001\022\024\n\017Hand" + "offSegments\020\376\001\022\030\n\023LoadBalanceSegments\020\377\001" + "\022\025\n\020DescribeSegments\020\200\002\022\034\n\027FederListInde" + "xedSegment\020\201\002\022\"\n\035FederDescribeSegmentInd" + "exData\020\202\002\022\020\n\013CreateIndex\020\254\002\022\022\n\rDescribeI" + "ndex\020\255\002\022\016\n\tDropIndex\020\256\002\022\027\n\022GetIndexStati" + "stics\020\257\002\022\013\n\006Insert\020\220\003\022\013\n\006Delete\020\221\003\022\n\n\005Fl" + "ush\020\222\003\022\027\n\022ResendSegmentStats\020\223\003\022\013\n\006Upser" + "t\020\224\003\022\013\n\006Search\020\364\003\022\021\n\014SearchResult\020\365\003\022\022\n\r" + "GetIndexState\020\366\003\022\032\n\025GetIndexBuildProgres" + "s\020\367\003\022\034\n\027GetCollectionStatistics\020\370\003\022\033\n\026Ge" + "tPartitionStatistics\020\371\003\022\r\n\010Retrieve\020\372\003\022\023" + "\n\016RetrieveResult\020\373\003\022\024\n\017WatchDmChannels\020\374" + "\003\022\025\n\020RemoveDmChannels\020\375\003\022\027\n\022WatchQueryCh" + "annels\020\376\003\022\030\n\023RemoveQueryChannels\020\377\003\022\035\n\030S" + "ealedSegmentsChangeInfo\020\200\004\022\027\n\022WatchDelta" + "Channels\020\201\004\022\024\n\017GetShardLeaders\020\202\004\022\020\n\013Get" + "Replicas\020\203\004\022\023\n\016UnsubDmChannel\020\204\004\022\024\n\017GetD" + "istribution\020\205\004\022\025\n\020SyncDistribution\020\206\004\022\020\n" + "\013SegmentInfo\020\330\004\022\017\n\nSystemInfo\020\331\004\022\024\n\017GetR" + "ecoveryInfo\020\332\004\022\024\n\017GetSegmentState\020\333\004\022\r\n\010" + "TimeTick\020\260\t\022\023\n\016QueryNodeStats\020\261\t\022\016\n\tLoad" + "Index\020\262\t\022\016\n\tRequestID\020\263\t\022\017\n\nRequestTSO\020\264" + "\t\022\024\n\017AllocateSegment\020\265\t\022\026\n\021SegmentStatis" + "tics\020\266\t\022\025\n\020SegmentFlushDone\020\267\t\022\017\n\nDataNo" + "deTt\020\270\t\022\014\n\007Connect\020\271\t\022\024\n\017ListClientInfos" + "\020\272\t\022\023\n\016AllocTimestamp\020\273\t\022\025\n\020CreateCreden" + "tial\020\334\013\022\022\n\rGetCredential\020\335\013\022\025\n\020DeleteCre" + "dential\020\336\013\022\025\n\020UpdateCredential\020\337\013\022\026\n\021Lis" + "tCredUsernames\020\340\013\022\017\n\nCreateRole\020\300\014\022\r\n\010Dr" + "opRole\020\301\014\022\024\n\017OperateUserRole\020\302\014\022\017\n\nSelec" + "tRole\020\303\014\022\017\n\nSelectUser\020\304\014\022\023\n\016SelectResou" + "rce\020\305\014\022\025\n\020OperatePrivilege\020\306\014\022\020\n\013SelectG" + "rant\020\307\014\022\033\n\026RefreshPolicyInfoCache\020\310\014\022\017\n\n" + "ListPolicy\020\311\014\022\030\n\023CreateResourceGroup\020\244\r\022" + "\026\n\021DropResourceGroup\020\245\r\022\027\n\022ListResourceG" + "roups\020\246\r\022\032\n\025DescribeResourceGroup\020\247\r\022\021\n\014" + "TransferNode\020\250\r\022\024\n\017TransferReplica\020\251\r\022\023\n" + "\016CreateDatabase\020\211\016\022\021\n\014DropDatabase\020\212\016\022\022\n" + "\rListDatabases\020\213\016*\"\n\007DslType\022\007\n\003Dsl\020\000\022\016\n" + "\nBoolExprV1\020\001*B\n\017CompactionState\022\021\n\rUnde" + "fiedState\020\000\022\r\n\tExecuting\020\001\022\r\n\tCompleted\020" + "\002*X\n\020ConsistencyLevel\022\n\n\006Strong\020\000\022\013\n\007Ses" + "sion\020\001\022\013\n\007Bounded\020\002\022\016\n\nEventually\020\003\022\016\n\nC" + "ustomized\020\004*\236\001\n\013ImportState\022\021\n\rImportPen" + "ding\020\000\022\020\n\014ImportFailed\020\001\022\021\n\rImportStarte" + "d\020\002\022\023\n\017ImportPersisted\020\005\022\021\n\rImportFlushe" + "d\020\010\022\023\n\017ImportCompleted\020\006\022\032\n\026ImportFailed" + "AndCleaned\020\007*2\n\nObjectType\022\016\n\nCollection" + "\020\000\022\n\n\006Global\020\001\022\010\n\004User\020\002*\241\010\n\017ObjectPrivi" + "lege\022\020\n\014PrivilegeAll\020\000\022\035\n\031PrivilegeCreat" + "eCollection\020\001\022\033\n\027PrivilegeDropCollection" + "\020\002\022\037\n\033PrivilegeDescribeCollection\020\003\022\034\n\030P" + "rivilegeShowCollections\020\004\022\021\n\rPrivilegeLo" + "ad\020\005\022\024\n\020PrivilegeRelease\020\006\022\027\n\023PrivilegeC" + "ompaction\020\007\022\023\n\017PrivilegeInsert\020\010\022\023\n\017Priv" + "ilegeDelete\020\t\022\032\n\026PrivilegeGetStatistics\020" + "\n\022\030\n\024PrivilegeCreateIndex\020\013\022\030\n\024Privilege" + "IndexDetail\020\014\022\026\n\022PrivilegeDropIndex\020\r\022\023\n" + "\017PrivilegeSearch\020\016\022\022\n\016PrivilegeFlush\020\017\022\022" + "\n\016PrivilegeQuery\020\020\022\030\n\024PrivilegeLoadBalan" + "ce\020\021\022\023\n\017PrivilegeImport\020\022\022\034\n\030PrivilegeCr" + "eateOwnership\020\023\022\027\n\023PrivilegeUpdateUser\020\024" + "\022\032\n\026PrivilegeDropOwnership\020\025\022\034\n\030Privileg" + "eSelectOwnership\020\026\022\034\n\030PrivilegeManageOwn" + "ership\020\027\022\027\n\023PrivilegeSelectUser\020\030\022\023\n\017Pri" + "vilegeUpsert\020\031\022 \n\034PrivilegeCreateResourc" + "eGroup\020\032\022\036\n\032PrivilegeDropResourceGroup\020\033" + "\022\"\n\036PrivilegeDescribeResourceGroup\020\034\022\037\n\033" + "PrivilegeListResourceGroups\020\035\022\031\n\025Privile" + "geTransferNode\020\036\022\034\n\030PrivilegeTransferRep" + "lica\020\037\022\037\n\033PrivilegeGetLoadingProgress\020 \022" + "\031\n\025PrivilegeGetLoadState\020!\022\035\n\031PrivilegeR" + "enameCollection\020\"\022\033\n\027PrivilegeCreateData" + "base\020#\022\031\n\025PrivilegeDropDatabase\020$\022\032\n\026Pri" + "vilegeListDatabases\020%\022\025\n\021PrivilegeFlushA" + "ll\020&*S\n\tStateCode\022\020\n\014Initializing\020\000\022\013\n\007H" + "ealthy\020\001\022\014\n\010Abnormal\020\002\022\013\n\007StandBy\020\003\022\014\n\010S" + "topping\020\004*c\n\tLoadState\022\025\n\021LoadStateNotEx" + "ist\020\000\022\024\n\020LoadStateNotLoad\020\001\022\024\n\020LoadState" + "Loading\020\002\022\023\n\017LoadStateLoaded\020\003:^\n\021privil" + "ege_ext_obj\022\037.google.protobuf.MessageOpt" + "ions\030\351\007 \001(\0132!.milvus.proto.common.Privil" + "egeExtBi\n\016io.milvus.grpcB\013CommonProtoP\001Z" + "4github.com/milvus-io/milvus-proto/go-ap" + "i/v2/commonpb\240\001\001\252\002\016IO.Milvus.Grpcb\006proto" + "3" ; static const ::_pbi::DescriptorTable* const descriptor_table_common_2eproto_deps[1] = { &::descriptor_table_google_2fprotobuf_2fdescriptor_2eproto, }; static ::_pbi::once_flag descriptor_table_common_2eproto_once; const ::_pbi::DescriptorTable descriptor_table_common_2eproto = { - false, false, 7036, descriptor_table_protodef_common_2eproto, + false, false, 7081, descriptor_table_protodef_common_2eproto, "common.proto", &descriptor_table_common_2eproto_once, descriptor_table_common_2eproto_deps, 1, 16, schemas, file_default_instances, TableStruct_common_2eproto::offsets, @@ -759,6 +761,8 @@ const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* PlaceholderType_descriptor() { bool PlaceholderType_IsValid(int value) { switch (value) { case 0: + case 5: + case 21: case 100: case 101: return true; @@ -848,6 +852,7 @@ bool MsgType_IsValid(int value) { case 1208: case 1209: case 1210: + case 1211: case 1500: case 1501: case 1502: diff --git a/internal/core/src/pb/common.pb.h b/internal/core/src/pb/common.pb.h index 23d22d1bff..fb109e35d3 100644 --- a/internal/core/src/pb/common.pb.h +++ b/internal/core/src/pb/common.pb.h @@ -271,6 +271,8 @@ enum PlaceholderType : int { None = 0, BinaryVector = 100, FloatVector = 101, + Int64 = 5, + VarChar = 21, PlaceholderType_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits::min(), PlaceholderType_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits::max() }; @@ -369,6 +371,7 @@ enum MsgType : int { DataNodeTt = 1208, Connect = 1209, ListClientInfos = 1210, + AllocTimestamp = 1211, CreateCredential = 1500, GetCredential = 1501, DeleteCredential = 1502, diff --git a/internal/datacoord/index_service.go b/internal/datacoord/index_service.go index 434a0d148e..f26f418d21 100644 --- a/internal/datacoord/index_service.go +++ b/internal/datacoord/index_service.go @@ -272,7 +272,7 @@ func (s *Server) GetIndexState(ctx context.Context, req *indexpb.GetIndexStateRe } s.completeIndexInfo(indexInfo, indexes[0], s.meta.SelectSegments(func(info *SegmentInfo) bool { return isFlush(info) && info.CollectionID == req.GetCollectionID() - }), false) + }), false, indexes[0].CreateTime) ret.State = indexInfo.State ret.FailReason = indexInfo.IndexStateFailReason @@ -379,7 +379,7 @@ func (s *Server) countIndexedRows(indexInfo *indexpb.IndexInfo, segments []*Segm // completeIndexInfo get the index row count and index task state // if realTime, calculate current statistics // if not realTime, which means get info of the prior `CreateIndex` action, skip segments created after index's create time -func (s *Server) completeIndexInfo(indexInfo *indexpb.IndexInfo, index *model.Index, segments []*SegmentInfo, realTime bool) { +func (s *Server) completeIndexInfo(indexInfo *indexpb.IndexInfo, index *model.Index, segments []*SegmentInfo, realTime bool, ts Timestamp) { var ( cntNone = 0 cntUnissued = 0 @@ -397,7 +397,7 @@ func (s *Server) completeIndexInfo(indexInfo *indexpb.IndexInfo, index *model.In segIdx, ok := seg.segmentIndexes[index.IndexID] if !ok { - if seg.GetStartPosition().GetTimestamp() <= index.CreateTime { + if seg.GetStartPosition().GetTimestamp() <= ts { cntUnissued++ } pendingIndexRows += seg.GetNumOfRows() @@ -409,7 +409,7 @@ func (s *Server) completeIndexInfo(indexInfo *indexpb.IndexInfo, index *model.In // if realTime, calculate current statistics // if not realTime, skip segments created after index create - if !realTime && seg.GetStartPosition().GetTimestamp() > index.CreateTime { + if !realTime && seg.GetStartPosition().GetTimestamp() > ts { continue } @@ -506,8 +506,9 @@ func (s *Server) GetIndexBuildProgress(ctx context.Context, req *indexpb.GetInde } s.completeIndexInfo(indexInfo, indexes[0], s.meta.SelectSegments(func(info *SegmentInfo) bool { return isFlush(info) && info.CollectionID == req.GetCollectionID() - }), false) - log.Info("GetIndexBuildProgress success", zap.String("indexName", req.GetIndexName())) + }), false, indexes[0].CreateTime) + log.Info("GetIndexBuildProgress success", zap.Int64("collectionID", req.GetCollectionID()), + zap.String("indexName", req.GetIndexName())) return &indexpb.GetIndexBuildProgressResponse{ Status: &commonpb.Status{ ErrorCode: commonpb.ErrorCode_Success, @@ -523,7 +524,8 @@ func (s *Server) DescribeIndex(ctx context.Context, req *indexpb.DescribeIndexRe log := log.Ctx(ctx).With( zap.Int64("collectionID", req.CollectionID), ) - log.Info("receive DescribeIndex request", zap.String("indexName", req.GetIndexName())) + log.Info("receive DescribeIndex request", zap.String("indexName", req.GetIndexName()), + zap.Uint64("timestamp", req.GetTimestamp())) errResp := &commonpb.Status{ ErrorCode: commonpb.ErrorCode_UnexpectedError, Reason: "", @@ -569,7 +571,11 @@ func (s *Server) DescribeIndex(ctx context.Context, req *indexpb.DescribeIndexRe IsAutoIndex: index.IsAutoIndex, UserIndexParams: index.UserIndexParams, } - s.completeIndexInfo(indexInfo, index, segments, false) + createTs := index.CreateTime + if req.GetTimestamp() != 0 { + createTs = req.GetTimestamp() + } + s.completeIndexInfo(indexInfo, index, segments, false, createTs) indexInfos = append(indexInfos, indexInfo) } log.Info("DescribeIndex success", zap.String("indexName", req.GetIndexName())) @@ -628,7 +634,7 @@ func (s *Server) GetIndexStatistics(ctx context.Context, req *indexpb.GetIndexSt IsAutoIndex: index.IsAutoIndex, UserIndexParams: index.UserIndexParams, } - s.completeIndexInfo(indexInfo, index, segments, true) + s.completeIndexInfo(indexInfo, index, segments, true, index.CreateTime) indexInfos = append(indexInfos, indexInfo) } log.Debug("GetIndexStatisticsResponse success", diff --git a/internal/datacoord/index_service_test.go b/internal/datacoord/index_service_test.go index dec715b07a..5dfc182996 100644 --- a/internal/datacoord/index_service_test.go +++ b/internal/datacoord/index_service_test.go @@ -691,6 +691,7 @@ func TestServer_DescribeIndex(t *testing.T) { req = &indexpb.DescribeIndexRequest{ CollectionID: collID, IndexName: "", + Timestamp: createTS, } ) diff --git a/internal/distributed/proxy/service.go b/internal/distributed/proxy/service.go index 96b49f3d8a..d224eb6c7e 100644 --- a/internal/distributed/proxy/service.go +++ b/internal/distributed/proxy/service.go @@ -958,3 +958,7 @@ func (s *Server) DropDatabase(ctx context.Context, request *milvuspb.DropDatabas func (s *Server) ListDatabases(ctx context.Context, request *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error) { return s.proxy.ListDatabases(ctx, request) } + +func (s *Server) AllocTimestamp(ctx context.Context, req *milvuspb.AllocTimestampRequest) (*milvuspb.AllocTimestampResponse, error) { + return s.proxy.AllocTimestamp(ctx, req) +} diff --git a/internal/distributed/proxy/service_test.go b/internal/distributed/proxy/service_test.go index 4f1ae5cfd3..bc21d8eef1 100644 --- a/internal/distributed/proxy/service_test.go +++ b/internal/distributed/proxy/service_test.go @@ -889,6 +889,10 @@ func (m *MockProxy) Connect(ctx context.Context, req *milvuspb.ConnectRequest) ( return nil, nil } +func (m *MockProxy) AllocTimestamp(ctx context.Context, req *milvuspb.AllocTimestampRequest) (*milvuspb.AllocTimestampResponse, error) { + return nil, nil +} + /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// type WaitOption struct { @@ -1380,6 +1384,11 @@ func Test_NewServer(t *testing.T) { _, err := server.ListDatabases(ctx, nil) assert.Nil(t, err) }) + + t.Run("AllocTimestamp", func(t *testing.T) { + _, err := server.AllocTimestamp(ctx, nil) + assert.Nil(t, err) + }) err = server.Stop() assert.NoError(t, err) diff --git a/internal/mocks/mock_proxy.go b/internal/mocks/mock_proxy.go index 25d90bbb1b..81436d88c4 100644 --- a/internal/mocks/mock_proxy.go +++ b/internal/mocks/mock_proxy.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.30.1. DO NOT EDIT. +// Code generated by mockery v2.16.0. DO NOT EDIT. package mocks @@ -32,15 +32,58 @@ func (_m *MockProxy) EXPECT() *MockProxy_Expecter { return &MockProxy_Expecter{mock: &_m.Mock} } +// AllocTimestamp provides a mock function with given fields: ctx, req +func (_m *MockProxy) AllocTimestamp(ctx context.Context, req *milvuspb.AllocTimestampRequest) (*milvuspb.AllocTimestampResponse, error) { + ret := _m.Called(ctx, req) + + var r0 *milvuspb.AllocTimestampResponse + if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AllocTimestampRequest) *milvuspb.AllocTimestampResponse); ok { + r0 = rf(ctx, req) + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(*milvuspb.AllocTimestampResponse) + } + } + + var r1 error + if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.AllocTimestampRequest) error); ok { + r1 = rf(ctx, req) + } else { + r1 = ret.Error(1) + } + + return r0, r1 +} + +// MockProxy_AllocTimestamp_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AllocTimestamp' +type MockProxy_AllocTimestamp_Call struct { + *mock.Call +} + +// AllocTimestamp is a helper method to define mock.On call +// - ctx context.Context +// - req *milvuspb.AllocTimestampRequest +func (_e *MockProxy_Expecter) AllocTimestamp(ctx interface{}, req interface{}) *MockProxy_AllocTimestamp_Call { + return &MockProxy_AllocTimestamp_Call{Call: _e.mock.On("AllocTimestamp", ctx, req)} +} + +func (_c *MockProxy_AllocTimestamp_Call) Run(run func(ctx context.Context, req *milvuspb.AllocTimestampRequest)) *MockProxy_AllocTimestamp_Call { + _c.Call.Run(func(args mock.Arguments) { + run(args[0].(context.Context), args[1].(*milvuspb.AllocTimestampRequest)) + }) + return _c +} + +func (_c *MockProxy_AllocTimestamp_Call) Return(_a0 *milvuspb.AllocTimestampResponse, _a1 error) *MockProxy_AllocTimestamp_Call { + _c.Call.Return(_a0, _a1) + return _c +} + // AlterAlias provides a mock function with given fields: ctx, request func (_m *MockProxy) AlterAlias(ctx context.Context, request *milvuspb.AlterAliasRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterAliasRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterAliasRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -49,6 +92,7 @@ func (_m *MockProxy) AlterAlias(ctx context.Context, request *milvuspb.AlterAlia } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.AlterAliasRequest) error); ok { r1 = rf(ctx, request) } else { @@ -64,8 +108,8 @@ type MockProxy_AlterAlias_Call struct { } // AlterAlias is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.AlterAliasRequest +// - ctx context.Context +// - request *milvuspb.AlterAliasRequest func (_e *MockProxy_Expecter) AlterAlias(ctx interface{}, request interface{}) *MockProxy_AlterAlias_Call { return &MockProxy_AlterAlias_Call{Call: _e.mock.On("AlterAlias", ctx, request)} } @@ -82,20 +126,11 @@ func (_c *MockProxy_AlterAlias_Call) Return(_a0 *commonpb.Status, _a1 error) *Mo return _c } -func (_c *MockProxy_AlterAlias_Call) RunAndReturn(run func(context.Context, *milvuspb.AlterAliasRequest) (*commonpb.Status, error)) *MockProxy_AlterAlias_Call { - _c.Call.Return(run) - return _c -} - // AlterCollection provides a mock function with given fields: ctx, request func (_m *MockProxy) AlterCollection(ctx context.Context, request *milvuspb.AlterCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterCollectionRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterCollectionRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -104,6 +139,7 @@ func (_m *MockProxy) AlterCollection(ctx context.Context, request *milvuspb.Alte } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.AlterCollectionRequest) error); ok { r1 = rf(ctx, request) } else { @@ -119,8 +155,8 @@ type MockProxy_AlterCollection_Call struct { } // AlterCollection is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.AlterCollectionRequest +// - ctx context.Context +// - request *milvuspb.AlterCollectionRequest func (_e *MockProxy_Expecter) AlterCollection(ctx interface{}, request interface{}) *MockProxy_AlterCollection_Call { return &MockProxy_AlterCollection_Call{Call: _e.mock.On("AlterCollection", ctx, request)} } @@ -137,20 +173,11 @@ func (_c *MockProxy_AlterCollection_Call) Return(_a0 *commonpb.Status, _a1 error return _c } -func (_c *MockProxy_AlterCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.AlterCollectionRequest) (*commonpb.Status, error)) *MockProxy_AlterCollection_Call { - _c.Call.Return(run) - return _c -} - // CalcDistance provides a mock function with given fields: ctx, request func (_m *MockProxy) CalcDistance(ctx context.Context, request *milvuspb.CalcDistanceRequest) (*milvuspb.CalcDistanceResults, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.CalcDistanceResults - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CalcDistanceRequest) (*milvuspb.CalcDistanceResults, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CalcDistanceRequest) *milvuspb.CalcDistanceResults); ok { r0 = rf(ctx, request) } else { @@ -159,6 +186,7 @@ func (_m *MockProxy) CalcDistance(ctx context.Context, request *milvuspb.CalcDis } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CalcDistanceRequest) error); ok { r1 = rf(ctx, request) } else { @@ -174,8 +202,8 @@ type MockProxy_CalcDistance_Call struct { } // CalcDistance is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.CalcDistanceRequest +// - ctx context.Context +// - request *milvuspb.CalcDistanceRequest func (_e *MockProxy_Expecter) CalcDistance(ctx interface{}, request interface{}) *MockProxy_CalcDistance_Call { return &MockProxy_CalcDistance_Call{Call: _e.mock.On("CalcDistance", ctx, request)} } @@ -192,20 +220,11 @@ func (_c *MockProxy_CalcDistance_Call) Return(_a0 *milvuspb.CalcDistanceResults, return _c } -func (_c *MockProxy_CalcDistance_Call) RunAndReturn(run func(context.Context, *milvuspb.CalcDistanceRequest) (*milvuspb.CalcDistanceResults, error)) *MockProxy_CalcDistance_Call { - _c.Call.Return(run) - return _c -} - // CheckHealth provides a mock function with given fields: ctx, req func (_m *MockProxy) CheckHealth(ctx context.Context, req *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.CheckHealthResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) *milvuspb.CheckHealthResponse); ok { r0 = rf(ctx, req) } else { @@ -214,6 +233,7 @@ func (_m *MockProxy) CheckHealth(ctx context.Context, req *milvuspb.CheckHealthR } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CheckHealthRequest) error); ok { r1 = rf(ctx, req) } else { @@ -229,8 +249,8 @@ type MockProxy_CheckHealth_Call struct { } // CheckHealth is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.CheckHealthRequest +// - ctx context.Context +// - req *milvuspb.CheckHealthRequest func (_e *MockProxy_Expecter) CheckHealth(ctx interface{}, req interface{}) *MockProxy_CheckHealth_Call { return &MockProxy_CheckHealth_Call{Call: _e.mock.On("CheckHealth", ctx, req)} } @@ -247,20 +267,11 @@ func (_c *MockProxy_CheckHealth_Call) Return(_a0 *milvuspb.CheckHealthResponse, return _c } -func (_c *MockProxy_CheckHealth_Call) RunAndReturn(run func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)) *MockProxy_CheckHealth_Call { - _c.Call.Return(run) - return _c -} - // Connect provides a mock function with given fields: ctx, req func (_m *MockProxy) Connect(ctx context.Context, req *milvuspb.ConnectRequest) (*milvuspb.ConnectResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.ConnectResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ConnectRequest) (*milvuspb.ConnectResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ConnectRequest) *milvuspb.ConnectResponse); ok { r0 = rf(ctx, req) } else { @@ -269,6 +280,7 @@ func (_m *MockProxy) Connect(ctx context.Context, req *milvuspb.ConnectRequest) } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ConnectRequest) error); ok { r1 = rf(ctx, req) } else { @@ -284,8 +296,8 @@ type MockProxy_Connect_Call struct { } // Connect is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.ConnectRequest +// - ctx context.Context +// - req *milvuspb.ConnectRequest func (_e *MockProxy_Expecter) Connect(ctx interface{}, req interface{}) *MockProxy_Connect_Call { return &MockProxy_Connect_Call{Call: _e.mock.On("Connect", ctx, req)} } @@ -302,20 +314,11 @@ func (_c *MockProxy_Connect_Call) Return(_a0 *milvuspb.ConnectResponse, _a1 erro return _c } -func (_c *MockProxy_Connect_Call) RunAndReturn(run func(context.Context, *milvuspb.ConnectRequest) (*milvuspb.ConnectResponse, error)) *MockProxy_Connect_Call { - _c.Call.Return(run) - return _c -} - // CreateAlias provides a mock function with given fields: ctx, request func (_m *MockProxy) CreateAlias(ctx context.Context, request *milvuspb.CreateAliasRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateAliasRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateAliasRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -324,6 +327,7 @@ func (_m *MockProxy) CreateAlias(ctx context.Context, request *milvuspb.CreateAl } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateAliasRequest) error); ok { r1 = rf(ctx, request) } else { @@ -339,8 +343,8 @@ type MockProxy_CreateAlias_Call struct { } // CreateAlias is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.CreateAliasRequest +// - ctx context.Context +// - request *milvuspb.CreateAliasRequest func (_e *MockProxy_Expecter) CreateAlias(ctx interface{}, request interface{}) *MockProxy_CreateAlias_Call { return &MockProxy_CreateAlias_Call{Call: _e.mock.On("CreateAlias", ctx, request)} } @@ -357,20 +361,11 @@ func (_c *MockProxy_CreateAlias_Call) Return(_a0 *commonpb.Status, _a1 error) *M return _c } -func (_c *MockProxy_CreateAlias_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateAliasRequest) (*commonpb.Status, error)) *MockProxy_CreateAlias_Call { - _c.Call.Return(run) - return _c -} - // CreateCollection provides a mock function with given fields: ctx, request func (_m *MockProxy) CreateCollection(ctx context.Context, request *milvuspb.CreateCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCollectionRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCollectionRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -379,6 +374,7 @@ func (_m *MockProxy) CreateCollection(ctx context.Context, request *milvuspb.Cre } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateCollectionRequest) error); ok { r1 = rf(ctx, request) } else { @@ -394,8 +390,8 @@ type MockProxy_CreateCollection_Call struct { } // CreateCollection is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.CreateCollectionRequest +// - ctx context.Context +// - request *milvuspb.CreateCollectionRequest func (_e *MockProxy_Expecter) CreateCollection(ctx interface{}, request interface{}) *MockProxy_CreateCollection_Call { return &MockProxy_CreateCollection_Call{Call: _e.mock.On("CreateCollection", ctx, request)} } @@ -412,20 +408,11 @@ func (_c *MockProxy_CreateCollection_Call) Return(_a0 *commonpb.Status, _a1 erro return _c } -func (_c *MockProxy_CreateCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateCollectionRequest) (*commonpb.Status, error)) *MockProxy_CreateCollection_Call { - _c.Call.Return(run) - return _c -} - // CreateCredential provides a mock function with given fields: ctx, req func (_m *MockProxy) CreateCredential(ctx context.Context, req *milvuspb.CreateCredentialRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCredentialRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCredentialRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -434,6 +421,7 @@ func (_m *MockProxy) CreateCredential(ctx context.Context, req *milvuspb.CreateC } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateCredentialRequest) error); ok { r1 = rf(ctx, req) } else { @@ -449,8 +437,8 @@ type MockProxy_CreateCredential_Call struct { } // CreateCredential is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.CreateCredentialRequest +// - ctx context.Context +// - req *milvuspb.CreateCredentialRequest func (_e *MockProxy_Expecter) CreateCredential(ctx interface{}, req interface{}) *MockProxy_CreateCredential_Call { return &MockProxy_CreateCredential_Call{Call: _e.mock.On("CreateCredential", ctx, req)} } @@ -467,20 +455,11 @@ func (_c *MockProxy_CreateCredential_Call) Return(_a0 *commonpb.Status, _a1 erro return _c } -func (_c *MockProxy_CreateCredential_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateCredentialRequest) (*commonpb.Status, error)) *MockProxy_CreateCredential_Call { - _c.Call.Return(run) - return _c -} - // CreateDatabase provides a mock function with given fields: ctx, req func (_m *MockProxy) CreateDatabase(ctx context.Context, req *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateDatabaseRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -489,6 +468,7 @@ func (_m *MockProxy) CreateDatabase(ctx context.Context, req *milvuspb.CreateDat } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateDatabaseRequest) error); ok { r1 = rf(ctx, req) } else { @@ -504,8 +484,8 @@ type MockProxy_CreateDatabase_Call struct { } // CreateDatabase is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.CreateDatabaseRequest +// - ctx context.Context +// - req *milvuspb.CreateDatabaseRequest func (_e *MockProxy_Expecter) CreateDatabase(ctx interface{}, req interface{}) *MockProxy_CreateDatabase_Call { return &MockProxy_CreateDatabase_Call{Call: _e.mock.On("CreateDatabase", ctx, req)} } @@ -522,20 +502,11 @@ func (_c *MockProxy_CreateDatabase_Call) Return(_a0 *commonpb.Status, _a1 error) return _c } -func (_c *MockProxy_CreateDatabase_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error)) *MockProxy_CreateDatabase_Call { - _c.Call.Return(run) - return _c -} - // CreateIndex provides a mock function with given fields: ctx, request func (_m *MockProxy) CreateIndex(ctx context.Context, request *milvuspb.CreateIndexRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateIndexRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateIndexRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -544,6 +515,7 @@ func (_m *MockProxy) CreateIndex(ctx context.Context, request *milvuspb.CreateIn } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateIndexRequest) error); ok { r1 = rf(ctx, request) } else { @@ -559,8 +531,8 @@ type MockProxy_CreateIndex_Call struct { } // CreateIndex is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.CreateIndexRequest +// - ctx context.Context +// - request *milvuspb.CreateIndexRequest func (_e *MockProxy_Expecter) CreateIndex(ctx interface{}, request interface{}) *MockProxy_CreateIndex_Call { return &MockProxy_CreateIndex_Call{Call: _e.mock.On("CreateIndex", ctx, request)} } @@ -577,20 +549,11 @@ func (_c *MockProxy_CreateIndex_Call) Return(_a0 *commonpb.Status, _a1 error) *M return _c } -func (_c *MockProxy_CreateIndex_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateIndexRequest) (*commonpb.Status, error)) *MockProxy_CreateIndex_Call { - _c.Call.Return(run) - return _c -} - // CreatePartition provides a mock function with given fields: ctx, request func (_m *MockProxy) CreatePartition(ctx context.Context, request *milvuspb.CreatePartitionRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreatePartitionRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreatePartitionRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -599,6 +562,7 @@ func (_m *MockProxy) CreatePartition(ctx context.Context, request *milvuspb.Crea } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreatePartitionRequest) error); ok { r1 = rf(ctx, request) } else { @@ -614,8 +578,8 @@ type MockProxy_CreatePartition_Call struct { } // CreatePartition is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.CreatePartitionRequest +// - ctx context.Context +// - request *milvuspb.CreatePartitionRequest func (_e *MockProxy_Expecter) CreatePartition(ctx interface{}, request interface{}) *MockProxy_CreatePartition_Call { return &MockProxy_CreatePartition_Call{Call: _e.mock.On("CreatePartition", ctx, request)} } @@ -632,20 +596,11 @@ func (_c *MockProxy_CreatePartition_Call) Return(_a0 *commonpb.Status, _a1 error return _c } -func (_c *MockProxy_CreatePartition_Call) RunAndReturn(run func(context.Context, *milvuspb.CreatePartitionRequest) (*commonpb.Status, error)) *MockProxy_CreatePartition_Call { - _c.Call.Return(run) - return _c -} - // CreateResourceGroup provides a mock function with given fields: ctx, req func (_m *MockProxy) CreateResourceGroup(ctx context.Context, req *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateResourceGroupRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -654,6 +609,7 @@ func (_m *MockProxy) CreateResourceGroup(ctx context.Context, req *milvuspb.Crea } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateResourceGroupRequest) error); ok { r1 = rf(ctx, req) } else { @@ -669,8 +625,8 @@ type MockProxy_CreateResourceGroup_Call struct { } // CreateResourceGroup is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.CreateResourceGroupRequest +// - ctx context.Context +// - req *milvuspb.CreateResourceGroupRequest func (_e *MockProxy_Expecter) CreateResourceGroup(ctx interface{}, req interface{}) *MockProxy_CreateResourceGroup_Call { return &MockProxy_CreateResourceGroup_Call{Call: _e.mock.On("CreateResourceGroup", ctx, req)} } @@ -687,20 +643,11 @@ func (_c *MockProxy_CreateResourceGroup_Call) Return(_a0 *commonpb.Status, _a1 e return _c } -func (_c *MockProxy_CreateResourceGroup_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error)) *MockProxy_CreateResourceGroup_Call { - _c.Call.Return(run) - return _c -} - // CreateRole provides a mock function with given fields: ctx, req func (_m *MockProxy) CreateRole(ctx context.Context, req *milvuspb.CreateRoleRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRoleRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRoleRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -709,6 +656,7 @@ func (_m *MockProxy) CreateRole(ctx context.Context, req *milvuspb.CreateRoleReq } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateRoleRequest) error); ok { r1 = rf(ctx, req) } else { @@ -724,8 +672,8 @@ type MockProxy_CreateRole_Call struct { } // CreateRole is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.CreateRoleRequest +// - ctx context.Context +// - req *milvuspb.CreateRoleRequest func (_e *MockProxy_Expecter) CreateRole(ctx interface{}, req interface{}) *MockProxy_CreateRole_Call { return &MockProxy_CreateRole_Call{Call: _e.mock.On("CreateRole", ctx, req)} } @@ -742,20 +690,11 @@ func (_c *MockProxy_CreateRole_Call) Return(_a0 *commonpb.Status, _a1 error) *Mo return _c } -func (_c *MockProxy_CreateRole_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateRoleRequest) (*commonpb.Status, error)) *MockProxy_CreateRole_Call { - _c.Call.Return(run) - return _c -} - // Delete provides a mock function with given fields: ctx, request func (_m *MockProxy) Delete(ctx context.Context, request *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.MutationResult - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteRequest) *milvuspb.MutationResult); ok { r0 = rf(ctx, request) } else { @@ -764,6 +703,7 @@ func (_m *MockProxy) Delete(ctx context.Context, request *milvuspb.DeleteRequest } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DeleteRequest) error); ok { r1 = rf(ctx, request) } else { @@ -779,8 +719,8 @@ type MockProxy_Delete_Call struct { } // Delete is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.DeleteRequest +// - ctx context.Context +// - request *milvuspb.DeleteRequest func (_e *MockProxy_Expecter) Delete(ctx interface{}, request interface{}) *MockProxy_Delete_Call { return &MockProxy_Delete_Call{Call: _e.mock.On("Delete", ctx, request)} } @@ -797,20 +737,11 @@ func (_c *MockProxy_Delete_Call) Return(_a0 *milvuspb.MutationResult, _a1 error) return _c } -func (_c *MockProxy_Delete_Call) RunAndReturn(run func(context.Context, *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error)) *MockProxy_Delete_Call { - _c.Call.Return(run) - return _c -} - // DeleteCredential provides a mock function with given fields: ctx, req func (_m *MockProxy) DeleteCredential(ctx context.Context, req *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteCredentialRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -819,6 +750,7 @@ func (_m *MockProxy) DeleteCredential(ctx context.Context, req *milvuspb.DeleteC } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DeleteCredentialRequest) error); ok { r1 = rf(ctx, req) } else { @@ -834,8 +766,8 @@ type MockProxy_DeleteCredential_Call struct { } // DeleteCredential is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.DeleteCredentialRequest +// - ctx context.Context +// - req *milvuspb.DeleteCredentialRequest func (_e *MockProxy_Expecter) DeleteCredential(ctx interface{}, req interface{}) *MockProxy_DeleteCredential_Call { return &MockProxy_DeleteCredential_Call{Call: _e.mock.On("DeleteCredential", ctx, req)} } @@ -852,20 +784,11 @@ func (_c *MockProxy_DeleteCredential_Call) Return(_a0 *commonpb.Status, _a1 erro return _c } -func (_c *MockProxy_DeleteCredential_Call) RunAndReturn(run func(context.Context, *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error)) *MockProxy_DeleteCredential_Call { - _c.Call.Return(run) - return _c -} - // DescribeCollection provides a mock function with given fields: ctx, request func (_m *MockProxy) DescribeCollection(ctx context.Context, request *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.DescribeCollectionResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest) *milvuspb.DescribeCollectionResponse); ok { r0 = rf(ctx, request) } else { @@ -874,6 +797,7 @@ func (_m *MockProxy) DescribeCollection(ctx context.Context, request *milvuspb.D } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DescribeCollectionRequest) error); ok { r1 = rf(ctx, request) } else { @@ -889,8 +813,8 @@ type MockProxy_DescribeCollection_Call struct { } // DescribeCollection is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.DescribeCollectionRequest +// - ctx context.Context +// - request *milvuspb.DescribeCollectionRequest func (_e *MockProxy_Expecter) DescribeCollection(ctx interface{}, request interface{}) *MockProxy_DescribeCollection_Call { return &MockProxy_DescribeCollection_Call{Call: _e.mock.On("DescribeCollection", ctx, request)} } @@ -907,20 +831,11 @@ func (_c *MockProxy_DescribeCollection_Call) Return(_a0 *milvuspb.DescribeCollec return _c } -func (_c *MockProxy_DescribeCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error)) *MockProxy_DescribeCollection_Call { - _c.Call.Return(run) - return _c -} - // DescribeIndex provides a mock function with given fields: ctx, request func (_m *MockProxy) DescribeIndex(ctx context.Context, request *milvuspb.DescribeIndexRequest) (*milvuspb.DescribeIndexResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.DescribeIndexResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeIndexRequest) (*milvuspb.DescribeIndexResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeIndexRequest) *milvuspb.DescribeIndexResponse); ok { r0 = rf(ctx, request) } else { @@ -929,6 +844,7 @@ func (_m *MockProxy) DescribeIndex(ctx context.Context, request *milvuspb.Descri } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DescribeIndexRequest) error); ok { r1 = rf(ctx, request) } else { @@ -944,8 +860,8 @@ type MockProxy_DescribeIndex_Call struct { } // DescribeIndex is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.DescribeIndexRequest +// - ctx context.Context +// - request *milvuspb.DescribeIndexRequest func (_e *MockProxy_Expecter) DescribeIndex(ctx interface{}, request interface{}) *MockProxy_DescribeIndex_Call { return &MockProxy_DescribeIndex_Call{Call: _e.mock.On("DescribeIndex", ctx, request)} } @@ -962,20 +878,11 @@ func (_c *MockProxy_DescribeIndex_Call) Return(_a0 *milvuspb.DescribeIndexRespon return _c } -func (_c *MockProxy_DescribeIndex_Call) RunAndReturn(run func(context.Context, *milvuspb.DescribeIndexRequest) (*milvuspb.DescribeIndexResponse, error)) *MockProxy_DescribeIndex_Call { - _c.Call.Return(run) - return _c -} - // DescribeResourceGroup provides a mock function with given fields: ctx, req func (_m *MockProxy) DescribeResourceGroup(ctx context.Context, req *milvuspb.DescribeResourceGroupRequest) (*milvuspb.DescribeResourceGroupResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.DescribeResourceGroupResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeResourceGroupRequest) (*milvuspb.DescribeResourceGroupResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeResourceGroupRequest) *milvuspb.DescribeResourceGroupResponse); ok { r0 = rf(ctx, req) } else { @@ -984,6 +891,7 @@ func (_m *MockProxy) DescribeResourceGroup(ctx context.Context, req *milvuspb.De } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DescribeResourceGroupRequest) error); ok { r1 = rf(ctx, req) } else { @@ -999,8 +907,8 @@ type MockProxy_DescribeResourceGroup_Call struct { } // DescribeResourceGroup is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.DescribeResourceGroupRequest +// - ctx context.Context +// - req *milvuspb.DescribeResourceGroupRequest func (_e *MockProxy_Expecter) DescribeResourceGroup(ctx interface{}, req interface{}) *MockProxy_DescribeResourceGroup_Call { return &MockProxy_DescribeResourceGroup_Call{Call: _e.mock.On("DescribeResourceGroup", ctx, req)} } @@ -1017,20 +925,11 @@ func (_c *MockProxy_DescribeResourceGroup_Call) Return(_a0 *milvuspb.DescribeRes return _c } -func (_c *MockProxy_DescribeResourceGroup_Call) RunAndReturn(run func(context.Context, *milvuspb.DescribeResourceGroupRequest) (*milvuspb.DescribeResourceGroupResponse, error)) *MockProxy_DescribeResourceGroup_Call { - _c.Call.Return(run) - return _c -} - // DropAlias provides a mock function with given fields: ctx, request func (_m *MockProxy) DropAlias(ctx context.Context, request *milvuspb.DropAliasRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropAliasRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropAliasRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -1039,6 +938,7 @@ func (_m *MockProxy) DropAlias(ctx context.Context, request *milvuspb.DropAliasR } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropAliasRequest) error); ok { r1 = rf(ctx, request) } else { @@ -1054,8 +954,8 @@ type MockProxy_DropAlias_Call struct { } // DropAlias is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.DropAliasRequest +// - ctx context.Context +// - request *milvuspb.DropAliasRequest func (_e *MockProxy_Expecter) DropAlias(ctx interface{}, request interface{}) *MockProxy_DropAlias_Call { return &MockProxy_DropAlias_Call{Call: _e.mock.On("DropAlias", ctx, request)} } @@ -1072,20 +972,11 @@ func (_c *MockProxy_DropAlias_Call) Return(_a0 *commonpb.Status, _a1 error) *Moc return _c } -func (_c *MockProxy_DropAlias_Call) RunAndReturn(run func(context.Context, *milvuspb.DropAliasRequest) (*commonpb.Status, error)) *MockProxy_DropAlias_Call { - _c.Call.Return(run) - return _c -} - // DropCollection provides a mock function with given fields: ctx, request func (_m *MockProxy) DropCollection(ctx context.Context, request *milvuspb.DropCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropCollectionRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropCollectionRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -1094,6 +985,7 @@ func (_m *MockProxy) DropCollection(ctx context.Context, request *milvuspb.DropC } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropCollectionRequest) error); ok { r1 = rf(ctx, request) } else { @@ -1109,8 +1001,8 @@ type MockProxy_DropCollection_Call struct { } // DropCollection is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.DropCollectionRequest +// - ctx context.Context +// - request *milvuspb.DropCollectionRequest func (_e *MockProxy_Expecter) DropCollection(ctx interface{}, request interface{}) *MockProxy_DropCollection_Call { return &MockProxy_DropCollection_Call{Call: _e.mock.On("DropCollection", ctx, request)} } @@ -1127,20 +1019,11 @@ func (_c *MockProxy_DropCollection_Call) Return(_a0 *commonpb.Status, _a1 error) return _c } -func (_c *MockProxy_DropCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.DropCollectionRequest) (*commonpb.Status, error)) *MockProxy_DropCollection_Call { - _c.Call.Return(run) - return _c -} - // DropDatabase provides a mock function with given fields: ctx, req func (_m *MockProxy) DropDatabase(ctx context.Context, req *milvuspb.DropDatabaseRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropDatabaseRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropDatabaseRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -1149,6 +1032,7 @@ func (_m *MockProxy) DropDatabase(ctx context.Context, req *milvuspb.DropDatabas } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropDatabaseRequest) error); ok { r1 = rf(ctx, req) } else { @@ -1164,8 +1048,8 @@ type MockProxy_DropDatabase_Call struct { } // DropDatabase is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.DropDatabaseRequest +// - ctx context.Context +// - req *milvuspb.DropDatabaseRequest func (_e *MockProxy_Expecter) DropDatabase(ctx interface{}, req interface{}) *MockProxy_DropDatabase_Call { return &MockProxy_DropDatabase_Call{Call: _e.mock.On("DropDatabase", ctx, req)} } @@ -1182,20 +1066,11 @@ func (_c *MockProxy_DropDatabase_Call) Return(_a0 *commonpb.Status, _a1 error) * return _c } -func (_c *MockProxy_DropDatabase_Call) RunAndReturn(run func(context.Context, *milvuspb.DropDatabaseRequest) (*commonpb.Status, error)) *MockProxy_DropDatabase_Call { - _c.Call.Return(run) - return _c -} - // DropIndex provides a mock function with given fields: ctx, request func (_m *MockProxy) DropIndex(ctx context.Context, request *milvuspb.DropIndexRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropIndexRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropIndexRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -1204,6 +1079,7 @@ func (_m *MockProxy) DropIndex(ctx context.Context, request *milvuspb.DropIndexR } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropIndexRequest) error); ok { r1 = rf(ctx, request) } else { @@ -1219,8 +1095,8 @@ type MockProxy_DropIndex_Call struct { } // DropIndex is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.DropIndexRequest +// - ctx context.Context +// - request *milvuspb.DropIndexRequest func (_e *MockProxy_Expecter) DropIndex(ctx interface{}, request interface{}) *MockProxy_DropIndex_Call { return &MockProxy_DropIndex_Call{Call: _e.mock.On("DropIndex", ctx, request)} } @@ -1237,20 +1113,11 @@ func (_c *MockProxy_DropIndex_Call) Return(_a0 *commonpb.Status, _a1 error) *Moc return _c } -func (_c *MockProxy_DropIndex_Call) RunAndReturn(run func(context.Context, *milvuspb.DropIndexRequest) (*commonpb.Status, error)) *MockProxy_DropIndex_Call { - _c.Call.Return(run) - return _c -} - // DropPartition provides a mock function with given fields: ctx, request func (_m *MockProxy) DropPartition(ctx context.Context, request *milvuspb.DropPartitionRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropPartitionRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropPartitionRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -1259,6 +1126,7 @@ func (_m *MockProxy) DropPartition(ctx context.Context, request *milvuspb.DropPa } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropPartitionRequest) error); ok { r1 = rf(ctx, request) } else { @@ -1274,8 +1142,8 @@ type MockProxy_DropPartition_Call struct { } // DropPartition is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.DropPartitionRequest +// - ctx context.Context +// - request *milvuspb.DropPartitionRequest func (_e *MockProxy_Expecter) DropPartition(ctx interface{}, request interface{}) *MockProxy_DropPartition_Call { return &MockProxy_DropPartition_Call{Call: _e.mock.On("DropPartition", ctx, request)} } @@ -1292,20 +1160,11 @@ func (_c *MockProxy_DropPartition_Call) Return(_a0 *commonpb.Status, _a1 error) return _c } -func (_c *MockProxy_DropPartition_Call) RunAndReturn(run func(context.Context, *milvuspb.DropPartitionRequest) (*commonpb.Status, error)) *MockProxy_DropPartition_Call { - _c.Call.Return(run) - return _c -} - // DropResourceGroup provides a mock function with given fields: ctx, req func (_m *MockProxy) DropResourceGroup(ctx context.Context, req *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropResourceGroupRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -1314,6 +1173,7 @@ func (_m *MockProxy) DropResourceGroup(ctx context.Context, req *milvuspb.DropRe } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropResourceGroupRequest) error); ok { r1 = rf(ctx, req) } else { @@ -1329,8 +1189,8 @@ type MockProxy_DropResourceGroup_Call struct { } // DropResourceGroup is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.DropResourceGroupRequest +// - ctx context.Context +// - req *milvuspb.DropResourceGroupRequest func (_e *MockProxy_Expecter) DropResourceGroup(ctx interface{}, req interface{}) *MockProxy_DropResourceGroup_Call { return &MockProxy_DropResourceGroup_Call{Call: _e.mock.On("DropResourceGroup", ctx, req)} } @@ -1347,20 +1207,11 @@ func (_c *MockProxy_DropResourceGroup_Call) Return(_a0 *commonpb.Status, _a1 err return _c } -func (_c *MockProxy_DropResourceGroup_Call) RunAndReturn(run func(context.Context, *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error)) *MockProxy_DropResourceGroup_Call { - _c.Call.Return(run) - return _c -} - // DropRole provides a mock function with given fields: ctx, req func (_m *MockProxy) DropRole(ctx context.Context, req *milvuspb.DropRoleRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRoleRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRoleRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -1369,6 +1220,7 @@ func (_m *MockProxy) DropRole(ctx context.Context, req *milvuspb.DropRoleRequest } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropRoleRequest) error); ok { r1 = rf(ctx, req) } else { @@ -1384,8 +1236,8 @@ type MockProxy_DropRole_Call struct { } // DropRole is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.DropRoleRequest +// - ctx context.Context +// - req *milvuspb.DropRoleRequest func (_e *MockProxy_Expecter) DropRole(ctx interface{}, req interface{}) *MockProxy_DropRole_Call { return &MockProxy_DropRole_Call{Call: _e.mock.On("DropRole", ctx, req)} } @@ -1402,20 +1254,11 @@ func (_c *MockProxy_DropRole_Call) Return(_a0 *commonpb.Status, _a1 error) *Mock return _c } -func (_c *MockProxy_DropRole_Call) RunAndReturn(run func(context.Context, *milvuspb.DropRoleRequest) (*commonpb.Status, error)) *MockProxy_DropRole_Call { - _c.Call.Return(run) - return _c -} - // Dummy provides a mock function with given fields: ctx, request func (_m *MockProxy) Dummy(ctx context.Context, request *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.DummyResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DummyRequest) *milvuspb.DummyResponse); ok { r0 = rf(ctx, request) } else { @@ -1424,6 +1267,7 @@ func (_m *MockProxy) Dummy(ctx context.Context, request *milvuspb.DummyRequest) } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DummyRequest) error); ok { r1 = rf(ctx, request) } else { @@ -1439,8 +1283,8 @@ type MockProxy_Dummy_Call struct { } // Dummy is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.DummyRequest +// - ctx context.Context +// - request *milvuspb.DummyRequest func (_e *MockProxy_Expecter) Dummy(ctx interface{}, request interface{}) *MockProxy_Dummy_Call { return &MockProxy_Dummy_Call{Call: _e.mock.On("Dummy", ctx, request)} } @@ -1457,20 +1301,11 @@ func (_c *MockProxy_Dummy_Call) Return(_a0 *milvuspb.DummyResponse, _a1 error) * return _c } -func (_c *MockProxy_Dummy_Call) RunAndReturn(run func(context.Context, *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error)) *MockProxy_Dummy_Call { - _c.Call.Return(run) - return _c -} - // Flush provides a mock function with given fields: ctx, request func (_m *MockProxy) Flush(ctx context.Context, request *milvuspb.FlushRequest) (*milvuspb.FlushResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.FlushResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushRequest) (*milvuspb.FlushResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushRequest) *milvuspb.FlushResponse); ok { r0 = rf(ctx, request) } else { @@ -1479,6 +1314,7 @@ func (_m *MockProxy) Flush(ctx context.Context, request *milvuspb.FlushRequest) } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.FlushRequest) error); ok { r1 = rf(ctx, request) } else { @@ -1494,8 +1330,8 @@ type MockProxy_Flush_Call struct { } // Flush is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.FlushRequest +// - ctx context.Context +// - request *milvuspb.FlushRequest func (_e *MockProxy_Expecter) Flush(ctx interface{}, request interface{}) *MockProxy_Flush_Call { return &MockProxy_Flush_Call{Call: _e.mock.On("Flush", ctx, request)} } @@ -1512,20 +1348,11 @@ func (_c *MockProxy_Flush_Call) Return(_a0 *milvuspb.FlushResponse, _a1 error) * return _c } -func (_c *MockProxy_Flush_Call) RunAndReturn(run func(context.Context, *milvuspb.FlushRequest) (*milvuspb.FlushResponse, error)) *MockProxy_Flush_Call { - _c.Call.Return(run) - return _c -} - // FlushAll provides a mock function with given fields: ctx, request func (_m *MockProxy) FlushAll(ctx context.Context, request *milvuspb.FlushAllRequest) (*milvuspb.FlushAllResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.FlushAllResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushAllRequest) (*milvuspb.FlushAllResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushAllRequest) *milvuspb.FlushAllResponse); ok { r0 = rf(ctx, request) } else { @@ -1534,6 +1361,7 @@ func (_m *MockProxy) FlushAll(ctx context.Context, request *milvuspb.FlushAllReq } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.FlushAllRequest) error); ok { r1 = rf(ctx, request) } else { @@ -1549,8 +1377,8 @@ type MockProxy_FlushAll_Call struct { } // FlushAll is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.FlushAllRequest +// - ctx context.Context +// - request *milvuspb.FlushAllRequest func (_e *MockProxy_Expecter) FlushAll(ctx interface{}, request interface{}) *MockProxy_FlushAll_Call { return &MockProxy_FlushAll_Call{Call: _e.mock.On("FlushAll", ctx, request)} } @@ -1567,11 +1395,6 @@ func (_c *MockProxy_FlushAll_Call) Return(_a0 *milvuspb.FlushAllResponse, _a1 er return _c } -func (_c *MockProxy_FlushAll_Call) RunAndReturn(run func(context.Context, *milvuspb.FlushAllRequest) (*milvuspb.FlushAllResponse, error)) *MockProxy_FlushAll_Call { - _c.Call.Return(run) - return _c -} - // GetAddress provides a mock function with given fields: func (_m *MockProxy) GetAddress() string { ret := _m.Called() @@ -1608,20 +1431,11 @@ func (_c *MockProxy_GetAddress_Call) Return(_a0 string) *MockProxy_GetAddress_Ca return _c } -func (_c *MockProxy_GetAddress_Call) RunAndReturn(run func() string) *MockProxy_GetAddress_Call { - _c.Call.Return(run) - return _c -} - // GetCollectionStatistics provides a mock function with given fields: ctx, request func (_m *MockProxy) GetCollectionStatistics(ctx context.Context, request *milvuspb.GetCollectionStatisticsRequest) (*milvuspb.GetCollectionStatisticsResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetCollectionStatisticsResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCollectionStatisticsRequest) (*milvuspb.GetCollectionStatisticsResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCollectionStatisticsRequest) *milvuspb.GetCollectionStatisticsResponse); ok { r0 = rf(ctx, request) } else { @@ -1630,6 +1444,7 @@ func (_m *MockProxy) GetCollectionStatistics(ctx context.Context, request *milvu } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetCollectionStatisticsRequest) error); ok { r1 = rf(ctx, request) } else { @@ -1645,8 +1460,8 @@ type MockProxy_GetCollectionStatistics_Call struct { } // GetCollectionStatistics is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetCollectionStatisticsRequest +// - ctx context.Context +// - request *milvuspb.GetCollectionStatisticsRequest func (_e *MockProxy_Expecter) GetCollectionStatistics(ctx interface{}, request interface{}) *MockProxy_GetCollectionStatistics_Call { return &MockProxy_GetCollectionStatistics_Call{Call: _e.mock.On("GetCollectionStatistics", ctx, request)} } @@ -1663,20 +1478,11 @@ func (_c *MockProxy_GetCollectionStatistics_Call) Return(_a0 *milvuspb.GetCollec return _c } -func (_c *MockProxy_GetCollectionStatistics_Call) RunAndReturn(run func(context.Context, *milvuspb.GetCollectionStatisticsRequest) (*milvuspb.GetCollectionStatisticsResponse, error)) *MockProxy_GetCollectionStatistics_Call { - _c.Call.Return(run) - return _c -} - // GetCompactionState provides a mock function with given fields: ctx, req func (_m *MockProxy) GetCompactionState(ctx context.Context, req *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.GetCompactionStateResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionStateRequest) *milvuspb.GetCompactionStateResponse); ok { r0 = rf(ctx, req) } else { @@ -1685,6 +1491,7 @@ func (_m *MockProxy) GetCompactionState(ctx context.Context, req *milvuspb.GetCo } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetCompactionStateRequest) error); ok { r1 = rf(ctx, req) } else { @@ -1700,8 +1507,8 @@ type MockProxy_GetCompactionState_Call struct { } // GetCompactionState is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.GetCompactionStateRequest +// - ctx context.Context +// - req *milvuspb.GetCompactionStateRequest func (_e *MockProxy_Expecter) GetCompactionState(ctx interface{}, req interface{}) *MockProxy_GetCompactionState_Call { return &MockProxy_GetCompactionState_Call{Call: _e.mock.On("GetCompactionState", ctx, req)} } @@ -1718,20 +1525,11 @@ func (_c *MockProxy_GetCompactionState_Call) Return(_a0 *milvuspb.GetCompactionS return _c } -func (_c *MockProxy_GetCompactionState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error)) *MockProxy_GetCompactionState_Call { - _c.Call.Return(run) - return _c -} - // GetCompactionStateWithPlans provides a mock function with given fields: ctx, req func (_m *MockProxy) GetCompactionStateWithPlans(ctx context.Context, req *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.GetCompactionPlansResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionPlansRequest) *milvuspb.GetCompactionPlansResponse); ok { r0 = rf(ctx, req) } else { @@ -1740,6 +1538,7 @@ func (_m *MockProxy) GetCompactionStateWithPlans(ctx context.Context, req *milvu } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetCompactionPlansRequest) error); ok { r1 = rf(ctx, req) } else { @@ -1755,8 +1554,8 @@ type MockProxy_GetCompactionStateWithPlans_Call struct { } // GetCompactionStateWithPlans is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.GetCompactionPlansRequest +// - ctx context.Context +// - req *milvuspb.GetCompactionPlansRequest func (_e *MockProxy_Expecter) GetCompactionStateWithPlans(ctx interface{}, req interface{}) *MockProxy_GetCompactionStateWithPlans_Call { return &MockProxy_GetCompactionStateWithPlans_Call{Call: _e.mock.On("GetCompactionStateWithPlans", ctx, req)} } @@ -1773,20 +1572,11 @@ func (_c *MockProxy_GetCompactionStateWithPlans_Call) Return(_a0 *milvuspb.GetCo return _c } -func (_c *MockProxy_GetCompactionStateWithPlans_Call) RunAndReturn(run func(context.Context, *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error)) *MockProxy_GetCompactionStateWithPlans_Call { - _c.Call.Return(run) - return _c -} - // GetComponentStates provides a mock function with given fields: ctx func (_m *MockProxy) GetComponentStates(ctx context.Context) (*milvuspb.ComponentStates, error) { ret := _m.Called(ctx) var r0 *milvuspb.ComponentStates - var r1 error - if rf, ok := ret.Get(0).(func(context.Context) (*milvuspb.ComponentStates, error)); ok { - return rf(ctx) - } if rf, ok := ret.Get(0).(func(context.Context) *milvuspb.ComponentStates); ok { r0 = rf(ctx) } else { @@ -1795,6 +1585,7 @@ func (_m *MockProxy) GetComponentStates(ctx context.Context) (*milvuspb.Componen } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context) error); ok { r1 = rf(ctx) } else { @@ -1810,7 +1601,7 @@ type MockProxy_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - ctx context.Context +// - ctx context.Context func (_e *MockProxy_Expecter) GetComponentStates(ctx interface{}) *MockProxy_GetComponentStates_Call { return &MockProxy_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", ctx)} } @@ -1827,20 +1618,11 @@ func (_c *MockProxy_GetComponentStates_Call) Return(_a0 *milvuspb.ComponentState return _c } -func (_c *MockProxy_GetComponentStates_Call) RunAndReturn(run func(context.Context) (*milvuspb.ComponentStates, error)) *MockProxy_GetComponentStates_Call { - _c.Call.Return(run) - return _c -} - // GetDdChannel provides a mock function with given fields: ctx, request func (_m *MockProxy) GetDdChannel(ctx context.Context, request *internalpb.GetDdChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.StringResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetDdChannelRequest) (*milvuspb.StringResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetDdChannelRequest) *milvuspb.StringResponse); ok { r0 = rf(ctx, request) } else { @@ -1849,6 +1631,7 @@ func (_m *MockProxy) GetDdChannel(ctx context.Context, request *internalpb.GetDd } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *internalpb.GetDdChannelRequest) error); ok { r1 = rf(ctx, request) } else { @@ -1864,8 +1647,8 @@ type MockProxy_GetDdChannel_Call struct { } // GetDdChannel is a helper method to define mock.On call -// - ctx context.Context -// - request *internalpb.GetDdChannelRequest +// - ctx context.Context +// - request *internalpb.GetDdChannelRequest func (_e *MockProxy_Expecter) GetDdChannel(ctx interface{}, request interface{}) *MockProxy_GetDdChannel_Call { return &MockProxy_GetDdChannel_Call{Call: _e.mock.On("GetDdChannel", ctx, request)} } @@ -1882,20 +1665,11 @@ func (_c *MockProxy_GetDdChannel_Call) Return(_a0 *milvuspb.StringResponse, _a1 return _c } -func (_c *MockProxy_GetDdChannel_Call) RunAndReturn(run func(context.Context, *internalpb.GetDdChannelRequest) (*milvuspb.StringResponse, error)) *MockProxy_GetDdChannel_Call { - _c.Call.Return(run) - return _c -} - // GetFlushAllState provides a mock function with given fields: ctx, req func (_m *MockProxy) GetFlushAllState(ctx context.Context, req *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.GetFlushAllStateResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushAllStateRequest) *milvuspb.GetFlushAllStateResponse); ok { r0 = rf(ctx, req) } else { @@ -1904,6 +1678,7 @@ func (_m *MockProxy) GetFlushAllState(ctx context.Context, req *milvuspb.GetFlus } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetFlushAllStateRequest) error); ok { r1 = rf(ctx, req) } else { @@ -1919,8 +1694,8 @@ type MockProxy_GetFlushAllState_Call struct { } // GetFlushAllState is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.GetFlushAllStateRequest +// - ctx context.Context +// - req *milvuspb.GetFlushAllStateRequest func (_e *MockProxy_Expecter) GetFlushAllState(ctx interface{}, req interface{}) *MockProxy_GetFlushAllState_Call { return &MockProxy_GetFlushAllState_Call{Call: _e.mock.On("GetFlushAllState", ctx, req)} } @@ -1937,20 +1712,11 @@ func (_c *MockProxy_GetFlushAllState_Call) Return(_a0 *milvuspb.GetFlushAllState return _c } -func (_c *MockProxy_GetFlushAllState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error)) *MockProxy_GetFlushAllState_Call { - _c.Call.Return(run) - return _c -} - // GetFlushState provides a mock function with given fields: ctx, req func (_m *MockProxy) GetFlushState(ctx context.Context, req *milvuspb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.GetFlushStateResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushStateRequest) *milvuspb.GetFlushStateResponse); ok { r0 = rf(ctx, req) } else { @@ -1959,6 +1725,7 @@ func (_m *MockProxy) GetFlushState(ctx context.Context, req *milvuspb.GetFlushSt } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetFlushStateRequest) error); ok { r1 = rf(ctx, req) } else { @@ -1974,8 +1741,8 @@ type MockProxy_GetFlushState_Call struct { } // GetFlushState is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.GetFlushStateRequest +// - ctx context.Context +// - req *milvuspb.GetFlushStateRequest func (_e *MockProxy_Expecter) GetFlushState(ctx interface{}, req interface{}) *MockProxy_GetFlushState_Call { return &MockProxy_GetFlushState_Call{Call: _e.mock.On("GetFlushState", ctx, req)} } @@ -1992,20 +1759,11 @@ func (_c *MockProxy_GetFlushState_Call) Return(_a0 *milvuspb.GetFlushStateRespon return _c } -func (_c *MockProxy_GetFlushState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error)) *MockProxy_GetFlushState_Call { - _c.Call.Return(run) - return _c -} - // GetImportState provides a mock function with given fields: ctx, req func (_m *MockProxy) GetImportState(ctx context.Context, req *milvuspb.GetImportStateRequest) (*milvuspb.GetImportStateResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.GetImportStateResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetImportStateRequest) (*milvuspb.GetImportStateResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetImportStateRequest) *milvuspb.GetImportStateResponse); ok { r0 = rf(ctx, req) } else { @@ -2014,6 +1772,7 @@ func (_m *MockProxy) GetImportState(ctx context.Context, req *milvuspb.GetImport } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetImportStateRequest) error); ok { r1 = rf(ctx, req) } else { @@ -2029,8 +1788,8 @@ type MockProxy_GetImportState_Call struct { } // GetImportState is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.GetImportStateRequest +// - ctx context.Context +// - req *milvuspb.GetImportStateRequest func (_e *MockProxy_Expecter) GetImportState(ctx interface{}, req interface{}) *MockProxy_GetImportState_Call { return &MockProxy_GetImportState_Call{Call: _e.mock.On("GetImportState", ctx, req)} } @@ -2047,20 +1806,11 @@ func (_c *MockProxy_GetImportState_Call) Return(_a0 *milvuspb.GetImportStateResp return _c } -func (_c *MockProxy_GetImportState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetImportStateRequest) (*milvuspb.GetImportStateResponse, error)) *MockProxy_GetImportState_Call { - _c.Call.Return(run) - return _c -} - // GetIndexBuildProgress provides a mock function with given fields: ctx, request func (_m *MockProxy) GetIndexBuildProgress(ctx context.Context, request *milvuspb.GetIndexBuildProgressRequest) (*milvuspb.GetIndexBuildProgressResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetIndexBuildProgressResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexBuildProgressRequest) (*milvuspb.GetIndexBuildProgressResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexBuildProgressRequest) *milvuspb.GetIndexBuildProgressResponse); ok { r0 = rf(ctx, request) } else { @@ -2069,6 +1819,7 @@ func (_m *MockProxy) GetIndexBuildProgress(ctx context.Context, request *milvusp } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetIndexBuildProgressRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2084,8 +1835,8 @@ type MockProxy_GetIndexBuildProgress_Call struct { } // GetIndexBuildProgress is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetIndexBuildProgressRequest +// - ctx context.Context +// - request *milvuspb.GetIndexBuildProgressRequest func (_e *MockProxy_Expecter) GetIndexBuildProgress(ctx interface{}, request interface{}) *MockProxy_GetIndexBuildProgress_Call { return &MockProxy_GetIndexBuildProgress_Call{Call: _e.mock.On("GetIndexBuildProgress", ctx, request)} } @@ -2102,20 +1853,11 @@ func (_c *MockProxy_GetIndexBuildProgress_Call) Return(_a0 *milvuspb.GetIndexBui return _c } -func (_c *MockProxy_GetIndexBuildProgress_Call) RunAndReturn(run func(context.Context, *milvuspb.GetIndexBuildProgressRequest) (*milvuspb.GetIndexBuildProgressResponse, error)) *MockProxy_GetIndexBuildProgress_Call { - _c.Call.Return(run) - return _c -} - // GetIndexState provides a mock function with given fields: ctx, request func (_m *MockProxy) GetIndexState(ctx context.Context, request *milvuspb.GetIndexStateRequest) (*milvuspb.GetIndexStateResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetIndexStateResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexStateRequest) (*milvuspb.GetIndexStateResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexStateRequest) *milvuspb.GetIndexStateResponse); ok { r0 = rf(ctx, request) } else { @@ -2124,6 +1866,7 @@ func (_m *MockProxy) GetIndexState(ctx context.Context, request *milvuspb.GetInd } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetIndexStateRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2139,8 +1882,8 @@ type MockProxy_GetIndexState_Call struct { } // GetIndexState is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetIndexStateRequest +// - ctx context.Context +// - request *milvuspb.GetIndexStateRequest func (_e *MockProxy_Expecter) GetIndexState(ctx interface{}, request interface{}) *MockProxy_GetIndexState_Call { return &MockProxy_GetIndexState_Call{Call: _e.mock.On("GetIndexState", ctx, request)} } @@ -2157,20 +1900,11 @@ func (_c *MockProxy_GetIndexState_Call) Return(_a0 *milvuspb.GetIndexStateRespon return _c } -func (_c *MockProxy_GetIndexState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetIndexStateRequest) (*milvuspb.GetIndexStateResponse, error)) *MockProxy_GetIndexState_Call { - _c.Call.Return(run) - return _c -} - // GetIndexStatistics provides a mock function with given fields: ctx, request func (_m *MockProxy) GetIndexStatistics(ctx context.Context, request *milvuspb.GetIndexStatisticsRequest) (*milvuspb.GetIndexStatisticsResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetIndexStatisticsResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexStatisticsRequest) (*milvuspb.GetIndexStatisticsResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexStatisticsRequest) *milvuspb.GetIndexStatisticsResponse); ok { r0 = rf(ctx, request) } else { @@ -2179,6 +1913,7 @@ func (_m *MockProxy) GetIndexStatistics(ctx context.Context, request *milvuspb.G } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetIndexStatisticsRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2194,8 +1929,8 @@ type MockProxy_GetIndexStatistics_Call struct { } // GetIndexStatistics is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetIndexStatisticsRequest +// - ctx context.Context +// - request *milvuspb.GetIndexStatisticsRequest func (_e *MockProxy_Expecter) GetIndexStatistics(ctx interface{}, request interface{}) *MockProxy_GetIndexStatistics_Call { return &MockProxy_GetIndexStatistics_Call{Call: _e.mock.On("GetIndexStatistics", ctx, request)} } @@ -2212,20 +1947,11 @@ func (_c *MockProxy_GetIndexStatistics_Call) Return(_a0 *milvuspb.GetIndexStatis return _c } -func (_c *MockProxy_GetIndexStatistics_Call) RunAndReturn(run func(context.Context, *milvuspb.GetIndexStatisticsRequest) (*milvuspb.GetIndexStatisticsResponse, error)) *MockProxy_GetIndexStatistics_Call { - _c.Call.Return(run) - return _c -} - // GetLoadState provides a mock function with given fields: ctx, request func (_m *MockProxy) GetLoadState(ctx context.Context, request *milvuspb.GetLoadStateRequest) (*milvuspb.GetLoadStateResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetLoadStateResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadStateRequest) (*milvuspb.GetLoadStateResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadStateRequest) *milvuspb.GetLoadStateResponse); ok { r0 = rf(ctx, request) } else { @@ -2234,6 +1960,7 @@ func (_m *MockProxy) GetLoadState(ctx context.Context, request *milvuspb.GetLoad } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetLoadStateRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2249,8 +1976,8 @@ type MockProxy_GetLoadState_Call struct { } // GetLoadState is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetLoadStateRequest +// - ctx context.Context +// - request *milvuspb.GetLoadStateRequest func (_e *MockProxy_Expecter) GetLoadState(ctx interface{}, request interface{}) *MockProxy_GetLoadState_Call { return &MockProxy_GetLoadState_Call{Call: _e.mock.On("GetLoadState", ctx, request)} } @@ -2267,20 +1994,11 @@ func (_c *MockProxy_GetLoadState_Call) Return(_a0 *milvuspb.GetLoadStateResponse return _c } -func (_c *MockProxy_GetLoadState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetLoadStateRequest) (*milvuspb.GetLoadStateResponse, error)) *MockProxy_GetLoadState_Call { - _c.Call.Return(run) - return _c -} - // GetLoadingProgress provides a mock function with given fields: ctx, request func (_m *MockProxy) GetLoadingProgress(ctx context.Context, request *milvuspb.GetLoadingProgressRequest) (*milvuspb.GetLoadingProgressResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetLoadingProgressResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadingProgressRequest) (*milvuspb.GetLoadingProgressResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadingProgressRequest) *milvuspb.GetLoadingProgressResponse); ok { r0 = rf(ctx, request) } else { @@ -2289,6 +2007,7 @@ func (_m *MockProxy) GetLoadingProgress(ctx context.Context, request *milvuspb.G } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetLoadingProgressRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2304,8 +2023,8 @@ type MockProxy_GetLoadingProgress_Call struct { } // GetLoadingProgress is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetLoadingProgressRequest +// - ctx context.Context +// - request *milvuspb.GetLoadingProgressRequest func (_e *MockProxy_Expecter) GetLoadingProgress(ctx interface{}, request interface{}) *MockProxy_GetLoadingProgress_Call { return &MockProxy_GetLoadingProgress_Call{Call: _e.mock.On("GetLoadingProgress", ctx, request)} } @@ -2322,20 +2041,11 @@ func (_c *MockProxy_GetLoadingProgress_Call) Return(_a0 *milvuspb.GetLoadingProg return _c } -func (_c *MockProxy_GetLoadingProgress_Call) RunAndReturn(run func(context.Context, *milvuspb.GetLoadingProgressRequest) (*milvuspb.GetLoadingProgressResponse, error)) *MockProxy_GetLoadingProgress_Call { - _c.Call.Return(run) - return _c -} - // GetMetrics provides a mock function with given fields: ctx, request func (_m *MockProxy) GetMetrics(ctx context.Context, request *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetMetricsResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) *milvuspb.GetMetricsResponse); ok { r0 = rf(ctx, request) } else { @@ -2344,6 +2054,7 @@ func (_m *MockProxy) GetMetrics(ctx context.Context, request *milvuspb.GetMetric } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetMetricsRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2359,8 +2070,8 @@ type MockProxy_GetMetrics_Call struct { } // GetMetrics is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetMetricsRequest +// - ctx context.Context +// - request *milvuspb.GetMetricsRequest func (_e *MockProxy_Expecter) GetMetrics(ctx interface{}, request interface{}) *MockProxy_GetMetrics_Call { return &MockProxy_GetMetrics_Call{Call: _e.mock.On("GetMetrics", ctx, request)} } @@ -2377,20 +2088,11 @@ func (_c *MockProxy_GetMetrics_Call) Return(_a0 *milvuspb.GetMetricsResponse, _a return _c } -func (_c *MockProxy_GetMetrics_Call) RunAndReturn(run func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)) *MockProxy_GetMetrics_Call { - _c.Call.Return(run) - return _c -} - // GetPartitionStatistics provides a mock function with given fields: ctx, request func (_m *MockProxy) GetPartitionStatistics(ctx context.Context, request *milvuspb.GetPartitionStatisticsRequest) (*milvuspb.GetPartitionStatisticsResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetPartitionStatisticsResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPartitionStatisticsRequest) (*milvuspb.GetPartitionStatisticsResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPartitionStatisticsRequest) *milvuspb.GetPartitionStatisticsResponse); ok { r0 = rf(ctx, request) } else { @@ -2399,6 +2101,7 @@ func (_m *MockProxy) GetPartitionStatistics(ctx context.Context, request *milvus } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetPartitionStatisticsRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2414,8 +2117,8 @@ type MockProxy_GetPartitionStatistics_Call struct { } // GetPartitionStatistics is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetPartitionStatisticsRequest +// - ctx context.Context +// - request *milvuspb.GetPartitionStatisticsRequest func (_e *MockProxy_Expecter) GetPartitionStatistics(ctx interface{}, request interface{}) *MockProxy_GetPartitionStatistics_Call { return &MockProxy_GetPartitionStatistics_Call{Call: _e.mock.On("GetPartitionStatistics", ctx, request)} } @@ -2432,20 +2135,11 @@ func (_c *MockProxy_GetPartitionStatistics_Call) Return(_a0 *milvuspb.GetPartiti return _c } -func (_c *MockProxy_GetPartitionStatistics_Call) RunAndReturn(run func(context.Context, *milvuspb.GetPartitionStatisticsRequest) (*milvuspb.GetPartitionStatisticsResponse, error)) *MockProxy_GetPartitionStatistics_Call { - _c.Call.Return(run) - return _c -} - // GetPersistentSegmentInfo provides a mock function with given fields: ctx, request func (_m *MockProxy) GetPersistentSegmentInfo(ctx context.Context, request *milvuspb.GetPersistentSegmentInfoRequest) (*milvuspb.GetPersistentSegmentInfoResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetPersistentSegmentInfoResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) (*milvuspb.GetPersistentSegmentInfoResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) *milvuspb.GetPersistentSegmentInfoResponse); ok { r0 = rf(ctx, request) } else { @@ -2454,6 +2148,7 @@ func (_m *MockProxy) GetPersistentSegmentInfo(ctx context.Context, request *milv } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2469,8 +2164,8 @@ type MockProxy_GetPersistentSegmentInfo_Call struct { } // GetPersistentSegmentInfo is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetPersistentSegmentInfoRequest +// - ctx context.Context +// - request *milvuspb.GetPersistentSegmentInfoRequest func (_e *MockProxy_Expecter) GetPersistentSegmentInfo(ctx interface{}, request interface{}) *MockProxy_GetPersistentSegmentInfo_Call { return &MockProxy_GetPersistentSegmentInfo_Call{Call: _e.mock.On("GetPersistentSegmentInfo", ctx, request)} } @@ -2487,20 +2182,11 @@ func (_c *MockProxy_GetPersistentSegmentInfo_Call) Return(_a0 *milvuspb.GetPersi return _c } -func (_c *MockProxy_GetPersistentSegmentInfo_Call) RunAndReturn(run func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) (*milvuspb.GetPersistentSegmentInfoResponse, error)) *MockProxy_GetPersistentSegmentInfo_Call { - _c.Call.Return(run) - return _c -} - // GetProxyMetrics provides a mock function with given fields: ctx, request func (_m *MockProxy) GetProxyMetrics(ctx context.Context, request *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetMetricsResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) *milvuspb.GetMetricsResponse); ok { r0 = rf(ctx, request) } else { @@ -2509,6 +2195,7 @@ func (_m *MockProxy) GetProxyMetrics(ctx context.Context, request *milvuspb.GetM } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetMetricsRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2524,8 +2211,8 @@ type MockProxy_GetProxyMetrics_Call struct { } // GetProxyMetrics is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetMetricsRequest +// - ctx context.Context +// - request *milvuspb.GetMetricsRequest func (_e *MockProxy_Expecter) GetProxyMetrics(ctx interface{}, request interface{}) *MockProxy_GetProxyMetrics_Call { return &MockProxy_GetProxyMetrics_Call{Call: _e.mock.On("GetProxyMetrics", ctx, request)} } @@ -2542,20 +2229,11 @@ func (_c *MockProxy_GetProxyMetrics_Call) Return(_a0 *milvuspb.GetMetricsRespons return _c } -func (_c *MockProxy_GetProxyMetrics_Call) RunAndReturn(run func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)) *MockProxy_GetProxyMetrics_Call { - _c.Call.Return(run) - return _c -} - // GetQuerySegmentInfo provides a mock function with given fields: ctx, request func (_m *MockProxy) GetQuerySegmentInfo(ctx context.Context, request *milvuspb.GetQuerySegmentInfoRequest) (*milvuspb.GetQuerySegmentInfoResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.GetQuerySegmentInfoResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) (*milvuspb.GetQuerySegmentInfoResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) *milvuspb.GetQuerySegmentInfoResponse); ok { r0 = rf(ctx, request) } else { @@ -2564,6 +2242,7 @@ func (_m *MockProxy) GetQuerySegmentInfo(ctx context.Context, request *milvuspb. } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2579,8 +2258,8 @@ type MockProxy_GetQuerySegmentInfo_Call struct { } // GetQuerySegmentInfo is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.GetQuerySegmentInfoRequest +// - ctx context.Context +// - request *milvuspb.GetQuerySegmentInfoRequest func (_e *MockProxy_Expecter) GetQuerySegmentInfo(ctx interface{}, request interface{}) *MockProxy_GetQuerySegmentInfo_Call { return &MockProxy_GetQuerySegmentInfo_Call{Call: _e.mock.On("GetQuerySegmentInfo", ctx, request)} } @@ -2597,20 +2276,11 @@ func (_c *MockProxy_GetQuerySegmentInfo_Call) Return(_a0 *milvuspb.GetQuerySegme return _c } -func (_c *MockProxy_GetQuerySegmentInfo_Call) RunAndReturn(run func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) (*milvuspb.GetQuerySegmentInfoResponse, error)) *MockProxy_GetQuerySegmentInfo_Call { - _c.Call.Return(run) - return _c -} - // GetRateLimiter provides a mock function with given fields: func (_m *MockProxy) GetRateLimiter() (types.Limiter, error) { ret := _m.Called() var r0 types.Limiter - var r1 error - if rf, ok := ret.Get(0).(func() (types.Limiter, error)); ok { - return rf() - } if rf, ok := ret.Get(0).(func() types.Limiter); ok { r0 = rf() } else { @@ -2619,6 +2289,7 @@ func (_m *MockProxy) GetRateLimiter() (types.Limiter, error) { } } + var r1 error if rf, ok := ret.Get(1).(func() error); ok { r1 = rf() } else { @@ -2650,20 +2321,11 @@ func (_c *MockProxy_GetRateLimiter_Call) Return(_a0 types.Limiter, _a1 error) *M return _c } -func (_c *MockProxy_GetRateLimiter_Call) RunAndReturn(run func() (types.Limiter, error)) *MockProxy_GetRateLimiter_Call { - _c.Call.Return(run) - return _c -} - // GetReplicas provides a mock function with given fields: ctx, req func (_m *MockProxy) GetReplicas(ctx context.Context, req *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.GetReplicasResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetReplicasRequest) *milvuspb.GetReplicasResponse); ok { r0 = rf(ctx, req) } else { @@ -2672,6 +2334,7 @@ func (_m *MockProxy) GetReplicas(ctx context.Context, req *milvuspb.GetReplicasR } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetReplicasRequest) error); ok { r1 = rf(ctx, req) } else { @@ -2687,8 +2350,8 @@ type MockProxy_GetReplicas_Call struct { } // GetReplicas is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.GetReplicasRequest +// - ctx context.Context +// - req *milvuspb.GetReplicasRequest func (_e *MockProxy_Expecter) GetReplicas(ctx interface{}, req interface{}) *MockProxy_GetReplicas_Call { return &MockProxy_GetReplicas_Call{Call: _e.mock.On("GetReplicas", ctx, req)} } @@ -2705,20 +2368,11 @@ func (_c *MockProxy_GetReplicas_Call) Return(_a0 *milvuspb.GetReplicasResponse, return _c } -func (_c *MockProxy_GetReplicas_Call) RunAndReturn(run func(context.Context, *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error)) *MockProxy_GetReplicas_Call { - _c.Call.Return(run) - return _c -} - // GetStatisticsChannel provides a mock function with given fields: ctx func (_m *MockProxy) GetStatisticsChannel(ctx context.Context) (*milvuspb.StringResponse, error) { ret := _m.Called(ctx) var r0 *milvuspb.StringResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context) (*milvuspb.StringResponse, error)); ok { - return rf(ctx) - } if rf, ok := ret.Get(0).(func(context.Context) *milvuspb.StringResponse); ok { r0 = rf(ctx) } else { @@ -2727,6 +2381,7 @@ func (_m *MockProxy) GetStatisticsChannel(ctx context.Context) (*milvuspb.String } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context) error); ok { r1 = rf(ctx) } else { @@ -2742,7 +2397,7 @@ type MockProxy_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - ctx context.Context +// - ctx context.Context func (_e *MockProxy_Expecter) GetStatisticsChannel(ctx interface{}) *MockProxy_GetStatisticsChannel_Call { return &MockProxy_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", ctx)} } @@ -2759,20 +2414,11 @@ func (_c *MockProxy_GetStatisticsChannel_Call) Return(_a0 *milvuspb.StringRespon return _c } -func (_c *MockProxy_GetStatisticsChannel_Call) RunAndReturn(run func(context.Context) (*milvuspb.StringResponse, error)) *MockProxy_GetStatisticsChannel_Call { - _c.Call.Return(run) - return _c -} - // HasCollection provides a mock function with given fields: ctx, request func (_m *MockProxy) HasCollection(ctx context.Context, request *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.BoolResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasCollectionRequest) *milvuspb.BoolResponse); ok { r0 = rf(ctx, request) } else { @@ -2781,6 +2427,7 @@ func (_m *MockProxy) HasCollection(ctx context.Context, request *milvuspb.HasCol } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.HasCollectionRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2796,8 +2443,8 @@ type MockProxy_HasCollection_Call struct { } // HasCollection is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.HasCollectionRequest +// - ctx context.Context +// - request *milvuspb.HasCollectionRequest func (_e *MockProxy_Expecter) HasCollection(ctx interface{}, request interface{}) *MockProxy_HasCollection_Call { return &MockProxy_HasCollection_Call{Call: _e.mock.On("HasCollection", ctx, request)} } @@ -2814,20 +2461,11 @@ func (_c *MockProxy_HasCollection_Call) Return(_a0 *milvuspb.BoolResponse, _a1 e return _c } -func (_c *MockProxy_HasCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error)) *MockProxy_HasCollection_Call { - _c.Call.Return(run) - return _c -} - // HasPartition provides a mock function with given fields: ctx, request func (_m *MockProxy) HasPartition(ctx context.Context, request *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.BoolResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasPartitionRequest) *milvuspb.BoolResponse); ok { r0 = rf(ctx, request) } else { @@ -2836,6 +2474,7 @@ func (_m *MockProxy) HasPartition(ctx context.Context, request *milvuspb.HasPart } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.HasPartitionRequest) error); ok { r1 = rf(ctx, request) } else { @@ -2851,8 +2490,8 @@ type MockProxy_HasPartition_Call struct { } // HasPartition is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.HasPartitionRequest +// - ctx context.Context +// - request *milvuspb.HasPartitionRequest func (_e *MockProxy_Expecter) HasPartition(ctx interface{}, request interface{}) *MockProxy_HasPartition_Call { return &MockProxy_HasPartition_Call{Call: _e.mock.On("HasPartition", ctx, request)} } @@ -2869,20 +2508,11 @@ func (_c *MockProxy_HasPartition_Call) Return(_a0 *milvuspb.BoolResponse, _a1 er return _c } -func (_c *MockProxy_HasPartition_Call) RunAndReturn(run func(context.Context, *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error)) *MockProxy_HasPartition_Call { - _c.Call.Return(run) - return _c -} - // Import provides a mock function with given fields: ctx, req func (_m *MockProxy) Import(ctx context.Context, req *milvuspb.ImportRequest) (*milvuspb.ImportResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.ImportResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ImportRequest) (*milvuspb.ImportResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ImportRequest) *milvuspb.ImportResponse); ok { r0 = rf(ctx, req) } else { @@ -2891,6 +2521,7 @@ func (_m *MockProxy) Import(ctx context.Context, req *milvuspb.ImportRequest) (* } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ImportRequest) error); ok { r1 = rf(ctx, req) } else { @@ -2906,8 +2537,8 @@ type MockProxy_Import_Call struct { } // Import is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.ImportRequest +// - ctx context.Context +// - req *milvuspb.ImportRequest func (_e *MockProxy_Expecter) Import(ctx interface{}, req interface{}) *MockProxy_Import_Call { return &MockProxy_Import_Call{Call: _e.mock.On("Import", ctx, req)} } @@ -2924,11 +2555,6 @@ func (_c *MockProxy_Import_Call) Return(_a0 *milvuspb.ImportResponse, _a1 error) return _c } -func (_c *MockProxy_Import_Call) RunAndReturn(run func(context.Context, *milvuspb.ImportRequest) (*milvuspb.ImportResponse, error)) *MockProxy_Import_Call { - _c.Call.Return(run) - return _c -} - // Init provides a mock function with given fields: func (_m *MockProxy) Init() error { ret := _m.Called() @@ -2965,20 +2591,11 @@ func (_c *MockProxy_Init_Call) Return(_a0 error) *MockProxy_Init_Call { return _c } -func (_c *MockProxy_Init_Call) RunAndReturn(run func() error) *MockProxy_Init_Call { - _c.Call.Return(run) - return _c -} - // Insert provides a mock function with given fields: ctx, request func (_m *MockProxy) Insert(ctx context.Context, request *milvuspb.InsertRequest) (*milvuspb.MutationResult, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.MutationResult - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.InsertRequest) (*milvuspb.MutationResult, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.InsertRequest) *milvuspb.MutationResult); ok { r0 = rf(ctx, request) } else { @@ -2987,6 +2604,7 @@ func (_m *MockProxy) Insert(ctx context.Context, request *milvuspb.InsertRequest } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.InsertRequest) error); ok { r1 = rf(ctx, request) } else { @@ -3002,8 +2620,8 @@ type MockProxy_Insert_Call struct { } // Insert is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.InsertRequest +// - ctx context.Context +// - request *milvuspb.InsertRequest func (_e *MockProxy_Expecter) Insert(ctx interface{}, request interface{}) *MockProxy_Insert_Call { return &MockProxy_Insert_Call{Call: _e.mock.On("Insert", ctx, request)} } @@ -3020,20 +2638,11 @@ func (_c *MockProxy_Insert_Call) Return(_a0 *milvuspb.MutationResult, _a1 error) return _c } -func (_c *MockProxy_Insert_Call) RunAndReturn(run func(context.Context, *milvuspb.InsertRequest) (*milvuspb.MutationResult, error)) *MockProxy_Insert_Call { - _c.Call.Return(run) - return _c -} - // InvalidateCollectionMetaCache provides a mock function with given fields: ctx, request func (_m *MockProxy) InvalidateCollectionMetaCache(ctx context.Context, request *proxypb.InvalidateCollMetaCacheRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -3042,6 +2651,7 @@ func (_m *MockProxy) InvalidateCollectionMetaCache(ctx context.Context, request } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest) error); ok { r1 = rf(ctx, request) } else { @@ -3057,8 +2667,8 @@ type MockProxy_InvalidateCollectionMetaCache_Call struct { } // InvalidateCollectionMetaCache is a helper method to define mock.On call -// - ctx context.Context -// - request *proxypb.InvalidateCollMetaCacheRequest +// - ctx context.Context +// - request *proxypb.InvalidateCollMetaCacheRequest func (_e *MockProxy_Expecter) InvalidateCollectionMetaCache(ctx interface{}, request interface{}) *MockProxy_InvalidateCollectionMetaCache_Call { return &MockProxy_InvalidateCollectionMetaCache_Call{Call: _e.mock.On("InvalidateCollectionMetaCache", ctx, request)} } @@ -3075,20 +2685,11 @@ func (_c *MockProxy_InvalidateCollectionMetaCache_Call) Return(_a0 *commonpb.Sta return _c } -func (_c *MockProxy_InvalidateCollectionMetaCache_Call) RunAndReturn(run func(context.Context, *proxypb.InvalidateCollMetaCacheRequest) (*commonpb.Status, error)) *MockProxy_InvalidateCollectionMetaCache_Call { - _c.Call.Return(run) - return _c -} - // InvalidateCredentialCache provides a mock function with given fields: ctx, request func (_m *MockProxy) InvalidateCredentialCache(ctx context.Context, request *proxypb.InvalidateCredCacheRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCredCacheRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCredCacheRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -3097,6 +2698,7 @@ func (_m *MockProxy) InvalidateCredentialCache(ctx context.Context, request *pro } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *proxypb.InvalidateCredCacheRequest) error); ok { r1 = rf(ctx, request) } else { @@ -3112,8 +2714,8 @@ type MockProxy_InvalidateCredentialCache_Call struct { } // InvalidateCredentialCache is a helper method to define mock.On call -// - ctx context.Context -// - request *proxypb.InvalidateCredCacheRequest +// - ctx context.Context +// - request *proxypb.InvalidateCredCacheRequest func (_e *MockProxy_Expecter) InvalidateCredentialCache(ctx interface{}, request interface{}) *MockProxy_InvalidateCredentialCache_Call { return &MockProxy_InvalidateCredentialCache_Call{Call: _e.mock.On("InvalidateCredentialCache", ctx, request)} } @@ -3130,20 +2732,11 @@ func (_c *MockProxy_InvalidateCredentialCache_Call) Return(_a0 *commonpb.Status, return _c } -func (_c *MockProxy_InvalidateCredentialCache_Call) RunAndReturn(run func(context.Context, *proxypb.InvalidateCredCacheRequest) (*commonpb.Status, error)) *MockProxy_InvalidateCredentialCache_Call { - _c.Call.Return(run) - return _c -} - // ListClientInfos provides a mock function with given fields: ctx, req func (_m *MockProxy) ListClientInfos(ctx context.Context, req *proxypb.ListClientInfosRequest) (*proxypb.ListClientInfosResponse, error) { ret := _m.Called(ctx, req) var r0 *proxypb.ListClientInfosResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *proxypb.ListClientInfosRequest) (*proxypb.ListClientInfosResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *proxypb.ListClientInfosRequest) *proxypb.ListClientInfosResponse); ok { r0 = rf(ctx, req) } else { @@ -3152,6 +2745,7 @@ func (_m *MockProxy) ListClientInfos(ctx context.Context, req *proxypb.ListClien } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *proxypb.ListClientInfosRequest) error); ok { r1 = rf(ctx, req) } else { @@ -3167,8 +2761,8 @@ type MockProxy_ListClientInfos_Call struct { } // ListClientInfos is a helper method to define mock.On call -// - ctx context.Context -// - req *proxypb.ListClientInfosRequest +// - ctx context.Context +// - req *proxypb.ListClientInfosRequest func (_e *MockProxy_Expecter) ListClientInfos(ctx interface{}, req interface{}) *MockProxy_ListClientInfos_Call { return &MockProxy_ListClientInfos_Call{Call: _e.mock.On("ListClientInfos", ctx, req)} } @@ -3185,20 +2779,11 @@ func (_c *MockProxy_ListClientInfos_Call) Return(_a0 *proxypb.ListClientInfosRes return _c } -func (_c *MockProxy_ListClientInfos_Call) RunAndReturn(run func(context.Context, *proxypb.ListClientInfosRequest) (*proxypb.ListClientInfosResponse, error)) *MockProxy_ListClientInfos_Call { - _c.Call.Return(run) - return _c -} - // ListCredUsers provides a mock function with given fields: ctx, req func (_m *MockProxy) ListCredUsers(ctx context.Context, req *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.ListCredUsersResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListCredUsersRequest) *milvuspb.ListCredUsersResponse); ok { r0 = rf(ctx, req) } else { @@ -3207,6 +2792,7 @@ func (_m *MockProxy) ListCredUsers(ctx context.Context, req *milvuspb.ListCredUs } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListCredUsersRequest) error); ok { r1 = rf(ctx, req) } else { @@ -3222,8 +2808,8 @@ type MockProxy_ListCredUsers_Call struct { } // ListCredUsers is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.ListCredUsersRequest +// - ctx context.Context +// - req *milvuspb.ListCredUsersRequest func (_e *MockProxy_Expecter) ListCredUsers(ctx interface{}, req interface{}) *MockProxy_ListCredUsers_Call { return &MockProxy_ListCredUsers_Call{Call: _e.mock.On("ListCredUsers", ctx, req)} } @@ -3240,20 +2826,11 @@ func (_c *MockProxy_ListCredUsers_Call) Return(_a0 *milvuspb.ListCredUsersRespon return _c } -func (_c *MockProxy_ListCredUsers_Call) RunAndReturn(run func(context.Context, *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error)) *MockProxy_ListCredUsers_Call { - _c.Call.Return(run) - return _c -} - // ListDatabases provides a mock function with given fields: ctx, req func (_m *MockProxy) ListDatabases(ctx context.Context, req *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.ListDatabasesResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListDatabasesRequest) *milvuspb.ListDatabasesResponse); ok { r0 = rf(ctx, req) } else { @@ -3262,6 +2839,7 @@ func (_m *MockProxy) ListDatabases(ctx context.Context, req *milvuspb.ListDataba } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListDatabasesRequest) error); ok { r1 = rf(ctx, req) } else { @@ -3277,8 +2855,8 @@ type MockProxy_ListDatabases_Call struct { } // ListDatabases is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.ListDatabasesRequest +// - ctx context.Context +// - req *milvuspb.ListDatabasesRequest func (_e *MockProxy_Expecter) ListDatabases(ctx interface{}, req interface{}) *MockProxy_ListDatabases_Call { return &MockProxy_ListDatabases_Call{Call: _e.mock.On("ListDatabases", ctx, req)} } @@ -3295,20 +2873,11 @@ func (_c *MockProxy_ListDatabases_Call) Return(_a0 *milvuspb.ListDatabasesRespon return _c } -func (_c *MockProxy_ListDatabases_Call) RunAndReturn(run func(context.Context, *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error)) *MockProxy_ListDatabases_Call { - _c.Call.Return(run) - return _c -} - // ListImportTasks provides a mock function with given fields: ctx, req func (_m *MockProxy) ListImportTasks(ctx context.Context, req *milvuspb.ListImportTasksRequest) (*milvuspb.ListImportTasksResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.ListImportTasksResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListImportTasksRequest) (*milvuspb.ListImportTasksResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListImportTasksRequest) *milvuspb.ListImportTasksResponse); ok { r0 = rf(ctx, req) } else { @@ -3317,6 +2886,7 @@ func (_m *MockProxy) ListImportTasks(ctx context.Context, req *milvuspb.ListImpo } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListImportTasksRequest) error); ok { r1 = rf(ctx, req) } else { @@ -3332,8 +2902,8 @@ type MockProxy_ListImportTasks_Call struct { } // ListImportTasks is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.ListImportTasksRequest +// - ctx context.Context +// - req *milvuspb.ListImportTasksRequest func (_e *MockProxy_Expecter) ListImportTasks(ctx interface{}, req interface{}) *MockProxy_ListImportTasks_Call { return &MockProxy_ListImportTasks_Call{Call: _e.mock.On("ListImportTasks", ctx, req)} } @@ -3350,20 +2920,11 @@ func (_c *MockProxy_ListImportTasks_Call) Return(_a0 *milvuspb.ListImportTasksRe return _c } -func (_c *MockProxy_ListImportTasks_Call) RunAndReturn(run func(context.Context, *milvuspb.ListImportTasksRequest) (*milvuspb.ListImportTasksResponse, error)) *MockProxy_ListImportTasks_Call { - _c.Call.Return(run) - return _c -} - // ListResourceGroups provides a mock function with given fields: ctx, req func (_m *MockProxy) ListResourceGroups(ctx context.Context, req *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.ListResourceGroupsResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListResourceGroupsRequest) *milvuspb.ListResourceGroupsResponse); ok { r0 = rf(ctx, req) } else { @@ -3372,6 +2933,7 @@ func (_m *MockProxy) ListResourceGroups(ctx context.Context, req *milvuspb.ListR } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListResourceGroupsRequest) error); ok { r1 = rf(ctx, req) } else { @@ -3387,8 +2949,8 @@ type MockProxy_ListResourceGroups_Call struct { } // ListResourceGroups is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.ListResourceGroupsRequest +// - ctx context.Context +// - req *milvuspb.ListResourceGroupsRequest func (_e *MockProxy_Expecter) ListResourceGroups(ctx interface{}, req interface{}) *MockProxy_ListResourceGroups_Call { return &MockProxy_ListResourceGroups_Call{Call: _e.mock.On("ListResourceGroups", ctx, req)} } @@ -3405,20 +2967,11 @@ func (_c *MockProxy_ListResourceGroups_Call) Return(_a0 *milvuspb.ListResourceGr return _c } -func (_c *MockProxy_ListResourceGroups_Call) RunAndReturn(run func(context.Context, *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error)) *MockProxy_ListResourceGroups_Call { - _c.Call.Return(run) - return _c -} - // LoadBalance provides a mock function with given fields: ctx, request func (_m *MockProxy) LoadBalance(ctx context.Context, request *milvuspb.LoadBalanceRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadBalanceRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadBalanceRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -3427,6 +2980,7 @@ func (_m *MockProxy) LoadBalance(ctx context.Context, request *milvuspb.LoadBala } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.LoadBalanceRequest) error); ok { r1 = rf(ctx, request) } else { @@ -3442,8 +2996,8 @@ type MockProxy_LoadBalance_Call struct { } // LoadBalance is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.LoadBalanceRequest +// - ctx context.Context +// - request *milvuspb.LoadBalanceRequest func (_e *MockProxy_Expecter) LoadBalance(ctx interface{}, request interface{}) *MockProxy_LoadBalance_Call { return &MockProxy_LoadBalance_Call{Call: _e.mock.On("LoadBalance", ctx, request)} } @@ -3460,20 +3014,11 @@ func (_c *MockProxy_LoadBalance_Call) Return(_a0 *commonpb.Status, _a1 error) *M return _c } -func (_c *MockProxy_LoadBalance_Call) RunAndReturn(run func(context.Context, *milvuspb.LoadBalanceRequest) (*commonpb.Status, error)) *MockProxy_LoadBalance_Call { - _c.Call.Return(run) - return _c -} - // LoadCollection provides a mock function with given fields: ctx, request func (_m *MockProxy) LoadCollection(ctx context.Context, request *milvuspb.LoadCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadCollectionRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadCollectionRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -3482,6 +3027,7 @@ func (_m *MockProxy) LoadCollection(ctx context.Context, request *milvuspb.LoadC } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.LoadCollectionRequest) error); ok { r1 = rf(ctx, request) } else { @@ -3497,8 +3043,8 @@ type MockProxy_LoadCollection_Call struct { } // LoadCollection is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.LoadCollectionRequest +// - ctx context.Context +// - request *milvuspb.LoadCollectionRequest func (_e *MockProxy_Expecter) LoadCollection(ctx interface{}, request interface{}) *MockProxy_LoadCollection_Call { return &MockProxy_LoadCollection_Call{Call: _e.mock.On("LoadCollection", ctx, request)} } @@ -3515,20 +3061,11 @@ func (_c *MockProxy_LoadCollection_Call) Return(_a0 *commonpb.Status, _a1 error) return _c } -func (_c *MockProxy_LoadCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.LoadCollectionRequest) (*commonpb.Status, error)) *MockProxy_LoadCollection_Call { - _c.Call.Return(run) - return _c -} - // LoadPartitions provides a mock function with given fields: ctx, request func (_m *MockProxy) LoadPartitions(ctx context.Context, request *milvuspb.LoadPartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadPartitionsRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadPartitionsRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -3537,6 +3074,7 @@ func (_m *MockProxy) LoadPartitions(ctx context.Context, request *milvuspb.LoadP } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.LoadPartitionsRequest) error); ok { r1 = rf(ctx, request) } else { @@ -3552,8 +3090,8 @@ type MockProxy_LoadPartitions_Call struct { } // LoadPartitions is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.LoadPartitionsRequest +// - ctx context.Context +// - request *milvuspb.LoadPartitionsRequest func (_e *MockProxy_Expecter) LoadPartitions(ctx interface{}, request interface{}) *MockProxy_LoadPartitions_Call { return &MockProxy_LoadPartitions_Call{Call: _e.mock.On("LoadPartitions", ctx, request)} } @@ -3570,20 +3108,11 @@ func (_c *MockProxy_LoadPartitions_Call) Return(_a0 *commonpb.Status, _a1 error) return _c } -func (_c *MockProxy_LoadPartitions_Call) RunAndReturn(run func(context.Context, *milvuspb.LoadPartitionsRequest) (*commonpb.Status, error)) *MockProxy_LoadPartitions_Call { - _c.Call.Return(run) - return _c -} - // ManualCompaction provides a mock function with given fields: ctx, req func (_m *MockProxy) ManualCompaction(ctx context.Context, req *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.ManualCompactionResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ManualCompactionRequest) *milvuspb.ManualCompactionResponse); ok { r0 = rf(ctx, req) } else { @@ -3592,6 +3121,7 @@ func (_m *MockProxy) ManualCompaction(ctx context.Context, req *milvuspb.ManualC } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ManualCompactionRequest) error); ok { r1 = rf(ctx, req) } else { @@ -3607,8 +3137,8 @@ type MockProxy_ManualCompaction_Call struct { } // ManualCompaction is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.ManualCompactionRequest +// - ctx context.Context +// - req *milvuspb.ManualCompactionRequest func (_e *MockProxy_Expecter) ManualCompaction(ctx interface{}, req interface{}) *MockProxy_ManualCompaction_Call { return &MockProxy_ManualCompaction_Call{Call: _e.mock.On("ManualCompaction", ctx, req)} } @@ -3625,20 +3155,11 @@ func (_c *MockProxy_ManualCompaction_Call) Return(_a0 *milvuspb.ManualCompaction return _c } -func (_c *MockProxy_ManualCompaction_Call) RunAndReturn(run func(context.Context, *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error)) *MockProxy_ManualCompaction_Call { - _c.Call.Return(run) - return _c -} - // OperatePrivilege provides a mock function with given fields: ctx, req func (_m *MockProxy) OperatePrivilege(ctx context.Context, req *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperatePrivilegeRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -3647,6 +3168,7 @@ func (_m *MockProxy) OperatePrivilege(ctx context.Context, req *milvuspb.Operate } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.OperatePrivilegeRequest) error); ok { r1 = rf(ctx, req) } else { @@ -3662,8 +3184,8 @@ type MockProxy_OperatePrivilege_Call struct { } // OperatePrivilege is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.OperatePrivilegeRequest +// - ctx context.Context +// - req *milvuspb.OperatePrivilegeRequest func (_e *MockProxy_Expecter) OperatePrivilege(ctx interface{}, req interface{}) *MockProxy_OperatePrivilege_Call { return &MockProxy_OperatePrivilege_Call{Call: _e.mock.On("OperatePrivilege", ctx, req)} } @@ -3680,20 +3202,11 @@ func (_c *MockProxy_OperatePrivilege_Call) Return(_a0 *commonpb.Status, _a1 erro return _c } -func (_c *MockProxy_OperatePrivilege_Call) RunAndReturn(run func(context.Context, *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error)) *MockProxy_OperatePrivilege_Call { - _c.Call.Return(run) - return _c -} - // OperateUserRole provides a mock function with given fields: ctx, req func (_m *MockProxy) OperateUserRole(ctx context.Context, req *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperateUserRoleRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -3702,6 +3215,7 @@ func (_m *MockProxy) OperateUserRole(ctx context.Context, req *milvuspb.OperateU } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.OperateUserRoleRequest) error); ok { r1 = rf(ctx, req) } else { @@ -3717,8 +3231,8 @@ type MockProxy_OperateUserRole_Call struct { } // OperateUserRole is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.OperateUserRoleRequest +// - ctx context.Context +// - req *milvuspb.OperateUserRoleRequest func (_e *MockProxy_Expecter) OperateUserRole(ctx interface{}, req interface{}) *MockProxy_OperateUserRole_Call { return &MockProxy_OperateUserRole_Call{Call: _e.mock.On("OperateUserRole", ctx, req)} } @@ -3735,20 +3249,11 @@ func (_c *MockProxy_OperateUserRole_Call) Return(_a0 *commonpb.Status, _a1 error return _c } -func (_c *MockProxy_OperateUserRole_Call) RunAndReturn(run func(context.Context, *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error)) *MockProxy_OperateUserRole_Call { - _c.Call.Return(run) - return _c -} - // Query provides a mock function with given fields: ctx, request func (_m *MockProxy) Query(ctx context.Context, request *milvuspb.QueryRequest) (*milvuspb.QueryResults, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.QueryResults - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.QueryRequest) (*milvuspb.QueryResults, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.QueryRequest) *milvuspb.QueryResults); ok { r0 = rf(ctx, request) } else { @@ -3757,6 +3262,7 @@ func (_m *MockProxy) Query(ctx context.Context, request *milvuspb.QueryRequest) } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.QueryRequest) error); ok { r1 = rf(ctx, request) } else { @@ -3772,8 +3278,8 @@ type MockProxy_Query_Call struct { } // Query is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.QueryRequest +// - ctx context.Context +// - request *milvuspb.QueryRequest func (_e *MockProxy_Expecter) Query(ctx interface{}, request interface{}) *MockProxy_Query_Call { return &MockProxy_Query_Call{Call: _e.mock.On("Query", ctx, request)} } @@ -3790,20 +3296,11 @@ func (_c *MockProxy_Query_Call) Return(_a0 *milvuspb.QueryResults, _a1 error) *M return _c } -func (_c *MockProxy_Query_Call) RunAndReturn(run func(context.Context, *milvuspb.QueryRequest) (*milvuspb.QueryResults, error)) *MockProxy_Query_Call { - _c.Call.Return(run) - return _c -} - // RefreshPolicyInfoCache provides a mock function with given fields: ctx, req func (_m *MockProxy) RefreshPolicyInfoCache(ctx context.Context, req *proxypb.RefreshPolicyInfoCacheRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *proxypb.RefreshPolicyInfoCacheRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *proxypb.RefreshPolicyInfoCacheRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -3812,6 +3309,7 @@ func (_m *MockProxy) RefreshPolicyInfoCache(ctx context.Context, req *proxypb.Re } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *proxypb.RefreshPolicyInfoCacheRequest) error); ok { r1 = rf(ctx, req) } else { @@ -3827,8 +3325,8 @@ type MockProxy_RefreshPolicyInfoCache_Call struct { } // RefreshPolicyInfoCache is a helper method to define mock.On call -// - ctx context.Context -// - req *proxypb.RefreshPolicyInfoCacheRequest +// - ctx context.Context +// - req *proxypb.RefreshPolicyInfoCacheRequest func (_e *MockProxy_Expecter) RefreshPolicyInfoCache(ctx interface{}, req interface{}) *MockProxy_RefreshPolicyInfoCache_Call { return &MockProxy_RefreshPolicyInfoCache_Call{Call: _e.mock.On("RefreshPolicyInfoCache", ctx, req)} } @@ -3845,11 +3343,6 @@ func (_c *MockProxy_RefreshPolicyInfoCache_Call) Return(_a0 *commonpb.Status, _a return _c } -func (_c *MockProxy_RefreshPolicyInfoCache_Call) RunAndReturn(run func(context.Context, *proxypb.RefreshPolicyInfoCacheRequest) (*commonpb.Status, error)) *MockProxy_RefreshPolicyInfoCache_Call { - _c.Call.Return(run) - return _c -} - // Register provides a mock function with given fields: func (_m *MockProxy) Register() error { ret := _m.Called() @@ -3886,20 +3379,11 @@ func (_c *MockProxy_Register_Call) Return(_a0 error) *MockProxy_Register_Call { return _c } -func (_c *MockProxy_Register_Call) RunAndReturn(run func() error) *MockProxy_Register_Call { - _c.Call.Return(run) - return _c -} - // RegisterLink provides a mock function with given fields: ctx, request func (_m *MockProxy) RegisterLink(ctx context.Context, request *milvuspb.RegisterLinkRequest) (*milvuspb.RegisterLinkResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.RegisterLinkResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RegisterLinkRequest) (*milvuspb.RegisterLinkResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RegisterLinkRequest) *milvuspb.RegisterLinkResponse); ok { r0 = rf(ctx, request) } else { @@ -3908,6 +3392,7 @@ func (_m *MockProxy) RegisterLink(ctx context.Context, request *milvuspb.Registe } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.RegisterLinkRequest) error); ok { r1 = rf(ctx, request) } else { @@ -3923,8 +3408,8 @@ type MockProxy_RegisterLink_Call struct { } // RegisterLink is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.RegisterLinkRequest +// - ctx context.Context +// - request *milvuspb.RegisterLinkRequest func (_e *MockProxy_Expecter) RegisterLink(ctx interface{}, request interface{}) *MockProxy_RegisterLink_Call { return &MockProxy_RegisterLink_Call{Call: _e.mock.On("RegisterLink", ctx, request)} } @@ -3941,20 +3426,11 @@ func (_c *MockProxy_RegisterLink_Call) Return(_a0 *milvuspb.RegisterLinkResponse return _c } -func (_c *MockProxy_RegisterLink_Call) RunAndReturn(run func(context.Context, *milvuspb.RegisterLinkRequest) (*milvuspb.RegisterLinkResponse, error)) *MockProxy_RegisterLink_Call { - _c.Call.Return(run) - return _c -} - // ReleaseCollection provides a mock function with given fields: ctx, request func (_m *MockProxy) ReleaseCollection(ctx context.Context, request *milvuspb.ReleaseCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleaseCollectionRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleaseCollectionRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -3963,6 +3439,7 @@ func (_m *MockProxy) ReleaseCollection(ctx context.Context, request *milvuspb.Re } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ReleaseCollectionRequest) error); ok { r1 = rf(ctx, request) } else { @@ -3978,8 +3455,8 @@ type MockProxy_ReleaseCollection_Call struct { } // ReleaseCollection is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.ReleaseCollectionRequest +// - ctx context.Context +// - request *milvuspb.ReleaseCollectionRequest func (_e *MockProxy_Expecter) ReleaseCollection(ctx interface{}, request interface{}) *MockProxy_ReleaseCollection_Call { return &MockProxy_ReleaseCollection_Call{Call: _e.mock.On("ReleaseCollection", ctx, request)} } @@ -3996,20 +3473,11 @@ func (_c *MockProxy_ReleaseCollection_Call) Return(_a0 *commonpb.Status, _a1 err return _c } -func (_c *MockProxy_ReleaseCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.ReleaseCollectionRequest) (*commonpb.Status, error)) *MockProxy_ReleaseCollection_Call { - _c.Call.Return(run) - return _c -} - // ReleasePartitions provides a mock function with given fields: ctx, request func (_m *MockProxy) ReleasePartitions(ctx context.Context, request *milvuspb.ReleasePartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleasePartitionsRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -4018,6 +3486,7 @@ func (_m *MockProxy) ReleasePartitions(ctx context.Context, request *milvuspb.Re } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ReleasePartitionsRequest) error); ok { r1 = rf(ctx, request) } else { @@ -4033,8 +3502,8 @@ type MockProxy_ReleasePartitions_Call struct { } // ReleasePartitions is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.ReleasePartitionsRequest +// - ctx context.Context +// - request *milvuspb.ReleasePartitionsRequest func (_e *MockProxy_Expecter) ReleasePartitions(ctx interface{}, request interface{}) *MockProxy_ReleasePartitions_Call { return &MockProxy_ReleasePartitions_Call{Call: _e.mock.On("ReleasePartitions", ctx, request)} } @@ -4051,20 +3520,11 @@ func (_c *MockProxy_ReleasePartitions_Call) Return(_a0 *commonpb.Status, _a1 err return _c } -func (_c *MockProxy_ReleasePartitions_Call) RunAndReturn(run func(context.Context, *milvuspb.ReleasePartitionsRequest) (*commonpb.Status, error)) *MockProxy_ReleasePartitions_Call { - _c.Call.Return(run) - return _c -} - // RenameCollection provides a mock function with given fields: ctx, req func (_m *MockProxy) RenameCollection(ctx context.Context, req *milvuspb.RenameCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RenameCollectionRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RenameCollectionRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -4073,6 +3533,7 @@ func (_m *MockProxy) RenameCollection(ctx context.Context, req *milvuspb.RenameC } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.RenameCollectionRequest) error); ok { r1 = rf(ctx, req) } else { @@ -4088,8 +3549,8 @@ type MockProxy_RenameCollection_Call struct { } // RenameCollection is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.RenameCollectionRequest +// - ctx context.Context +// - req *milvuspb.RenameCollectionRequest func (_e *MockProxy_Expecter) RenameCollection(ctx interface{}, req interface{}) *MockProxy_RenameCollection_Call { return &MockProxy_RenameCollection_Call{Call: _e.mock.On("RenameCollection", ctx, req)} } @@ -4106,20 +3567,11 @@ func (_c *MockProxy_RenameCollection_Call) Return(_a0 *commonpb.Status, _a1 erro return _c } -func (_c *MockProxy_RenameCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.RenameCollectionRequest) (*commonpb.Status, error)) *MockProxy_RenameCollection_Call { - _c.Call.Return(run) - return _c -} - // Search provides a mock function with given fields: ctx, request func (_m *MockProxy) Search(ctx context.Context, request *milvuspb.SearchRequest) (*milvuspb.SearchResults, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.SearchResults - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SearchRequest) (*milvuspb.SearchResults, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SearchRequest) *milvuspb.SearchResults); ok { r0 = rf(ctx, request) } else { @@ -4128,6 +3580,7 @@ func (_m *MockProxy) Search(ctx context.Context, request *milvuspb.SearchRequest } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SearchRequest) error); ok { r1 = rf(ctx, request) } else { @@ -4143,8 +3596,8 @@ type MockProxy_Search_Call struct { } // Search is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.SearchRequest +// - ctx context.Context +// - request *milvuspb.SearchRequest func (_e *MockProxy_Expecter) Search(ctx interface{}, request interface{}) *MockProxy_Search_Call { return &MockProxy_Search_Call{Call: _e.mock.On("Search", ctx, request)} } @@ -4161,20 +3614,11 @@ func (_c *MockProxy_Search_Call) Return(_a0 *milvuspb.SearchResults, _a1 error) return _c } -func (_c *MockProxy_Search_Call) RunAndReturn(run func(context.Context, *milvuspb.SearchRequest) (*milvuspb.SearchResults, error)) *MockProxy_Search_Call { - _c.Call.Return(run) - return _c -} - // SelectGrant provides a mock function with given fields: ctx, req func (_m *MockProxy) SelectGrant(ctx context.Context, req *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.SelectGrantResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectGrantRequest) *milvuspb.SelectGrantResponse); ok { r0 = rf(ctx, req) } else { @@ -4183,6 +3627,7 @@ func (_m *MockProxy) SelectGrant(ctx context.Context, req *milvuspb.SelectGrantR } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SelectGrantRequest) error); ok { r1 = rf(ctx, req) } else { @@ -4198,8 +3643,8 @@ type MockProxy_SelectGrant_Call struct { } // SelectGrant is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.SelectGrantRequest +// - ctx context.Context +// - req *milvuspb.SelectGrantRequest func (_e *MockProxy_Expecter) SelectGrant(ctx interface{}, req interface{}) *MockProxy_SelectGrant_Call { return &MockProxy_SelectGrant_Call{Call: _e.mock.On("SelectGrant", ctx, req)} } @@ -4216,20 +3661,11 @@ func (_c *MockProxy_SelectGrant_Call) Return(_a0 *milvuspb.SelectGrantResponse, return _c } -func (_c *MockProxy_SelectGrant_Call) RunAndReturn(run func(context.Context, *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error)) *MockProxy_SelectGrant_Call { - _c.Call.Return(run) - return _c -} - // SelectRole provides a mock function with given fields: ctx, req func (_m *MockProxy) SelectRole(ctx context.Context, req *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.SelectRoleResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectRoleRequest) *milvuspb.SelectRoleResponse); ok { r0 = rf(ctx, req) } else { @@ -4238,6 +3674,7 @@ func (_m *MockProxy) SelectRole(ctx context.Context, req *milvuspb.SelectRoleReq } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SelectRoleRequest) error); ok { r1 = rf(ctx, req) } else { @@ -4253,8 +3690,8 @@ type MockProxy_SelectRole_Call struct { } // SelectRole is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.SelectRoleRequest +// - ctx context.Context +// - req *milvuspb.SelectRoleRequest func (_e *MockProxy_Expecter) SelectRole(ctx interface{}, req interface{}) *MockProxy_SelectRole_Call { return &MockProxy_SelectRole_Call{Call: _e.mock.On("SelectRole", ctx, req)} } @@ -4271,20 +3708,11 @@ func (_c *MockProxy_SelectRole_Call) Return(_a0 *milvuspb.SelectRoleResponse, _a return _c } -func (_c *MockProxy_SelectRole_Call) RunAndReturn(run func(context.Context, *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error)) *MockProxy_SelectRole_Call { - _c.Call.Return(run) - return _c -} - // SelectUser provides a mock function with given fields: ctx, req func (_m *MockProxy) SelectUser(ctx context.Context, req *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error) { ret := _m.Called(ctx, req) var r0 *milvuspb.SelectUserResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectUserRequest) *milvuspb.SelectUserResponse); ok { r0 = rf(ctx, req) } else { @@ -4293,6 +3721,7 @@ func (_m *MockProxy) SelectUser(ctx context.Context, req *milvuspb.SelectUserReq } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SelectUserRequest) error); ok { r1 = rf(ctx, req) } else { @@ -4308,8 +3737,8 @@ type MockProxy_SelectUser_Call struct { } // SelectUser is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.SelectUserRequest +// - ctx context.Context +// - req *milvuspb.SelectUserRequest func (_e *MockProxy_Expecter) SelectUser(ctx interface{}, req interface{}) *MockProxy_SelectUser_Call { return &MockProxy_SelectUser_Call{Call: _e.mock.On("SelectUser", ctx, req)} } @@ -4326,11 +3755,6 @@ func (_c *MockProxy_SelectUser_Call) Return(_a0 *milvuspb.SelectUserResponse, _a return _c } -func (_c *MockProxy_SelectUser_Call) RunAndReturn(run func(context.Context, *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error)) *MockProxy_SelectUser_Call { - _c.Call.Return(run) - return _c -} - // SetAddress provides a mock function with given fields: address func (_m *MockProxy) SetAddress(address string) { _m.Called(address) @@ -4342,7 +3766,7 @@ type MockProxy_SetAddress_Call struct { } // SetAddress is a helper method to define mock.On call -// - address string +// - address string func (_e *MockProxy_Expecter) SetAddress(address interface{}) *MockProxy_SetAddress_Call { return &MockProxy_SetAddress_Call{Call: _e.mock.On("SetAddress", address)} } @@ -4359,11 +3783,6 @@ func (_c *MockProxy_SetAddress_Call) Return() *MockProxy_SetAddress_Call { return _c } -func (_c *MockProxy_SetAddress_Call) RunAndReturn(run func(string)) *MockProxy_SetAddress_Call { - _c.Call.Return(run) - return _c -} - // SetDataCoordClient provides a mock function with given fields: dataCoord func (_m *MockProxy) SetDataCoordClient(dataCoord types.DataCoord) { _m.Called(dataCoord) @@ -4375,7 +3794,7 @@ type MockProxy_SetDataCoordClient_Call struct { } // SetDataCoordClient is a helper method to define mock.On call -// - dataCoord types.DataCoord +// - dataCoord types.DataCoord func (_e *MockProxy_Expecter) SetDataCoordClient(dataCoord interface{}) *MockProxy_SetDataCoordClient_Call { return &MockProxy_SetDataCoordClient_Call{Call: _e.mock.On("SetDataCoordClient", dataCoord)} } @@ -4392,11 +3811,6 @@ func (_c *MockProxy_SetDataCoordClient_Call) Return() *MockProxy_SetDataCoordCli return _c } -func (_c *MockProxy_SetDataCoordClient_Call) RunAndReturn(run func(types.DataCoord)) *MockProxy_SetDataCoordClient_Call { - _c.Call.Return(run) - return _c -} - // SetEtcdClient provides a mock function with given fields: etcdClient func (_m *MockProxy) SetEtcdClient(etcdClient *clientv3.Client) { _m.Called(etcdClient) @@ -4408,7 +3822,7 @@ type MockProxy_SetEtcdClient_Call struct { } // SetEtcdClient is a helper method to define mock.On call -// - etcdClient *clientv3.Client +// - etcdClient *clientv3.Client func (_e *MockProxy_Expecter) SetEtcdClient(etcdClient interface{}) *MockProxy_SetEtcdClient_Call { return &MockProxy_SetEtcdClient_Call{Call: _e.mock.On("SetEtcdClient", etcdClient)} } @@ -4425,11 +3839,6 @@ func (_c *MockProxy_SetEtcdClient_Call) Return() *MockProxy_SetEtcdClient_Call { return _c } -func (_c *MockProxy_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockProxy_SetEtcdClient_Call { - _c.Call.Return(run) - return _c -} - // SetQueryCoordClient provides a mock function with given fields: queryCoord func (_m *MockProxy) SetQueryCoordClient(queryCoord types.QueryCoord) { _m.Called(queryCoord) @@ -4441,7 +3850,7 @@ type MockProxy_SetQueryCoordClient_Call struct { } // SetQueryCoordClient is a helper method to define mock.On call -// - queryCoord types.QueryCoord +// - queryCoord types.QueryCoord func (_e *MockProxy_Expecter) SetQueryCoordClient(queryCoord interface{}) *MockProxy_SetQueryCoordClient_Call { return &MockProxy_SetQueryCoordClient_Call{Call: _e.mock.On("SetQueryCoordClient", queryCoord)} } @@ -4458,11 +3867,6 @@ func (_c *MockProxy_SetQueryCoordClient_Call) Return() *MockProxy_SetQueryCoordC return _c } -func (_c *MockProxy_SetQueryCoordClient_Call) RunAndReturn(run func(types.QueryCoord)) *MockProxy_SetQueryCoordClient_Call { - _c.Call.Return(run) - return _c -} - // SetQueryNodeCreator provides a mock function with given fields: _a0 func (_m *MockProxy) SetQueryNodeCreator(_a0 func(context.Context, string) (types.QueryNode, error)) { _m.Called(_a0) @@ -4474,7 +3878,7 @@ type MockProxy_SetQueryNodeCreator_Call struct { } // SetQueryNodeCreator is a helper method to define mock.On call -// - _a0 func(context.Context , string)(types.QueryNode , error) +// - _a0 func(context.Context , string)(types.QueryNode , error) func (_e *MockProxy_Expecter) SetQueryNodeCreator(_a0 interface{}) *MockProxy_SetQueryNodeCreator_Call { return &MockProxy_SetQueryNodeCreator_Call{Call: _e.mock.On("SetQueryNodeCreator", _a0)} } @@ -4491,20 +3895,11 @@ func (_c *MockProxy_SetQueryNodeCreator_Call) Return() *MockProxy_SetQueryNodeCr return _c } -func (_c *MockProxy_SetQueryNodeCreator_Call) RunAndReturn(run func(func(context.Context, string) (types.QueryNode, error))) *MockProxy_SetQueryNodeCreator_Call { - _c.Call.Return(run) - return _c -} - // SetRates provides a mock function with given fields: ctx, req func (_m *MockProxy) SetRates(ctx context.Context, req *proxypb.SetRatesRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *proxypb.SetRatesRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *proxypb.SetRatesRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -4513,6 +3908,7 @@ func (_m *MockProxy) SetRates(ctx context.Context, req *proxypb.SetRatesRequest) } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *proxypb.SetRatesRequest) error); ok { r1 = rf(ctx, req) } else { @@ -4528,8 +3924,8 @@ type MockProxy_SetRates_Call struct { } // SetRates is a helper method to define mock.On call -// - ctx context.Context -// - req *proxypb.SetRatesRequest +// - ctx context.Context +// - req *proxypb.SetRatesRequest func (_e *MockProxy_Expecter) SetRates(ctx interface{}, req interface{}) *MockProxy_SetRates_Call { return &MockProxy_SetRates_Call{Call: _e.mock.On("SetRates", ctx, req)} } @@ -4546,11 +3942,6 @@ func (_c *MockProxy_SetRates_Call) Return(_a0 *commonpb.Status, _a1 error) *Mock return _c } -func (_c *MockProxy_SetRates_Call) RunAndReturn(run func(context.Context, *proxypb.SetRatesRequest) (*commonpb.Status, error)) *MockProxy_SetRates_Call { - _c.Call.Return(run) - return _c -} - // SetRootCoordClient provides a mock function with given fields: rootCoord func (_m *MockProxy) SetRootCoordClient(rootCoord types.RootCoord) { _m.Called(rootCoord) @@ -4562,7 +3953,7 @@ type MockProxy_SetRootCoordClient_Call struct { } // SetRootCoordClient is a helper method to define mock.On call -// - rootCoord types.RootCoord +// - rootCoord types.RootCoord func (_e *MockProxy_Expecter) SetRootCoordClient(rootCoord interface{}) *MockProxy_SetRootCoordClient_Call { return &MockProxy_SetRootCoordClient_Call{Call: _e.mock.On("SetRootCoordClient", rootCoord)} } @@ -4579,20 +3970,11 @@ func (_c *MockProxy_SetRootCoordClient_Call) Return() *MockProxy_SetRootCoordCli return _c } -func (_c *MockProxy_SetRootCoordClient_Call) RunAndReturn(run func(types.RootCoord)) *MockProxy_SetRootCoordClient_Call { - _c.Call.Return(run) - return _c -} - // ShowCollections provides a mock function with given fields: ctx, request func (_m *MockProxy) ShowCollections(ctx context.Context, request *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.ShowCollectionsResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowCollectionsRequest) *milvuspb.ShowCollectionsResponse); ok { r0 = rf(ctx, request) } else { @@ -4601,6 +3983,7 @@ func (_m *MockProxy) ShowCollections(ctx context.Context, request *milvuspb.Show } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ShowCollectionsRequest) error); ok { r1 = rf(ctx, request) } else { @@ -4616,8 +3999,8 @@ type MockProxy_ShowCollections_Call struct { } // ShowCollections is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.ShowCollectionsRequest +// - ctx context.Context +// - request *milvuspb.ShowCollectionsRequest func (_e *MockProxy_Expecter) ShowCollections(ctx interface{}, request interface{}) *MockProxy_ShowCollections_Call { return &MockProxy_ShowCollections_Call{Call: _e.mock.On("ShowCollections", ctx, request)} } @@ -4634,20 +4017,11 @@ func (_c *MockProxy_ShowCollections_Call) Return(_a0 *milvuspb.ShowCollectionsRe return _c } -func (_c *MockProxy_ShowCollections_Call) RunAndReturn(run func(context.Context, *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error)) *MockProxy_ShowCollections_Call { - _c.Call.Return(run) - return _c -} - // ShowPartitions provides a mock function with given fields: ctx, request func (_m *MockProxy) ShowPartitions(ctx context.Context, request *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.ShowPartitionsResponse - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest) *milvuspb.ShowPartitionsResponse); ok { r0 = rf(ctx, request) } else { @@ -4656,6 +4030,7 @@ func (_m *MockProxy) ShowPartitions(ctx context.Context, request *milvuspb.ShowP } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ShowPartitionsRequest) error); ok { r1 = rf(ctx, request) } else { @@ -4671,8 +4046,8 @@ type MockProxy_ShowPartitions_Call struct { } // ShowPartitions is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.ShowPartitionsRequest +// - ctx context.Context +// - request *milvuspb.ShowPartitionsRequest func (_e *MockProxy_Expecter) ShowPartitions(ctx interface{}, request interface{}) *MockProxy_ShowPartitions_Call { return &MockProxy_ShowPartitions_Call{Call: _e.mock.On("ShowPartitions", ctx, request)} } @@ -4689,11 +4064,6 @@ func (_c *MockProxy_ShowPartitions_Call) Return(_a0 *milvuspb.ShowPartitionsResp return _c } -func (_c *MockProxy_ShowPartitions_Call) RunAndReturn(run func(context.Context, *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error)) *MockProxy_ShowPartitions_Call { - _c.Call.Return(run) - return _c -} - // Start provides a mock function with given fields: func (_m *MockProxy) Start() error { ret := _m.Called() @@ -4730,11 +4100,6 @@ func (_c *MockProxy_Start_Call) Return(_a0 error) *MockProxy_Start_Call { return _c } -func (_c *MockProxy_Start_Call) RunAndReturn(run func() error) *MockProxy_Start_Call { - _c.Call.Return(run) - return _c -} - // Stop provides a mock function with given fields: func (_m *MockProxy) Stop() error { ret := _m.Called() @@ -4771,20 +4136,11 @@ func (_c *MockProxy_Stop_Call) Return(_a0 error) *MockProxy_Stop_Call { return _c } -func (_c *MockProxy_Stop_Call) RunAndReturn(run func() error) *MockProxy_Stop_Call { - _c.Call.Return(run) - return _c -} - // TransferNode provides a mock function with given fields: ctx, req func (_m *MockProxy) TransferNode(ctx context.Context, req *milvuspb.TransferNodeRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferNodeRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferNodeRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -4793,6 +4149,7 @@ func (_m *MockProxy) TransferNode(ctx context.Context, req *milvuspb.TransferNod } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.TransferNodeRequest) error); ok { r1 = rf(ctx, req) } else { @@ -4808,8 +4165,8 @@ type MockProxy_TransferNode_Call struct { } // TransferNode is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.TransferNodeRequest +// - ctx context.Context +// - req *milvuspb.TransferNodeRequest func (_e *MockProxy_Expecter) TransferNode(ctx interface{}, req interface{}) *MockProxy_TransferNode_Call { return &MockProxy_TransferNode_Call{Call: _e.mock.On("TransferNode", ctx, req)} } @@ -4826,20 +4183,11 @@ func (_c *MockProxy_TransferNode_Call) Return(_a0 *commonpb.Status, _a1 error) * return _c } -func (_c *MockProxy_TransferNode_Call) RunAndReturn(run func(context.Context, *milvuspb.TransferNodeRequest) (*commonpb.Status, error)) *MockProxy_TransferNode_Call { - _c.Call.Return(run) - return _c -} - // TransferReplica provides a mock function with given fields: ctx, req func (_m *MockProxy) TransferReplica(ctx context.Context, req *milvuspb.TransferReplicaRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferReplicaRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferReplicaRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -4848,6 +4196,7 @@ func (_m *MockProxy) TransferReplica(ctx context.Context, req *milvuspb.Transfer } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.TransferReplicaRequest) error); ok { r1 = rf(ctx, req) } else { @@ -4863,8 +4212,8 @@ type MockProxy_TransferReplica_Call struct { } // TransferReplica is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.TransferReplicaRequest +// - ctx context.Context +// - req *milvuspb.TransferReplicaRequest func (_e *MockProxy_Expecter) TransferReplica(ctx interface{}, req interface{}) *MockProxy_TransferReplica_Call { return &MockProxy_TransferReplica_Call{Call: _e.mock.On("TransferReplica", ctx, req)} } @@ -4881,20 +4230,11 @@ func (_c *MockProxy_TransferReplica_Call) Return(_a0 *commonpb.Status, _a1 error return _c } -func (_c *MockProxy_TransferReplica_Call) RunAndReturn(run func(context.Context, *milvuspb.TransferReplicaRequest) (*commonpb.Status, error)) *MockProxy_TransferReplica_Call { - _c.Call.Return(run) - return _c -} - // UpdateCredential provides a mock function with given fields: ctx, req func (_m *MockProxy) UpdateCredential(ctx context.Context, req *milvuspb.UpdateCredentialRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, req) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpdateCredentialRequest) (*commonpb.Status, error)); ok { - return rf(ctx, req) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpdateCredentialRequest) *commonpb.Status); ok { r0 = rf(ctx, req) } else { @@ -4903,6 +4243,7 @@ func (_m *MockProxy) UpdateCredential(ctx context.Context, req *milvuspb.UpdateC } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.UpdateCredentialRequest) error); ok { r1 = rf(ctx, req) } else { @@ -4918,8 +4259,8 @@ type MockProxy_UpdateCredential_Call struct { } // UpdateCredential is a helper method to define mock.On call -// - ctx context.Context -// - req *milvuspb.UpdateCredentialRequest +// - ctx context.Context +// - req *milvuspb.UpdateCredentialRequest func (_e *MockProxy_Expecter) UpdateCredential(ctx interface{}, req interface{}) *MockProxy_UpdateCredential_Call { return &MockProxy_UpdateCredential_Call{Call: _e.mock.On("UpdateCredential", ctx, req)} } @@ -4936,20 +4277,11 @@ func (_c *MockProxy_UpdateCredential_Call) Return(_a0 *commonpb.Status, _a1 erro return _c } -func (_c *MockProxy_UpdateCredential_Call) RunAndReturn(run func(context.Context, *milvuspb.UpdateCredentialRequest) (*commonpb.Status, error)) *MockProxy_UpdateCredential_Call { - _c.Call.Return(run) - return _c -} - // UpdateCredentialCache provides a mock function with given fields: ctx, request func (_m *MockProxy) UpdateCredentialCache(ctx context.Context, request *proxypb.UpdateCredCacheRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, request) var r0 *commonpb.Status - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *proxypb.UpdateCredCacheRequest) (*commonpb.Status, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *proxypb.UpdateCredCacheRequest) *commonpb.Status); ok { r0 = rf(ctx, request) } else { @@ -4958,6 +4290,7 @@ func (_m *MockProxy) UpdateCredentialCache(ctx context.Context, request *proxypb } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *proxypb.UpdateCredCacheRequest) error); ok { r1 = rf(ctx, request) } else { @@ -4973,8 +4306,8 @@ type MockProxy_UpdateCredentialCache_Call struct { } // UpdateCredentialCache is a helper method to define mock.On call -// - ctx context.Context -// - request *proxypb.UpdateCredCacheRequest +// - ctx context.Context +// - request *proxypb.UpdateCredCacheRequest func (_e *MockProxy_Expecter) UpdateCredentialCache(ctx interface{}, request interface{}) *MockProxy_UpdateCredentialCache_Call { return &MockProxy_UpdateCredentialCache_Call{Call: _e.mock.On("UpdateCredentialCache", ctx, request)} } @@ -4991,11 +4324,6 @@ func (_c *MockProxy_UpdateCredentialCache_Call) Return(_a0 *commonpb.Status, _a1 return _c } -func (_c *MockProxy_UpdateCredentialCache_Call) RunAndReturn(run func(context.Context, *proxypb.UpdateCredCacheRequest) (*commonpb.Status, error)) *MockProxy_UpdateCredentialCache_Call { - _c.Call.Return(run) - return _c -} - // UpdateStateCode provides a mock function with given fields: stateCode func (_m *MockProxy) UpdateStateCode(stateCode commonpb.StateCode) { _m.Called(stateCode) @@ -5007,7 +4335,7 @@ type MockProxy_UpdateStateCode_Call struct { } // UpdateStateCode is a helper method to define mock.On call -// - stateCode commonpb.StateCode +// - stateCode commonpb.StateCode func (_e *MockProxy_Expecter) UpdateStateCode(stateCode interface{}) *MockProxy_UpdateStateCode_Call { return &MockProxy_UpdateStateCode_Call{Call: _e.mock.On("UpdateStateCode", stateCode)} } @@ -5024,20 +4352,11 @@ func (_c *MockProxy_UpdateStateCode_Call) Return() *MockProxy_UpdateStateCode_Ca return _c } -func (_c *MockProxy_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockProxy_UpdateStateCode_Call { - _c.Call.Return(run) - return _c -} - // Upsert provides a mock function with given fields: ctx, request func (_m *MockProxy) Upsert(ctx context.Context, request *milvuspb.UpsertRequest) (*milvuspb.MutationResult, error) { ret := _m.Called(ctx, request) var r0 *milvuspb.MutationResult - var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpsertRequest) (*milvuspb.MutationResult, error)); ok { - return rf(ctx, request) - } if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpsertRequest) *milvuspb.MutationResult); ok { r0 = rf(ctx, request) } else { @@ -5046,6 +4365,7 @@ func (_m *MockProxy) Upsert(ctx context.Context, request *milvuspb.UpsertRequest } } + var r1 error if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.UpsertRequest) error); ok { r1 = rf(ctx, request) } else { @@ -5061,8 +4381,8 @@ type MockProxy_Upsert_Call struct { } // Upsert is a helper method to define mock.On call -// - ctx context.Context -// - request *milvuspb.UpsertRequest +// - ctx context.Context +// - request *milvuspb.UpsertRequest func (_e *MockProxy_Expecter) Upsert(ctx interface{}, request interface{}) *MockProxy_Upsert_Call { return &MockProxy_Upsert_Call{Call: _e.mock.On("Upsert", ctx, request)} } @@ -5079,17 +4399,13 @@ func (_c *MockProxy_Upsert_Call) Return(_a0 *milvuspb.MutationResult, _a1 error) return _c } -func (_c *MockProxy_Upsert_Call) RunAndReturn(run func(context.Context, *milvuspb.UpsertRequest) (*milvuspb.MutationResult, error)) *MockProxy_Upsert_Call { - _c.Call.Return(run) - return _c +type mockConstructorTestingTNewMockProxy interface { + mock.TestingT + Cleanup(func()) } // NewMockProxy creates a new instance of MockProxy. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewMockProxy(t interface { - mock.TestingT - Cleanup(func()) -}) *MockProxy { +func NewMockProxy(t mockConstructorTestingTNewMockProxy) *MockProxy { mock := &MockProxy{} mock.Mock.Test(t) diff --git a/internal/proto/index_coord.proto b/internal/proto/index_coord.proto index 55728ebee7..2b531ecb6c 100644 --- a/internal/proto/index_coord.proto +++ b/internal/proto/index_coord.proto @@ -176,6 +176,7 @@ message DropIndexRequest { message DescribeIndexRequest { int64 collectionID = 1; string index_name = 2; + uint64 timestamp = 3; } message DescribeIndexResponse { diff --git a/internal/proto/indexpb/index_coord.pb.go b/internal/proto/indexpb/index_coord.pb.go index f3631560f1..bb42e95543 100644 --- a/internal/proto/indexpb/index_coord.pb.go +++ b/internal/proto/indexpb/index_coord.pb.go @@ -1169,6 +1169,7 @@ func (m *DropIndexRequest) GetDropAll() bool { type DescribeIndexRequest struct { CollectionID int64 `protobuf:"varint,1,opt,name=collectionID,proto3" json:"collectionID,omitempty"` IndexName string `protobuf:"bytes,2,opt,name=index_name,json=indexName,proto3" json:"index_name,omitempty"` + Timestamp uint64 `protobuf:"varint,3,opt,name=timestamp,proto3" json:"timestamp,omitempty"` XXX_NoUnkeyedLiteral struct{} `json:"-"` XXX_unrecognized []byte `json:"-"` XXX_sizecache int32 `json:"-"` @@ -1213,6 +1214,13 @@ func (m *DescribeIndexRequest) GetIndexName() string { return "" } +func (m *DescribeIndexRequest) GetTimestamp() uint64 { + if m != nil { + return m.Timestamp + } + return 0 +} + type DescribeIndexResponse struct { Status *commonpb.Status `protobuf:"bytes,1,opt,name=status,proto3" json:"status,omitempty"` IndexInfos []*IndexInfo `protobuf:"bytes,2,rep,name=index_infos,json=indexInfos,proto3" json:"index_infos,omitempty"` @@ -2141,148 +2149,149 @@ func init() { func init() { proto.RegisterFile("index_coord.proto", fileDescriptor_f9e019eb3fda53c2) } var fileDescriptor_f9e019eb3fda53c2 = []byte{ - // 2249 bytes of a gzipped FileDescriptorProto - 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe4, 0x59, 0xcd, 0x8f, 0x1b, 0x49, - 0x15, 0x4f, 0xbb, 0x3d, 0x33, 0xee, 0xd7, 0xf6, 0x7c, 0x54, 0xb2, 0xe0, 0x38, 0x09, 0x99, 0x74, - 0x36, 0xc9, 0x80, 0xc8, 0x24, 0xcc, 0xb2, 0x68, 0x41, 0x80, 0x34, 0x99, 0xd9, 0x24, 0x4e, 0x36, - 0xd1, 0xd0, 0x8e, 0x56, 0x62, 0x85, 0x30, 0x6d, 0x77, 0x79, 0xa6, 0x76, 0xda, 0x5d, 0x4e, 0x57, - 0x75, 0x92, 0x09, 0x12, 0x82, 0xc3, 0x1e, 0x40, 0x2b, 0x21, 0x10, 0x12, 0xff, 0x00, 0xa7, 0xe5, - 0x3f, 0xe0, 0xc2, 0x85, 0x23, 0x27, 0xee, 0x5c, 0xf8, 0x27, 0xb8, 0xa2, 0xfa, 0xe8, 0x76, 0x77, - 0xbb, 0x3d, 0xf6, 0x7c, 0x20, 0x24, 0xf6, 0xe6, 0x7a, 0xf5, 0xea, 0xa3, 0xdf, 0xfb, 0xbd, 0xf7, - 0x7b, 0xaf, 0x0c, 0x6b, 0x24, 0xf4, 0xf1, 0x9b, 0x6e, 0x9f, 0xd2, 0xc8, 0xdf, 0x1c, 0x45, 0x94, - 0x53, 0x84, 0x86, 0x24, 0x78, 0x15, 0x33, 0x35, 0xda, 0x94, 0xf3, 0xad, 0x7a, 0x9f, 0x0e, 0x87, - 0x34, 0x54, 0xb2, 0xd6, 0x32, 0x09, 0x39, 0x8e, 0x42, 0x2f, 0xd0, 0xe3, 0x7a, 0x76, 0x85, 0xf3, - 0xcf, 0x2a, 0x58, 0x6d, 0xb1, 0xaa, 0x1d, 0x0e, 0x28, 0x72, 0xa0, 0xde, 0xa7, 0x41, 0x80, 0xfb, - 0x9c, 0xd0, 0xb0, 0xbd, 0xdb, 0x34, 0xd6, 0x8d, 0x0d, 0xd3, 0xcd, 0xc9, 0x50, 0x13, 0x96, 0x06, - 0x04, 0x07, 0x7e, 0x7b, 0xb7, 0x59, 0x91, 0xd3, 0xc9, 0x10, 0x5d, 0x03, 0x50, 0x17, 0x0c, 0xbd, - 0x21, 0x6e, 0x9a, 0xeb, 0xc6, 0x86, 0xe5, 0x5a, 0x52, 0xf2, 0xdc, 0x1b, 0x62, 0xb1, 0x50, 0x0e, - 0xda, 0xbb, 0xcd, 0xaa, 0x5a, 0xa8, 0x87, 0xe8, 0x01, 0xd8, 0xfc, 0x68, 0x84, 0xbb, 0x23, 0x2f, - 0xf2, 0x86, 0xac, 0xb9, 0xb0, 0x6e, 0x6e, 0xd8, 0x5b, 0x37, 0x36, 0x73, 0x9f, 0xa6, 0xbf, 0xe9, - 0x29, 0x3e, 0xfa, 0xd8, 0x0b, 0x62, 0xbc, 0xe7, 0x91, 0xc8, 0x05, 0xb1, 0x6a, 0x4f, 0x2e, 0x42, - 0xbb, 0x50, 0x57, 0x87, 0xeb, 0x4d, 0x16, 0xe7, 0xdd, 0xc4, 0x96, 0xcb, 0xf4, 0x2e, 0x37, 0xf4, - 0x2e, 0xd8, 0xef, 0x46, 0xf4, 0x35, 0x6b, 0x2e, 0xc9, 0x8b, 0xda, 0x5a, 0xe6, 0xd2, 0xd7, 0x4c, - 0x7c, 0x25, 0xa7, 0xdc, 0x0b, 0x94, 0x42, 0x4d, 0x2a, 0x58, 0x52, 0x22, 0xa7, 0xdf, 0x87, 0x05, - 0xc6, 0x3d, 0x8e, 0x9b, 0xd6, 0xba, 0xb1, 0xb1, 0xbc, 0x75, 0xbd, 0xf4, 0x02, 0xd2, 0xe2, 0x1d, - 0xa1, 0xe6, 0x2a, 0x6d, 0xf4, 0x3e, 0x7c, 0x55, 0x5d, 0x5f, 0x0e, 0xbb, 0x03, 0x8f, 0x04, 0xdd, - 0x08, 0x7b, 0x8c, 0x86, 0x4d, 0x90, 0x86, 0xbc, 0x44, 0xd2, 0x35, 0x0f, 0x3d, 0x12, 0xb8, 0x72, - 0x0e, 0x39, 0xd0, 0x20, 0xac, 0xeb, 0xc5, 0x9c, 0x76, 0xe5, 0x7c, 0xd3, 0x5e, 0x37, 0x36, 0x6a, - 0xae, 0x4d, 0xd8, 0x76, 0xcc, 0xa9, 0x3c, 0x06, 0x3d, 0x83, 0xb5, 0x98, 0xe1, 0xa8, 0x9b, 0x33, - 0x4f, 0x7d, 0x5e, 0xf3, 0xac, 0x88, 0xb5, 0xed, 0x8c, 0x89, 0xbe, 0x09, 0x68, 0x84, 0x43, 0x9f, - 0x84, 0xfb, 0x7a, 0x47, 0x69, 0x87, 0x86, 0xb4, 0xc3, 0xaa, 0x9e, 0x91, 0xfa, 0xc2, 0x1c, 0xce, - 0x67, 0x06, 0xc0, 0x43, 0x89, 0x0f, 0x79, 0x97, 0xef, 0x27, 0x10, 0x21, 0xe1, 0x80, 0x4a, 0x78, - 0xd9, 0x5b, 0xd7, 0x36, 0x27, 0x31, 0xbc, 0x99, 0x62, 0x52, 0x23, 0x48, 0xc2, 0xb3, 0x09, 0x4b, - 0x3e, 0x0e, 0x30, 0xc7, 0xbe, 0x84, 0x5e, 0xcd, 0x4d, 0x86, 0xe8, 0x3a, 0xd8, 0xfd, 0x08, 0x0b, - 0xcb, 0x71, 0xa2, 0xb1, 0x57, 0x75, 0x41, 0x89, 0x5e, 0x90, 0x21, 0x76, 0x3e, 0xab, 0x42, 0xbd, - 0x83, 0xf7, 0x87, 0x38, 0xe4, 0xea, 0x26, 0xf3, 0x40, 0x7d, 0x1d, 0xec, 0x91, 0x17, 0x71, 0xa2, - 0x55, 0x14, 0xdc, 0xb3, 0x22, 0x74, 0x15, 0x2c, 0xa6, 0x77, 0xdd, 0x95, 0xa7, 0x9a, 0xee, 0x58, - 0x80, 0x2e, 0x43, 0x2d, 0x8c, 0x87, 0xca, 0x40, 0x1a, 0xf2, 0x61, 0x3c, 0x94, 0x30, 0xc9, 0x04, - 0xc3, 0x42, 0x3e, 0x18, 0x9a, 0xb0, 0xd4, 0x8b, 0x89, 0x8c, 0xaf, 0x45, 0x35, 0xa3, 0x87, 0xe8, - 0x2b, 0xb0, 0x18, 0x52, 0x1f, 0xb7, 0x77, 0x35, 0x2c, 0xf5, 0x08, 0xdd, 0x84, 0x86, 0x32, 0xea, - 0x2b, 0x1c, 0x31, 0x42, 0x43, 0x0d, 0x4a, 0x85, 0xe4, 0x8f, 0x95, 0xec, 0xb4, 0xb8, 0xbc, 0x0e, - 0xf6, 0x24, 0x16, 0x61, 0x30, 0x46, 0xe0, 0x6d, 0x58, 0x51, 0x87, 0x0f, 0x48, 0x80, 0xbb, 0x87, - 0xf8, 0x88, 0x35, 0xed, 0x75, 0x73, 0xc3, 0x72, 0xd5, 0x9d, 0x1e, 0x92, 0x00, 0x3f, 0xc5, 0x47, - 0x2c, 0xeb, 0xbb, 0xfa, 0xb1, 0xbe, 0x6b, 0x14, 0x7d, 0x87, 0x6e, 0xc1, 0x32, 0xc3, 0x11, 0xf1, - 0x02, 0xf2, 0x16, 0x77, 0x19, 0x79, 0x8b, 0x9b, 0xcb, 0x52, 0xa7, 0x91, 0x4a, 0x3b, 0xe4, 0x2d, - 0x16, 0x66, 0x78, 0x1d, 0x11, 0x8e, 0xbb, 0x07, 0x5e, 0xe8, 0xd3, 0xc1, 0xa0, 0xb9, 0x22, 0xcf, - 0xa9, 0x4b, 0xe1, 0x63, 0x25, 0x73, 0xfe, 0x68, 0xc0, 0x45, 0x17, 0xef, 0x13, 0xc6, 0x71, 0xf4, - 0x9c, 0xfa, 0xd8, 0xc5, 0x2f, 0x63, 0xcc, 0x38, 0xba, 0x0f, 0xd5, 0x9e, 0xc7, 0xb0, 0x86, 0xe4, - 0xd5, 0x52, 0xeb, 0x3c, 0x63, 0xfb, 0x0f, 0x3c, 0x86, 0x5d, 0xa9, 0x89, 0xbe, 0x03, 0x4b, 0x9e, - 0xef, 0x47, 0x98, 0x31, 0x09, 0x8c, 0x69, 0x8b, 0xb6, 0x95, 0x8e, 0x9b, 0x28, 0x67, 0xbc, 0x68, - 0x66, 0xbd, 0xe8, 0xfc, 0xd6, 0x80, 0x4b, 0xf9, 0x9b, 0xb1, 0x11, 0x0d, 0x19, 0x46, 0xef, 0xc1, - 0xa2, 0xf0, 0x45, 0xcc, 0xf4, 0xe5, 0xae, 0x94, 0x9e, 0xd3, 0x91, 0x2a, 0xae, 0x56, 0x15, 0x29, - 0x95, 0x84, 0x84, 0x27, 0xe1, 0xae, 0x6e, 0x78, 0xa3, 0x18, 0x69, 0x9a, 0x18, 0xda, 0x21, 0xe1, - 0x2a, 0xba, 0x5d, 0x20, 0xe9, 0x6f, 0xe7, 0xc7, 0x70, 0xe9, 0x11, 0xe6, 0x19, 0x4c, 0x68, 0x5b, - 0xcd, 0x13, 0x3a, 0x79, 0x2e, 0xa8, 0x14, 0xb8, 0xc0, 0xf9, 0x93, 0x01, 0xef, 0x14, 0xf6, 0x3e, - 0xcb, 0xd7, 0xa6, 0xe0, 0xae, 0x9c, 0x05, 0xdc, 0x66, 0x11, 0xdc, 0xce, 0x2f, 0x0d, 0xb8, 0xf2, - 0x08, 0xf3, 0x6c, 0xe2, 0x38, 0x67, 0x4b, 0xa0, 0xaf, 0x01, 0xa4, 0x09, 0x83, 0x35, 0xcd, 0x75, - 0x73, 0xc3, 0x74, 0x33, 0x12, 0xe7, 0xd7, 0x06, 0xac, 0x4d, 0x9c, 0x9f, 0xcf, 0x3b, 0x46, 0x31, - 0xef, 0xfc, 0xb7, 0xcc, 0xf1, 0x7b, 0x03, 0xae, 0x96, 0x9b, 0xe3, 0x2c, 0xce, 0xfb, 0x81, 0x5a, - 0x84, 0x05, 0x4a, 0x05, 0x29, 0xdd, 0x2a, 0xe3, 0x83, 0xc9, 0x33, 0xf5, 0x22, 0xe7, 0x73, 0x13, - 0xd0, 0x8e, 0x4c, 0x16, 0x8a, 0x75, 0x4e, 0xe0, 0x9a, 0x53, 0x97, 0x32, 0x85, 0x82, 0xa5, 0x7a, - 0x1e, 0x05, 0xcb, 0xc2, 0xa9, 0x0a, 0x96, 0xab, 0x60, 0x89, 0xac, 0xc9, 0xb8, 0x37, 0x1c, 0x49, - 0xbe, 0xa8, 0xba, 0x63, 0xc1, 0x64, 0x79, 0xb0, 0x34, 0x67, 0x79, 0x50, 0x3b, 0x6d, 0x79, 0xe0, - 0xbc, 0x81, 0x8b, 0x49, 0x60, 0x4b, 0xfa, 0x3e, 0x81, 0x3b, 0xf2, 0xa1, 0x50, 0x29, 0x86, 0xc2, - 0x0c, 0xa7, 0x38, 0xff, 0xae, 0xc0, 0x5a, 0x3b, 0xe1, 0x9c, 0x3d, 0x8f, 0x1f, 0xc8, 0x9a, 0xe1, - 0xf8, 0x48, 0x99, 0x8e, 0x80, 0x0c, 0x41, 0x9b, 0x53, 0x09, 0xba, 0x9a, 0x27, 0xe8, 0xfc, 0x05, - 0x17, 0x8a, 0xa8, 0x39, 0x9f, 0x12, 0x75, 0x03, 0x56, 0x33, 0x84, 0x3b, 0xf2, 0xf8, 0x81, 0x28, - 0x53, 0x05, 0xe3, 0x2e, 0x93, 0xec, 0xd7, 0x33, 0x74, 0x07, 0x56, 0x52, 0x86, 0xf4, 0x15, 0x71, - 0xd6, 0x24, 0x42, 0xc6, 0x74, 0xea, 0x27, 0xcc, 0x99, 0x2f, 0x20, 0xac, 0x92, 0x02, 0x22, 0x5b, - 0xcc, 0x40, 0xae, 0x98, 0x71, 0xfe, 0x62, 0x80, 0x9d, 0x06, 0xe8, 0x9c, 0x6d, 0x44, 0xce, 0x2f, - 0x95, 0xa2, 0x5f, 0x6e, 0x40, 0x1d, 0x87, 0x5e, 0x2f, 0xc0, 0x1a, 0xb7, 0xa6, 0xc2, 0xad, 0x92, - 0x29, 0xdc, 0x3e, 0x04, 0x7b, 0x5c, 0x4a, 0x26, 0x31, 0x78, 0x6b, 0x6a, 0x2d, 0x99, 0x05, 0x85, - 0x0b, 0x69, 0x4d, 0xc9, 0x9c, 0xdf, 0x54, 0xc6, 0x34, 0xa7, 0x10, 0x7b, 0x96, 0x64, 0xf6, 0x13, - 0xa8, 0xeb, 0xaf, 0x50, 0x25, 0xae, 0x4a, 0x69, 0xdf, 0x2d, 0xbb, 0x56, 0xd9, 0xa1, 0x9b, 0x19, - 0x33, 0x7e, 0x18, 0xf2, 0xe8, 0xc8, 0xb5, 0xd9, 0x58, 0xd2, 0xea, 0xc2, 0x6a, 0x51, 0x01, 0xad, - 0x82, 0x79, 0x88, 0x8f, 0xb4, 0x8d, 0xc5, 0x4f, 0x91, 0xfe, 0x5f, 0x09, 0xec, 0x68, 0xd6, 0xbf, - 0x7e, 0x6c, 0x3e, 0x1d, 0x50, 0x57, 0x69, 0x7f, 0xaf, 0xf2, 0x81, 0xe1, 0xfc, 0xc1, 0x80, 0xd5, - 0xdd, 0x88, 0x8e, 0x4e, 0x9c, 0x4a, 0x1d, 0xa8, 0x67, 0xea, 0xe2, 0x24, 0x7a, 0x73, 0xb2, 0x59, - 0x49, 0xf5, 0x32, 0xd4, 0xfc, 0x88, 0x8e, 0xba, 0x5e, 0x10, 0xc8, 0xc0, 0x12, 0x25, 0x62, 0x44, - 0x47, 0xdb, 0x41, 0x20, 0x2a, 0x91, 0x5d, 0xcc, 0xfa, 0x11, 0xe9, 0x9d, 0x3c, 0xc9, 0xcf, 0xa8, - 0x44, 0x3e, 0x37, 0xe0, 0x9d, 0xc2, 0xde, 0x67, 0xf1, 0xff, 0x0f, 0xf3, 0xa8, 0x54, 0xee, 0x9f, - 0xd1, 0xe1, 0x64, 0xd1, 0xe8, 0x49, 0x86, 0x95, 0x73, 0x0f, 0x44, 0x56, 0xd9, 0x8b, 0xe8, 0xbe, - 0xac, 0x1f, 0xcf, 0xef, 0x8b, 0xff, 0x66, 0xc0, 0xb5, 0x29, 0x67, 0x9c, 0xe5, 0xcb, 0x8b, 0xad, - 0x73, 0x65, 0x56, 0xeb, 0x6c, 0x16, 0x5b, 0xe7, 0xf2, 0xce, 0xb2, 0x3a, 0xa5, 0xb3, 0xfc, 0x73, - 0x05, 0x1a, 0x1d, 0x4e, 0x23, 0x6f, 0x1f, 0xef, 0xd0, 0x70, 0x40, 0xf6, 0x45, 0x62, 0x4e, 0x2a, - 0x72, 0x43, 0x7e, 0x74, 0x5a, 0x73, 0xdf, 0x80, 0xba, 0xd7, 0xef, 0x63, 0xc6, 0x44, 0x83, 0xa2, - 0xf3, 0x8d, 0xe5, 0xda, 0x4a, 0xf6, 0x54, 0x88, 0xd0, 0x37, 0x60, 0x8d, 0xe1, 0x7e, 0x84, 0x79, - 0x77, 0xac, 0xa9, 0x31, 0xba, 0xa2, 0x26, 0xb6, 0x13, 0x6d, 0x51, 0xc2, 0xc7, 0x0c, 0x77, 0x3a, - 0x1f, 0x69, 0x9c, 0xea, 0x91, 0x28, 0xa0, 0x7a, 0x71, 0xff, 0x10, 0xf3, 0x2c, 0x01, 0x80, 0x12, - 0x49, 0x88, 0x5f, 0x01, 0x2b, 0xa2, 0x94, 0xcb, 0xac, 0x2d, 0xd9, 0xda, 0x72, 0x6b, 0x42, 0x20, - 0x12, 0x93, 0xde, 0xb5, 0xbd, 0xfd, 0x4c, 0xb3, 0xb4, 0x1e, 0x89, 0x2e, 0xb4, 0xbd, 0xfd, 0xec, - 0xc3, 0xd0, 0x1f, 0x51, 0x12, 0x72, 0x99, 0xc2, 0x2d, 0x37, 0x2b, 0x12, 0x9f, 0xc7, 0x94, 0x25, - 0xba, 0xa2, 0xc0, 0x90, 0xe9, 0xdb, 0x72, 0x6d, 0x2d, 0x7b, 0x71, 0x34, 0xc2, 0xce, 0xbf, 0x4c, - 0x58, 0x55, 0x55, 0xd2, 0x13, 0xda, 0x4b, 0xc0, 0x74, 0x15, 0xac, 0x7e, 0x10, 0x8b, 0x86, 0x43, - 0x23, 0xc9, 0x72, 0xc7, 0x02, 0x61, 0x91, 0x2c, 0xd1, 0x44, 0x78, 0x40, 0xde, 0x68, 0xcb, 0xad, - 0x8c, 0x99, 0x46, 0x8a, 0xb3, 0x9c, 0x68, 0x4e, 0x70, 0xa2, 0xef, 0x71, 0x4f, 0x13, 0x55, 0x55, - 0x12, 0x95, 0x25, 0x24, 0x8a, 0xa3, 0x26, 0xa8, 0x67, 0xa1, 0x84, 0x7a, 0x32, 0x5c, 0xbc, 0x98, - 0xe7, 0xe2, 0x3c, 0xd4, 0x97, 0x8a, 0x29, 0xe5, 0x31, 0x2c, 0x27, 0x86, 0xe9, 0x4b, 0x8c, 0x48, - 0xeb, 0x95, 0x34, 0x42, 0x32, 0x25, 0x66, 0xc1, 0xe4, 0x36, 0x58, 0x0e, 0x5b, 0x45, 0xee, 0xb6, - 0x4e, 0xc5, 0xdd, 0x85, 0xba, 0x11, 0x4e, 0x53, 0x37, 0x66, 0x79, 0xd8, 0xce, 0xf3, 0xf0, 0x47, - 0xb0, 0xfa, 0xa3, 0x18, 0x47, 0x47, 0x4f, 0x68, 0x8f, 0xcd, 0xe7, 0xe3, 0x16, 0xd4, 0xb4, 0xa3, - 0x92, 0x94, 0x9d, 0x8e, 0x9d, 0x7f, 0x18, 0xd0, 0x90, 0xe1, 0xf6, 0xc2, 0x63, 0x87, 0xc9, 0xfb, - 0x4b, 0xe2, 0x65, 0x23, 0xef, 0xe5, 0x53, 0x76, 0x1c, 0x25, 0x8f, 0x07, 0x66, 0xd9, 0xe3, 0x41, - 0x49, 0x25, 0x53, 0x2d, 0xad, 0x64, 0x0a, 0x2d, 0xcc, 0xc2, 0x44, 0x0b, 0xf3, 0x85, 0x01, 0x6b, - 0x19, 0x1b, 0x9d, 0x25, 0xe1, 0xe5, 0x2c, 0x5b, 0x29, 0x5a, 0xf6, 0x41, 0x9e, 0x08, 0xcc, 0x32, - 0x57, 0x67, 0x88, 0x20, 0xb1, 0x71, 0x8e, 0x0c, 0x9e, 0xc2, 0x8a, 0x20, 0xe3, 0xf3, 0x71, 0xe7, - 0xdf, 0x0d, 0x58, 0x7a, 0x42, 0x7b, 0xd2, 0x91, 0x59, 0x0c, 0x19, 0xf9, 0x87, 0xa9, 0x55, 0x30, - 0x7d, 0x32, 0xd4, 0xd9, 0x5b, 0xfc, 0x14, 0x31, 0xc6, 0xb8, 0x17, 0xf1, 0xf1, 0xd3, 0x9a, 0x28, - 0xd5, 0x84, 0x44, 0xbe, 0xce, 0x5c, 0x86, 0x1a, 0x0e, 0x7d, 0x35, 0xa9, 0xeb, 0x61, 0x1c, 0xfa, - 0x72, 0xea, 0x7c, 0x5a, 0x9c, 0x4b, 0xb0, 0x30, 0xa2, 0xe3, 0xe7, 0x30, 0x35, 0x70, 0x2e, 0x01, - 0x7a, 0x84, 0xf9, 0x13, 0xda, 0x13, 0x5e, 0x49, 0xcc, 0xe3, 0xfc, 0xb5, 0x22, 0xdb, 0x8f, 0xb1, - 0xf8, 0x2c, 0x0e, 0x76, 0xa0, 0xa1, 0xe8, 0xea, 0x53, 0xda, 0xeb, 0x86, 0x71, 0x62, 0x14, 0x5b, - 0x0a, 0x9f, 0xd0, 0xde, 0xf3, 0x78, 0x88, 0xee, 0xc2, 0x45, 0x12, 0x76, 0x47, 0x9a, 0x41, 0x53, - 0x4d, 0x65, 0xa5, 0x55, 0x12, 0x26, 0xdc, 0xaa, 0xd5, 0x6f, 0xc3, 0x0a, 0x0e, 0x5f, 0xc6, 0x38, - 0xc6, 0xa9, 0xaa, 0xb2, 0x59, 0x43, 0x8b, 0xb5, 0x9e, 0x60, 0x4a, 0x8f, 0x1d, 0x76, 0x59, 0x40, - 0x39, 0xd3, 0x39, 0xd1, 0x12, 0x92, 0x8e, 0x10, 0xa0, 0x0f, 0xc0, 0x12, 0xcb, 0x15, 0xb4, 0x54, - 0x1b, 0x71, 0xa5, 0x0c, 0x5a, 0xda, 0xdf, 0x6e, 0xed, 0x53, 0xf5, 0x83, 0x89, 0x00, 0xd1, 0x85, - 0xb5, 0x4f, 0xd8, 0xa1, 0x66, 0x1a, 0x50, 0xa2, 0x5d, 0xc2, 0x0e, 0x9d, 0x9f, 0xc2, 0xe5, 0xec, - 0xc3, 0x0c, 0x61, 0x9c, 0xf4, 0xcf, 0xb3, 0xfa, 0xf8, 0x9d, 0x01, 0xad, 0xb2, 0x03, 0xfe, 0x87, - 0x45, 0xd7, 0xd6, 0xaf, 0x6c, 0x00, 0x39, 0xb3, 0x43, 0x69, 0xe4, 0xa3, 0x40, 0x42, 0x6b, 0x87, - 0x0e, 0x47, 0x34, 0xc4, 0x21, 0x97, 0x19, 0x8b, 0xa1, 0xcd, 0xfc, 0x7e, 0x7a, 0x30, 0xa9, 0xa8, - 0x6d, 0xd5, 0x7a, 0xb7, 0x54, 0xbf, 0xa0, 0xec, 0x5c, 0x40, 0x2f, 0x65, 0xfb, 0x31, 0x36, 0xc5, - 0xce, 0x81, 0x17, 0x86, 0x38, 0x40, 0x5b, 0x53, 0x1e, 0xeb, 0xca, 0x94, 0x93, 0x33, 0x6f, 0x96, - 0x9e, 0xd9, 0xe1, 0x11, 0x09, 0xf7, 0x13, 0x13, 0x3b, 0x17, 0xd0, 0x0b, 0xb0, 0x33, 0x2f, 0x26, - 0xe8, 0x76, 0x99, 0xa5, 0x26, 0x9f, 0x54, 0x5a, 0xc7, 0xf9, 0xc2, 0xb9, 0x80, 0x06, 0xd0, 0xc8, - 0x3d, 0xe9, 0xa1, 0x8d, 0xe3, 0xba, 0x9e, 0xec, 0x3b, 0x5a, 0xeb, 0xeb, 0x73, 0x68, 0xa6, 0xb7, - 0xff, 0xb9, 0x32, 0xd8, 0xc4, 0x9b, 0xd8, 0xbd, 0x29, 0x9b, 0x4c, 0x7b, 0xbd, 0x6b, 0xdd, 0x9f, - 0x7f, 0x41, 0x7a, 0xb8, 0x3f, 0xfe, 0x48, 0x15, 0x50, 0x77, 0x66, 0xb7, 0x76, 0xea, 0xb4, 0x8d, - 0x79, 0x7b, 0x40, 0xe7, 0x02, 0xda, 0x03, 0x2b, 0xed, 0xc2, 0xd0, 0xbb, 0x65, 0x0b, 0x8b, 0x4d, - 0xda, 0x1c, 0xce, 0xc9, 0x75, 0x39, 0xe5, 0xce, 0x29, 0x6b, 0xb2, 0xca, 0x9d, 0x53, 0xda, 0x32, - 0x39, 0x17, 0x50, 0x2c, 0x63, 0xa7, 0x10, 0xdd, 0xe8, 0xee, 0x2c, 0xff, 0xe6, 0xd2, 0x4c, 0x6b, - 0x73, 0x5e, 0xf5, 0xf4, 0xd8, 0x5f, 0x8c, 0x9f, 0x93, 0x73, 0x2d, 0x0d, 0xba, 0x7f, 0xdc, 0x56, - 0x65, 0x1d, 0x56, 0xeb, 0x5b, 0x27, 0x58, 0x91, 0xc1, 0x24, 0xea, 0x1c, 0xd0, 0xd7, 0xaa, 0x58, - 0x8c, 0x23, 0x4f, 0xe4, 0xc2, 0x92, 0xc3, 0x75, 0x08, 0x4f, 0xaa, 0x4e, 0x3d, 0xfc, 0x98, 0x15, - 0xe9, 0xe1, 0x5d, 0x80, 0x47, 0x98, 0x3f, 0xc3, 0x3c, 0x12, 0xb6, 0xbe, 0x3d, 0x2d, 0x4f, 0x69, - 0x85, 0xe4, 0xa8, 0x3b, 0x33, 0xf5, 0xd2, 0x03, 0x7a, 0x60, 0xef, 0x1c, 0xe0, 0xfe, 0xe1, 0x63, - 0xec, 0x05, 0xfc, 0x00, 0x95, 0xaf, 0xcc, 0x68, 0x4c, 0x81, 0x7c, 0x99, 0x62, 0x72, 0xc6, 0xd6, - 0x17, 0x8b, 0xfa, 0x8f, 0xe8, 0xe7, 0xd4, 0xc7, 0xff, 0xff, 0x29, 0x78, 0x0f, 0xac, 0xb4, 0x1d, - 0x2b, 0x8f, 0xf0, 0x62, 0xb7, 0x36, 0x2b, 0xc2, 0x3f, 0x01, 0x2b, 0x2d, 0x6c, 0xcb, 0x77, 0x2c, - 0xf6, 0x06, 0xad, 0x5b, 0x33, 0xb4, 0xd2, 0xdb, 0x3e, 0x87, 0x5a, 0x52, 0x88, 0xa2, 0x9b, 0xd3, - 0xd2, 0x51, 0x76, 0xe7, 0x19, 0x77, 0xfd, 0x19, 0xd8, 0x99, 0x2a, 0xad, 0x9c, 0x80, 0x26, 0xab, - 0xbb, 0xd6, 0x9d, 0x99, 0x7a, 0x5f, 0x8e, 0x80, 0x7c, 0xf0, 0xed, 0x4f, 0xb6, 0xf6, 0x09, 0x3f, - 0x88, 0x7b, 0xc2, 0xb2, 0xf7, 0x94, 0xe6, 0x5d, 0x42, 0xf5, 0xaf, 0x7b, 0xc9, 0x2d, 0xef, 0xc9, - 0x9d, 0xee, 0x49, 0x3b, 0x8d, 0x7a, 0xbd, 0x45, 0x39, 0x7c, 0xef, 0x3f, 0x01, 0x00, 0x00, 0xff, - 0xff, 0x65, 0x43, 0xa0, 0xdd, 0x47, 0x22, 0x00, 0x00, + // 2257 bytes of a gzipped FileDescriptorProto + 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe4, 0x59, 0x4f, 0x8f, 0x1b, 0x49, + 0x15, 0x4f, 0xbb, 0x3d, 0x33, 0xee, 0xd7, 0xf6, 0xfc, 0xa9, 0xcc, 0x82, 0xe3, 0x24, 0x64, 0xd2, + 0xd9, 0x24, 0x03, 0x22, 0x93, 0x30, 0xcb, 0xa2, 0x05, 0x01, 0xd2, 0x64, 0x66, 0x93, 0x38, 0xd9, + 0x44, 0x43, 0x3b, 0x5a, 0x89, 0x15, 0xc2, 0xb4, 0xdd, 0xe5, 0x99, 0xda, 0x69, 0x77, 0x39, 0x5d, + 0xd5, 0x49, 0x26, 0x48, 0x08, 0x0e, 0x7b, 0x00, 0xad, 0x84, 0x40, 0x48, 0x7c, 0x01, 0x4e, 0xcb, + 0x37, 0xe0, 0xc2, 0x85, 0x23, 0x27, 0xee, 0x5c, 0xf8, 0x12, 0x5c, 0x51, 0xfd, 0xe9, 0x76, 0x77, + 0xbb, 0x3d, 0x76, 0x66, 0x06, 0x21, 0xb1, 0x37, 0xd7, 0xab, 0x57, 0x7f, 0xfa, 0xbd, 0xdf, 0x7b, + 0xbf, 0xf7, 0xca, 0xb0, 0x46, 0x42, 0x1f, 0xbf, 0xee, 0xf6, 0x29, 0x8d, 0xfc, 0xad, 0x51, 0x44, + 0x39, 0x45, 0x68, 0x48, 0x82, 0x97, 0x31, 0x53, 0xa3, 0x2d, 0x39, 0xdf, 0xaa, 0xf7, 0xe9, 0x70, + 0x48, 0x43, 0x25, 0x6b, 0x2d, 0x93, 0x90, 0xe3, 0x28, 0xf4, 0x02, 0x3d, 0xae, 0x67, 0x57, 0x38, + 0xff, 0xac, 0x82, 0xd5, 0x16, 0xab, 0xda, 0xe1, 0x80, 0x22, 0x07, 0xea, 0x7d, 0x1a, 0x04, 0xb8, + 0xcf, 0x09, 0x0d, 0xdb, 0x7b, 0x4d, 0x63, 0xc3, 0xd8, 0x34, 0xdd, 0x9c, 0x0c, 0x35, 0x61, 0x69, + 0x40, 0x70, 0xe0, 0xb7, 0xf7, 0x9a, 0x15, 0x39, 0x9d, 0x0c, 0xd1, 0x55, 0x00, 0x75, 0xc1, 0xd0, + 0x1b, 0xe2, 0xa6, 0xb9, 0x61, 0x6c, 0x5a, 0xae, 0x25, 0x25, 0xcf, 0xbc, 0x21, 0x16, 0x0b, 0xe5, + 0xa0, 0xbd, 0xd7, 0xac, 0xaa, 0x85, 0x7a, 0x88, 0xee, 0x83, 0xcd, 0x8f, 0x47, 0xb8, 0x3b, 0xf2, + 0x22, 0x6f, 0xc8, 0x9a, 0x0b, 0x1b, 0xe6, 0xa6, 0xbd, 0x7d, 0x7d, 0x2b, 0xf7, 0x69, 0xfa, 0x9b, + 0x9e, 0xe0, 0xe3, 0x8f, 0xbd, 0x20, 0xc6, 0xfb, 0x1e, 0x89, 0x5c, 0x10, 0xab, 0xf6, 0xe5, 0x22, + 0xb4, 0x07, 0x75, 0x75, 0xb8, 0xde, 0x64, 0x71, 0xde, 0x4d, 0x6c, 0xb9, 0x4c, 0xef, 0x72, 0x5d, + 0xef, 0x82, 0xfd, 0x6e, 0x44, 0x5f, 0xb1, 0xe6, 0x92, 0xbc, 0xa8, 0xad, 0x65, 0x2e, 0x7d, 0xc5, + 0xc4, 0x57, 0x72, 0xca, 0xbd, 0x40, 0x29, 0xd4, 0xa4, 0x82, 0x25, 0x25, 0x72, 0xfa, 0x7d, 0x58, + 0x60, 0xdc, 0xe3, 0xb8, 0x69, 0x6d, 0x18, 0x9b, 0xcb, 0xdb, 0xd7, 0x4a, 0x2f, 0x20, 0x2d, 0xde, + 0x11, 0x6a, 0xae, 0xd2, 0x46, 0xef, 0xc3, 0x57, 0xd5, 0xf5, 0xe5, 0xb0, 0x3b, 0xf0, 0x48, 0xd0, + 0x8d, 0xb0, 0xc7, 0x68, 0xd8, 0x04, 0x69, 0xc8, 0x75, 0x92, 0xae, 0x79, 0xe0, 0x91, 0xc0, 0x95, + 0x73, 0xc8, 0x81, 0x06, 0x61, 0x5d, 0x2f, 0xe6, 0xb4, 0x2b, 0xe7, 0x9b, 0xf6, 0x86, 0xb1, 0x59, + 0x73, 0x6d, 0xc2, 0x76, 0x62, 0x4e, 0xe5, 0x31, 0xe8, 0x29, 0xac, 0xc5, 0x0c, 0x47, 0xdd, 0x9c, + 0x79, 0xea, 0xf3, 0x9a, 0x67, 0x45, 0xac, 0x6d, 0x67, 0x4c, 0xf4, 0x4d, 0x40, 0x23, 0x1c, 0xfa, + 0x24, 0x3c, 0xd0, 0x3b, 0x4a, 0x3b, 0x34, 0xa4, 0x1d, 0x56, 0xf5, 0x8c, 0xd4, 0x17, 0xe6, 0x70, + 0x3e, 0x33, 0x00, 0x1e, 0x48, 0x7c, 0xc8, 0xbb, 0x7c, 0x3f, 0x81, 0x08, 0x09, 0x07, 0x54, 0xc2, + 0xcb, 0xde, 0xbe, 0xba, 0x35, 0x89, 0xe1, 0xad, 0x14, 0x93, 0x1a, 0x41, 0x12, 0x9e, 0x4d, 0x58, + 0xf2, 0x71, 0x80, 0x39, 0xf6, 0x25, 0xf4, 0x6a, 0x6e, 0x32, 0x44, 0xd7, 0xc0, 0xee, 0x47, 0x58, + 0x58, 0x8e, 0x13, 0x8d, 0xbd, 0xaa, 0x0b, 0x4a, 0xf4, 0x9c, 0x0c, 0xb1, 0xf3, 0x59, 0x15, 0xea, + 0x1d, 0x7c, 0x30, 0xc4, 0x21, 0x57, 0x37, 0x99, 0x07, 0xea, 0x1b, 0x60, 0x8f, 0xbc, 0x88, 0x13, + 0xad, 0xa2, 0xe0, 0x9e, 0x15, 0xa1, 0x2b, 0x60, 0x31, 0xbd, 0xeb, 0x9e, 0x3c, 0xd5, 0x74, 0xc7, + 0x02, 0x74, 0x09, 0x6a, 0x61, 0x3c, 0x54, 0x06, 0xd2, 0x90, 0x0f, 0xe3, 0xa1, 0x84, 0x49, 0x26, + 0x18, 0x16, 0xf2, 0xc1, 0xd0, 0x84, 0xa5, 0x5e, 0x4c, 0x64, 0x7c, 0x2d, 0xaa, 0x19, 0x3d, 0x44, + 0x5f, 0x81, 0xc5, 0x90, 0xfa, 0xb8, 0xbd, 0xa7, 0x61, 0xa9, 0x47, 0xe8, 0x06, 0x34, 0x94, 0x51, + 0x5f, 0xe2, 0x88, 0x11, 0x1a, 0x6a, 0x50, 0x2a, 0x24, 0x7f, 0xac, 0x64, 0xa7, 0xc5, 0xe5, 0x35, + 0xb0, 0x27, 0xb1, 0x08, 0x83, 0x31, 0x02, 0x6f, 0xc1, 0x8a, 0x3a, 0x7c, 0x40, 0x02, 0xdc, 0x3d, + 0xc2, 0xc7, 0xac, 0x69, 0x6f, 0x98, 0x9b, 0x96, 0xab, 0xee, 0xf4, 0x80, 0x04, 0xf8, 0x09, 0x3e, + 0x66, 0x59, 0xdf, 0xd5, 0x4f, 0xf4, 0x5d, 0xa3, 0xe8, 0x3b, 0x74, 0x13, 0x96, 0x19, 0x8e, 0x88, + 0x17, 0x90, 0x37, 0xb8, 0xcb, 0xc8, 0x1b, 0xdc, 0x5c, 0x96, 0x3a, 0x8d, 0x54, 0xda, 0x21, 0x6f, + 0xb0, 0x30, 0xc3, 0xab, 0x88, 0x70, 0xdc, 0x3d, 0xf4, 0x42, 0x9f, 0x0e, 0x06, 0xcd, 0x15, 0x79, + 0x4e, 0x5d, 0x0a, 0x1f, 0x29, 0x99, 0xf3, 0x47, 0x03, 0x2e, 0xba, 0xf8, 0x80, 0x30, 0x8e, 0xa3, + 0x67, 0xd4, 0xc7, 0x2e, 0x7e, 0x11, 0x63, 0xc6, 0xd1, 0x3d, 0xa8, 0xf6, 0x3c, 0x86, 0x35, 0x24, + 0xaf, 0x94, 0x5a, 0xe7, 0x29, 0x3b, 0xb8, 0xef, 0x31, 0xec, 0x4a, 0x4d, 0xf4, 0x1d, 0x58, 0xf2, + 0x7c, 0x3f, 0xc2, 0x8c, 0x49, 0x60, 0x4c, 0x5b, 0xb4, 0xa3, 0x74, 0xdc, 0x44, 0x39, 0xe3, 0x45, + 0x33, 0xeb, 0x45, 0xe7, 0xb7, 0x06, 0xac, 0xe7, 0x6f, 0xc6, 0x46, 0x34, 0x64, 0x18, 0xbd, 0x07, + 0x8b, 0xc2, 0x17, 0x31, 0xd3, 0x97, 0xbb, 0x5c, 0x7a, 0x4e, 0x47, 0xaa, 0xb8, 0x5a, 0x55, 0xa4, + 0x54, 0x12, 0x12, 0x9e, 0x84, 0xbb, 0xba, 0xe1, 0xf5, 0x62, 0xa4, 0x69, 0x62, 0x68, 0x87, 0x84, + 0xab, 0xe8, 0x76, 0x81, 0xa4, 0xbf, 0x9d, 0x1f, 0xc3, 0xfa, 0x43, 0xcc, 0x33, 0x98, 0xd0, 0xb6, + 0x9a, 0x27, 0x74, 0xf2, 0x5c, 0x50, 0x29, 0x70, 0x81, 0xf3, 0x27, 0x03, 0xde, 0x29, 0xec, 0x7d, + 0x96, 0xaf, 0x4d, 0xc1, 0x5d, 0x39, 0x0b, 0xb8, 0xcd, 0x22, 0xb8, 0x9d, 0x5f, 0x1a, 0x70, 0xf9, + 0x21, 0xe6, 0xd9, 0xc4, 0x71, 0xce, 0x96, 0x40, 0x5f, 0x03, 0x48, 0x13, 0x06, 0x6b, 0x9a, 0x1b, + 0xe6, 0xa6, 0xe9, 0x66, 0x24, 0xce, 0xaf, 0x0d, 0x58, 0x9b, 0x38, 0x3f, 0x9f, 0x77, 0x8c, 0x62, + 0xde, 0xf9, 0x6f, 0x99, 0xe3, 0xf7, 0x06, 0x5c, 0x29, 0x37, 0xc7, 0x59, 0x9c, 0xf7, 0x03, 0xb5, + 0x08, 0x0b, 0x94, 0x0a, 0x52, 0xba, 0x59, 0xc6, 0x07, 0x93, 0x67, 0xea, 0x45, 0xce, 0xe7, 0x26, + 0xa0, 0x5d, 0x99, 0x2c, 0x14, 0xeb, 0xbc, 0x85, 0x6b, 0x4e, 0x5d, 0xca, 0x14, 0x0a, 0x96, 0xea, + 0x79, 0x14, 0x2c, 0x0b, 0xa7, 0x2a, 0x58, 0xae, 0x80, 0x25, 0xb2, 0x26, 0xe3, 0xde, 0x70, 0x24, + 0xf9, 0xa2, 0xea, 0x8e, 0x05, 0x93, 0xe5, 0xc1, 0xd2, 0x9c, 0xe5, 0x41, 0xed, 0xb4, 0xe5, 0x81, + 0xf3, 0x1a, 0x2e, 0x26, 0x81, 0x2d, 0xe9, 0xfb, 0x2d, 0xdc, 0x91, 0x0f, 0x85, 0x4a, 0x31, 0x14, + 0x66, 0x38, 0xc5, 0xf9, 0x77, 0x05, 0xd6, 0xda, 0x09, 0xe7, 0xec, 0x7b, 0xfc, 0x50, 0xd6, 0x0c, + 0x27, 0x47, 0xca, 0x74, 0x04, 0x64, 0x08, 0xda, 0x9c, 0x4a, 0xd0, 0xd5, 0x3c, 0x41, 0xe7, 0x2f, + 0xb8, 0x50, 0x44, 0xcd, 0xf9, 0x94, 0xa8, 0x9b, 0xb0, 0x9a, 0x21, 0xdc, 0x91, 0xc7, 0x0f, 0x45, + 0x99, 0x2a, 0x18, 0x77, 0x99, 0x64, 0xbf, 0x9e, 0xa1, 0xdb, 0xb0, 0x92, 0x32, 0xa4, 0xaf, 0x88, + 0xb3, 0x26, 0x11, 0x32, 0xa6, 0x53, 0x3f, 0x61, 0xce, 0x7c, 0x01, 0x61, 0x95, 0x14, 0x10, 0xd9, + 0x62, 0x06, 0x72, 0xc5, 0x8c, 0xf3, 0x17, 0x03, 0xec, 0x34, 0x40, 0xe7, 0x6c, 0x23, 0x72, 0x7e, + 0xa9, 0x14, 0xfd, 0x72, 0x1d, 0xea, 0x38, 0xf4, 0x7a, 0x01, 0xd6, 0xb8, 0x35, 0x15, 0x6e, 0x95, + 0x4c, 0xe1, 0xf6, 0x01, 0xd8, 0xe3, 0x52, 0x32, 0x89, 0xc1, 0x9b, 0x53, 0x6b, 0xc9, 0x2c, 0x28, + 0x5c, 0x48, 0x6b, 0x4a, 0xe6, 0xfc, 0xa6, 0x32, 0xa6, 0x39, 0x85, 0xd8, 0xb3, 0x24, 0xb3, 0x9f, + 0x40, 0x5d, 0x7f, 0x85, 0x2a, 0x71, 0x55, 0x4a, 0xfb, 0x6e, 0xd9, 0xb5, 0xca, 0x0e, 0xdd, 0xca, + 0x98, 0xf1, 0xc3, 0x90, 0x47, 0xc7, 0xae, 0xcd, 0xc6, 0x92, 0x56, 0x17, 0x56, 0x8b, 0x0a, 0x68, + 0x15, 0xcc, 0x23, 0x7c, 0xac, 0x6d, 0x2c, 0x7e, 0x8a, 0xf4, 0xff, 0x52, 0x60, 0x47, 0xb3, 0xfe, + 0xb5, 0x13, 0xf3, 0xe9, 0x80, 0xba, 0x4a, 0xfb, 0x7b, 0x95, 0x0f, 0x0c, 0xe7, 0x0f, 0x06, 0xac, + 0xee, 0x45, 0x74, 0xf4, 0xd6, 0xa9, 0xd4, 0x81, 0x7a, 0xa6, 0x2e, 0x4e, 0xa2, 0x37, 0x27, 0x9b, + 0x95, 0x54, 0x2f, 0x41, 0xcd, 0x8f, 0xe8, 0xa8, 0xeb, 0x05, 0x81, 0x0c, 0x2c, 0x51, 0x22, 0x46, + 0x74, 0xb4, 0x13, 0x04, 0xce, 0x2b, 0x58, 0xdf, 0xc3, 0xac, 0x1f, 0x91, 0xde, 0xdb, 0x27, 0xf9, + 0x19, 0xfc, 0x9b, 0x4b, 0xa0, 0x66, 0x21, 0x81, 0x3a, 0x9f, 0x1b, 0xf0, 0x4e, 0xe1, 0xe4, 0xb3, + 0xa0, 0xe3, 0x87, 0x79, 0xcc, 0x2a, 0x70, 0xcc, 0xe8, 0x7f, 0xb2, 0x58, 0xf5, 0x24, 0xff, 0xca, + 0xb9, 0xfb, 0x22, 0xe7, 0xec, 0x47, 0xf4, 0x40, 0x56, 0x97, 0xe7, 0x57, 0x99, 0xfd, 0xcd, 0x80, + 0xab, 0x53, 0xce, 0x38, 0xcb, 0x97, 0x17, 0x1b, 0xeb, 0xca, 0xac, 0xc6, 0xda, 0x2c, 0x36, 0xd6, + 0xe5, 0x7d, 0x67, 0x75, 0x4a, 0xdf, 0xf9, 0xe7, 0x0a, 0x34, 0x3a, 0x9c, 0x46, 0xde, 0x01, 0xde, + 0xa5, 0xe1, 0x80, 0x1c, 0x88, 0xb4, 0x9d, 0xd4, 0xeb, 0x86, 0xfc, 0xe8, 0xb4, 0x22, 0xbf, 0x0e, + 0x75, 0xaf, 0xdf, 0xc7, 0x8c, 0x89, 0xf6, 0x45, 0x67, 0x23, 0xcb, 0xb5, 0x95, 0xec, 0x89, 0x10, + 0xa1, 0x6f, 0xc0, 0x1a, 0xc3, 0xfd, 0x08, 0xf3, 0xee, 0x58, 0x53, 0x23, 0x78, 0x45, 0x4d, 0xec, + 0x24, 0xda, 0xa2, 0xc0, 0x8f, 0x19, 0xee, 0x74, 0x3e, 0xd2, 0x28, 0xd6, 0x23, 0x51, 0x5e, 0xf5, + 0xe2, 0xfe, 0x11, 0xe6, 0x59, 0x7a, 0x00, 0x25, 0x92, 0x50, 0xbc, 0x0c, 0x56, 0x44, 0x29, 0x97, + 0x39, 0x5d, 0x72, 0xb9, 0xe5, 0xd6, 0x84, 0x40, 0xa4, 0x2d, 0xbd, 0x6b, 0x7b, 0xe7, 0xa9, 0xe6, + 0x70, 0x3d, 0x12, 0x3d, 0x6a, 0x7b, 0xe7, 0xe9, 0x87, 0xa1, 0x3f, 0xa2, 0x24, 0xe4, 0x32, 0xc1, + 0x5b, 0x6e, 0x56, 0x24, 0x3e, 0x8f, 0x29, 0x4b, 0x74, 0x45, 0xf9, 0x21, 0x93, 0xbb, 0xe5, 0xda, + 0x5a, 0xf6, 0xfc, 0x78, 0x84, 0x9d, 0x7f, 0x99, 0xb0, 0xaa, 0x6a, 0xa8, 0xc7, 0xb4, 0x97, 0x80, + 0xe9, 0x0a, 0x58, 0xfd, 0x20, 0x16, 0xed, 0x88, 0x46, 0x92, 0xe5, 0x8e, 0x05, 0xc2, 0x22, 0x59, + 0x1a, 0x8a, 0xf0, 0x80, 0xbc, 0xd6, 0x96, 0x5b, 0x19, 0xf3, 0x90, 0x14, 0x67, 0x19, 0xd3, 0x9c, + 0x60, 0x4c, 0xdf, 0xe3, 0x9e, 0xa6, 0xb1, 0xaa, 0xa4, 0x31, 0x4b, 0x48, 0x14, 0x83, 0x4d, 0x10, + 0xd3, 0x42, 0x09, 0x31, 0x65, 0x98, 0x7a, 0x31, 0xcf, 0xd4, 0x79, 0xa8, 0x2f, 0x15, 0x43, 0xff, + 0x11, 0x2c, 0x27, 0x86, 0xe9, 0x4b, 0x8c, 0x48, 0xeb, 0x95, 0xb4, 0x49, 0x32, 0x61, 0x66, 0xc1, + 0xe4, 0x36, 0x58, 0x0e, 0x5b, 0x45, 0x66, 0xb7, 0x4e, 0xc5, 0xec, 0x85, 0xaa, 0x12, 0x4e, 0x53, + 0x55, 0x66, 0x59, 0xda, 0xce, 0xb3, 0xf4, 0x47, 0xb0, 0xfa, 0xa3, 0x18, 0x47, 0xc7, 0x8f, 0x69, + 0x8f, 0xcd, 0xe7, 0xe3, 0x16, 0xd4, 0xb4, 0xa3, 0x92, 0x84, 0x9e, 0x8e, 0x9d, 0x7f, 0x18, 0xd0, + 0x90, 0xe1, 0xf6, 0xdc, 0x63, 0x47, 0xc9, 0xeb, 0x4c, 0xe2, 0x65, 0x23, 0xef, 0xe5, 0x53, 0xf6, + 0x23, 0x25, 0x4f, 0x0b, 0x66, 0xd9, 0xd3, 0x42, 0x49, 0x9d, 0x53, 0x2d, 0xad, 0x73, 0x0a, 0x0d, + 0xce, 0xc2, 0x44, 0x83, 0xf3, 0x85, 0x01, 0x6b, 0x19, 0x1b, 0x9d, 0x25, 0xe1, 0xe5, 0x2c, 0x5b, + 0x29, 0x5a, 0xf6, 0x7e, 0x9e, 0x08, 0xcc, 0x32, 0x57, 0x67, 0x88, 0x20, 0xb1, 0x71, 0x8e, 0x0c, + 0x9e, 0xc0, 0x8a, 0xa0, 0xea, 0xf3, 0x71, 0xe7, 0xdf, 0x0d, 0x58, 0x7a, 0x4c, 0x7b, 0xd2, 0x91, + 0x59, 0x0c, 0x19, 0xf9, 0x67, 0xab, 0x55, 0x30, 0x7d, 0x32, 0xd4, 0xd9, 0x5b, 0xfc, 0x14, 0x31, + 0xc6, 0xb8, 0x17, 0xf1, 0xf1, 0xc3, 0x9b, 0x28, 0xe4, 0x84, 0x44, 0xbe, 0xdd, 0x5c, 0x82, 0x1a, + 0x0e, 0x7d, 0x35, 0xa9, 0xab, 0x65, 0x1c, 0xfa, 0x72, 0xea, 0x7c, 0x1a, 0xa0, 0x75, 0x58, 0x18, + 0xd1, 0xf1, 0x63, 0x99, 0x1a, 0x38, 0xeb, 0x80, 0x1e, 0x62, 0xfe, 0x98, 0xf6, 0x84, 0x57, 0x12, + 0xf3, 0x38, 0x7f, 0xad, 0xc8, 0xe6, 0x64, 0x2c, 0x3e, 0x8b, 0x83, 0x1d, 0x68, 0x28, 0xba, 0xfa, + 0x94, 0xf6, 0xba, 0x61, 0x9c, 0x18, 0xc5, 0x96, 0xc2, 0xc7, 0xb4, 0xf7, 0x2c, 0x1e, 0xa2, 0x3b, + 0x70, 0x91, 0x84, 0xdd, 0x91, 0x66, 0xd0, 0x54, 0x53, 0x59, 0x69, 0x95, 0x84, 0x09, 0xb7, 0x6a, + 0xf5, 0x5b, 0xb0, 0x82, 0xc3, 0x17, 0x31, 0x8e, 0x71, 0xaa, 0xaa, 0x6c, 0xd6, 0xd0, 0x62, 0xad, + 0x27, 0x98, 0xd2, 0x63, 0x47, 0x5d, 0x16, 0x50, 0xce, 0x74, 0x4e, 0xb4, 0x84, 0xa4, 0x23, 0x04, + 0xe8, 0x03, 0xb0, 0xc4, 0x72, 0x05, 0x2d, 0xd5, 0x64, 0x5c, 0x2e, 0x83, 0x96, 0xf6, 0xb7, 0x5b, + 0xfb, 0x54, 0xfd, 0x60, 0x22, 0x40, 0x74, 0xd9, 0xed, 0x13, 0x76, 0xa4, 0x99, 0x06, 0x94, 0x68, + 0x8f, 0xb0, 0x23, 0xe7, 0xa7, 0x70, 0x29, 0xfb, 0x6c, 0x43, 0x18, 0x27, 0xfd, 0xf3, 0xac, 0x3e, + 0x7e, 0x67, 0x40, 0xab, 0xec, 0x80, 0xff, 0x61, 0xd1, 0xb5, 0xfd, 0x2b, 0x1b, 0x40, 0xce, 0xec, + 0x52, 0x1a, 0xf9, 0x28, 0x90, 0xd0, 0xda, 0xa5, 0xc3, 0x11, 0x0d, 0x71, 0xc8, 0x65, 0xc6, 0x62, + 0x68, 0x2b, 0xbf, 0x9f, 0x1e, 0x4c, 0x2a, 0x6a, 0x5b, 0xb5, 0xde, 0x2d, 0xd5, 0x2f, 0x28, 0x3b, + 0x17, 0xd0, 0x0b, 0xd9, 0x9c, 0x8c, 0x4d, 0xb1, 0x7b, 0xe8, 0x85, 0x21, 0x0e, 0xd0, 0xf6, 0x94, + 0xa7, 0xbc, 0x32, 0xe5, 0xe4, 0xcc, 0x1b, 0xa5, 0x67, 0x76, 0x78, 0x44, 0xc2, 0x83, 0xc4, 0xc4, + 0xce, 0x05, 0xf4, 0x1c, 0xec, 0xcc, 0x7b, 0x0a, 0xba, 0x55, 0x66, 0xa9, 0xc9, 0x07, 0x97, 0xd6, + 0x49, 0xbe, 0x70, 0x2e, 0xa0, 0x01, 0x34, 0x72, 0x0f, 0x7e, 0x68, 0xf3, 0xa4, 0x9e, 0x28, 0xfb, + 0xca, 0xd6, 0xfa, 0xfa, 0x1c, 0x9a, 0xe9, 0xed, 0x7f, 0xae, 0x0c, 0x36, 0xf1, 0x62, 0x76, 0x77, + 0xca, 0x26, 0xd3, 0xde, 0xf6, 0x5a, 0xf7, 0xe6, 0x5f, 0x90, 0x1e, 0xee, 0x8f, 0x3f, 0x52, 0x05, + 0xd4, 0xed, 0xd9, 0x8d, 0x9f, 0x3a, 0x6d, 0x73, 0xde, 0x0e, 0xd1, 0xb9, 0x80, 0xf6, 0xc1, 0x4a, + 0x7b, 0x34, 0xf4, 0x6e, 0xd9, 0xc2, 0x62, 0x0b, 0x37, 0x87, 0x73, 0x72, 0x5d, 0x4e, 0xb9, 0x73, + 0xca, 0x5a, 0xb0, 0x72, 0xe7, 0x94, 0xb6, 0x4c, 0xce, 0x05, 0x14, 0xcb, 0xd8, 0x29, 0x44, 0x37, + 0xba, 0x33, 0xcb, 0xbf, 0xb9, 0x34, 0xd3, 0xda, 0x9a, 0x57, 0x3d, 0x3d, 0xf6, 0x17, 0xe3, 0xc7, + 0xe6, 0x5c, 0x4b, 0x83, 0xee, 0x9d, 0xb4, 0x55, 0x59, 0x87, 0xd5, 0xfa, 0xd6, 0x5b, 0xac, 0xc8, + 0x60, 0x12, 0x75, 0x0e, 0xe9, 0x2b, 0x55, 0x2c, 0xc6, 0x91, 0x27, 0x72, 0x61, 0xc9, 0xe1, 0x3a, + 0x84, 0x27, 0x55, 0xa7, 0x1e, 0x7e, 0xc2, 0x8a, 0xf4, 0xf0, 0x2e, 0xc0, 0x43, 0xcc, 0x9f, 0x62, + 0x1e, 0x09, 0x5b, 0xdf, 0x9a, 0x96, 0xa7, 0xb4, 0x42, 0x72, 0xd4, 0xed, 0x99, 0x7a, 0xe9, 0x01, + 0x3d, 0xb0, 0x77, 0x0f, 0x71, 0xff, 0xe8, 0x11, 0xf6, 0x02, 0x7e, 0x88, 0xca, 0x57, 0x66, 0x34, + 0xa6, 0x40, 0xbe, 0x4c, 0x31, 0x39, 0x63, 0xfb, 0x8b, 0x45, 0xfd, 0x37, 0xf5, 0x33, 0xea, 0xe3, + 0xff, 0xff, 0x14, 0xbc, 0x0f, 0x56, 0xda, 0x8e, 0x95, 0x47, 0x78, 0xb1, 0x5b, 0x9b, 0x15, 0xe1, + 0x9f, 0x80, 0x95, 0x16, 0xb6, 0xe5, 0x3b, 0x16, 0x7b, 0x83, 0xd6, 0xcd, 0x19, 0x5a, 0xe9, 0x6d, + 0x9f, 0x41, 0x2d, 0x29, 0x44, 0xd1, 0x8d, 0x69, 0xe9, 0x28, 0xbb, 0xf3, 0x8c, 0xbb, 0xfe, 0x0c, + 0xec, 0x4c, 0x95, 0x56, 0x4e, 0x40, 0x93, 0xd5, 0x5d, 0xeb, 0xf6, 0x4c, 0xbd, 0x2f, 0x47, 0x40, + 0xde, 0xff, 0xf6, 0x27, 0xdb, 0x07, 0x84, 0x1f, 0xc6, 0x3d, 0x61, 0xd9, 0xbb, 0x4a, 0xf3, 0x0e, + 0xa1, 0xfa, 0xd7, 0xdd, 0xe4, 0x96, 0x77, 0xe5, 0x4e, 0x77, 0xa5, 0x9d, 0x46, 0xbd, 0xde, 0xa2, + 0x1c, 0xbe, 0xf7, 0x9f, 0x00, 0x00, 0x00, 0xff, 0xff, 0xc2, 0x46, 0x93, 0xfc, 0x65, 0x22, 0x00, + 0x00, } // Reference imports to suppress errors if they are not otherwise used. diff --git a/internal/proxy/impl.go b/internal/proxy/impl.go index b58093015a..f4a71fcf62 100644 --- a/internal/proxy/impl.go +++ b/internal/proxy/impl.go @@ -5201,3 +5201,28 @@ func (node *Proxy) ListClientInfos(ctx context.Context, req *proxypb.ListClientI ClientInfos: clients, }, nil } + +func (node *Proxy) AllocTimestamp(ctx context.Context, req *milvuspb.AllocTimestampRequest) (*milvuspb.AllocTimestampResponse, error) { + if !node.checkHealthy() { + return &milvuspb.AllocTimestampResponse{Status: unhealthyStatus()}, nil + } + + log.Info("AllocTimestamp request receive") + ts, err := node.tsoAllocator.AllocOne(ctx) + if err != nil { + log.Info("AllocTimestamp failed", zap.Error(err)) + return &milvuspb.AllocTimestampResponse{ + Status: &commonpb.Status{ + ErrorCode: commonpb.ErrorCode_UnexpectedError, + Reason: err.Error(), + }, + }, nil + } + + log.Info("AllocTimestamp request success", zap.Uint64("timestamp", ts)) + + return &milvuspb.AllocTimestampResponse{ + Status: &commonpb.Status{ErrorCode: commonpb.ErrorCode_Success}, + Timestamp: ts, + }, nil +} diff --git a/internal/proxy/impl_test.go b/internal/proxy/impl_test.go index 8f202ddd45..eedfc89b0d 100644 --- a/internal/proxy/impl_test.go +++ b/internal/proxy/impl_test.go @@ -815,3 +815,52 @@ func TestProxyListDatabase(t *testing.T) { assert.Equal(t, commonpb.ErrorCode_Success, resp.GetStatus().GetErrorCode()) }) } + +func TestProxy_AllocTimestamp(t *testing.T) { + t.Run("proxy unhealthy", func(t *testing.T) { + node := &Proxy{} + node.UpdateStateCode(commonpb.StateCode_Abnormal) + + resp, err := node.AllocTimestamp(context.TODO(), nil) + assert.NoError(t, err) + assert.NotEqual(t, commonpb.ErrorCode_Success, resp.GetStatus().GetErrorCode()) + }) + + t.Run("success", func(t *testing.T) { + node := &Proxy{} + node.UpdateStateCode(commonpb.StateCode_Healthy) + + node.tsoAllocator = ×tampAllocator{ + tso: newMockTimestampAllocatorInterface(), + } + + resp, err := node.AllocTimestamp(context.TODO(), nil) + assert.NoError(t, err) + assert.Equal(t, commonpb.ErrorCode_Success, resp.GetStatus().GetErrorCode()) + }) + + t.Run("failed", func(t *testing.T) { + m := newMockTimestampAllocator(t) + m.On("AllocTimestamp", + mock.Anything, + mock.Anything, + ).Return(&rootcoordpb.AllocTimestampResponse{ + Status: &commonpb.Status{ + ErrorCode: commonpb.ErrorCode_UnexpectedError, + Reason: "failed", + }, + Timestamp: 20230518, + Count: 1, + }, nil) + + alloc, _ := newTimestampAllocator(m, 199) + node := Proxy{ + tsoAllocator: alloc, + } + node.UpdateStateCode(commonpb.StateCode_Healthy) + + resp, err := node.AllocTimestamp(context.TODO(), nil) + assert.NoError(t, err) + assert.Equal(t, commonpb.ErrorCode_UnexpectedError, resp.GetStatus().GetErrorCode()) + }) +} diff --git a/internal/proxy/task_index.go b/internal/proxy/task_index.go index edd46d5cba..4dd328ce03 100644 --- a/internal/proxy/task_index.go +++ b/internal/proxy/task_index.go @@ -487,7 +487,7 @@ func (dit *describeIndexTask) Execute(ctx context.Context) error { return fmt.Errorf("failed to parse collection schema: %s", err) } - resp, err := dit.datacoord.DescribeIndex(ctx, &indexpb.DescribeIndexRequest{CollectionID: dit.collectionID, IndexName: dit.IndexName}) + resp, err := dit.datacoord.DescribeIndex(ctx, &indexpb.DescribeIndexRequest{CollectionID: dit.collectionID, IndexName: dit.IndexName, Timestamp: dit.Timestamp}) if err != nil || resp == nil { return err } diff --git a/internal/types/types.go b/internal/types/types.go index 0fc06887fe..abe130a711 100644 --- a/internal/types/types.go +++ b/internal/types/types.go @@ -1418,6 +1418,8 @@ type ProxyComponent interface { DescribeResourceGroup(ctx context.Context, req *milvuspb.DescribeResourceGroupRequest) (*milvuspb.DescribeResourceGroupResponse, error) Connect(ctx context.Context, req *milvuspb.ConnectRequest) (*milvuspb.ConnectResponse, error) + + AllocTimestamp(ctx context.Context, req *milvuspb.AllocTimestampRequest) (*milvuspb.AllocTimestampResponse, error) } // QueryNode is the interface `querynode` package implements