使用read_to_string时如何在新行上停止?

时间:2017-03-28 01:26:01

标签: sockets tcp rust

我在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());
}

服务器将在显示任何数据之前等待断开连接。

如何才能使响应在换行符上进行解析?

1 个答案:

答案 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() {}

原始代码的附加说明:

  1. 不要忽略错误。如果您想忽略错误,那么Rust可能不是您喜欢使用的语言。错误旨在被忽视。

  2. 没有理由将stream重新绑定到conn,只是让它变得可变。

  3. 使用write!writeln!宏将格式简化为实现Read的内容。

  4. 不要使用"".to_string,只需使用String::new

  5. 我没有理由列出response两次的类型(: String= String::new()),只需让类型推断处理它。

    < / LI>
  6. 显式return语句不是惯用的Rust。