@@ -63,8 +63,8 @@ impl AdminCommand {
6363 // Write the message header
6464 conn. buffer . size_buffer ( ) ?;
6565 let size = conn. buffer . data_offset ;
66- conn. buffer . reset_offset ( ) ? ;
67- AdminCommand :: write_size ( & mut conn, size as i64 ) ? ;
66+ conn. buffer . reset_offset ( ) ;
67+ AdminCommand :: write_size ( & mut conn, size as i64 ) ;
6868
6969 // Send command.
7070 if let Err ( err) = conn. flush ( ) . await {
@@ -78,7 +78,7 @@ impl AdminCommand {
7878 return Err ( err) ;
7979 }
8080
81- let result_code = conn. buffer . read_u8 ( Some ( RESULT_CODE ) ) ? ;
81+ let result_code = conn. buffer . read_u8 ( Some ( RESULT_CODE ) ) ;
8282 let result_code = ResultCode :: from ( result_code) ;
8383 if result_code != ResultCode :: Ok {
8484 bail ! ( ErrorKind :: ServerError ( result_code) ) ;
@@ -89,25 +89,25 @@ impl AdminCommand {
8989
9090 pub async fn authenticate ( conn : & mut Connection , user : & str , password : & str ) -> Result < ( ) > {
9191 conn. buffer . resize_buffer ( 1024 ) ?;
92- conn. buffer . reset_offset ( ) ? ;
93- AdminCommand :: write_header ( conn, LOGIN , 2 ) ? ;
94- AdminCommand :: write_field_str ( conn, USER , user) ? ;
95- AdminCommand :: write_field_bytes ( conn, CREDENTIAL , password. as_bytes ( ) ) ? ;
92+ conn. buffer . reset_offset ( ) ;
93+ AdminCommand :: write_header ( conn, LOGIN , 2 ) ;
94+ AdminCommand :: write_field_str ( conn, USER , user) ;
95+ AdminCommand :: write_field_bytes ( conn, CREDENTIAL , password. as_bytes ( ) ) ;
9696 conn. buffer . size_buffer ( ) ?;
9797 let size = conn. buffer . data_offset ;
98- conn. buffer . reset_offset ( ) ? ;
99- AdminCommand :: write_size ( conn, size as i64 ) ? ;
98+ conn. buffer . reset_offset ( ) ;
99+ AdminCommand :: write_size ( conn, size as i64 ) ;
100100
101101 conn. flush ( ) . await ?;
102102 conn. read_buffer ( HEADER_SIZE ) . await ?;
103- let result_code = conn. buffer . read_u8 ( Some ( RESULT_CODE ) ) ? ;
103+ let result_code = conn. buffer . read_u8 ( Some ( RESULT_CODE ) ) ;
104104 let result_code = ResultCode :: from ( result_code) ;
105105 if ResultCode :: SecurityNotEnabled != result_code && ResultCode :: Ok != result_code {
106106 bail ! ( ErrorKind :: ServerError ( result_code) ) ;
107107 }
108108
109109 // consume the rest of the buffer
110- let sz = conn. buffer . read_u64 ( Some ( 0 ) ) ? ;
110+ let sz = conn. buffer . read_u64 ( Some ( 0 ) ) ;
111111 let receive_size = ( sz & 0xFFFF_FFFF_FFFF ) - HEADER_REMAINING as u64 ;
112112 conn. read_buffer ( receive_size as usize ) . await ?;
113113
@@ -124,15 +124,11 @@ impl AdminCommand {
124124 let mut conn = node. get_connection ( ) . await ?;
125125
126126 conn. buffer . resize_buffer ( 1024 ) ?;
127- conn. buffer . reset_offset ( ) ?;
128- AdminCommand :: write_header ( & mut conn, CREATE_USER , 3 ) ?;
129- AdminCommand :: write_field_str ( & mut conn, USER , user) ?;
130- AdminCommand :: write_field_str (
131- & mut conn,
132- PASSWORD ,
133- & AdminCommand :: hash_password ( password) ?,
134- ) ?;
135- AdminCommand :: write_roles ( & mut conn, roles) ?;
127+ conn. buffer . reset_offset ( ) ;
128+ AdminCommand :: write_header ( & mut conn, CREATE_USER , 3 ) ;
129+ AdminCommand :: write_field_str ( & mut conn, USER , user) ;
130+ AdminCommand :: write_field_str ( & mut conn, PASSWORD , & AdminCommand :: hash_password ( password) ?) ;
131+ AdminCommand :: write_roles ( & mut conn, roles) ;
136132
137133 AdminCommand :: execute ( conn) . await
138134 }
@@ -142,9 +138,9 @@ impl AdminCommand {
142138 let mut conn = node. get_connection ( ) . await ?;
143139
144140 conn. buffer . resize_buffer ( 1024 ) ?;
145- conn. buffer . reset_offset ( ) ? ;
146- AdminCommand :: write_header ( & mut conn, DROP_USER , 1 ) ? ;
147- AdminCommand :: write_field_str ( & mut conn, USER , user) ? ;
141+ conn. buffer . reset_offset ( ) ;
142+ AdminCommand :: write_header ( & mut conn, DROP_USER , 1 ) ;
143+ AdminCommand :: write_field_str ( & mut conn, USER , user) ;
148144
149145 AdminCommand :: execute ( conn) . await
150146 }
@@ -154,14 +150,10 @@ impl AdminCommand {
154150 let mut conn = node. get_connection ( ) . await ?;
155151
156152 conn. buffer . resize_buffer ( 1024 ) ?;
157- conn. buffer . reset_offset ( ) ?;
158- AdminCommand :: write_header ( & mut conn, SET_PASSWORD , 2 ) ?;
159- AdminCommand :: write_field_str ( & mut conn, USER , user) ?;
160- AdminCommand :: write_field_str (
161- & mut conn,
162- PASSWORD ,
163- & AdminCommand :: hash_password ( password) ?,
164- ) ?;
153+ conn. buffer . reset_offset ( ) ;
154+ AdminCommand :: write_header ( & mut conn, SET_PASSWORD , 2 ) ;
155+ AdminCommand :: write_field_str ( & mut conn, USER , user) ;
156+ AdminCommand :: write_field_str ( & mut conn, PASSWORD , & AdminCommand :: hash_password ( password) ?) ;
165157
166158 AdminCommand :: execute ( conn) . await
167159 }
@@ -171,26 +163,22 @@ impl AdminCommand {
171163 let mut conn = node. get_connection ( ) . await ?;
172164
173165 conn. buffer . resize_buffer ( 1024 ) ?;
174- conn. buffer . reset_offset ( ) ? ;
175- AdminCommand :: write_header ( & mut conn, CHANGE_PASSWORD , 3 ) ? ;
176- AdminCommand :: write_field_str ( & mut conn, USER , user) ? ;
166+ conn. buffer . reset_offset ( ) ;
167+ AdminCommand :: write_header ( & mut conn, CHANGE_PASSWORD , 3 ) ;
168+ AdminCommand :: write_field_str ( & mut conn, USER , user) ;
177169 match cluster. client_policy ( ) . user_password {
178170 Some ( ( _, ref password) ) => {
179171 AdminCommand :: write_field_str (
180172 & mut conn,
181173 OLD_PASSWORD ,
182174 & AdminCommand :: hash_password ( password) ?,
183- ) ? ;
175+ ) ;
184176 }
185177
186- None => AdminCommand :: write_field_str ( & mut conn, OLD_PASSWORD , "" ) ? ,
178+ None => AdminCommand :: write_field_str ( & mut conn, OLD_PASSWORD , "" ) ,
187179 } ;
188180
189- AdminCommand :: write_field_str (
190- & mut conn,
191- PASSWORD ,
192- & AdminCommand :: hash_password ( password) ?,
193- ) ?;
181+ AdminCommand :: write_field_str ( & mut conn, PASSWORD , & AdminCommand :: hash_password ( password) ?) ;
194182
195183 AdminCommand :: execute ( conn) . await
196184 }
@@ -200,10 +188,10 @@ impl AdminCommand {
200188 let mut conn = node. get_connection ( ) . await ?;
201189
202190 conn. buffer . resize_buffer ( 1024 ) ?;
203- conn. buffer . reset_offset ( ) ? ;
204- AdminCommand :: write_header ( & mut conn, GRANT_ROLES , 2 ) ? ;
205- AdminCommand :: write_field_str ( & mut conn, USER , user) ? ;
206- AdminCommand :: write_roles ( & mut conn, roles) ? ;
191+ conn. buffer . reset_offset ( ) ;
192+ AdminCommand :: write_header ( & mut conn, GRANT_ROLES , 2 ) ;
193+ AdminCommand :: write_field_str ( & mut conn, USER , user) ;
194+ AdminCommand :: write_roles ( & mut conn, roles) ;
207195
208196 AdminCommand :: execute ( conn) . await
209197 }
@@ -213,71 +201,62 @@ impl AdminCommand {
213201 let mut conn = node. get_connection ( ) . await ?;
214202
215203 conn. buffer . resize_buffer ( 1024 ) ?;
216- conn. buffer . reset_offset ( ) ? ;
217- AdminCommand :: write_header ( & mut conn, REVOKE_ROLES , 2 ) ? ;
218- AdminCommand :: write_field_str ( & mut conn, USER , user) ? ;
219- AdminCommand :: write_roles ( & mut conn, roles) ? ;
204+ conn. buffer . reset_offset ( ) ;
205+ AdminCommand :: write_header ( & mut conn, REVOKE_ROLES , 2 ) ;
206+ AdminCommand :: write_field_str ( & mut conn, USER , user) ;
207+ AdminCommand :: write_roles ( & mut conn, roles) ;
220208
221209 AdminCommand :: execute ( conn) . await
222210 }
223211
224212 // Utility methods
225213
226- fn write_size ( conn : & mut Connection , size : i64 ) -> Result < ( ) > {
214+ fn write_size ( conn : & mut Connection , size : i64 ) {
227215 // Write total size of message which is the current offset.
228216 let size = ( size - 8 ) | ( MSG_VERSION << 56 ) | ( MSG_TYPE << 48 ) ;
229- conn. buffer . write_i64 ( size) ?;
230-
231- Ok ( ( ) )
217+ conn. buffer . write_i64 ( size) ;
232218 }
233219
234- fn write_header ( conn : & mut Connection , command : u8 , field_count : u8 ) -> Result < ( ) > {
220+ fn write_header ( conn : & mut Connection , command : u8 , field_count : u8 ) {
235221 conn. buffer . data_offset = 8 ;
236- conn. buffer . write_u8 ( 0 ) ? ;
237- conn. buffer . write_u8 ( 0 ) ? ;
238- conn. buffer . write_u8 ( command) ? ;
239- conn. buffer . write_u8 ( field_count) ? ;
222+ conn. buffer . write_u8 ( 0 ) ;
223+ conn. buffer . write_u8 ( 0 ) ;
224+ conn. buffer . write_u8 ( command) ;
225+ conn. buffer . write_u8 ( field_count) ;
240226
241227 // Authenticate header is almost all zeros
242228 for _ in 0 ..( 16 - 4 ) {
243- conn. buffer . write_u8 ( 0 ) ? ;
229+ conn. buffer . write_u8 ( 0 ) ;
244230 }
245-
246- Ok ( ( ) )
247231 }
248232
249- fn write_field_header ( conn : & mut Connection , id : u8 , size : usize ) -> Result < ( ) > {
250- conn. buffer . write_u32 ( size as u32 + 1 ) ?;
251- conn. buffer . write_u8 ( id) ?;
252- Ok ( ( ) )
233+ fn write_field_header ( conn : & mut Connection , id : u8 , size : usize ) {
234+ conn. buffer . write_u32 ( size as u32 + 1 ) ;
235+ conn. buffer . write_u8 ( id) ;
253236 }
254237
255- fn write_field_str ( conn : & mut Connection , id : u8 , s : & str ) -> Result < ( ) > {
256- AdminCommand :: write_field_header ( conn, id, s. len ( ) ) ?;
257- conn. buffer . write_str ( s) ?;
258- Ok ( ( ) )
238+ fn write_field_str ( conn : & mut Connection , id : u8 , s : & str ) {
239+ AdminCommand :: write_field_header ( conn, id, s. len ( ) ) ;
240+ conn. buffer . write_str ( s) ;
259241 }
260242
261- fn write_field_bytes ( conn : & mut Connection , id : u8 , b : & [ u8 ] ) -> Result < ( ) > {
262- AdminCommand :: write_field_header ( conn, id, b. len ( ) ) ?;
263- conn. buffer . write_bytes ( b) ?;
264- Ok ( ( ) )
243+ fn write_field_bytes ( conn : & mut Connection , id : u8 , b : & [ u8 ] ) {
244+ AdminCommand :: write_field_header ( conn, id, b. len ( ) ) ;
245+ conn. buffer . write_bytes ( b) ;
265246 }
266247
267- fn write_roles ( conn : & mut Connection , roles : & [ & str ] ) -> Result < ( ) > {
248+ fn write_roles ( conn : & mut Connection , roles : & [ & str ] ) {
268249 let mut size = 0 ;
269250 for role in roles {
270251 size += role. len ( ) + 1 ; // size + len
271252 }
272253
273- AdminCommand :: write_field_header ( conn, ROLES , size) ? ;
274- conn. buffer . write_u8 ( roles. len ( ) as u8 ) ? ;
254+ AdminCommand :: write_field_header ( conn, ROLES , size) ;
255+ conn. buffer . write_u8 ( roles. len ( ) as u8 ) ;
275256 for role in roles {
276- conn. buffer . write_u8 ( role. len ( ) as u8 ) ? ;
277- conn. buffer . write_str ( role) ? ;
257+ conn. buffer . write_u8 ( role. len ( ) as u8 ) ;
258+ conn. buffer . write_str ( role) ;
278259 }
279-
280- Ok ( ( ) )
281260 }
282261
283262 pub fn hash_password ( password : & str ) -> Result < String > {
0 commit comments