我正在寻找 Rust 中的 TCP 服务器示例。
'hello world' 或 echo 服务器都很棒。
这是一个非常简单的示例,使用std::net
. 它是针对当前的 Rust 大师开发的,也应该适用于 1.*。
小心这个例子;它被简化了;如果绑定、侦听或接受产生错误,您可能不希望它恐慌。
use std::io::Write;
use std::net::TcpListener;
use std::thread;
fn main() {
let listener = TcpListener::bind("127.0.0.1:9123").unwrap();
println!("listening started, ready to accept");
for stream in listener.incoming() {
thread::spawn(|| {
let mut stream = stream.unwrap();
stream.write(b"Hello World\r\n").unwrap();
});
}
}
注意关于接受的范式;您必须自己发起accept()
请求(在此示例中,我们使用的是每次incoming()
只调用的迭代器),因此您可以真正控制有哪些任务。accept()
因此,我将实际的流处理代码放到了一个单独的线程中,但它不需要用于非常短的请求(它只是意味着在处理第一个请求时您将无法处理第二个请求);您也可以删除thread::spawn(|| { ... })
关于这两行的内容。额外线程的使用也提供了一定程度的隔离;如果线程展开,整个服务器不会死掉(但是请记住,内存不足或展开时出现析构函数恐慌会导致整个服务器死掉)。
简单的 TCP 回显服务器https://gist.github.com/seriyps/fd6d29442e16c44ba400
#![feature(phase)]
#[phase(plugin, link)] extern crate log;
extern crate green;
extern crate rustuv;
use std::io;
use std::os;
use std::io::{Listener,Acceptor,TcpStream};
// This is for green threads. If removed, will spawn 1 OS thread per client.
#[start]
fn start(argc: int, argv: *const *const u8) -> int {
green::start(argc, argv, rustuv::event_loop, main)
}
fn main() {
let host = "127.0.0.1";
let port = 8080;
let sock = io::TcpListener::bind(host, port).unwrap();
let mut acceptor = sock.listen();
for stream in acceptor.incoming() {
match stream {
Err(e) => warn!("Accept err {}", e),
Ok(stream) => {
spawn(proc() {
debug!("{}", handle_client(stream));
})
}
}
}
}
fn handle_client(mut stream: io::TcpStream) -> io::IoResult<()> {
info!("New client {}", stream.peer_name());
let mut buf = [0u8, ..4096];
loop {
let got = try!(stream.read(buf));
if got == 0 {
// Is it possible? Or IoError will be raised anyway?
break
}
try!(stream.write(buf.slice(0, got)));
}
Ok(())
}