ws-tool 0.7.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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

use std::sync::mpsc;

use bytes::BytesMut;
use clap::Parser;
use tracing_subscriber::util::SubscriberInitExt;
use ws_tool::{
    codec::{default_handshake_handler, BytesCodec},
    frame::OpCode,
    Message, ServerBuilder,
};

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

    /// level
    #[arg(short, long, default_value = "info")]
    level: tracing::Level,
}

fn main() -> Result<(), ()> {
    let args = Args::parse();
    tracing_subscriber::fmt::fmt()
        .with_max_level(args.level)
        .finish()
        .try_init()
        .expect("failed to init log");
    tracing::info!("binding on {}:{}", args.host, args.port);
    let listener = std::net::TcpListener::bind(format!("{}:{}", args.host, args.port)).unwrap();
    loop {
        let (stream, addr) = listener.accept().unwrap();
        stream.set_nodelay(true).unwrap();
        std::thread::spawn(move || {
            tracing::info!("got connect from {:?}", addr);
            let server =
                ServerBuilder::accept(stream, default_handshake_handler, BytesCodec::factory)
                    .unwrap();
            let (mut read, mut write) = server.split();
            let (tx, rx) = mpsc::channel::<Message<BytesMut>>();
            std::thread::spawn(move || loop {
                let msg = read.receive().unwrap();
                if msg.code == OpCode::Close {
                    break;
                }
                tx.send(msg).unwrap();
            });

            loop {
                match rx.recv() {
                    Ok(msg) => {
                        write.send(&msg.data[..]).unwrap();
                    }
                    Err(e) => {
                        dbg!(e);
                        break;
                    }
                }
            }
            tracing::info!("one conn down");
        });
    }
}