在同一端口上运行websocket和http服务器(Rust,hyper)

时间:2017-08-21 22:23:57

标签: websocket rust hyper

我想使用Rust,Hyper和websocket-rs编写一个Web服务器。 Web服务器必须能够在同一端口上处理http请求和websocket请求 。我使用了官方示例(async-server.rshttps://github.com/cyderize/rust-websocket/blob/master/examples/async-server.rs)并尝试修改它。我的想法是改变错误处理。如果客户端的请求是一个简单的http请求,那么不会返回错误,而是产生一个处理请求的未来。

基于另一个SO问题(How do I handle an error when using rust-websocket so that only that connection fails and not the entire program?),我从样本中更改了错误处理。

这是旧代码:

.map_err(|InvalidConnection {error, ..}| {
    println!("Error:{:?}",error);
    return error;
})

这是我的新代码:

.map(Some).or_else(|_| -> Result<_, ()> {
    // SPAWN THE FUTURE
    Ok(None)
})

这是我被困的地方。示例代码调用函数spawn_future,该函数需要一个在我的代码中不可用的参数(upgrade.reject()),因为我无法访问upgrade结构。我尝试了一些其他的东西(即调用handle.spawn_fn),但编译器从不喜欢我做的事情。 (我可以粘贴一些编译器错误,但我认为它们不会提高我帖子的质量。)

这是我的问题:编写可以在同一端口上处理http请求和websocket请求的异步服务器的推荐方法是什么?如果我的方法是正确的:您能帮助我填写SPAWN THE FUTURE行吗?

附加上下文(如果你还没有注意到):我是一个完全的Rust初学者。

1 个答案:

答案 0 :(得分:7)

在rust-websocket文档here

中似乎有一些文档

从文档中粘贴:

use hyper::server::{Server, Request, Response};
use websocket::Message;
use websocket::sync::server::upgrade::IntoWs;
use websocket::sync::server::upgrade::HyperRequest;

Server::http("0.0.0.0:80").unwrap().handle(move |req: Request, res: Response| {
    match HyperRequest(req).into_ws() {
        Ok(upgrade) => {
            // `accept` sends a successful handshake, no need to worry about res
            let mut client = match upgrade.accept() {
                Ok(c) => c,
                Err(_) => panic!(),
            };

            client.send_message(&Message::text("its free real estate"));
        },

        Err((request, err)) => {
            // continue using the request as normal, "echo uri"
            res.send(b"Try connecting over ws instead.").unwrap();
        },
    };
})
.unwrap();