1
- use std:: path:: PathBuf ;
2
-
3
- use bench_vortex:: tpch:: duckdb:: { generate_tpch, DuckdbTpchOptions } ;
4
- use bench_vortex:: tpch:: load_datasets;
5
- use bench_vortex:: Format ;
6
1
use clap:: Parser ;
7
- use tokio:: runtime:: Builder ;
8
- use url:: Url ;
9
2
use xshell:: { cmd, Shell } ;
10
3
11
4
static FLATC_BIN : & str = "flatc" ;
@@ -22,15 +15,6 @@ enum Commands {
22
15
GenerateFlatbuffers ,
23
16
#[ command( name = "generate-proto" ) ]
24
17
GenerateProto ,
25
- #[ command( name = "generate-tpch-csvs" ) ]
26
- GenerateTpchCsvs {
27
- scale_factor : Option < u8 > ,
28
- output_dir : Option < PathBuf > ,
29
- } ,
30
- #[ command( name = "tpch-csv-to-parquet" ) ]
31
- TpchCsvToParquet { base_dir : Option < PathBuf > } ,
32
- #[ command( name = "tpch-csv-to-vortex" ) ]
33
- TpchCsvToVortex { base_dir : Option < PathBuf > } ,
34
18
}
35
19
36
20
fn execute_generate_fbs ( ) -> anyhow:: Result < ( ) > {
@@ -80,50 +64,11 @@ fn execute_generate_proto() -> anyhow::Result<()> {
80
64
Ok ( ( ) )
81
65
}
82
66
83
- fn execute_generate_tpch_csv (
84
- scale_factor : Option < u8 > ,
85
- base_dir : Option < PathBuf > ,
86
- ) -> anyhow:: Result < ( ) > {
87
- let default = DuckdbTpchOptions :: default ( ) ;
88
- let conf = DuckdbTpchOptions {
89
- scale_factor : scale_factor. unwrap_or ( default. scale_factor ) ,
90
- base_dir : base_dir. unwrap_or ( default. base_dir ) ,
91
- } ;
92
- generate_tpch ( conf) ?;
93
- Ok ( ( ) )
94
- }
95
-
96
- fn execute_from_tpch_csv ( base_dir : Option < PathBuf > , format : Format ) -> anyhow:: Result < ( ) > {
97
- let runtime = Builder :: new_multi_thread ( ) . enable_all ( ) . build ( ) ?;
98
- let base_dir = base_dir. unwrap_or_else ( || DuckdbTpchOptions :: default ( ) . csvs_dir ( ) ) ;
99
- // add a trailing slash to bse_dir so path concat works as expected
100
- let base_url = Url :: parse (
101
- ( "file:" . to_owned ( )
102
- + base_dir
103
- . to_str ( )
104
- . ok_or_else ( || anyhow:: anyhow!( "must be utf8" ) ) ?
105
- + "/" )
106
- . as_ref ( ) ,
107
- ) ?;
108
- runtime. block_on ( load_datasets ( & base_url, format, false ) ) ?;
109
- Ok ( ( ) )
110
- }
111
-
112
67
fn main ( ) -> anyhow:: Result < ( ) > {
113
68
let cli = Xtask :: parse ( ) ;
114
69
match cli. command {
115
70
Commands :: GenerateFlatbuffers => execute_generate_fbs ( ) ?,
116
71
Commands :: GenerateProto => execute_generate_proto ( ) ?,
117
- Commands :: GenerateTpchCsvs {
118
- scale_factor,
119
- output_dir,
120
- } => execute_generate_tpch_csv ( scale_factor, output_dir) ?,
121
- Commands :: TpchCsvToParquet { base_dir } => {
122
- execute_from_tpch_csv ( base_dir, Format :: Parquet ) ?
123
- }
124
- Commands :: TpchCsvToVortex { base_dir } => {
125
- execute_from_tpch_csv ( base_dir, Format :: OnDiskVortex ) ?
126
- }
127
72
}
128
73
Ok ( ( ) )
129
74
}
0 commit comments