mirror of
https://github.com/GreptimeTeam/greptimedb.git
synced 2026-01-05 21:02:58 +00:00
5bdea1a755
* fix: correct `is_exceeded_size_limit` behavior for in-memory store * chore: rename `MetaClientExceededSizeLimit` to `ResponseExceededSizeLimit`
74 lines
2.2 KiB
Rust
74 lines
2.2 KiB
Rust
// Copyright 2023 Greptime Team
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
use common_grpc::channel_manager::ChannelManager;
|
|
use common_meta::kv_backend::{KvBackendRef, ResettableKvBackendRef};
|
|
use meta_srv::mocks as server_mock;
|
|
use meta_srv::mocks::MockInfo;
|
|
|
|
use crate::client::{MetaClient, MetaClientBuilder};
|
|
|
|
pub struct MockMetaContext {
|
|
pub kv_backend: KvBackendRef,
|
|
pub in_memory: Option<ResettableKvBackendRef>,
|
|
}
|
|
|
|
pub async fn mock_client_with_memstore() -> (MetaClient, MockMetaContext) {
|
|
let MockInfo {
|
|
server_addr,
|
|
channel_manager,
|
|
kv_backend,
|
|
in_memory,
|
|
..
|
|
} = server_mock::mock_with_memstore().await;
|
|
(
|
|
mock_client_by(server_addr, channel_manager).await,
|
|
MockMetaContext {
|
|
kv_backend,
|
|
in_memory,
|
|
},
|
|
)
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
pub async fn mock_client_with_etcdstore(addr: &str) -> (MetaClient, MockMetaContext) {
|
|
let MockInfo {
|
|
server_addr,
|
|
channel_manager,
|
|
kv_backend,
|
|
in_memory,
|
|
..
|
|
} = server_mock::mock_with_etcdstore(addr).await;
|
|
(
|
|
mock_client_by(server_addr, channel_manager).await,
|
|
MockMetaContext {
|
|
kv_backend,
|
|
in_memory,
|
|
},
|
|
)
|
|
}
|
|
|
|
pub async fn mock_client_by(server_addr: String, channel_manager: ChannelManager) -> MetaClient {
|
|
let id = (1000u64, 2000u64);
|
|
let mut meta_client = MetaClientBuilder::datanode_default_options(id.0, id.1)
|
|
.enable_access_cluster_info()
|
|
.channel_manager(channel_manager)
|
|
.build();
|
|
meta_client.start(&[&server_addr]).await.unwrap();
|
|
// required only when the heartbeat_client is enabled
|
|
meta_client.ask_leader().await.unwrap();
|
|
|
|
meta_client
|
|
}
|