ws-tool 0.3.0-alpha

an easy to use websocket tool
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use structopt::StructOpt;
use ws_tool::{frame::OpCode, Server};

/// websocket client connect to binance futures websocket
#[derive(StructOpt)]
struct Args {
    /// server host
    #[structopt(long, default_value = "127.0.0.1")]
    host: String,
    /// server port
    #[structopt(short, long, default_value = "9000")]
    port: u16,
}

#[tokio::main]
async fn main() -> Result<(), ()> {
    pretty_env_logger::init();
    let args = Args::from_args();
    log::info!("binding on {}:{}", args.host, args.port);
    let listener = tokio::net::TcpListener::bind(format!("{}:{}", args.host, args.port))
        .await
        .unwrap();
    for (stream, addr) in listener.accept().await {
        log::info!("got connect from {:?}", addr);
        let mut server = Server::from_stream(stream);
        server.handle_handshake().await.unwrap();
        while let Some(x) = server.read().await {
            let frame = x.unwrap();
            if frame.opcode() == OpCode::Close {
                log::info!("client close");
                break;
            }
            let payload = frame.payload_data_unmask();
            let mut frame = ws_tool::frame::Frame::new_with_opcode(ws_tool::frame::OpCode::Text);
            frame.set_mask(false);
            frame.set_payload(String::from_utf8(payload.to_vec()).unwrap().as_bytes());
            server.write(frame).await.unwrap();
        }
    }
    Ok(())
}