mirror of
https://github.com/GreptimeTeam/greptimedb.git
synced 2026-05-18 05:50:41 +00:00
b00b49284e
* feat: manager kafka cluster in sqlness runner Signed-off-by: Ruihang Xia <waynestxia@gmail.com> * pull up clippy config Signed-off-by: Ruihang Xia <waynestxia@gmail.com> * Apply suggestions from code review Co-authored-by: niebayes <niebayes@gmail.com> --------- Signed-off-by: Ruihang Xia <waynestxia@gmail.com> Co-authored-by: niebayes <niebayes@gmail.com>
100 lines
2.9 KiB
Rust
100 lines
2.9 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.
|
|
|
|
#![allow(clippy::print_stdout)]
|
|
|
|
use std::path::PathBuf;
|
|
|
|
use clap::{Parser, ValueEnum};
|
|
use env::{Env, WalConfig};
|
|
use sqlness::{ConfigBuilder, Runner};
|
|
|
|
mod env;
|
|
mod util;
|
|
|
|
#[derive(ValueEnum, Debug, Clone)]
|
|
#[clap(rename_all = "snake_case")]
|
|
enum Wal {
|
|
RaftEngine,
|
|
Kafka,
|
|
}
|
|
|
|
#[derive(Parser, Debug)]
|
|
#[clap(author, version, about, long_about = None)]
|
|
/// SQL Harness for GrepTimeDB
|
|
struct Args {
|
|
/// Directory of test cases
|
|
#[clap(short, long)]
|
|
case_dir: Option<PathBuf>,
|
|
|
|
/// Fail this run as soon as one case fails if true
|
|
#[arg(short, long, default_value = "false")]
|
|
fail_fast: bool,
|
|
|
|
/// Environment Configuration File
|
|
#[clap(short, long, default_value = "config.toml")]
|
|
env_config_file: String,
|
|
|
|
/// Name of test cases to run. Accept as a regexp.
|
|
#[clap(short, long, default_value = ".*")]
|
|
test_filter: String,
|
|
|
|
/// Address of the server.
|
|
#[clap(short, long)]
|
|
server_addr: Option<String>,
|
|
|
|
/// The type of Wal.
|
|
#[clap(short, long, default_value = "raft_engine")]
|
|
wal: Wal,
|
|
|
|
/// The kafka wal broker endpoints. This config will suppress sqlness runner
|
|
/// from starting a kafka cluster, and use the given endpoint as kafka backend.
|
|
#[clap(short, long)]
|
|
kafka_wal_broker_endpoints: Option<String>,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let args = Args::parse();
|
|
|
|
#[cfg(windows)]
|
|
let data_home = std::env::temp_dir();
|
|
#[cfg(not(windows))]
|
|
let data_home = std::path::PathBuf::from("/tmp");
|
|
|
|
let config = ConfigBuilder::default()
|
|
.case_dir(util::get_case_dir(args.case_dir))
|
|
.fail_fast(args.fail_fast)
|
|
.test_filter(args.test_filter)
|
|
.follow_links(true)
|
|
.env_config_file(args.env_config_file)
|
|
.build()
|
|
.unwrap();
|
|
|
|
let wal = match args.wal {
|
|
Wal::RaftEngine => WalConfig::RaftEngine,
|
|
Wal::Kafka => WalConfig::Kafka {
|
|
needs_kafka_cluster: args.kafka_wal_broker_endpoints.is_none(),
|
|
broker_endpoints: args
|
|
.kafka_wal_broker_endpoints
|
|
.map(|s| s.split(',').map(|s| s.to_string()).collect())
|
|
// otherwise default to the same port in `kafka-cluster.yml`
|
|
.unwrap_or(vec!["127.0.0.1:9092".to_string()]),
|
|
},
|
|
};
|
|
|
|
let runner = Runner::new(config, Env::new(data_home, args.server_addr, wal));
|
|
runner.run().await.unwrap();
|
|
}
|