You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.3 KiB
39 lines
1.3 KiB
7 years ago
|
extern crate tungstenite;
|
||
|
|
|
||
|
|
use std::thread::spawn;
|
||
|
|
use std::net::TcpListener;
|
||
|
|
|
||
|
|
use tungstenite::accept;
|
||
|
|
use tungstenite::handshake::server::Request;
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
let server = TcpListener::bind("127.0.0.1:3012").unwrap();
|
||
|
|
for stream in server.incoming() {
|
||
|
|
spawn(move || {
|
||
|
|
let callback = |req: &Request| {
|
||
|
|
println!("Received a new ws handshake");
|
||
|
|
println!("The request's path is: {}", req.path);
|
||
|
|
println!("The request's headers are:");
|
||
|
|
for &(ref header, _ /* value */) in req.headers.iter() {
|
||
|
|
println!("* {}", header);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Let's add an additional header to our response to the client.
|
||
|
|
let extra_headers = vec![
|
||
|
|
(String::from("MyCustomHeader"), String::from(":)")),
|
||
|
|
(String::from("SOME_TUNGSTENITE_HEADER"), String::from("header_value")),
|
||
|
|
];
|
||
|
|
Ok(Some(extra_headers))
|
||
|
|
};
|
||
|
|
let mut websocket = accept(stream.unwrap(), Some(Box::new(callback))).unwrap();
|
||
|
|
|
||
|
|
loop {
|
||
|
|
let msg = websocket.read_message().unwrap();
|
||
|
|
if msg.is_binary() || msg.is_text() {
|
||
|
|
websocket.write_message(msg).unwrap();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|