summa 0.8.8

Fast full-text search server
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

//! Search GRPC API
//!
//! Search GRPC API is using for querying indices

use crate::errors::SummaResult;
use crate::proto;
use crate::services::IndexService;

use std::time::Instant;
use tonic::{Request, Response, Status};
use tracing::{info_span, Instrument};

#[derive(Debug)]
pub struct SearchApiImpl {
    index_service: IndexService,
}

impl SearchApiImpl {
    pub fn new(index_service: &IndexService) -> SummaResult<SearchApiImpl> {
        Ok(SearchApiImpl {
            index_service: index_service.clone(),
        })
    }
}

#[tonic::async_trait]
impl proto::search_api_server::SearchApi for SearchApiImpl {
    async fn search(&self, proto_request: Request<proto::SearchRequest>) -> Result<Response<proto::SearchResponse>, Status> {
        let proto_request = proto_request.into_inner();
        let index_holder = self.index_service.get_index_holder(&proto_request.index_alias)?;

        let query = proto_request.query.unwrap_or(proto::Query {
            query: Some(proto::query::Query::All(proto::AllQuery {})),
        });
        let now = Instant::now();

        let collector_outputs = index_holder
            .search(&query, proto_request.collectors)
            .instrument(info_span!("search", tags = ?proto_request.tags))
            .await?;
        let elapsed_secs = now.elapsed().as_secs_f64();
        Ok(Response::new(proto::SearchResponse {
            index_name: index_holder.index_name().to_owned(),
            collector_outputs,
            elapsed_secs,
        }))
    }
}