mirror of
https://github.com/langchain-ai/datafusion.git
synced 2026-07-01 21:24:06 -04:00
张林伟
465c89f7f1
* Update github repo link * Format markdown --------- Co-authored-by: Andrew Lamb <andrew@nerdnetworks.org>
61 lines
2.0 KiB
Rust
61 lines
2.0 KiB
Rust
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you 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.
|
|
|
|
//! tpch binary only entrypoint
|
|
|
|
use datafusion::error::Result;
|
|
use datafusion_benchmarks::tpch;
|
|
use structopt::StructOpt;
|
|
|
|
#[cfg(feature = "snmalloc")]
|
|
#[global_allocator]
|
|
static ALLOC: snmalloc_rs::SnMalloc = snmalloc_rs::SnMalloc;
|
|
|
|
#[cfg(feature = "mimalloc")]
|
|
#[global_allocator]
|
|
static ALLOC: mimalloc::MiMalloc = mimalloc::MiMalloc;
|
|
|
|
#[derive(Debug, StructOpt)]
|
|
#[structopt(about = "benchmark command")]
|
|
enum BenchmarkSubCommandOpt {
|
|
#[structopt(name = "datafusion")]
|
|
DataFusionBenchmark(tpch::RunOpt),
|
|
}
|
|
|
|
#[derive(Debug, StructOpt)]
|
|
#[structopt(name = "TPC-H", about = "TPC-H Benchmarks.")]
|
|
enum TpchOpt {
|
|
Benchmark(BenchmarkSubCommandOpt),
|
|
Convert(tpch::ConvertOpt),
|
|
}
|
|
|
|
/// 'tpch' entry point, with tortured command line arguments. Please
|
|
/// use `dbbench` instead.
|
|
///
|
|
/// Note: this is kept to be backwards compatible with the benchmark names prior to
|
|
/// <https://github.com/apache/datafusion/issues/6994>
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
env_logger::init();
|
|
match TpchOpt::from_args() {
|
|
TpchOpt::Benchmark(BenchmarkSubCommandOpt::DataFusionBenchmark(opt)) => {
|
|
opt.run().await
|
|
}
|
|
TpchOpt::Convert(opt) => opt.run().await,
|
|
}
|
|
}
|