mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 05:18:52 +08:00
83 lines
2.2 KiB
Go
83 lines
2.2 KiB
Go
package indexbuilder
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/zilliztech/milvus-distributed/internal/proto/commonpb"
|
|
"github.com/zilliztech/milvus-distributed/internal/proto/indexbuilderpb"
|
|
)
|
|
|
|
const (
|
|
reqTimeoutInterval = time.Second * 10
|
|
)
|
|
|
|
func (b *Builder) BuildIndex(ctx context.Context, request *indexbuilderpb.BuildIndexRequest) (*indexbuilderpb.BuildIndexResponse, error) {
|
|
t := NewIndexAddTask()
|
|
t.req = request
|
|
t.idAllocator = b.idAllocator
|
|
t.buildQueue = b.sched.IndexBuildQueue
|
|
t.table = b.metaTable
|
|
t.kv = b.kv
|
|
var cancel func()
|
|
t.ctx, cancel = context.WithTimeout(ctx, reqTimeoutInterval)
|
|
defer cancel()
|
|
|
|
fn := func() error {
|
|
select {
|
|
case <-ctx.Done():
|
|
return errors.New("insert timeout")
|
|
default:
|
|
return b.sched.IndexAddQueue.Enqueue(t)
|
|
}
|
|
}
|
|
ret := &indexbuilderpb.BuildIndexResponse{
|
|
Status: &commonpb.Status{
|
|
ErrorCode: commonpb.ErrorCode_SUCCESS,
|
|
},
|
|
}
|
|
|
|
err := fn()
|
|
if err != nil {
|
|
ret.Status.ErrorCode = commonpb.ErrorCode_UNEXPECTED_ERROR
|
|
ret.Status.Reason = err.Error()
|
|
return ret, nil
|
|
}
|
|
|
|
err = t.WaitToFinish()
|
|
if err != nil {
|
|
ret.Status.ErrorCode = commonpb.ErrorCode_UNEXPECTED_ERROR
|
|
ret.Status.Reason = err.Error()
|
|
return ret, nil
|
|
}
|
|
ret.IndexID = t.indexID
|
|
return ret, nil
|
|
}
|
|
|
|
func (b *Builder) DescribeIndex(ctx context.Context, request *indexbuilderpb.DescribleIndexRequest) (*indexbuilderpb.DescribleIndexResponse, error) {
|
|
indexID := request.IndexID
|
|
ret, err := b.metaTable.GetIndexDescription(indexID)
|
|
ret.Status = &commonpb.Status{ErrorCode: commonpb.ErrorCode_SUCCESS}
|
|
ret.IndexID = indexID
|
|
if err != nil {
|
|
ret.Status.ErrorCode = commonpb.ErrorCode_UNEXPECTED_ERROR
|
|
ret.Status.Reason = err.Error()
|
|
}
|
|
return ret, nil
|
|
}
|
|
|
|
func (b *Builder) GetIndexFilePaths(ctx context.Context, request *indexbuilderpb.GetIndexFilePathsRequest) (*indexbuilderpb.GetIndexFilePathsResponse, error) {
|
|
ret := &indexbuilderpb.GetIndexFilePathsResponse{
|
|
Status: &commonpb.Status{ErrorCode: commonpb.ErrorCode_SUCCESS},
|
|
IndexID: request.IndexID,
|
|
}
|
|
filePaths, err := b.metaTable.GetIndexFilePaths(request.IndexID)
|
|
if err != nil {
|
|
ret.Status.ErrorCode = commonpb.ErrorCode_UNEXPECTED_ERROR
|
|
ret.Status.Reason = err.Error()
|
|
}
|
|
ret.IndexFilePaths = filePaths
|
|
return ret, nil
|
|
}
|