@@ -6,7 +6,7 @@ use bytes::{Bytes, BytesMut};
66use futures:: {
77 stream:: futures_unordered:: FuturesUnordered , FutureExt , StreamExt , TryStreamExt ,
88} ;
9- use rand:: { rngs:: SmallRng , thread_rng , Rng , SeedableRng } ;
9+ use rand:: { rngs:: SmallRng , Rng , SeedableRng } ;
1010use tokio_postgres:: { connect, types:: ToSql , Client , NoTls , Statement } ;
1111
1212use crate :: {
@@ -101,9 +101,9 @@ impl PgConnection {
101101 }
102102
103103 pub async fn get_world ( & self ) -> Result < Bytes , PgError > {
104- let mut rng = SmallRng :: from_rng ( & mut thread_rng ( ) ) . unwrap ( ) ;
104+ let mut rng = SmallRng :: from_rng ( & mut rand :: rng ( ) ) ;
105105
106- let random_id = ( rng. gen :: < u32 > ( ) % 10_000 + 1 ) as i32 ;
106+ let random_id = ( rng. random :: < u32 > ( ) % 10_000 + 1 ) as i32 ;
107107
108108 let world = self . query_one_world ( random_id) . await ?;
109109 let mut body = BytesMut :: with_capacity ( 40 ) ;
@@ -113,26 +113,26 @@ impl PgConnection {
113113 }
114114
115115 pub async fn get_worlds ( & self , num : usize ) -> Result < Vec < World > , PgError > {
116- let mut rng = SmallRng :: from_rng ( & mut thread_rng ( ) ) . unwrap ( ) ;
116+ let mut rng = SmallRng :: from_rng ( & mut rand :: rng ( ) ) ;
117117
118118 let worlds = FuturesUnordered :: new ( ) ;
119119
120120 for _ in 0 ..num {
121- let w_id = ( rng. gen :: < u32 > ( ) % 10_000 + 1 ) as i32 ;
121+ let w_id = ( rng. random :: < u32 > ( ) % 10_000 + 1 ) as i32 ;
122122 worlds. push ( self . query_one_world ( w_id) ) ;
123123 }
124124
125125 worlds. try_collect ( ) . await
126126 }
127127
128128 pub async fn update ( & self , num : u16 ) -> Result < Vec < World > , PgError > {
129- let mut rng = SmallRng :: from_rng ( & mut thread_rng ( ) ) . unwrap ( ) ;
129+ let mut rng = SmallRng :: from_rng ( & mut rand :: rng ( ) ) ;
130130
131131 let worlds = FuturesUnordered :: new ( ) ;
132132
133133 for _ in 0 ..num {
134- let id = ( rng. gen :: < u32 > ( ) % 10_000 + 1 ) as i32 ;
135- let w_id = ( rng. gen :: < u32 > ( ) % 10_000 + 1 ) as i32 ;
134+ let id = ( rng. random :: < u32 > ( ) % 10_000 + 1 ) as i32 ;
135+ let w_id = ( rng. random :: < u32 > ( ) % 10_000 + 1 ) as i32 ;
136136
137137 worlds. push ( self . query_one_world ( w_id) . map ( move |res| match res {
138138 Ok ( mut world) => {
0 commit comments