@@ -6,7 +6,7 @@ use bytes::{Bytes, BytesMut};
6
6
use futures:: {
7
7
stream:: futures_unordered:: FuturesUnordered , FutureExt , StreamExt , TryStreamExt ,
8
8
} ;
9
- use rand:: { rngs:: SmallRng , thread_rng , Rng , SeedableRng } ;
9
+ use rand:: { rngs:: SmallRng , Rng , SeedableRng } ;
10
10
use tokio_postgres:: { connect, types:: ToSql , Client , NoTls , Statement } ;
11
11
12
12
use crate :: {
@@ -101,9 +101,9 @@ impl PgConnection {
101
101
}
102
102
103
103
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 ( ) ) ;
105
105
106
- let random_id = ( rng. gen :: < u32 > ( ) % 10_000 + 1 ) as i32 ;
106
+ let random_id = ( rng. random :: < u32 > ( ) % 10_000 + 1 ) as i32 ;
107
107
108
108
let world = self . query_one_world ( random_id) . await ?;
109
109
let mut body = BytesMut :: with_capacity ( 40 ) ;
@@ -113,26 +113,26 @@ impl PgConnection {
113
113
}
114
114
115
115
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 ( ) ) ;
117
117
118
118
let worlds = FuturesUnordered :: new ( ) ;
119
119
120
120
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 ;
122
122
worlds. push ( self . query_one_world ( w_id) ) ;
123
123
}
124
124
125
125
worlds. try_collect ( ) . await
126
126
}
127
127
128
128
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 ( ) ) ;
130
130
131
131
let worlds = FuturesUnordered :: new ( ) ;
132
132
133
133
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 ;
136
136
137
137
worlds. push ( self . query_one_world ( w_id) . map ( move |res| match res {
138
138
Ok ( mut world) => {
0 commit comments