我在Rust中编写套接字服务器,并已实现响应/发送代码。主要问题是方法read_to_string
一直有效,直到连接关闭,直到出现换行符。
例如,在此代码中,
use std::net::{TcpListener, TcpStream};
use std::io::Write;
use std::io::Read;
fn handle_client(stream: TcpStream) {
let conn = stream;
let conn = send(conn, "Hello!");
let mut conn = send(conn, "Bye!");
fn send(mut stream: TcpStream, s: &str) -> TcpStream {
let giver = format!("{}\n", s);
let _ = stream.write(giver.into_bytes().as_slice());
return stream;
}
let mut response: String = "".to_string();
let length = conn.read_to_string(&mut response); // ignore here too
println!("Message:\n{}\n\nLength: {}", response, length.unwrap());
}
服务器将在显示任何数据之前等待断开连接。
如何才能使响应在换行符上进行解析?
答案 0 :(得分:2)
您想要读取一行,其方法称为BufRead::read_line
。您可以在BufReader
中包含实现Read
的内容:
use std::io::{Write, BufRead, BufReader};
use std::net::TcpStream;
fn handle_client(mut conn: TcpStream) {
writeln!(&mut conn, "Hello!").expect("Unable to write");
writeln!(&mut conn, "Bye!").expect("Unable to write");
let mut response = String::new();
let mut conn = BufReader::new(conn);
let length = conn.read_line(&mut response).expect("unable to read");
println!("Message:\n{}\n\nLength: {}", response, length);
}
fn main() {}
原始代码的附加说明:
不要忽略错误。如果您想忽略错误,那么Rust可能不是您喜欢使用的语言。错误旨在被忽视。
没有理由将stream
重新绑定到conn
,只是让它变得可变。
使用write!
和writeln!
宏将格式简化为实现Read
的内容。
不要使用"".to_string
,只需使用String::new
。
我没有理由列出response
两次的类型(: String
,= String::new()
),只需让类型推断处理它。
显式return
语句不是惯用的Rust。