AAOS项目中如何用Protocol Buffers映射Rust嵌套枚举？

由于AAOS的Protobuf IDL仅支持简单枚举类型，无法直接映射Rust中的嵌套枚举结构，最直接的解决方式是将嵌套层级的枚举变体扁平化到单一Protobuf枚举中，通过命名前缀区分原有的分类，再在Rust端实现双向转换逻辑。

步骤1：定义符合AAOS要求的Protobuf枚举 #

将所有车辆类型的变体合并到一个枚举中，用前缀（如CAR_、TRUCK_）保留原有的分类信息，同时遵循Protobuf枚举的规范（必须包含值为0的UNKNOWN类型）：

syntax = "proto3";

package vehicle;

enum Vehicle {
  UNKNOWN_VEHICLE = 0; // 必须的默认值
  CAR_SEDAN = 1;
  CAR_COUPE = 2;
  CAR_HATCHBACK = 3;
  TRUCK_PICKUP = 4;
  TRUCK_SEMI = 5;
}

步骤2：Rust端的类型转换实现 #

如果需要保留原有嵌套枚举的业务逻辑，可以通过实现From/TryFrom trait完成自定义枚举与Protobuf生成枚举的双向转换（假设使用prost作为Protobuf Rust代码生成工具）：

// 由prost根据上述proto文件生成的枚举类型
#[derive(Debug, Clone, PartialEq, prost::Enumeration)]
#[repr(i32)]
pub enum VehicleProto {
    UnknownVehicle = 0,
    CarSedan = 1,
    CarCoupe = 2,
    CarHatchback = 3,
    TruckPickup = 4,
    TruckSemi = 5,
}

// 你的原有嵌套枚举结构
enum Vehicle {
    Car(CarType),
    Truck(TruckType),
}

enum CarType {
    Sedan,
    Coupe,
    Hatchback,
}

enum TruckType {
    Pickup,
    Semi,
}

// 自定义Vehicle转Protobuf枚举
impl From<Vehicle> for VehicleProto {
    fn from(vehicle: Vehicle) -> Self {
        match vehicle {
            Vehicle::Car(CarType::Sedan) => VehicleProto::CarSedan,
            Vehicle::Car(CarType::Coupe) => VehicleProto::CarCoupe,
            Vehicle::Car(CarType::Hatchback) => VehicleProto::CarHatchback,
            Vehicle::Truck(TruckType::Pickup) => VehicleProto::TruckPickup,
            Vehicle::Truck(TruckType::Semi) => VehicleProto::TruckSemi,
        }
    }
}

// Protobuf枚举转自定义Vehicle（处理未知类型）
impl TryFrom<VehicleProto> for Vehicle {
    type Error = String;

    fn try_from(proto: VehicleProto) -> Result<Self, Self::Error> {
        match proto {
            VehicleProto::CarSedan => Ok(Vehicle::Car(CarType::Sedan)),
            VehicleProto::CarCoupe => Ok(Vehicle::Car(CarType::Coupe)),
            VehicleProto::CarHatchback => Ok(Vehicle::Car(CarType::Hatchback)),
            VehicleProto::TruckPickup => Ok(Vehicle::Truck(TruckType::Pickup)),
            VehicleProto::TruckSemi => Ok(Vehicle::Truck(TruckType::Semi)),
            VehicleProto::UnknownVehicle => Err("未知车辆类型".into()),
        }
    }
}

// 原有车队创建函数可保持不变
fn create_vehicle_fleet() -> [Vehicle; 8] {
    [
        Vehicle::Car(CarType::Sedan),
        Vehicle::Truck(TruckType::Pickup),
        Vehicle::Car(CarType::Coupe),
        Vehicle::Truck(TruckType::Semi),
        Vehicle::Car(CarType::Hatchback),
        Vehicle::Car(CarType::Sedan),
        Vehicle::Truck(TruckType::Pickup),
        Vehicle::Car(CarType::Coupe),
    ]
}

// 示例：将车队转换为Protobuf枚举数组
fn fleet_to_proto(fleet: &[Vehicle]) -> Vec<VehicleProto> {
    fleet.iter().map(|v| v.clone().into()).collect()
}

替代方案：直接对齐Protobuf枚举结构 #

如果原有代码对嵌套枚举的依赖不强，可以直接修改Rust枚举结构与Protobuf定义对齐，省去转换步骤：

#[derive(Debug, Clone, PartialEq)]
enum Vehicle {
    UnknownVehicle,
    CarSedan,
    CarCoupe,
    CarHatchback,
    TruckPickup,
    TruckSemi,
}

fn create_vehicle_fleet() -> [Vehicle; 8] {
    [
        Vehicle::CarSedan,
        Vehicle::TruckPickup,
        Vehicle::CarCoupe,
        Vehicle::TruckSemi,
        Vehicle::CarHatchback,
        Vehicle::CarSedan,
        Vehicle::TruckPickup,
        Vehicle::CarCoupe,
    ]
}

内容的提问来源于stack exchange，提问作者Js7