@@ -10,18 +10,18 @@ use std::path::Path;
1010use std:: sync:: Arc ;
1111
1212const FIRST_NAMES : & [ & str ] = & [
13- "Alice" , "Bob" , "Charlie" , "Diana" , "Eve" , "Frank" , "Grace" , "Hank" , "Ivy" , "Jack" ,
14- "Karen" , " Leo", "Mia" , "Noah" , "Olivia" , "Paul" , "Quinn" , "Rachel" , "Sam" , "Tina" ,
15- "Uma" , "Victor" , " Wendy", "Xander" , "Yara" , "Zane" , "Anna" , "Ben" , "Clara" , "David" ,
16- "Elena " , "Felix " , "Gina " , "Hugo " , "Iris " , "James " , "Kate " , "Liam " , "Maya " , "Nora " ,
17- "Oscar" , "Pia" , "Raj" , "Sara" , "Tom" , " Vera", "Will" , "Xena" , "Yuki" , "Zara" ,
13+ "Alice" , "Bob" , "Charlie" , "Diana" , "Eve" , "Frank" , "Grace" , "Hank" , "Ivy" , "Jack" , "Karen" ,
14+ "Leo" , "Mia" , "Noah" , "Olivia" , "Paul" , "Quinn" , "Rachel" , "Sam" , "Tina" , "Uma" , "Victor ",
15+ "Wendy" , "Xander" , "Yara" , "Zane" , "Anna" , "Ben" , "Clara" , "David" , "Elena" , "Felix" , "Gina ",
16+ "Hugo " , "Iris " , "James " , "Kate " , "Liam " , "Maya " , "Nora " , "Oscar " , "Pia " , "Raj" , "Sara" , "Tom ",
17+ "Vera" , "Will" , "Xena" , "Yuki" , "Zara" ,
1818] ;
1919
2020const LAST_NAMES : & [ & str ] = & [
21- "Smith" , "Johnson" , "Brown" , "Taylor" , "Anderson" , "Wilson" , "Moore" , "Clark" ,
22- "Walker" , " Hall", "Allen" , "Young" , "King" , "Wright" , "Hill" , "Scott" , "Adams" ,
23- "Baker " , "Carter " , "Evans " , "Garcia " , "Harris " , "Lee " , "Martin " , "Nelson " ,
24- "Patel" , "Roberts" , "Thomas" , "Turner" , " White",
21+ "Smith" , "Johnson" , "Brown" , "Taylor" , "Anderson" , "Wilson" , "Moore" , "Clark" , "Walker" ,
22+ "Hall" , "Allen" , "Young" , "King" , "Wright" , "Hill" , "Scott" , "Adams" , "Baker" , "Carter ",
23+ "Evans " , "Garcia " , "Harris " , "Lee " , "Martin " , "Nelson " , "Patel " , "Roberts" , "Thomas" , "Turner ",
24+ "White" ,
2525] ;
2626
2727const DEPARTMENTS : & [ & str ] = & [
@@ -36,8 +36,21 @@ const DEPARTMENTS: &[&str] = &[
3636] ;
3737
3838const PRODUCTS : & [ & str ] = & [
39- "Laptop" , "Keyboard" , "Mouse" , "Monitor" , "Headphones" , "Webcam" , "Dock" , "Cable" ,
40- "Charger" , "Tablet" , "Phone" , "Speaker" , "SSD" , "RAM" , "GPU" ,
39+ "Laptop" ,
40+ "Keyboard" ,
41+ "Mouse" ,
42+ "Monitor" ,
43+ "Headphones" ,
44+ "Webcam" ,
45+ "Dock" ,
46+ "Cable" ,
47+ "Charger" ,
48+ "Tablet" ,
49+ "Phone" ,
50+ "Speaker" ,
51+ "SSD" ,
52+ "RAM" ,
53+ "GPU" ,
4154] ;
4255
4356const STATUSES : & [ & str ] = & [ "pending" , "shipped" , "delivered" , "cancelled" ] ;
@@ -117,16 +130,18 @@ fn generate_users(dir: &Path, rng: &mut StdRng) {
117130 // Timestamps: 2023-01-01 to 2025-03-01 in millis
118131 let ts_start: i64 = 1_672_531_200_000 ; // 2023-01-01T00:00:00Z
119132 let ts_end: i64 = 1_740_787_200_000 ; // 2025-03-01T00:00:00Z
120- let timestamps: Vec < i64 > = ( 0 ..n)
121- . map ( |_| rng. random_range ( ts_start..ts_end) )
122- . collect ( ) ;
133+ let timestamps: Vec < i64 > = ( 0 ..n) . map ( |_| rng. random_range ( ts_start..ts_end) ) . collect ( ) ;
123134
124135 let batch = arrow:: record_batch:: RecordBatch :: try_new (
125136 schema,
126137 vec ! [
127138 Arc :: new( Int64Array :: from( ids) ) ,
128- Arc :: new( StringArray :: from( names. iter( ) . map( |s| s. as_str( ) ) . collect:: <Vec <_>>( ) ) ) ,
129- Arc :: new( StringArray :: from( emails. iter( ) . map( |s| s. as_str( ) ) . collect:: <Vec <_>>( ) ) ) ,
139+ Arc :: new( StringArray :: from(
140+ names. iter( ) . map( |s| s. as_str( ) ) . collect:: <Vec <_>>( ) ,
141+ ) ) ,
142+ Arc :: new( StringArray :: from(
143+ emails. iter( ) . map( |s| s. as_str( ) ) . collect:: <Vec <_>>( ) ,
144+ ) ) ,
130145 Arc :: new( Int32Array :: from( ages) ) ,
131146 Arc :: new( Float64Array :: from( salaries) ) ,
132147 Arc :: new( BooleanArray :: from( actives) ) ,
@@ -192,8 +207,5 @@ fn generate_orders(dir: &Path, rng: &mut StdRng, filename: &str, n: usize, year:
192207
193208 writer. write ( & batch) . unwrap ( ) ;
194209 writer. close ( ) . unwrap ( ) ;
195- eprintln ! (
196- " {} ({n} rows, 1 row group, snappy)" ,
197- path. display( )
198- ) ;
210+ eprintln ! ( " {} ({n} rows, 1 row group, snappy)" , path. display( ) ) ;
199211}
0 commit comments