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::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        self.region_server
68            .handle_region_request(request)
69            .await
70            .context(InvokeRegionServerSnafu)
71    }
72}
73
74#[async_trait]
75impl Datanode for RegionInvoker {
76    async fn handle_request(&self, request: Arc<RegionRequest>) -> MetaResult<RegionResponse> {
77        // Local execution consumes the request; standalone does not retain it for retries.
78        let request = Arc::try_unwrap(request)
79            .map_err(|_| {
80                InvalidRegionRequestSnafu {
81                    reason: "local region request has multiple owners",
82                }
83                .build()
84            })
85            .map_err(BoxedError::new)
86            .context(meta_error::ExternalSnafu)?;
87        let span = request
88            .header
89            .as_ref()
90            .map(|h| TracingContext::from_w3c(&h.tracing_context))
91            .unwrap_or_default()
92            .attach(tracing::info_span!("RegionInvoker::handle_region_request"));
93        let response = self
94            .handle_inner(request)
95            .trace(span)
96            .await
97            .map_err(BoxedError::new)
98            .context(meta_error::ExternalSnafu)?;
99        check_response_header(&response.header)
100            .map_err(BoxedError::new)
101            .context(meta_error::ExternalSnafu)?;
102        Ok(RegionResponse::from_region_response(response))
103    }
104
105    async fn handle_query(&self, request: QueryRequest) -> MetaResult<SendableRecordBatchStream> {
106        let region_id = request.region_id.to_string();
107        let span = request
108            .header
109            .as_ref()
110            .map(|h| TracingContext::from_w3c(&h.tracing_context))
111            .unwrap_or_default()
112            .attach(tracing::info_span!(
113                "RegionInvoker::handle_query",
114                region_id = region_id
115            ));
116        self.region_server
117            .handle_read(request)
118            .trace(span)
119            .await
120            .map_err(BoxedError::new)
121            .context(meta_error::ExternalSnafu)
122    }
123}
standalone/datanode_manager.rs

standalone/
datanode_manager.rs
