mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-08 01:58:34 +08:00
Remove querycoord grpc error (#12535)
Signed-off-by: godchen <qingxiang.chen@zilliz.com>
This commit is contained in:
parent
3f060f4028
commit
a76f490f12
@ -101,7 +101,7 @@ func (qc *QueryCoord) ShowCollections(ctx context.Context, req *querypb.ShowColl
|
|||||||
log.Debug("show collection end with query coordinator not healthy")
|
log.Debug("show collection end with query coordinator not healthy")
|
||||||
return &querypb.ShowCollectionsResponse{
|
return &querypb.ShowCollectionsResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
collectionInfos := qc.meta.showCollections()
|
collectionInfos := qc.meta.showCollections()
|
||||||
ID2collectionInfo := make(map[UniqueID]*querypb.CollectionInfo)
|
ID2collectionInfo := make(map[UniqueID]*querypb.CollectionInfo)
|
||||||
@ -129,7 +129,7 @@ func (qc *QueryCoord) ShowCollections(ctx context.Context, req *querypb.ShowColl
|
|||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return &querypb.ShowCollectionsResponse{
|
return &querypb.ShowCollectionsResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
inMemoryPercentages = append(inMemoryPercentages, ID2collectionInfo[id].InMemoryPercentage)
|
inMemoryPercentages = append(inMemoryPercentages, ID2collectionInfo[id].InMemoryPercentage)
|
||||||
}
|
}
|
||||||
@ -155,7 +155,7 @@ func (qc *QueryCoord) LoadCollection(ctx context.Context, req *querypb.LoadColle
|
|||||||
err := errors.New("query coordinator is not healthy")
|
err := errors.New("query coordinator is not healthy")
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
log.Debug("load collection end with query coordinator not healthy")
|
log.Debug("load collection end with query coordinator not healthy")
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
baseTask := newBaseTask(qc.loopCtx, querypb.TriggerCondition_grpcRequest)
|
baseTask := newBaseTask(qc.loopCtx, querypb.TriggerCondition_grpcRequest)
|
||||||
@ -172,14 +172,14 @@ func (qc *QueryCoord) LoadCollection(ctx context.Context, req *querypb.LoadColle
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
err = loadCollectionTask.waitToFinish()
|
err = loadCollectionTask.waitToFinish()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("LoadCollectionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", collectionID), zap.Any("status", status))
|
log.Debug("LoadCollectionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", collectionID), zap.Any("status", status))
|
||||||
@ -199,7 +199,7 @@ func (qc *QueryCoord) ReleaseCollection(ctx context.Context, req *querypb.Releas
|
|||||||
err := errors.New("query coordinator is not healthy")
|
err := errors.New("query coordinator is not healthy")
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
log.Debug("release collection end with query coordinator not healthy")
|
log.Debug("release collection end with query coordinator not healthy")
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
hasCollection := qc.meta.hasCollection(collectionID)
|
hasCollection := qc.meta.hasCollection(collectionID)
|
||||||
@ -220,14 +220,14 @@ func (qc *QueryCoord) ReleaseCollection(ctx context.Context, req *querypb.Releas
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
err = releaseCollectionTask.waitToFinish()
|
err = releaseCollectionTask.waitToFinish()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("ReleaseCollectionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", collectionID))
|
log.Debug("ReleaseCollectionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", collectionID))
|
||||||
@ -250,7 +250,7 @@ func (qc *QueryCoord) ShowPartitions(ctx context.Context, req *querypb.ShowParti
|
|||||||
log.Debug("show partition end with query coordinator not healthy")
|
log.Debug("show partition end with query coordinator not healthy")
|
||||||
return &querypb.ShowPartitionsResponse{
|
return &querypb.ShowPartitionsResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
partitionStates, err := qc.meta.showPartitions(collectionID)
|
partitionStates, err := qc.meta.showPartitions(collectionID)
|
||||||
@ -259,7 +259,7 @@ func (qc *QueryCoord) ShowPartitions(ctx context.Context, req *querypb.ShowParti
|
|||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return &querypb.ShowPartitionsResponse{
|
return &querypb.ShowPartitionsResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
ID2PartitionState := make(map[UniqueID]*querypb.PartitionStates)
|
ID2PartitionState := make(map[UniqueID]*querypb.PartitionStates)
|
||||||
inMemoryPartitionIDs := make([]UniqueID, 0)
|
inMemoryPartitionIDs := make([]UniqueID, 0)
|
||||||
@ -286,7 +286,7 @@ func (qc *QueryCoord) ShowPartitions(ctx context.Context, req *querypb.ShowParti
|
|||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return &querypb.ShowPartitionsResponse{
|
return &querypb.ShowPartitionsResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
inMemoryPercentages = append(inMemoryPercentages, ID2PartitionState[id].InMemoryPercentage)
|
inMemoryPercentages = append(inMemoryPercentages, ID2PartitionState[id].InMemoryPercentage)
|
||||||
}
|
}
|
||||||
@ -313,7 +313,7 @@ func (qc *QueryCoord) LoadPartitions(ctx context.Context, req *querypb.LoadParti
|
|||||||
err := errors.New("query coordinator is not healthy")
|
err := errors.New("query coordinator is not healthy")
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
log.Debug("load partition end with query coordinator not healthy")
|
log.Debug("load partition end with query coordinator not healthy")
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(partitionIDs) == 0 {
|
if len(partitionIDs) == 0 {
|
||||||
@ -321,7 +321,7 @@ func (qc *QueryCoord) LoadPartitions(ctx context.Context, req *querypb.LoadParti
|
|||||||
err := errors.New("partitionIDs are empty")
|
err := errors.New("partitionIDs are empty")
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
log.Debug("LoadPartitionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", req.CollectionID))
|
log.Debug("LoadPartitionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", req.CollectionID))
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
hasCollection := qc.meta.hasCollection(collectionID)
|
hasCollection := qc.meta.hasCollection(collectionID)
|
||||||
@ -365,7 +365,7 @@ func (qc *QueryCoord) LoadPartitions(ctx context.Context, req *querypb.LoadParti
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
err = loadPartitionTask.waitToFinish()
|
err = loadPartitionTask.waitToFinish()
|
||||||
@ -373,7 +373,7 @@ func (qc *QueryCoord) LoadPartitions(ctx context.Context, req *querypb.LoadParti
|
|||||||
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
log.Debug("LoadPartitionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", req.CollectionID))
|
log.Debug("LoadPartitionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", req.CollectionID))
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("LoadPartitionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", req.CollectionID))
|
log.Debug("LoadPartitionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", req.CollectionID))
|
||||||
@ -394,7 +394,7 @@ func (qc *QueryCoord) ReleasePartitions(ctx context.Context, req *querypb.Releas
|
|||||||
err := errors.New("query coordinator is not healthy")
|
err := errors.New("query coordinator is not healthy")
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
log.Debug("release partition end with query coordinator not healthy")
|
log.Debug("release partition end with query coordinator not healthy")
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
hasCollection := qc.meta.hasCollection(collectionID)
|
hasCollection := qc.meta.hasCollection(collectionID)
|
||||||
@ -408,7 +408,7 @@ func (qc *QueryCoord) ReleasePartitions(ctx context.Context, req *querypb.Releas
|
|||||||
err := errors.New("partitionIDs are empty")
|
err := errors.New("partitionIDs are empty")
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
log.Debug("releasePartitionsRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", req.CollectionID))
|
log.Debug("releasePartitionsRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", req.CollectionID))
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
toReleasedPartitions := make([]UniqueID, 0)
|
toReleasedPartitions := make([]UniqueID, 0)
|
||||||
@ -434,14 +434,14 @@ func (qc *QueryCoord) ReleasePartitions(ctx context.Context, req *querypb.Releas
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
err = releasePartitionTask.waitToFinish()
|
err = releasePartitionTask.waitToFinish()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
log.Debug("ReleasePartitionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", collectionID), zap.Int64s("partitionIDs", partitionIDs))
|
log.Debug("ReleasePartitionRequest completed", zap.String("role", Params.RoleName), zap.Int64("msgID", req.Base.MsgID), zap.Int64("collectionID", collectionID), zap.Int64s("partitionIDs", partitionIDs))
|
||||||
//qc.MetaReplica.printMeta()
|
//qc.MetaReplica.printMeta()
|
||||||
@ -461,7 +461,7 @@ func (qc *QueryCoord) CreateQueryChannel(ctx context.Context, req *querypb.Creat
|
|||||||
log.Debug("createQueryChannel end with query coordinator not healthy")
|
log.Debug("createQueryChannel end with query coordinator not healthy")
|
||||||
return &querypb.CreateQueryChannelResponse{
|
return &querypb.CreateQueryChannelResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
collectionID := req.CollectionID
|
collectionID := req.CollectionID
|
||||||
@ -472,7 +472,7 @@ func (qc *QueryCoord) CreateQueryChannel(ctx context.Context, req *querypb.Creat
|
|||||||
log.Debug("createQueryChannel end with error")
|
log.Debug("createQueryChannel end with error")
|
||||||
return &querypb.CreateQueryChannelResponse{
|
return &querypb.CreateQueryChannelResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return &querypb.CreateQueryChannelResponse{
|
return &querypb.CreateQueryChannelResponse{
|
||||||
@ -494,7 +494,7 @@ func (qc *QueryCoord) GetPartitionStates(ctx context.Context, req *querypb.GetPa
|
|||||||
log.Debug("getPartitionStates end with query coordinator not healthy")
|
log.Debug("getPartitionStates end with query coordinator not healthy")
|
||||||
return &querypb.GetPartitionStatesResponse{
|
return &querypb.GetPartitionStatesResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
partitionIDs := req.PartitionIDs
|
partitionIDs := req.PartitionIDs
|
||||||
@ -506,7 +506,7 @@ func (qc *QueryCoord) GetPartitionStates(ctx context.Context, req *querypb.GetPa
|
|||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return &querypb.GetPartitionStatesResponse{
|
return &querypb.GetPartitionStatesResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
partitionState := &querypb.PartitionStates{
|
partitionState := &querypb.PartitionStates{
|
||||||
PartitionID: partitionID,
|
PartitionID: partitionID,
|
||||||
@ -533,7 +533,7 @@ func (qc *QueryCoord) GetSegmentInfo(ctx context.Context, req *querypb.GetSegmen
|
|||||||
log.Debug("getSegmentInfo end with query coordinator not healthy")
|
log.Debug("getSegmentInfo end with query coordinator not healthy")
|
||||||
return &querypb.GetSegmentInfoResponse{
|
return &querypb.GetSegmentInfoResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
totalMemSize := int64(0)
|
totalMemSize := int64(0)
|
||||||
@ -547,7 +547,7 @@ func (qc *QueryCoord) GetSegmentInfo(ctx context.Context, req *querypb.GetSegmen
|
|||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return &querypb.GetSegmentInfoResponse{
|
return &querypb.GetSegmentInfoResponse{
|
||||||
Status: status,
|
Status: status,
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
for _, info := range segmentInfos {
|
for _, info := range segmentInfos {
|
||||||
totalNumRows += info.NumRows
|
totalNumRows += info.NumRows
|
||||||
@ -599,7 +599,7 @@ func (qc *QueryCoord) LoadBalance(ctx context.Context, req *querypb.LoadBalanceR
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
status.ErrorCode = commonpb.ErrorCode_UnexpectedError
|
||||||
status.Reason = err.Error()
|
status.Reason = err.Error()
|
||||||
return status, err
|
return status, nil
|
||||||
}
|
}
|
||||||
log.Debug("LoadBalanceRequest completed",
|
log.Debug("LoadBalanceRequest completed",
|
||||||
zap.String("role", Params.RoleName),
|
zap.String("role", Params.RoleName),
|
||||||
@ -648,7 +648,7 @@ func (qc *QueryCoord) GetMetrics(ctx context.Context, req *milvuspb.GetMetricsRe
|
|||||||
Reason: err.Error(),
|
Reason: err.Error(),
|
||||||
},
|
},
|
||||||
Response: "",
|
Response: "",
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("QueryCoord.GetMetrics",
|
log.Debug("QueryCoord.GetMetrics",
|
||||||
@ -673,7 +673,7 @@ func (qc *QueryCoord) GetMetrics(ctx context.Context, req *milvuspb.GetMetricsRe
|
|||||||
|
|
||||||
qc.metricsCacheManager.UpdateSystemInfoMetrics(metrics)
|
qc.metricsCacheManager.UpdateSystemInfoMetrics(metrics)
|
||||||
|
|
||||||
return metrics, err
|
return metrics, nil
|
||||||
}
|
}
|
||||||
err = errors.New(metricsinfo.MsgUnimplementedMetric)
|
err = errors.New(metricsinfo.MsgUnimplementedMetric)
|
||||||
log.Debug("QueryCoord.GetMetrics failed",
|
log.Debug("QueryCoord.GetMetrics failed",
|
||||||
@ -688,5 +688,5 @@ func (qc *QueryCoord) GetMetrics(ctx context.Context, req *milvuspb.GetMetricsRe
|
|||||||
Reason: err.Error(),
|
Reason: err.Error(),
|
||||||
},
|
},
|
||||||
Response: "",
|
Response: "",
|
||||||
}, err
|
}, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@ -55,7 +55,7 @@ func TestGrpcTask(t *testing.T) {
|
|||||||
PartitionIDs: []UniqueID{defaultPartitionID},
|
PartitionIDs: []UniqueID{defaultPartitionID},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test LoadEmptyPartition", func(t *testing.T) {
|
t.Run("Test LoadEmptyPartition", func(t *testing.T) {
|
||||||
@ -67,7 +67,7 @@ func TestGrpcTask(t *testing.T) {
|
|||||||
Schema: genCollectionSchema(defaultCollectionID, false),
|
Schema: genCollectionSchema(defaultCollectionID, false),
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test LoadPartition", func(t *testing.T) {
|
t.Run("Test LoadPartition", func(t *testing.T) {
|
||||||
@ -104,7 +104,7 @@ func TestGrpcTask(t *testing.T) {
|
|||||||
PartitionIDs: []UniqueID{-1},
|
PartitionIDs: []UniqueID{-1},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test ShowAllPartitions", func(t *testing.T) {
|
t.Run("Test ShowAllPartitions", func(t *testing.T) {
|
||||||
@ -174,7 +174,7 @@ func TestGrpcTask(t *testing.T) {
|
|||||||
CollectionIDs: []UniqueID{-1},
|
CollectionIDs: []UniqueID{-1},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test ShowAllCollections", func(t *testing.T) {
|
t.Run("Test ShowAllCollections", func(t *testing.T) {
|
||||||
@ -218,7 +218,7 @@ func TestGrpcTask(t *testing.T) {
|
|||||||
CollectionID: defaultCollectionID,
|
CollectionID: defaultCollectionID,
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test ReleaseNotExistPartition", func(t *testing.T) {
|
t.Run("Test ReleaseNotExistPartition", func(t *testing.T) {
|
||||||
@ -299,7 +299,7 @@ func TestGrpcTask(t *testing.T) {
|
|||||||
},
|
},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test GetMetrics", func(t *testing.T) {
|
t.Run("Test GetMetrics", func(t *testing.T) {
|
||||||
@ -326,7 +326,7 @@ func TestGrpcTask(t *testing.T) {
|
|||||||
Request: string(req),
|
Request: string(req),
|
||||||
})
|
})
|
||||||
|
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
|
|
||||||
metricReq = make(map[string]string)
|
metricReq = make(map[string]string)
|
||||||
@ -338,7 +338,7 @@ func TestGrpcTask(t *testing.T) {
|
|||||||
Request: string(req),
|
Request: string(req),
|
||||||
})
|
})
|
||||||
|
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
})
|
})
|
||||||
|
|
||||||
@ -376,7 +376,7 @@ func TestGrpcTaskEnqueueFail(t *testing.T) {
|
|||||||
Schema: genCollectionSchema(defaultCollectionID, false),
|
Schema: genCollectionSchema(defaultCollectionID, false),
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test LoadCollection", func(t *testing.T) {
|
t.Run("Test LoadCollection", func(t *testing.T) {
|
||||||
@ -388,7 +388,7 @@ func TestGrpcTaskEnqueueFail(t *testing.T) {
|
|||||||
Schema: genCollectionSchema(defaultCollectionID, false),
|
Schema: genCollectionSchema(defaultCollectionID, false),
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
queryCoord.scheduler.taskIDAllocator = taskIDAllocator
|
queryCoord.scheduler.taskIDAllocator = taskIDAllocator
|
||||||
@ -412,7 +412,7 @@ func TestGrpcTaskEnqueueFail(t *testing.T) {
|
|||||||
PartitionIDs: []UniqueID{defaultPartitionID},
|
PartitionIDs: []UniqueID{defaultPartitionID},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test ReleaseCollection", func(t *testing.T) {
|
t.Run("Test ReleaseCollection", func(t *testing.T) {
|
||||||
@ -423,7 +423,7 @@ func TestGrpcTaskEnqueueFail(t *testing.T) {
|
|||||||
CollectionID: defaultCollectionID,
|
CollectionID: defaultCollectionID,
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test LoadBalance", func(t *testing.T) {
|
t.Run("Test LoadBalance", func(t *testing.T) {
|
||||||
@ -529,7 +529,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
Schema: genCollectionSchema(defaultCollectionID, false),
|
Schema: genCollectionSchema(defaultCollectionID, false),
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test ShowPartitions", func(t *testing.T) {
|
t.Run("Test ShowPartitions", func(t *testing.T) {
|
||||||
@ -541,7 +541,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
PartitionIDs: []UniqueID{defaultPartitionID},
|
PartitionIDs: []UniqueID{defaultPartitionID},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test ShowAllPartitions", func(t *testing.T) {
|
t.Run("Test ShowAllPartitions", func(t *testing.T) {
|
||||||
@ -552,7 +552,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
CollectionID: defaultCollectionID,
|
CollectionID: defaultCollectionID,
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test GetPartitionStates", func(t *testing.T) {
|
t.Run("Test GetPartitionStates", func(t *testing.T) {
|
||||||
@ -564,7 +564,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
PartitionIDs: []UniqueID{defaultPartitionID},
|
PartitionIDs: []UniqueID{defaultPartitionID},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test LoadCollection", func(t *testing.T) {
|
t.Run("Test LoadCollection", func(t *testing.T) {
|
||||||
@ -576,7 +576,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
Schema: genCollectionSchema(defaultCollectionID, false),
|
Schema: genCollectionSchema(defaultCollectionID, false),
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test ShowCollections", func(t *testing.T) {
|
t.Run("Test ShowCollections", func(t *testing.T) {
|
||||||
@ -587,7 +587,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
CollectionIDs: []UniqueID{defaultCollectionID},
|
CollectionIDs: []UniqueID{defaultCollectionID},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test ShowAllCollections", func(t *testing.T) {
|
t.Run("Test ShowAllCollections", func(t *testing.T) {
|
||||||
@ -597,7 +597,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
},
|
},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test GetSegmentInfo", func(t *testing.T) {
|
t.Run("Test GetSegmentInfo", func(t *testing.T) {
|
||||||
@ -608,7 +608,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
SegmentIDs: []UniqueID{defaultSegmentID},
|
SegmentIDs: []UniqueID{defaultSegmentID},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test LoadBalance", func(t *testing.T) {
|
t.Run("Test LoadBalance", func(t *testing.T) {
|
||||||
@ -630,7 +630,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
PartitionIDs: []UniqueID{defaultPartitionID},
|
PartitionIDs: []UniqueID{defaultPartitionID},
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
|
|
||||||
})
|
})
|
||||||
|
|
||||||
@ -642,7 +642,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
CollectionID: defaultCollectionID,
|
CollectionID: defaultCollectionID,
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test GetComponentStates", func(t *testing.T) {
|
t.Run("Test GetComponentStates", func(t *testing.T) {
|
||||||
@ -657,7 +657,7 @@ func TestGrpcTaskBeforeHealthy(t *testing.T) {
|
|||||||
CollectionID: defaultCollectionID,
|
CollectionID: defaultCollectionID,
|
||||||
})
|
})
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Test GetMetrics", func(t *testing.T) {
|
t.Run("Test GetMetrics", func(t *testing.T) {
|
||||||
@ -694,7 +694,7 @@ func Test_GrpcGetQueryChannelFail(t *testing.T) {
|
|||||||
res, err := queryCoord.CreateQueryChannel(context.Background(), &querypb.CreateQueryChannelRequest{
|
res, err := queryCoord.CreateQueryChannel(context.Background(), &querypb.CreateQueryChannelRequest{
|
||||||
CollectionID: defaultCollectionID,
|
CollectionID: defaultCollectionID,
|
||||||
})
|
})
|
||||||
assert.NotNil(t, err)
|
assert.Nil(t, err)
|
||||||
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
assert.Equal(t, commonpb.ErrorCode_UnexpectedError, res.Status.ErrorCode)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user