log_store/test_util/
log_store_util.rs1use std::path::Path;
16
17use common_base::readable_size::ReadableSize;
18use common_wal::config::kafka::common::KafkaConnectionConfig;
19use common_wal::config::kafka::DatanodeKafkaConfig;
20use common_wal::config::raft_engine::RaftEngineConfig;
21
22use crate::kafka::log_store::KafkaLogStore;
23use crate::raft_engine::log_store::RaftEngineLogStore;
24
25pub async fn create_tmp_local_file_log_store<P: AsRef<Path>>(path: P) -> RaftEngineLogStore {
27 let path = path.as_ref().display().to_string();
28 let cfg = RaftEngineConfig {
29 file_size: ReadableSize::kb(128),
30 ..Default::default()
31 };
32 RaftEngineLogStore::try_new(path, &cfg).await.unwrap()
33}
34
35pub async fn create_kafka_log_store(broker_endpoints: Vec<String>) -> KafkaLogStore {
37 KafkaLogStore::try_new(
38 &DatanodeKafkaConfig {
39 connection: KafkaConnectionConfig {
40 broker_endpoints,
41 ..Default::default()
42 },
43 ..Default::default()
44 },
45 None,
46 )
47 .await
48 .unwrap()
49}