@@ -58,10 +58,9 @@ pub async fn insert_records() {
5858 return ;
5959 }
6060 let missing_count: i64 = limit - count;
61- let mut rng: ThreadRng = rng ( ) ;
6261 let mut values: Vec < String > = Vec :: new ( ) ;
6362 for _ in 0 ..missing_count {
64- let random_number: i32 = rng . random_range ( 1 ..= RANDOM_MAX ) ;
63+ let random_number: i32 = get_random_id ( ) ;
6564 values. push ( format ! ( "(DEFAULT, {})" , random_number) ) ;
6665 }
6766 let sql: String = format ! (
@@ -72,7 +71,7 @@ pub async fn insert_records() {
7271 let _ = query ( & sql) . execute ( & db_pool) . await ;
7372 let mut values: Vec < String > = Vec :: new ( ) ;
7473 for _ in 0 ..missing_count {
75- let random_number: String = rng . random_range ( 1 ..= RANDOM_MAX ) . to_string ( ) ;
74+ let random_number: i32 = get_random_id ( ) ;
7675 values. push ( format ! ( "(DEFAULT, {})" , random_number) ) ;
7776 }
7877 let sql: String = format ! (
@@ -138,7 +137,7 @@ pub async fn get_update_data(limit: Queries) -> (String, Vec<QueryRow>) {
138137 let mut id_in_clause: String = format ! ( "{}" , rows[ 0 ] . id) ;
139138 let last_idx: usize = rows. len ( ) - 1 ;
140139 for ( i, row) in rows. iter ( ) . enumerate ( ) {
141- let new_random_number: i32 = rng ( ) . random_range ( 1 .. RANDOM_MAX ) ;
140+ let new_random_number: i32 = get_random_id ( ) ;
142141 let id: i32 = row. id ;
143142 id_list. push ( id) ;
144143 value_list. push_str ( & format ! ( "WHEN {} THEN {} " , id, new_random_number) ) ;
@@ -148,7 +147,10 @@ pub async fn get_update_data(limit: Queries) -> (String, Vec<QueryRow>) {
148147 query_res_list. push ( QueryRow :: new ( id, new_random_number) ) ;
149148 }
150149 sql. push_str ( & value_list) ;
151- sql. push_str ( & format ! ( "END WHERE id IN ({})" , id_in_clause) ) ;
150+ sql. push_str ( & format ! (
151+ "ELSE randomNumber END WHERE id IN ({})" ,
152+ id_in_clause
153+ ) ) ;
152154 ( sql, query_res_list)
153155}
154156
@@ -169,7 +171,7 @@ pub async fn init_db() {
169171
170172#[ inline]
171173pub async fn random_world_row ( db_pool : & DbPoolConnection ) -> QueryRow {
172- let random_id: i32 = rng ( ) . random_range ( 1 ..= RANDOM_MAX ) ;
174+ let random_id: i32 = get_random_id ( ) ;
173175 let sql: String = format ! (
174176 "SELECT id, randomNumber FROM {} WHERE id = {}" ,
175177 TABLE_NAME_WORLD , random_id
0 commit comments