要确保AsyncWriteExt将所有字节写入Rust TcpStream，可以使用循环来持续写入数据直到所有字节成功发送。下面是一个示例代码：

use std::io::{self, ErrorKind};
use tokio::io::AsyncWriteExt;
use tokio::net::TcpStream;

async fn write_all_data(stream: &mut TcpStream, data: &[u8]) -> io::Result<()> {
    let mut remaining = data.len();

    while remaining > 0 {
        // 将数据写入TcpStream
        let n = stream.write(&data[data.len() - remaining..]).await?;

        if n == 0 {
            // 如果写入的字节数为0，则意味着TcpStream已关闭
            return Err(ErrorKind::BrokenPipe.into());
        }

        remaining -= n;
    }

    Ok(())
}

#[tokio::main]
async fn main() -> io::Result<()> {
    let mut stream = TcpStream::connect("127.0.0.1:8080").await?;
    let data = b"Hello, world!";

    write_all_data(&mut stream, data).await?;

    println!("All data has been written successfully.");

    Ok(())
}

在上述示例中，我们定义了一个名为write_all_data的异步函数，它接受一个TcpStream和要写入的数据作为参数。该函数使用循环来持续写入数据，直到所有字节都成功发送。在每一次写入操作后，我们检查实际写入的字节数是否为0，如果是0，则意味着TcpStream已关闭，我们将返回一个BrokenPipe错误。