rust- 结构体与二进制数组转换

将结构体当二进制流传输是做网络编程时传输协议的常用功能。golang语言可以使用包 encoding/binary实现,例如

go 复制代码
import (
	"encoding/binary"
	"os"
)

...
err := binary.Write(f, binary.LittleEndian, p)
...

rust中可以使用 deku将结构体实例转换为bytes数组。

安装依赖

rust 复制代码
use anyhow;
use anyhow::bail;
use deku::prelude::*;
use std::net::TcpStream;
use std::{
    io::{Read, Write},
};

定义结构体

rust 复制代码
#[derive(Debug, PartialEq, DekuRead, DekuWrite, Default)]
struct Message {
    #[deku(endian = "big")]
    msgtype: u32,
    #[deku(endian = "big")]
    taskid: i32,
    #[deku(endian = "big")]
    utctime: u32,
    #[deku(endian = "big", update = "self.data.len()")]
    bodylen: u32,
    #[deku(count = "bodylen", endian = "little")]
    data: Vec<u8>,
}

注意上面的结构体,整数字段有些是使用大段规则,有些是使用小段规则,可以通过如下的宏实现

rust 复制代码
#[deku(endian = "big")]` 

或

#[deku(endian = "little")]

通常在定义协议时,需要知道协议体的长度,但是协议的body部分通常是可变的,例如data: Vec<u8>,只有在结构体初始化时才知道body的长度,所以这里使用如下宏延迟计算了协议的长度,如下

rust 复制代码
#[deku(endian = "big", update = "self.data.len()")]
#[deku(count = "bodylen", endian = "little")]

将结构体转换为字节流发送

添加如下宏后

rust 复制代码
#[derive(Debug, PartialEq, DekuRead, DekuWrite, Default)]

则可以调用 to_bytes()方法进行转换。

将转换后的二进制发送到tcp服务器,如下

rust 复制代码
impl Message {

    fn send(&self) -> anyhow::Result<()> {
        // 将结构体转换为二进制
        let binary_data = self.to_bytes()?;

        // tcp发送二进制消息
        let mut stream =
            match TcpStream::connect(self.host) {
                Ok(stream) => stream,
                Err(e) => {
                    bail!(
                        "Couldn't connect to server {} {}",
                        self.host,
                        e
                    )
                }
            };
        stream.write_all(&binary_data)?;
        stream.flush()?;

        Ok(())
    }
}
相关推荐
小喷友16 小时前
阶段三:进阶(Rust 高级特性)
前端·rust
Python私教18 小时前
源滚滚Rust全栈班v1.02 无符号整数详解
开发语言·后端·rust
专注VB编程开发20年1 天前
CSS 的命名方式像是 PowerShell 的动词-名词结构,缺乏面向对象的层级关系
开发语言·后端·rust
伍哥的传说1 天前
Tailwind CSS v4 终极指南:体验 Rust 驱动的闪电般性能与现代化 CSS 工作流
前端·css·rust·tailwindcss·tailwind css v4·lightning css·utility-first
专注VB编程开发20年1 天前
rust语言-对象多级访问
服务器·前端·rust
编码浪子2 天前
趣味学RUST基础篇(构建一个命令行程序2重构)
开发语言·重构·rust
susnm2 天前
组件生命周期
rust·全栈
bruce541102 天前
Axum 最佳实践:如何构建优雅的 Rust 错误处理系统?(三)
rust
Source.Liu3 天前
【Python基础】 15 Rust 与 Python 基本类型对比笔记
笔记·python·rust
咸甜适中3 天前
rust语言 (1.88) egui (0.32.1) 学习笔记(逐行注释)(二十六)windows平台运行时隐藏控制台
笔记·学习·rust·egui