Make bincode.write respect "BincodeSize" for enums

Syndica · Jun 13, 2024 · b807264 · b807264
1 parent 94f2212
commit b807264
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/bincode/bincode.zig b/src/bincode/bincode.zig
@@ -419,13 +419,17 @@ pub fn write(writer: anytype, data: anytype, params: bincode.Params) !void {
         .Type, .Void, .NoReturn, .Undefined, .Null, .Fn, .Opaque, .Frame, .AnyFrame => return,
         .Bool => return writer.writeByte(@intFromBool(data)),
         .Enum => |_| {
+            comptime var SerializedSize = u32;
+            comptime if (@hasDecl(T, "BincodeSize")) {
+                SerializedSize = T.BincodeSize;
+            };
             if (getConfig(T)) |type_config| {
                 if (type_config.serializer) |serialize_fcn| {
                     return serialize_fcn(writer, data, params);
                 }
             }
 
-            return bincode.write(writer, @as(u32, @intFromEnum(data)), params);
+            return bincode.write(writer, @as(SerializedSize, @intFromEnum(data)), params);
         },
         .Union => |info| {
             try bincode.write(writer, @as(u32, @intFromEnum(data)), params);