Update proto (#226)

Signed-off-by: yhmo <yihua.mo@zilliz.com>

src/main/proto/common.proto

@@ -6,6 +6,7 @@ option java_package = "io.milvus.grpc";
 option java_outer_classname = "CommonProto";
 option java_generate_equals_and_hash = true;
 
+
 enum ErrorCode {
     Success = 0;
     UnexpectedError = 1;
@@ -53,6 +54,7 @@ enum SegmentState {
     Sealed = 3;
     Flushed = 4;
     Flushing = 5;
+    Dropped = 6;
 }
 
 message Status {
@@ -75,8 +77,8 @@ message Blob {
 }
 
 message Address {
-    string ip = 1;
-    int64 port = 2;
+  string ip = 1;
+  int64 port = 2;
 }
 
 enum MsgType {
@@ -136,9 +138,11 @@ enum MsgType {
     WatchQueryChannels = 510;
     RemoveQueryChannels = 511;
     SealedSegmentsChangeInfo = 512;
+    WatchDeltaChannels = 513;
 
     /* DATA SERVICE */
     SegmentInfo = 600;
+    SystemInfo = 601;
 
     /* SYSTEM CONTROL */
     TimeTick = 1200;
@@ -175,3 +179,9 @@ message DMLMsgHeader {
     common.MsgBase base = 1;
     string shardName = 2;
 }
+
+enum CompactionState {
+  UndefiedState = 0;
+  Executing = 1;
+  Completed = 2;
+}

src/main/proto/milvus.proto

@@ -11,298 +11,302 @@ option java_generate_equals_and_hash = true;
 package milvus.proto.milvus;
 
 service MilvusService {
-    rpc CreateCollection(CreateCollectionRequest) returns (common.Status) {}
-    rpc DropCollection(DropCollectionRequest) returns (common.Status) {}
-    rpc HasCollection(HasCollectionRequest) returns (BoolResponse) {}
-    rpc LoadCollection(LoadCollectionRequest) returns (common.Status) {}
-    rpc ReleaseCollection(ReleaseCollectionRequest) returns (common.Status) {}
-    rpc DescribeCollection(DescribeCollectionRequest) returns (DescribeCollectionResponse) {}
-    rpc GetCollectionStatistics(GetCollectionStatisticsRequest) returns (GetCollectionStatisticsResponse) {}
-    rpc ShowCollections(ShowCollectionsRequest) returns (ShowCollectionsResponse) {}
-
-    rpc CreatePartition(CreatePartitionRequest) returns (common.Status) {}
-    rpc DropPartition(DropPartitionRequest) returns (common.Status) {}
-    rpc HasPartition(HasPartitionRequest) returns (BoolResponse) {}
-    rpc LoadPartitions(LoadPartitionsRequest) returns (common.Status) {}
-    rpc ReleasePartitions(ReleasePartitionsRequest) returns (common.Status) {}
-    rpc GetPartitionStatistics(GetPartitionStatisticsRequest) returns (GetPartitionStatisticsResponse) {}
-    rpc ShowPartitions(ShowPartitionsRequest) returns (ShowPartitionsResponse) {}
-
-    rpc CreateAlias(CreateAliasRequest) returns (common.Status) {}
-    rpc DropAlias(DropAliasRequest) returns (common.Status) {}
-    rpc AlterAlias(AlterAliasRequest) returns (common.Status) {}
-
-    rpc CreateIndex(CreateIndexRequest) returns (common.Status) {}
-    rpc DescribeIndex(DescribeIndexRequest) returns (DescribeIndexResponse) {}
-    rpc GetIndexState(GetIndexStateRequest) returns (GetIndexStateResponse) {}
-    rpc GetIndexBuildProgress(GetIndexBuildProgressRequest) returns (GetIndexBuildProgressResponse) {}
-    rpc DropIndex(DropIndexRequest) returns (common.Status) {}
-
-    rpc Insert(InsertRequest) returns (MutationResult) {}
-    rpc Delete(DeleteRequest) returns (MutationResult) {}
-    rpc Search(SearchRequest) returns (SearchResults) {}
-    rpc Flush(FlushRequest) returns (FlushResponse) {}
-    rpc Query(QueryRequest) returns (QueryResults) {}
-    rpc CalcDistance(CalcDistanceRequest) returns (CalcDistanceResults) {}
-
-    rpc GetPersistentSegmentInfo(GetPersistentSegmentInfoRequest) returns (GetPersistentSegmentInfoResponse) {}
-    rpc GetQuerySegmentInfo(GetQuerySegmentInfoRequest) returns (GetQuerySegmentInfoResponse) {}
-
-    rpc Dummy(DummyRequest) returns (DummyResponse) {}
-
-    // TODO: remove
-    rpc RegisterLink(RegisterLinkRequest) returns (RegisterLinkResponse) {}
-
-    // https://wiki.lfaidata.foundation/display/MIL/MEP+8+--+Add+metrics+for+proxy
-    rpc GetMetrics(GetMetricsRequest) returns (GetMetricsResponse) {}
+  rpc CreateCollection(CreateCollectionRequest) returns (common.Status) {}
+  rpc DropCollection(DropCollectionRequest) returns (common.Status) {}
+  rpc HasCollection(HasCollectionRequest) returns (BoolResponse) {}
+  rpc LoadCollection(LoadCollectionRequest) returns (common.Status) {}
+  rpc ReleaseCollection(ReleaseCollectionRequest) returns (common.Status) {}
+  rpc DescribeCollection(DescribeCollectionRequest) returns (DescribeCollectionResponse) {}
+  rpc GetCollectionStatistics(GetCollectionStatisticsRequest) returns (GetCollectionStatisticsResponse) {}
+  rpc ShowCollections(ShowCollectionsRequest) returns (ShowCollectionsResponse) {}
+
+  rpc CreatePartition(CreatePartitionRequest) returns (common.Status) {}
+  rpc DropPartition(DropPartitionRequest) returns (common.Status) {}
+  rpc HasPartition(HasPartitionRequest) returns (BoolResponse) {}
+  rpc LoadPartitions(LoadPartitionsRequest) returns (common.Status) {}
+  rpc ReleasePartitions(ReleasePartitionsRequest) returns (common.Status) {}
+  rpc GetPartitionStatistics(GetPartitionStatisticsRequest) returns (GetPartitionStatisticsResponse) {}
+  rpc ShowPartitions(ShowPartitionsRequest) returns (ShowPartitionsResponse) {}
+
+  rpc CreateAlias(CreateAliasRequest) returns (common.Status) {}
+  rpc DropAlias(DropAliasRequest) returns (common.Status) {}
+  rpc AlterAlias(AlterAliasRequest) returns (common.Status) {}
+
+  rpc CreateIndex(CreateIndexRequest) returns (common.Status) {}
+  rpc DescribeIndex(DescribeIndexRequest) returns (DescribeIndexResponse) {}
+  rpc GetIndexState(GetIndexStateRequest) returns (GetIndexStateResponse) {}
+  rpc GetIndexBuildProgress(GetIndexBuildProgressRequest) returns (GetIndexBuildProgressResponse) {}
+  rpc DropIndex(DropIndexRequest) returns (common.Status) {}
+
+  rpc Insert(InsertRequest) returns (MutationResult) {}
+  rpc Delete(DeleteRequest) returns (MutationResult) {}
+  rpc Search(SearchRequest) returns (SearchResults) {}
+  rpc Flush(FlushRequest) returns (FlushResponse) {}
+  rpc Query(QueryRequest) returns (QueryResults) {}
+  rpc CalcDistance(CalcDistanceRequest) returns (CalcDistanceResults) {}
+
+  rpc GetPersistentSegmentInfo(GetPersistentSegmentInfoRequest) returns (GetPersistentSegmentInfoResponse) {}
+  rpc GetQuerySegmentInfo(GetQuerySegmentInfoRequest) returns (GetQuerySegmentInfoResponse) {}
+
+  rpc Dummy(DummyRequest) returns (DummyResponse) {}
+
+  // TODO: remove
+  rpc RegisterLink(RegisterLinkRequest) returns (RegisterLinkResponse) {}
+
+  // https://wiki.lfaidata.foundation/display/MIL/MEP+8+--+Add+metrics+for+proxy
+  rpc GetMetrics(GetMetricsRequest) returns (GetMetricsResponse) {}
+  rpc LoadBalance(LoadBalanceRequest) returns (common.Status) {}
+  rpc GetCompactionState(GetCompactionStateRequest) returns (GetCompactionStateResponse) {}
+  rpc ManualCompaction(ManualCompactionRequest) returns (ManualCompactionResponse) {}
+  rpc GetCompactionStateWithPlans(GetCompactionPlansRequest) returns (GetCompactionPlansResponse) {}
 }
 
 message CreateAliasRequest {
-    common.MsgBase base = 1;
-    string db_name = 2;
-    string collection_name = 3;
-    string alias = 4;
+  common.MsgBase base = 1;
+  string db_name = 2;
+  string collection_name = 3;
+  string alias = 4;
 }
 
 message DropAliasRequest {
-    common.MsgBase base = 1;
-    string db_name = 2;
-    string alias = 3;
+  common.MsgBase base = 1;
+  string db_name = 2;
+  string alias = 3;
 }
 
 message AlterAliasRequest{
-    common.MsgBase base = 1;
-    string db_name = 2;
-    string collection_name = 3;
-    string alias = 4;
+  common.MsgBase base = 1;
+  string db_name = 2;
+  string collection_name = 3;
+  string alias = 4;
 }
 
 /**
 * Create collection in milvus
 */
 message CreateCollectionRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The unique collection name in milvus.(Required)
-    string collection_name = 3;
-    // The serialized `schema.CollectionSchema`(Required)
-    bytes schema = 4;
-    // Once set, no modification is allowed (Optional)
-    // https://github.com/milvus-io/milvus/issues/6690
-    int32 shards_num = 5;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The unique collection name in milvus.(Required)
+  string collection_name = 3; 
+  // The serialized `schema.CollectionSchema`(Required)
+  bytes schema = 4; 
+  // Once set, no modification is allowed (Optional)
+  // https://github.com/milvus-io/milvus/issues/6690
+  int32 shards_num = 5;
 }
 
 /**
-* Drop collection in milvus, also will drop data in collection.
+* Drop collection in milvus, also will drop data in collection. 
 */
 message DropCollectionRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The unique collection name in milvus.(Required)
-    string collection_name = 3;
+  // Not useful for now
+  common.MsgBase base = 1; 
+  // Not useful for now
+  string db_name = 2;
+  // The unique collection name in milvus.(Required)
+  string collection_name = 3;
 }
 
 /**
 * Check collection exist in milvus or not.
 */
 message HasCollectionRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name you want to check.
-    string collection_name = 3;
-    // If time_stamp is not zero, will return true when time_stamp >= created collection timestamp, otherwise will return false.
-    uint64 time_stamp = 4;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name you want to check.
+  string collection_name = 3; 
+  // If time_stamp is not zero, will return true when time_stamp >= created collection timestamp, otherwise will return false.
+  uint64 time_stamp = 4;
 }
 
 
 message BoolResponse {
-    common.Status status = 1;
-    bool value = 2;
+  common.Status status = 1;
+  bool value = 2;
 }
 
 message StringResponse {
-    common.Status status = 1;
-    string value = 2;
+  common.Status status = 1;
+  string value = 2;
 }
 
 /**
 * Get collection meta datas like: schema, collectionID, shards number ...
 */
 message DescribeCollectionRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name you want to describe, you can pass collection_name or collectionID
-    string collection_name = 3;
-    // The collection ID you want to describe
-    int64 collectionID = 4;
-    // If time_stamp is not zero, will describe collection success when time_stamp >= created collection timestamp, otherwise will throw error.
-    uint64 time_stamp = 5;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name you want to describe, you can pass collection_name or collectionID
+  string collection_name = 3;
+  // The collection ID you want to describe
+  int64 collectionID = 4;
+  // If time_stamp is not zero, will describe collection success when time_stamp >= created collection timestamp, otherwise will throw error.
+  uint64 time_stamp = 5;
 }
 
 /**
 * DescribeCollection Response
 */
 message DescribeCollectionResponse {
-    // Contain error_code and reason
-    common.Status status = 1;
-    // The schema param when you created collection.
-    schema.CollectionSchema schema = 2;
-    // The collection id
-    int64 collectionID = 3;
-    // System design related, users should not perceive
-    repeated string virtual_channel_names = 4;
-    // System design related, users should not perceive
-    repeated string physical_channel_names = 5;
-    // Hybrid timestamp in milvus
-    uint64 created_timestamp = 6;
-    // The utc timestamp calculated by created_timestamp
-    uint64 created_utc_timestamp = 7;
-    // The shards number you set.
-    int32 shards_num = 8;
-    // The aliases of this collection
-    repeated string aliases = 9;
-    // The message ID/posititon when collection is created
-    repeated common.KeyDataPair start_positions = 10;
+  // Contain error_code and reason
+  common.Status status = 1;
+  // The schema param when you created collection.
+  schema.CollectionSchema schema = 2;
+  // The collection id
+  int64 collectionID = 3;
+  // System design related, users should not perceive
+  repeated string virtual_channel_names = 4;
+  // System design related, users should not perceive
+  repeated string physical_channel_names = 5;
+  // Hybrid timestamp in milvus
+  uint64 created_timestamp = 6;
+  // The utc timestamp calculated by created_timestamp
+  uint64 created_utc_timestamp = 7;
+  // The shards number you set.
+  int32 shards_num = 8;
+  // The aliases of this collection
+  repeated string aliases = 9;
+  // The message ID/posititon when collection is created
+  repeated common.KeyDataPair start_positions = 10;
 }
 
 /**
 * Load collection data into query nodes, then you can do vector search on this collection.
 */
 message LoadCollectionRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name you want to load
-    string collection_name = 3;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name you want to load
+  string collection_name = 3;
 }
 
 /**
 * Release collection data from query nodes, then you can't do vector search on this collection.
 */
 message ReleaseCollectionRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name you want to release
-    string collection_name = 3;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name you want to release
+  string collection_name = 3;
 }
 
 /**
 * Get collection statistics like row_count.
 */
 message GetCollectionStatisticsRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name you want get statistics
-    string collection_name = 3;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name you want get statistics
+  string collection_name = 3;
 }
 
 /**
 * Will return collection statistics in stats field like [{key:"row_count",value:"1"}]
 */
 message GetCollectionStatisticsResponse {
-    // Contain error_code and reason
-    common.Status status = 1;
-    // Collection statistics data
-    repeated common.KeyValuePair stats = 2;
+  // Contain error_code and reason
+  common.Status status = 1;
+  // Collection statistics data
+  repeated common.KeyValuePair stats = 2;
 }
 
 /*
 * This is for ShowCollectionsRequest type field.
 */
 enum ShowType {
-    // Will return all colloections
-    All = 0;
-    // Will return loaded collections with their inMemory_percentages
-    InMemory = 1;
+  // Will return all colloections 
+  All = 0;
+  // Will return loaded collections with their inMemory_percentages
+  InMemory = 1;
 }
 
 /*
-* List collections
+* List collections 
 */
 message ShowCollectionsRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // Not useful for now
-    uint64 time_stamp = 3;
-    // Decide return Loaded collections or All collections(Optional)
-    ShowType type = 4;
-    // When type is InMemory, will return these collection's inMemory_percentages.(Optional)
-    repeated string collection_names = 5;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // Not useful for now
+  uint64 time_stamp = 3;
+  // Decide return Loaded collections or All collections(Optional)
+  ShowType type = 4;
+  // When type is InMemory, will return these collection's inMemory_percentages.(Optional)
+  repeated string collection_names = 5; 
 }
 
 /*
 * Return basic collection infos.
 */
 message ShowCollectionsResponse {
-    // Contain error_code and reason
-    common.Status status = 1;
-    // Collection name array
-    repeated string collection_names = 2;
-    // Collection Id array
-    repeated int64 collection_ids = 3;
-    // Hybrid timestamps in milvus
-    repeated uint64 created_timestamps = 4;
-    // The utc timestamp calculated by created_timestamp
-    repeated uint64 created_utc_timestamps = 5;
-    // Load percentage on querynode when type is InMemory
-    repeated int64 inMemory_percentages = 6;
+  // Contain error_code and reason
+  common.Status status = 1;
+  // Collection name array
+  repeated string collection_names = 2;
+  // Collection Id array
+  repeated int64 collection_ids = 3;
+  // Hybrid timestamps in milvus
+  repeated uint64 created_timestamps = 4;
+  // The utc timestamp calculated by created_timestamp
+  repeated uint64 created_utc_timestamps = 5;
+  // Load percentage on querynode when type is InMemory
+  repeated int64 inMemory_percentages = 6; 
 }
 
 /*
 * Create partition in created collection.
 */
 message CreatePartitionRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name in milvus
-    string collection_name = 3;
-    // The partition name you want to create.
-    string partition_name = 4;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name in milvus
+  string collection_name = 3;
+  // The partition name you want to create.
+  string partition_name = 4;
 }
 
 /*
 * Drop partition in created collection.
 */
 message DropPartitionRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name in milvus
-    string collection_name = 3;
-    // The partition name you want to drop
-    string partition_name = 4;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name in milvus
+  string collection_name = 3;
+  // The partition name you want to drop
+  string partition_name = 4; 
 }
 
 /*
 * Check if partition exist in collection or not.
 */
 message HasPartitionRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name in milvus
-    string collection_name = 3;
-    // The partition name you want to check
-    string partition_name = 4;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name in milvus
+  string collection_name = 3;
+  // The partition name you want to check
+  string partition_name = 4;
 }
 
 /*
@@ -310,14 +314,14 @@ message HasPartitionRequest {
 * Then you can get these data as result when you do vector search on this collection.
 */
 message LoadPartitionsRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name in milvus
-    string collection_name = 3;
-    // The partition names you want to load
-    repeated string partition_names = 4;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name in milvus
+  string collection_name = 3;
+  // The partition names you want to load
+  repeated string partition_names = 4;
 }
 
 /*
@@ -325,51 +329,51 @@ message LoadPartitionsRequest {
 * Then you can not get these data as result when you do vector search on this collection.
 */
 message ReleasePartitionsRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name in milvus
-    string collection_name = 3;
-    // The partition names you want to release
-    repeated string partition_names = 4;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name in milvus
+  string collection_name = 3;
+  // The partition names you want to release
+  repeated string partition_names = 4;
 }
 
 /*
 * Get partition statistics like row_count.
 */
 message GetPartitionStatisticsRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name in milvus
-    string collection_name = 3;
-    // The partition name you want to collect statistics
-    string partition_name = 4;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name in milvus
+  string collection_name = 3;
+  // The partition name you want to collect statistics
+  string partition_name = 4; 
 }
 
 message GetPartitionStatisticsResponse {
-    common.Status status = 1;
-    repeated common.KeyValuePair stats = 2;
+  common.Status status = 1;
+  repeated common.KeyValuePair stats = 2;
 }
 
 /*
 * List all partitions for particular collection
 */
 message ShowPartitionsRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The collection name you want to describe, you can pass collection_name or collectionID
-    string collection_name = 3;
-    // The collection id in milvus
-    int64 collectionID = 4;
-    // When type is InMemory, will return these patitions's inMemory_percentages.(Optional)
-    repeated string partition_names = 5;
-    // Decide return Loaded partitions or All partitions(Optional)
-    ShowType type = 6;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The collection name you want to describe, you can pass collection_name or collectionID
+  string collection_name = 3;
+  // The collection id in milvus
+  int64 collectionID = 4;
+  // When type is InMemory, will return these patitions's inMemory_percentages.(Optional)
+  repeated string partition_names = 5;
+  // Decide return Loaded partitions or All partitions(Optional)
+  ShowType type = 6;
 }
 
 /*
@@ -377,339 +381,393 @@ message ShowPartitionsRequest {
 * The returned datas are all rows, we can format to columns by therir index.
 */
 message ShowPartitionsResponse {
-    // Contain error_code and reason
-    common.Status status = 1;
-    // All partition names for this collection
-    repeated string partition_names = 2;
-    // All partition ids for this collection
-    repeated int64 partitionIDs = 3;
-    // All hybrid timestamps
-    repeated uint64 created_timestamps = 4;
-    // All utc timestamps calculated by created_timestamps
-    repeated uint64 created_utc_timestamps = 5;
-    // Load percentage on querynode
-    repeated int64 inMemory_percentages = 6;
+  // Contain error_code and reason
+  common.Status status = 1;
+  // All partition names for this collection
+  repeated string partition_names = 2;
+  // All partition ids for this collection
+  repeated int64 partitionIDs = 3;
+  // All hybrid timestamps
+  repeated uint64 created_timestamps = 4;
+  // All utc timestamps calculated by created_timestamps
+  repeated uint64 created_utc_timestamps = 5;
+  // Load percentage on querynode
+  repeated int64 inMemory_percentages = 6;
 }
 
 message DescribeSegmentRequest {
-    common.MsgBase base = 1;
-    int64 collectionID = 2;
-    int64 segmentID = 3;
+  common.MsgBase base = 1;
+  int64 collectionID = 2;
+  int64 segmentID = 3;
 }
 
 message DescribeSegmentResponse {
-    common.Status status = 1;
-    int64 indexID = 2;
-    int64 buildID = 3;
-    bool enable_index = 4;
+  common.Status status = 1;
+  int64 indexID = 2;
+  int64 buildID = 3;
+  bool enable_index = 4;
 }
 
 message ShowSegmentsRequest {
-    common.MsgBase base = 1;
-    int64 collectionID = 2;
-    int64 partitionID = 3;
+  common.MsgBase base = 1;
+  int64 collectionID = 2;
+  int64 partitionID = 3;
 }
 
 message ShowSegmentsResponse {
-    common.Status status = 1;
-    repeated int64 segmentIDs = 2;
+  common.Status status = 1;
+  repeated int64 segmentIDs = 2;
 }
 
 /*
 * Create index for vector datas
 */
 message CreateIndexRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The particular collection name you want to create index.
-    string collection_name = 3;
-    // The vector field name in this particular collection
-    string field_name = 4;
-    // Support keys: index_type,metric_type, params. Different index_type may has different params.
-    repeated common.KeyValuePair extra_params = 5;
+  // Not useful for now
+  common.MsgBase base = 1; 
+  // Not useful for now
+  string db_name = 2;
+  // The particular collection name you want to create index.
+  string collection_name = 3;
+  // The vector field name in this particular collection
+  string field_name = 4;
+  // Support keys: index_type,metric_type, params. Different index_type may has different params.
+  repeated common.KeyValuePair extra_params = 5; 
 }
 
 /*
-* Get created index information.
+* Get created index information. 
 * Current release of Milvus only supports showing latest built index.
 */
 message DescribeIndexRequest {
-    // Not useful for now
-    common.MsgBase base = 1;
-    // Not useful for now
-    string db_name = 2;
-    // The particular collection name in Milvus
-    string collection_name = 3;
-    // The vector field name in this particular collection
-    string field_name = 4;
-    // No need to set up for now @2021.06.30
-    string index_name = 5;
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2;
+  // The particular collection name in Milvus
+  string collection_name = 3;
+   // The vector field name in this particular collection
+  string field_name = 4;
+  // No need to set up for now @2021.06.30
+  string index_name = 5; 
 }
 
 /*
 * Index informations
 */
 message IndexDescription {
-    // Index name
-    string index_name = 1;
-    // Index id
-    int64 indexID = 2;
-    // Will return index_type, metric_type, params(like nlist).
-    repeated common.KeyValuePair params = 3;
-    // The vector field name
-    string field_name = 4;
+  // Index name
+  string index_name = 1;
+  // Index id
+  int64 indexID = 2;
+  // Will return index_type, metric_type, params(like nlist).
+  repeated common.KeyValuePair params = 3;
+  // The vector field name
+  string field_name = 4;
 }
 
 /*
 * Describe index response
 */
 message DescribeIndexResponse {
-    // Response status
-    common.Status status = 1;
-    // All index informations, for now only return tha latest index you created for the collection.
-    repeated IndexDescription index_descriptions = 2;
+  // Response status
+  common.Status status = 1;
+  // All index informations, for now only return tha latest index you created for the collection.
+  repeated IndexDescription index_descriptions = 2;
 }
 
+/*
+*  Get index building progress 
+*/
 message GetIndexBuildProgressRequest {
-    common.MsgBase base = 1; // must
-    string db_name = 2 ;
-    string collection_name = 3; // must
-    string field_name = 4;
-    string index_name = 5; // must
+  // Not useful for now
+  common.MsgBase base = 1;
+  // Not useful for now
+  string db_name = 2 ;
+  // The collection name in milvus
+  string collection_name = 3;
+  // The vector field name in this collection
+  string field_name = 4;
+  // Not useful for now
+  string index_name = 5;
 }
 
 message GetIndexBuildProgressResponse {
-    common.Status status = 1;
-    int64 indexed_rows = 2;
-    int64 total_rows = 3;
+  common.Status status = 1;
+  int64 indexed_rows = 2;
+  int64 total_rows = 3;
 }
 
 message GetIndexStateRequest {
-    common.MsgBase base = 1; // must
-    string db_name = 2 ;
-    string collection_name = 3; // must
-    string field_name = 4;
-    string index_name = 5; // No need to set up for now @2021.06.30
+  common.MsgBase base = 1; // must
+  string db_name = 2 ;
+  string collection_name = 3; // must
+  string field_name = 4;
+  string index_name = 5; // No need to set up for now @2021.06.30
 }
 
 message GetIndexStateResponse {
-    common.Status status = 1;
-    common.IndexState state = 2;
-    string fail_reason = 3;
+  common.Status status = 1;
+  common.IndexState state = 2;
+  string fail_reason = 3;
 }
 
 message DropIndexRequest {
-    common.MsgBase base = 1; // must
-    string db_name = 2;
-    string collection_name = 3; // must
-    string field_name = 4;
-    string index_name = 5; // No need to set up for now @2021.06.30
+  common.MsgBase base = 1; // must
+  string db_name = 2;
+  string collection_name = 3; // must
+  string field_name = 4;
+  string index_name = 5; // No need to set up for now @2021.06.30
 }
 
 message InsertRequest {
-    common.MsgBase base = 1;
-    string db_name = 2;
-    string collection_name = 3;
-    string partition_name = 4;
-    repeated schema.FieldData fields_data = 5;
-    repeated uint32 hash_keys = 6;
-    uint32 num_rows = 7;
+  common.MsgBase base = 1;
+  string db_name = 2;
+  string collection_name = 3;
+  string partition_name = 4;
+  repeated schema.FieldData fields_data = 5;
+  repeated uint32 hash_keys = 6;
+  uint32 num_rows = 7;
 }
 
 message MutationResult {
-    common.Status status = 1;
-    schema.IDs IDs = 2; // required for insert, delete
-    repeated uint32 succ_index = 3; // error indexes indicate
-    repeated uint32 err_index = 4; // error indexes indicate
-    bool acknowledged = 5;
-    int64 insert_cnt = 6;
-    int64 delete_cnt = 7;
-    int64 upsert_cnt = 8;
-    uint64 timestamp = 9;
+  common.Status status = 1;
+  schema.IDs IDs = 2; // required for insert, delete
+  repeated uint32 succ_index = 3; // error indexes indicate
+  repeated uint32 err_index = 4; // error indexes indicate
+  bool acknowledged = 5;
+  int64 insert_cnt = 6;
+  int64 delete_cnt = 7;
+  int64 upsert_cnt = 8;
+  uint64 timestamp = 9;
 }
 
 message DeleteRequest {
-    common.MsgBase base = 1;
-    string db_name = 2;
-    string collection_name = 3;
-    string partition_name = 4;
-    string expr = 5;
-    repeated uint32 hash_keys = 6;
+  common.MsgBase base = 1;
+  string db_name = 2;
+  string collection_name = 3;
+  string partition_name = 4;
+  string expr = 5;
+  repeated uint32 hash_keys = 6;
 }
 
 enum PlaceholderType {
-    None = 0;
-    BinaryVector = 100;
-    FloatVector = 101;
+  None = 0;
+  BinaryVector = 100;
+  FloatVector = 101;
 }
 
 message PlaceholderValue {
-    string tag = 1;
-    PlaceholderType type = 2;
-    // values is a 2d-array, every array contains a vector
-    repeated bytes values = 3;
+  string tag = 1;
+  PlaceholderType type = 2;
+  // values is a 2d-array, every array contains a vector
+  repeated bytes values = 3;
 }
 
 message PlaceholderGroup {
-    repeated PlaceholderValue placeholders = 1;
+  repeated PlaceholderValue placeholders = 1;
 }
 
 message SearchRequest {
-    common.MsgBase base = 1; // must
-    string db_name = 2;
-    string collection_name = 3; // must
-    repeated string partition_names = 4; // must
-    string dsl = 5; // must
-    // serialized `PlaceholderGroup`
-    bytes placeholder_group = 6; // must
-    common.DslType dsl_type = 7; // must
-    repeated string output_fields = 8;
-    repeated common.KeyValuePair search_params = 9; // must
-    uint64 travel_timestamp = 10;
-    uint64 guarantee_timestamp = 11; // guarantee_timestamp
+  common.MsgBase base = 1; // must
+  string db_name = 2;
+  string collection_name = 3; // must
+  repeated string partition_names = 4; // must
+  string dsl = 5; // must
+  // serialized `PlaceholderGroup`
+  bytes placeholder_group = 6; // must
+  common.DslType dsl_type = 7; // must
+  repeated string output_fields = 8;
+  repeated common.KeyValuePair search_params = 9; // must
+  uint64 travel_timestamp = 10;
+  uint64 guarantee_timestamp = 11; // guarantee_timestamp
 }
 
 message Hits {
-    repeated int64 IDs = 1;
-    repeated bytes row_data = 2;
-    repeated float scores = 3;
+  repeated int64 IDs = 1;
+  repeated bytes row_data = 2;
+  repeated float scores = 3;
 }
 
 message SearchResults {
-    common.Status status = 1;
-    schema.SearchResultData results = 2;
+  common.Status status = 1;
+  schema.SearchResultData results = 2;
 }
 
 message FlushRequest {
-    common.MsgBase base = 1;
-    string db_name = 2;
-    repeated string collection_names = 3;
+  common.MsgBase base = 1;
+  string db_name = 2;
+  repeated string collection_names = 3;
 }
 
 message FlushResponse{
-    common.Status status = 1;
-    string db_name = 2;
-    map<string, schema.LongArray> coll_segIDs = 3;
+  common.Status status = 1;
+  string db_name = 2;
+  map<string, schema.LongArray> coll_segIDs = 3;
 }
 
 message QueryRequest {
-    common.MsgBase base = 1;
-    string db_name = 2;
-    string collection_name = 3;
-    string expr = 4;
-    repeated string output_fields = 5;
-    repeated string partition_names = 6;
-    uint64 travel_timestamp = 7;
-    uint64 guarantee_timestamp = 8; // guarantee_timestamp
+  common.MsgBase base = 1;
+  string db_name = 2;
+  string collection_name = 3;
+  string expr = 4;
+  repeated string output_fields = 5;
+  repeated string partition_names = 6;
+  uint64 travel_timestamp = 7;
+  uint64 guarantee_timestamp = 8; // guarantee_timestamp
 }
 
 message QueryResults {
-    common.Status status = 1;
-    repeated schema.FieldData fields_data = 2;
+  common.Status status = 1;
+  repeated schema.FieldData fields_data = 2;
 }
 
 message VectorIDs {
-    string collection_name = 1;
-    string field_name = 2;
-    schema.IDs id_array = 3;
-    repeated string partition_names = 4;
+  string collection_name = 1;
+  string field_name = 2;
+  schema.IDs id_array = 3;
+  repeated string partition_names = 4;
 }
 
 message VectorsArray {
-    oneof array {
-        VectorIDs id_array = 1; // vector ids
-        schema.VectorField data_array = 2; // vectors data
-    }
+  oneof array {
+    VectorIDs id_array = 1; // vector ids
+    schema.VectorField data_array = 2; // vectors data
+  } 
 }
 
 message CalcDistanceRequest {
-    common.MsgBase base = 1;
-    VectorsArray op_left = 2; // vectors on the left of operator
-    VectorsArray op_right = 3; // vectors on the right of operator
-    repeated common.KeyValuePair params = 4; // "metric":"L2"/"IP"/"HAMMIN"/"TANIMOTO"
+  common.MsgBase base = 1;
+  VectorsArray op_left = 2; // vectors on the left of operator
+  VectorsArray op_right = 3; // vectors on the right of operator
+  repeated common.KeyValuePair params = 4; // "metric":"L2"/"IP"/"HAMMIN"/"TANIMOTO"
 }
 
 message CalcDistanceResults {
-    common.Status status = 1;
-    // num(op_left)*num(op_right) distance values, "HAMMIN" return integer distance
-    oneof array {
-        schema.IntArray int_dist = 2;
-        schema.FloatArray float_dist = 3;
-    }
+  common.Status status = 1;
+  // num(op_left)*num(op_right) distance values, "HAMMIN" return integer distance
+  oneof array {
+    	schema.IntArray int_dist = 2;
+	schema.FloatArray float_dist = 3;
+  }
 }
 
 message PersistentSegmentInfo {
-    int64 segmentID = 1;
-    int64 collectionID = 2;
-    int64 partitionID = 3;
-    int64 num_rows = 4;
-    common.SegmentState state = 5;
+  int64 segmentID = 1;
+  int64 collectionID = 2;
+  int64 partitionID = 3;
+  int64 num_rows = 4;
+  common.SegmentState state = 5;
 }
 
 message GetPersistentSegmentInfoRequest {
-    common.MsgBase base = 1; // must
-    string dbName = 2;
-    string collectionName = 3; // must
+  common.MsgBase base = 1; // must
+  string dbName = 2;
+  string collectionName = 3; // must
 }
 
 message GetPersistentSegmentInfoResponse {
-    common.Status status = 1;
-    repeated PersistentSegmentInfo infos = 2;
+  common.Status status = 1;
+  repeated PersistentSegmentInfo infos = 2;
 }
 
 message QuerySegmentInfo {
-    int64 segmentID = 1;
-    int64 collectionID = 2;
-    int64 partitionID = 3;
-    int64 mem_size = 4;
-    int64 num_rows = 5;
-    string index_name = 6;
-    int64 indexID = 7;
-    int64 nodeID = 8;
-    common.SegmentState state = 9;
+  int64 segmentID = 1;
+  int64 collectionID = 2;
+  int64 partitionID = 3;
+  int64 mem_size = 4;
+  int64 num_rows = 5;
+  string index_name = 6;
+  int64 indexID = 7;
+  int64 nodeID = 8;
+  common.SegmentState state = 9;
 }
 
 message GetQuerySegmentInfoRequest {
-    common.MsgBase base = 1; // must
-    string dbName = 2;
-    string collectionName = 3; // must
+  common.MsgBase base = 1; // must
+  string dbName = 2;
+  string collectionName = 3; // must
 }
 
 message GetQuerySegmentInfoResponse {
-    common.Status status = 1;
-    repeated QuerySegmentInfo infos = 2;
+  common.Status status = 1;
+  repeated QuerySegmentInfo infos = 2;
 }
 
 message DummyRequest {
-    string request_type = 1;
+  string request_type = 1;
 }
 
 message DummyResponse {
-    string response = 1;
+  string response = 1;
 }
 
 message RegisterLinkRequest {
 }
 
 message RegisterLinkResponse {
-    common.Address address = 1;
-    common.Status status = 2;
+  common.Address address = 1;
+  common.Status status = 2;
 }
 
 message GetMetricsRequest {
-    common.MsgBase base = 1;
-    string request = 2; // request is of jsonic format
+  common.MsgBase base = 1;
+  string request = 2; // request is of jsonic format
 }
 
 message GetMetricsResponse {
-    common.Status status = 1;
-    string response = 2;  // response is of jsonic format
-    string component_name = 3; // metrics from which component
+  common.Status status = 1;
+  string response = 2;  // response is of jsonic format
+  string component_name = 3; // metrics from which component
 }
 
-service ProxyService {
-    rpc RegisterLink(RegisterLinkRequest) returns (RegisterLinkResponse) {}
+/*
+* Do load balancing operation from src_nodeID to dst_nodeID.
+*/
+message LoadBalanceRequest {
+  common.MsgBase base = 1;
+  int64 src_nodeID = 2;
+  repeated int64 dst_nodeIDs = 3;
+  repeated int64 sealed_segmentIDs = 4;
+}
+
+message ManualCompactionRequest {
+  int64 collectionID = 1;
+  uint64 timetravel = 2;
+}
+
+message ManualCompactionResponse {
+  common.Status status = 1;
+  int64 compactionID = 2;
+}
+
+message GetCompactionStateRequest {
+  int64 compactionID = 1;
 }
 
+message GetCompactionStateResponse {
+  common.Status status = 1;
+  common.CompactionState state = 2;
+  int64 executingPlanNo = 3;
+  int64 timeoutPlanNo = 4;
+  int64 completedPlanNo = 5;
+}
+
+message GetCompactionPlansRequest {
+  int64 compactionID = 1;
+}
+
+message GetCompactionPlansResponse {
+  common.Status status = 1;
+  common.CompactionState state = 2;
+  repeated CompactionMergeInfo mergeInfos = 3;
+}
+
+message CompactionMergeInfo {
+  repeated int64 sources = 1;
+  int64 target = 2;
+}
+
+service ProxyService {
+  rpc RegisterLink(RegisterLinkRequest) returns (RegisterLinkResponse) {}
+}

src/test/java/io/milvus/client/MilvusServiceClientTest.java

@@ -67,6 +67,7 @@ class MilvusServiceClientTest {
         return new MilvusServiceClient(connectParam);
     }
 
+    @SuppressWarnings("unchecked")
     private <T, P> void testFuncByName(String funcName, T param) {
         try {
             Class<?> clientClass = MilvusServiceClient.class;