| 59 | } |
| 60 | |
| 61 | func (s *SearchService) DocCount(ctx context.Context, |
| 62 | req *pb.DocCountRequest) (*pb.DocCountResult, error) { |
| 63 | pindex := s.mgr.GetPIndex(req.IndexName) |
| 64 | if pindex == nil { |
| 65 | return &pb.DocCountResult{DocCount: 0}, fmt.Errorf("grpc_server: "+ |
| 66 | "CountPIndex, no pindex, pindexName: %s", req.IndexName) |
| 67 | } |
| 68 | if pindex.Dest == nil { |
| 69 | return &pb.DocCountResult{DocCount: 0}, fmt.Errorf("grpc_server: "+ |
| 70 | "CountPIndex, no pindex.Dest, pindexName: %s", req.IndexName) |
| 71 | |
| 72 | } |
| 73 | |
| 74 | if req.IndexUUID != "" && pindex.UUID != req.IndexUUID { |
| 75 | return &pb.DocCountResult{DocCount: 0}, fmt.Errorf("grpc_server: "+ |
| 76 | "CountPIndex, wrong pindexUUID: %s, pindex.UUID: %s, pindexName: %s", |
| 77 | req.IndexUUID, pindex.UUID, req.IndexName) |
| 78 | } |
| 79 | |
| 80 | count, err := pindex.Dest.Count(pindex, nil) |
| 81 | if err != nil { |
| 82 | return &pb.DocCountResult{DocCount: 0}, fmt.Errorf("grpc_server: "+ |
| 83 | "CountPIndex, pindexName: %s, req: %#v, err: %v", |
| 84 | req.IndexName, req, err) |
| 85 | |
| 86 | } |
| 87 | |
| 88 | return &pb.DocCountResult{DocCount: int64(count)}, nil |
| 89 | } |
| 90 | |
| 91 | func (s *SearchService) Search(req *pb.SearchRequest, |
| 92 | stream pb.SearchService_SearchServer) (err error) { |