mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 09:38:39 +08:00
99 lines
2.3 KiB
Go
99 lines
2.3 KiB
Go
package indexbuilder
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
"github.com/zilliztech/milvus-distributed/internal/proto/commonpb"
|
|
"github.com/zilliztech/milvus-distributed/internal/proto/indexbuilderpb"
|
|
"github.com/zilliztech/milvus-distributed/internal/util/typeutil"
|
|
)
|
|
|
|
type UniqueID = typeutil.UniqueID
|
|
|
|
type Client struct {
|
|
client indexbuilderpb.IndexBuildServiceClient
|
|
}
|
|
|
|
type IndexStatus int32
|
|
|
|
const (
|
|
NOTEXIST IndexStatus = 0
|
|
UNISSUED IndexStatus = 1
|
|
INPROGRESS IndexStatus = 2
|
|
FINISHED IndexStatus = 3
|
|
)
|
|
|
|
type IndexDescription struct {
|
|
ID UniqueID
|
|
Status IndexStatus
|
|
EnqueueTime time.Time
|
|
ScheduleTime time.Time
|
|
BuildCompleteTime time.Time
|
|
}
|
|
|
|
func NewBuildIndexClient(conn *grpc.ClientConn) *Client {
|
|
return &Client{
|
|
client: indexbuilderpb.NewIndexBuildServiceClient(conn),
|
|
}
|
|
}
|
|
|
|
func (c *Client) BuildIndexWithoutID(columnDataPaths []string, typeParams map[string]string, indexParams map[string]string) (UniqueID, error) {
|
|
//first new a build service client
|
|
|
|
var typeParamsKV []*commonpb.KeyValuePair
|
|
for typeParam := range typeParams {
|
|
typeParamsKV = append(typeParamsKV, &commonpb.KeyValuePair{
|
|
Key: typeParam,
|
|
Value: typeParams[typeParam],
|
|
})
|
|
}
|
|
|
|
var indexParamsKV []*commonpb.KeyValuePair
|
|
for indexParam := range indexParams {
|
|
indexParamsKV = append(indexParamsKV, &commonpb.KeyValuePair{
|
|
Key: indexParam,
|
|
Value: indexParams[indexParam],
|
|
})
|
|
}
|
|
|
|
ctx := context.TODO()
|
|
requset := &indexbuilderpb.BuildIndexRequest{
|
|
DataPaths: columnDataPaths,
|
|
TypeParams: typeParamsKV,
|
|
IndexParams: indexParamsKV,
|
|
}
|
|
response, err := c.client.BuildIndex(ctx, requset)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
indexID := response.IndexID
|
|
return indexID, err
|
|
}
|
|
|
|
func (c *Client) DescribeIndex(indexID UniqueID) (IndexDescription, error) {
|
|
//ctx := context.TODO()
|
|
//request := &indexbuilderpb.DescribleIndexRequest{
|
|
// IndexID: indexID,
|
|
//}
|
|
//response, err := c.client.DescribeIndex(ctx, request)
|
|
//if err != nil {
|
|
// return IndexDescription{}, err
|
|
//}
|
|
//
|
|
//indexDescription := &IndexDescription{
|
|
// ID: indexID,
|
|
// Status: response.IndexStatus,
|
|
// EnqueueTime: time.Unix(),
|
|
//}
|
|
return IndexDescription{}, nil
|
|
}
|
|
|
|
func (c *Client) GetIndexFilePaths(IndexID UniqueID) ([]string, error) {
|
|
|
|
return nil, nil
|
|
}
|