syntax = "proto3";

package milvus.proto.index;

option go_package = "github.com/milvus-io/milvus/internal/proto/indexpb";

import "common.proto";
import "internal.proto";
import "milvus.proto";

service IndexCoord {
  rpc GetComponentStates(milvus.GetComponentStatesRequest) returns (milvus.ComponentStates) {}
  rpc GetStatisticsChannel(internal.GetStatisticsChannelRequest) returns(milvus.StringResponse){}
  rpc CreateIndex(CreateIndexRequest) returns (common.Status){}
  // Deprecated: use DescribeIndex instead
  rpc GetIndexState(GetIndexStateRequest) returns (GetIndexStateResponse) {}
  rpc GetSegmentIndexState(GetSegmentIndexStateRequest) returns (GetSegmentIndexStateResponse) {}
  rpc GetIndexInfos(GetIndexInfoRequest) returns (GetIndexInfoResponse){}
  rpc DropIndex(DropIndexRequest) returns (common.Status) {}
  rpc DescribeIndex(DescribeIndexRequest) returns (DescribeIndexResponse) {}
  rpc GetIndexStatistics(GetIndexStatisticsRequest) returns (GetIndexStatisticsResponse) {}
  // Deprecated: use DescribeIndex instead
  rpc GetIndexBuildProgress(GetIndexBuildProgressRequest) returns (GetIndexBuildProgressResponse) {}

  rpc ShowConfigurations(internal.ShowConfigurationsRequest) returns (internal.ShowConfigurationsResponse){}
  // https://wiki.lfaidata.foundation/display/MIL/MEP+8+--+Add+metrics+for+proxy
  rpc GetMetrics(milvus.GetMetricsRequest) returns (milvus.GetMetricsResponse) {}

  rpc CheckHealth(milvus.CheckHealthRequest) returns (milvus.CheckHealthResponse) {}
}

service IndexNode {
  rpc GetComponentStates(milvus.GetComponentStatesRequest) returns (milvus.ComponentStates) {}
  rpc GetStatisticsChannel(internal.GetStatisticsChannelRequest) returns(milvus.StringResponse){}
  rpc CreateJob(CreateJobRequest) returns (common.Status) {}
  rpc QueryJobs(QueryJobsRequest) returns (QueryJobsResponse) {}
  rpc DropJobs(DropJobsRequest) returns (common.Status) {}
  rpc GetJobStats(GetJobStatsRequest) returns (GetJobStatsResponse) {}

  rpc ShowConfigurations(internal.ShowConfigurationsRequest) returns (internal.ShowConfigurationsResponse){}
  // https://wiki.lfaidata.foundation/display/MIL/MEP+8+--+Add+metrics+for+proxy
  rpc GetMetrics(milvus.GetMetricsRequest) returns (milvus.GetMetricsResponse) {}
}

message IndexInfo {
  int64 collectionID = 1;
  int64 fieldID = 2;
  string index_name = 3;
  int64 indexID = 4;
  repeated common.KeyValuePair type_params = 5;
  repeated common.KeyValuePair index_params = 6;
  // index build progress
  // The real-time statistics may not be expected due to the existence of the compaction mechanism.
  int64 indexed_rows = 7;
  int64 total_rows = 8;
  // index state
  common.IndexState state = 9;
  string index_state_fail_reason = 10;
  bool is_auto_index = 11;
  repeated common.KeyValuePair user_index_params = 12;
  int64 pending_index_rows = 13;
}

message FieldIndex {
  IndexInfo index_info = 1;
  bool deleted = 2;
  uint64 create_time = 3;
}

message SegmentIndex {
  int64 collectionID = 1;
  int64 partitionID = 2;
  int64 segmentID = 3;
  int64 num_rows = 4;
  int64 indexID = 5;
  int64 buildID = 6;
  int64 nodeID = 7;
  int64 index_version = 8;
  common.IndexState state = 9;
  string fail_reason = 10;
  repeated string index_file_keys = 11;
  bool deleted = 12;
  uint64 create_time = 13;
  uint64 serialize_size = 14;
  bool write_handoff = 15;
  int32 current_index_version = 16;
}

message RegisterNodeRequest {
  common.MsgBase base = 1;
  common.Address address = 2;
  int64 nodeID = 3;
}

message RegisterNodeResponse {
  common.Status status = 1;
  internal.InitParams init_params = 2;
}

message GetIndexStateRequest {
  int64 collectionID = 1;
  string index_name = 2;
}

message GetIndexStateResponse {
  common.Status status = 1;
  common.IndexState state = 2;
  string fail_reason = 3;
}

message GetSegmentIndexStateRequest {
  int64 collectionID = 1;
  string index_name = 2;
  repeated int64 segmentIDs = 3;
}

message SegmentIndexState {
  int64 segmentID = 1;
  common.IndexState state = 2;
  string fail_reason = 3;
}

message GetSegmentIndexStateResponse {
  common.Status status = 1;
  repeated SegmentIndexState states = 2;
}

message CreateIndexRequest {
  int64 collectionID = 1;
  int64 fieldID = 2;
  string index_name = 3;
  repeated common.KeyValuePair type_params = 4;
  repeated common.KeyValuePair index_params = 5;
  uint64 timestamp = 6;
  bool  is_auto_index = 7;
  repeated common.KeyValuePair user_index_params = 8;
}

message GetIndexInfoRequest {
  int64 collectionID = 1;
  repeated int64 segmentIDs = 2;
  string index_name = 3;
}

message IndexFilePathInfo {
  int64 segmentID = 1;
  int64 fieldID = 2;
  int64 indexID = 3;
  int64 buildID = 4;
  string index_name = 5;
  repeated common.KeyValuePair index_params = 6;
  repeated string index_file_paths = 7;
  uint64 serialized_size = 8;
  int64 index_version = 9;
  int64 num_rows = 10;
  int32 current_index_version = 11;
}

message SegmentInfo {
  int64 collectionID = 1;
  int64 segmentID = 2;
  bool enable_index = 3;
  repeated IndexFilePathInfo index_infos = 4;
}

message GetIndexInfoResponse {
  common.Status status = 1;
  map<int64, SegmentInfo> segment_info = 2;
}

message DropIndexRequest {
  int64 collectionID = 1;
  repeated int64 partitionIDs = 2;
  string index_name = 3;
  bool drop_all = 4;
}

message DescribeIndexRequest {
  int64 collectionID = 1;
  string index_name = 2;
  uint64 timestamp = 3;
}

message DescribeIndexResponse {
  common.Status status = 1;
  repeated IndexInfo index_infos = 2;
}

message GetIndexBuildProgressRequest {
  int64 collectionID = 1;
  string index_name = 2;
}

message GetIndexBuildProgressResponse {
  common.Status status = 1;
  int64 indexed_rows = 2;
  int64 total_rows = 3;
  int64 pending_index_rows = 4;
}

message StorageConfig {
  string address = 1;
  string access_keyID = 2;
  string secret_access_key = 3;
  bool useSSL = 4;
  string bucket_name = 5;
  string root_path = 6;
  bool useIAM = 7;
  string IAMEndpoint = 8;
  string storage_type = 9;
  bool use_virtual_host = 10;
  string region = 11;
  string cloud_provider = 12;
  int64 request_timeout_ms = 13;
}

message CreateJobRequest {
  string clusterID = 1;
  string index_file_prefix = 2;
  int64 buildID = 3;
  repeated string data_paths = 4;
  int64 index_version = 5;
  int64 indexID = 6;
  string index_name = 7;
  StorageConfig storage_config = 8;
  repeated common.KeyValuePair index_params = 9;
  repeated common.KeyValuePair type_params = 10;
  int64 num_rows = 11;
  int32 current_index_version = 12;
}

message QueryJobsRequest {
  string clusterID = 1;
  repeated int64 buildIDs = 2;
}

message IndexTaskInfo {
  int64 buildID = 1;
  common.IndexState state = 2;
  repeated string index_file_keys = 3;
  uint64 serialized_size = 4;
  string fail_reason = 5;
  int32 current_index_version = 6;
}

message QueryJobsResponse {
  common.Status status = 1;
  string clusterID = 2;
  repeated IndexTaskInfo index_infos = 3;
}

message DropJobsRequest {
  string clusterID = 1;
  repeated int64 buildIDs = 2;
}

message JobInfo {
  int64 num_rows = 1;
  int64 dim = 2;
  int64 start_time = 3;
  int64 end_time = 4;
  repeated common.KeyValuePair index_params = 5;
  int64 podID = 6;
}

message GetJobStatsRequest {
}

message GetJobStatsResponse {
  common.Status status = 1;
  int64 total_job_num = 2;
  int64 in_progress_job_num = 3;
  int64 enqueue_job_num = 4;
  int64 task_slots = 5;
  repeated JobInfo job_infos = 6;
  bool enable_disk = 7;
}

message GetIndexStatisticsRequest {
  int64 collectionID = 1;
  string index_name = 2;
}

message GetIndexStatisticsResponse {
  common.Status status = 1;
  repeated IndexInfo index_infos = 2;
}