standalone/
datanode_manager.rs

1// Copyright 2023 Greptime Team
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use std::sync::Arc;
16
17use api::region::RegionResponse;
18use api::v1::region::{RegionRequest, RegionResponse as RegionResponseV1};
19use async_trait::async_trait;
20use client::region::check_response_header;
21use common_error::ext::BoxedError;
22use common_meta::error::{self as meta_error, Result as MetaResult};
23use common_meta::node_manager::{
24    Datanode, DatanodeManager, DatanodeRef, FlownodeManager, FlownodeRef,
25};
26use common_meta::peer::Peer;
27use common_query::request::QueryRequest;
28use common_recordbatch::SendableRecordBatchStream;
29use common_telemetry::tracing;
30use common_telemetry::tracing_context::{FutureExt, TracingContext};
31use datanode::region_server::RegionServer;
32use servers::grpc::region_server::RegionServerHandler;
33use snafu::{OptionExt, ResultExt};
34
35use crate::error::{InvalidRegionRequestSnafu, InvokeRegionServerSnafu, Result};
36
37pub struct StandaloneDatanodeManager {
38    pub region_server: RegionServer,
39    pub flow_server: FlownodeRef,
40}
41
42#[async_trait]
43impl DatanodeManager for StandaloneDatanodeManager {
44    async fn datanode(&self, _datanode: &Peer) -> DatanodeRef {
45        RegionInvoker::arc(self.region_server.clone())
46    }
47}
48
49#[async_trait]
50impl FlownodeManager for StandaloneDatanodeManager {
51    async fn flownode(&self, _node: &Peer) -> FlownodeRef {
52        self.flow_server.clone()
53    }
54}
55
56/// Relative to [client::region::RegionRequester]
57struct RegionInvoker {
58    region_server: RegionServer,
59}
60
61impl RegionInvoker {
62    fn arc(region_server: RegionServer) -> Arc<Self> {
63        Arc::new(Self { region_server })
64    }
65
66    async fn handle_inner(&self, request: RegionRequest) -> Result<RegionResponseV1> {
67        let body = request.body.with_context(|| InvalidRegionRequestSnafu {
68            reason: "body not found",
69        })?;
70
71        self.region_server
72            .handle(body)
73            .await
74            .context(InvokeRegionServerSnafu)
75    }
76}
77
78#[async_trait]
79impl Datanode for RegionInvoker {
80    async fn handle_request(&self, request: Arc<RegionRequest>) -> MetaResult<RegionResponse> {
81        // Local execution consumes the request; standalone does not retain it for retries.
82        let request = Arc::try_unwrap(request)
83            .map_err(|_| {
84                InvalidRegionRequestSnafu {
85                    reason: "local region request has multiple owners",
86                }
87                .build()
88            })
89            .map_err(BoxedError::new)
90            .context(meta_error::ExternalSnafu)?;
91        let span = request
92            .header
93            .as_ref()
94            .map(|h| TracingContext::from_w3c(&h.tracing_context))
95            .unwrap_or_default()
96            .attach(tracing::info_span!("RegionInvoker::handle_region_request"));
97        let response = self
98            .handle_inner(request)
99            .trace(span)
100            .await
101            .map_err(BoxedError::new)
102            .context(meta_error::ExternalSnafu)?;
103        check_response_header(&response.header)
104            .map_err(BoxedError::new)
105            .context(meta_error::ExternalSnafu)?;
106        Ok(RegionResponse::from_region_response(response))
107    }
108
109    async fn handle_query(&self, request: QueryRequest) -> MetaResult<SendableRecordBatchStream> {
110        let region_id = request.region_id.to_string();
111        let span = request
112            .header
113            .as_ref()
114            .map(|h| TracingContext::from_w3c(&h.tracing_context))
115            .unwrap_or_default()
116            .attach(tracing::info_span!(
117                "RegionInvoker::handle_query",
118                region_id = region_id
119            ));
120        self.region_server
121            .handle_read(request)
122            .trace(span)
123            .await
124            .map_err(BoxedError::new)
125            .context(meta_error::ExternalSnafu)
126    }
127}
standalone/datanode_manager.rs

standalone/
datanode_manager.rs
