fix search failed msg (#22036)

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
This commit is contained in:
wei liu 2023-02-08 15:44:31 +08:00 committed by GitHub
parent c21e11eb8a
commit 137e620a8c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -755,12 +755,10 @@ func (node *QueryNode) Search(ctx context.Context, req *querypb.SearchRequest) (
mu.Lock()
defer mu.Unlock()
if err != nil {
failRet.Status.Reason = err.Error()
failRet.Status.ErrorCode = commonpb.ErrorCode_UnexpectedError
return err
}
if ret.GetStatus().GetErrorCode() != commonpb.ErrorCode_Success {
failRet.Status.Reason = ret.Status.Reason
failRet.Status.ErrorCode = ret.Status.ErrorCode
return fmt.Errorf("%s", ret.Status.Reason)
}
@ -769,6 +767,8 @@ func (node *QueryNode) Search(ctx context.Context, req *querypb.SearchRequest) (
})
}
if err := runningGp.Wait(); err != nil {
// make fail reason comes from first err
failRet.Status.Reason = err.Error()
return failRet, nil
}
@ -1137,12 +1137,10 @@ func (node *QueryNode) Query(ctx context.Context, req *querypb.QueryRequest) (*i
mu.Lock()
defer mu.Unlock()
if err != nil {
failRet.Status.Reason = err.Error()
failRet.Status.ErrorCode = commonpb.ErrorCode_UnexpectedError
return err
}
if ret.GetStatus().GetErrorCode() != commonpb.ErrorCode_Success {
failRet.Status.Reason = ret.Status.Reason
failRet.Status.ErrorCode = ret.Status.ErrorCode
return fmt.Errorf("%s", ret.Status.Reason)
}
@ -1151,6 +1149,7 @@ func (node *QueryNode) Query(ctx context.Context, req *querypb.QueryRequest) (*i
})
}
if err := runningGp.Wait(); err != nil {
failRet.Status.Reason = err.Error()
return failRet, nil
}
ret, err := mergeInternalRetrieveResultsAndFillIfEmpty(ctx, toMergeResults, req.GetReq().GetLimit(), req.GetReq().GetOutputFieldsId(), coll.Schema())