Skip to content

Commit 2e3ac59

Browse files
committed
rustfmt
1 parent d51eaed commit 2e3ac59

37 files changed

+1705
-1406
lines changed

.rustfmt.toml

+2
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
use_try_shorthand = true
2+
error_on_line_overflow = false

examples/autobahn-client.rs

+19-34
Original file line numberDiff line numberDiff line change
@@ -1,54 +1,47 @@
11
/// WebSocket client used for testing against the Autobahn Test Suite
2-
32
extern crate ws;
43

54
use std::rc::Rc;
65
use std::cell::Cell;
76
use ws::{connect, CloseCode, Message, Result};
87

9-
#[cfg(feature="permessage-deflate")]
8+
#[cfg(feature = "permessage-deflate")]
109
use ws::deflate::DeflateHandler;
1110

1211
const AGENT: &'static str = "WS-RS";
1312

14-
#[cfg(not(feature="permessage-deflate"))]
15-
fn main () {
16-
13+
#[cfg(not(feature = "permessage-deflate"))]
14+
fn main() {
1715
let total = get_case_count().unwrap();
1816
let mut case_id = 1;
1917

20-
2118
while case_id <= total {
19+
let case_url = format!(
20+
"ws://127.0.0.1:9001/runCase?case={}&agent={}",
21+
case_id, AGENT
22+
);
2223

23-
let case_url = format!("ws://127.0.0.1:9001/runCase?case={}&agent={}", case_id, AGENT);
24-
25-
connect(case_url, |out| {
26-
move |msg| {
27-
out.send(msg)
28-
}
29-
}).unwrap();
24+
connect(case_url, |out| move |msg| out.send(msg)).unwrap();
3025

3126
case_id += 1
3227
}
3328

3429
update_reports().unwrap();
3530
}
3631

37-
#[cfg(feature="permessage-deflate")]
38-
fn main () {
39-
32+
#[cfg(feature = "permessage-deflate")]
33+
fn main() {
4034
let total = get_case_count().unwrap();
4135
let mut case_id = 1;
4236

43-
4437
while case_id <= total {
45-
46-
let case_url = format!("ws://127.0.0.1:9001/runCase?case={}&agent={}", case_id, AGENT);
38+
let case_url = format!(
39+
"ws://127.0.0.1:9001/runCase?case={}&agent={}",
40+
case_id, AGENT
41+
);
4742

4843
connect(case_url, |out| {
49-
DeflateHandler::new(move |msg| {
50-
out.send(msg)
51-
})
44+
DeflateHandler::new(move |msg| out.send(msg))
5245
}).unwrap();
5346

5447
case_id += 1
@@ -58,34 +51,26 @@ fn main () {
5851
}
5952

6053
fn get_case_count() -> Result<u32> {
61-
6254
// sadly we need to use a Cell because we need to set the total, and RC is immutable
6355
let total = Rc::new(Cell::new(0));
6456

65-
try!(connect("ws://127.0.0.1:9001/getCaseCount", |out| {
66-
57+
connect("ws://127.0.0.1:9001/getCaseCount", |out| {
6758
let my_total = total.clone();
6859

6960
move |msg: Message| {
70-
71-
let count = try!(msg.as_text());
61+
let count = msg.as_text()?;
7262

7363
my_total.set(count.parse::<u32>().unwrap());
7464

7565
out.close(CloseCode::Normal)
7666
}
77-
78-
}));
67+
})?;
7968

8069
Ok(total.get())
8170
}
8271

8372
fn update_reports() -> Result<()> {
8473
let report_url = format!("ws://127.0.0.1:9001/updateReports?agent={}", AGENT);
8574

86-
connect(report_url, |out| {
87-
move |_| {
88-
out.close(CloseCode::Normal)
89-
}
90-
})
75+
connect(report_url, |out| move |_| out.close(CloseCode::Normal))
9176
}

examples/autobahn-server.rs

+8-13
Original file line numberDiff line numberDiff line change
@@ -1,30 +1,25 @@
1+
extern crate env_logger;
12
/// WebSocket server used for testing against the Autobahn Test Suite. This is basically the server
23
/// example without printing output or comments.
3-
44
extern crate ws;
5-
extern crate env_logger;
65

7-
#[cfg(feature="permessage-deflate")]
6+
#[cfg(feature = "permessage-deflate")]
87
use ws::deflate::DeflateHandler;
98

10-
#[cfg(not(feature="permessage-deflate"))]
11-
fn main () {
9+
#[cfg(not(feature = "permessage-deflate"))]
10+
fn main() {
1211
env_logger::init();
1312

1413
ws::listen("127.0.0.1:3012", |out| {
15-
move |msg| {
16-
out.send(msg)
17-
}
14+
move |msg| out.send(msg)
1815
}).unwrap()
1916
}
2017

21-
#[cfg(feature="permessage-deflate")]
22-
fn main () {
18+
#[cfg(feature = "permessage-deflate")]
19+
fn main() {
2320
env_logger::init();
2421

2522
ws::listen("127.0.0.1:3012", |out| {
26-
DeflateHandler::new(move |msg| {
27-
out.send(msg)
28-
})
23+
DeflateHandler::new(move |msg| out.send(msg))
2924
}).unwrap();
3025
}

examples/bench-server.rs

+11-11
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
1-
21
/// WebSocket server used for testing the bench example.
32
extern crate ws;
43

5-
use ws::{Builder, Settings, Sender};
4+
use ws::{Builder, Sender, Settings};
65

7-
fn main () {
8-
Builder::new().with_settings(Settings {
9-
max_connections: 10_000,
10-
..Settings::default()
11-
}).build(|out: Sender| {
12-
move |msg| {
13-
out.send(msg)
14-
}
15-
}).unwrap().listen("127.0.0.1:3012").unwrap();
6+
fn main() {
7+
Builder::new()
8+
.with_settings(Settings {
9+
max_connections: 10_000,
10+
..Settings::default()
11+
})
12+
.build(|out: Sender| move |msg| out.send(msg))
13+
.unwrap()
14+
.listen("127.0.0.1:3012")
15+
.unwrap();
1616
}

examples/bench.rs

+20-19
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,24 @@
1+
extern crate env_logger;
2+
extern crate time;
3+
extern crate url;
14
/// A simple, but immature, benchmark client for destroying other WebSocket frameworks and proving
25
/// WS-RS's performance excellence. ;)
36
/// Make sure you allow for enough connections in your OS (e.g. ulimit -Sn 10000).
4-
57
extern crate ws;
6-
extern crate url;
7-
extern crate time;
8-
extern crate env_logger;
98

109
// Try this against node for some fun
1110

1211
// TODO: Separate this example into a separate lib
1312
// TODO: num threads, num connections per thread, num concurrent connections per thread, num
1413
// messages per connection, length of message, text or binary
1514

16-
use ws::{Builder, Settings, Sender, CloseCode, Handler, Message, Handshake, Result};
15+
use ws::{Builder, CloseCode, Handler, Handshake, Message, Result, Sender, Settings};
1716

1817
const CONNECTIONS: usize = 10_000; // simultaneous
1918
const MESSAGES: u32 = 10;
2019
static MESSAGE: &'static str = "TEST TEST TEST TEST TEST TEST TEST TEST";
2120

22-
fn main () {
21+
fn main() {
2322
env_logger::init();
2423

2524
let url = url::Url::parse("ws://127.0.0.1:3012").unwrap();
@@ -32,19 +31,18 @@ fn main () {
3231
}
3332

3433
impl Handler for Connection {
35-
3634
fn on_open(&mut self, _: Handshake) -> Result<()> {
37-
try!(self.out.send(MESSAGE));
35+
self.out.send(MESSAGE)?;
3836
self.count += 1;
3937
Ok(self.time = time::precise_time_ns())
4038
}
4139

4240
fn on_message(&mut self, msg: Message) -> Result<()> {
4341
assert_eq!(msg.as_text().unwrap(), MESSAGE);
4442
if self.count > MESSAGES {
45-
try!(self.out.close(CloseCode::Normal));
43+
self.out.close(CloseCode::Normal)?;
4644
} else {
47-
try!(self.out.send(MESSAGE));
45+
self.out.send(MESSAGE)?;
4846
let time = time::precise_time_ns();
4947
// println!("time {}", time -self.time);
5048
self.total += time - self.time;
@@ -55,23 +53,26 @@ fn main () {
5553
}
5654
}
5755

58-
let mut ws = Builder::new().with_settings(Settings {
59-
max_connections: CONNECTIONS,
60-
..Settings::default()
61-
}).build(|out| {
62-
Connection {
56+
let mut ws = Builder::new()
57+
.with_settings(Settings {
58+
max_connections: CONNECTIONS,
59+
..Settings::default()
60+
})
61+
.build(|out| Connection {
6362
out: out,
6463
count: 0,
6564
time: 0,
6665
total: 0,
67-
}
68-
}).unwrap();
69-
66+
})
67+
.unwrap();
7068

7169
for _ in 0..CONNECTIONS {
7270
ws.connect(url.clone()).unwrap();
7371
}
7472
let start = time::precise_time_ns();
7573
ws.run().unwrap();
76-
println!("Total time. {}", (time::precise_time_ns() - start) / 1_000_000)
74+
println!(
75+
"Total time. {}",
76+
(time::precise_time_ns() - start) / 1_000_000
77+
)
7778
}

0 commit comments

Comments
 (0)