diff options
| -rw-r--r-- | src/protobuf/gen.nim | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/protobuf/gen.nim b/src/protobuf/gen.nim index 53e7b4e..512c0c5 100644 --- a/src/protobuf/gen.nim +++ b/src/protobuf/gen.nim @@ -307,11 +307,14 @@ proc generateWriteMessageProc(desc: NimNode): NimNode = procName = postfix(ident("write" & getMessageName(desc)), "*") body = newStmtList() stream = ident("stream") + sizeproc = postfix(ident("sizeOf" & getMessageName(desc)), "*") for field in fields(desc): add(body, genWriteField(field)) result = quote do: + proc `sizeproc`(`messageId`: `mtype`): uint64 + proc `procName`(`stream`: ProtobufStream, `messageId`: `mtype`) = `body` |