@@ -37,11 +37,11 @@ pub enum ReadResult<'a> {
37
37
38
38
#[ derive( Debug ) ]
39
39
pub enum FramerError {
40
+ Io ( core2:: io:: Error ) ,
40
41
FrameTooLarge ( usize ) ,
41
42
Utf8 ( Utf8Error ) ,
42
43
HttpHeader ( httparse:: Error ) ,
43
44
WebSocket ( crate :: Error ) ,
44
- Io ( anyhow:: Error ) ,
45
45
}
46
46
47
47
pub struct Framer < ' a , TRng , TWebSocketType >
@@ -72,15 +72,13 @@ where
72
72
. map_err ( FramerError :: WebSocket ) ?;
73
73
stream
74
74
. write_all ( & self . write_buf [ ..len] )
75
- . map_err ( anyhow:: Error :: new)
76
75
. map_err ( FramerError :: Io ) ?;
77
76
* self . read_cursor = 0 ;
78
77
79
78
loop {
80
79
// read the response from the server and check it to complete the opening handshake
81
80
let received_size = stream
82
81
. read ( & mut self . read_buf [ * self . read_cursor ..] )
83
- . map_err ( anyhow:: Error :: new)
84
82
. map_err ( FramerError :: Io ) ?;
85
83
86
84
match self . websocket . client_accept (
@@ -119,7 +117,6 @@ where
119
117
120
118
stream
121
119
. write_all ( & self . write_buf [ ..len] )
122
- . map_err ( anyhow:: Error :: new)
123
120
. map_err ( FramerError :: Io ) ?;
124
121
Ok ( ( ) )
125
122
}
@@ -182,7 +179,6 @@ where
182
179
let len = self . close_len ( close_status, status_description) ?;
183
180
stream
184
181
. write_all ( & self . write_buf [ ..len] )
185
- . map_err ( anyhow:: Error :: new)
186
182
. map_err ( FramerError :: Io ) ?;
187
183
Ok ( ( ) )
188
184
}
@@ -208,7 +204,6 @@ where
208
204
let len = self . write_len ( message_type, end_of_message, frame_buf) ?;
209
205
stream
210
206
. write_all ( & self . write_buf [ ..len] )
211
- . map_err ( anyhow:: Error :: new)
212
207
. map_err ( FramerError :: Io ) ?;
213
208
Ok ( ( ) )
214
209
}
@@ -223,10 +218,7 @@ where
223
218
) -> Result < ReadResult < ' b > , FramerError > {
224
219
loop {
225
220
if * self . read_cursor == 0 || * self . read_cursor == self . read_len {
226
- self . read_len = stream
227
- . read ( self . read_buf )
228
- . map_err ( anyhow:: Error :: new)
229
- . map_err ( FramerError :: Io ) ?;
221
+ self . read_len = stream. read ( self . read_buf ) . map_err ( FramerError :: Io ) ?;
230
222
* self . read_cursor = 0 ;
231
223
}
232
224
@@ -309,7 +301,6 @@ where
309
301
let len = self . send_back_data ( & frame_buf, len_to, send_message_type) ?;
310
302
stream
311
303
. write_all ( & self . write_buf [ ..len] )
312
- . map_err ( anyhow:: Error :: new)
313
304
. map_err ( FramerError :: Io ) ?;
314
305
Ok ( ( ) )
315
306
}
@@ -345,7 +336,6 @@ where
345
336
stream
346
337
. write_all ( & self . write_buf [ ..len] )
347
338
. await
348
- . map_err ( anyhow:: Error :: new)
349
339
. map_err ( FramerError :: Io ) ?;
350
340
* self . read_cursor = 0 ;
351
341
@@ -354,7 +344,6 @@ where
354
344
let received_size = stream
355
345
. read ( & mut self . read_buf [ * self . read_cursor ..] )
356
346
. await
357
- . map_err ( anyhow:: Error :: new)
358
347
. map_err ( FramerError :: Io ) ?;
359
348
360
349
match self . websocket . client_accept (
@@ -395,7 +384,6 @@ where
395
384
stream
396
385
. write_all ( & self . write_buf [ ..len] )
397
386
. await
398
- . map_err ( anyhow:: Error :: new)
399
387
. map_err ( FramerError :: Io ) ?;
400
388
Ok ( ( ) )
401
389
}
@@ -418,7 +406,6 @@ where
418
406
stream
419
407
. write_all ( & self . write_buf [ ..len] )
420
408
. await
421
- . map_err ( anyhow:: Error :: new)
422
409
. map_err ( FramerError :: Io ) ?;
423
410
Ok ( ( ) )
424
411
}
@@ -435,7 +422,6 @@ where
435
422
stream
436
423
. write_all ( & self . write_buf [ ..len] )
437
424
. await
438
- . map_err ( anyhow:: Error :: new)
439
425
. map_err ( FramerError :: Io ) ?;
440
426
Ok ( ( ) )
441
427
}
@@ -447,11 +433,7 @@ where
447
433
) -> Result < ReadResult < ' b > , FramerError > {
448
434
loop {
449
435
if * self . read_cursor == 0 || * self . read_cursor == self . read_len {
450
- self . read_len = stream
451
- . read ( self . read_buf )
452
- . await
453
- . map_err ( anyhow:: Error :: new)
454
- . map_err ( FramerError :: Io ) ?;
436
+ self . read_len = stream. read ( self . read_buf ) . await . map_err ( FramerError :: Io ) ?;
455
437
* self . read_cursor = 0 ;
456
438
}
457
439
@@ -537,7 +519,6 @@ where
537
519
stream
538
520
. write_all ( & self . write_buf [ ..len] )
539
521
. await
540
- . map_err ( anyhow:: Error :: new)
541
522
. map_err ( FramerError :: Io ) ?;
542
523
Ok ( ( ) )
543
524
}
0 commit comments