diff options
| author | Oskari Timperi <oskari.timperi@iki.fi> | 2018-03-25 13:31:35 +0300 |
|---|---|---|
| committer | Oskari Timperi <oskari.timperi@iki.fi> | 2018-03-25 13:31:35 +0300 |
| commit | f26641d78d7ff7a9d19114ec77a9f97378d1f68c (patch) | |
| tree | 97ca459ac533669d3ced7aa4729e53f898c89b38 /src/protobuf/gen.nim | |
| parent | a3ae66e1aa6a79c3b1c5e23ccdce778bba829c09 (diff) | |
| download | nimpb-f26641d78d7ff7a9d19114ec77a9f97378d1f68c.tar.gz nimpb-f26641d78d7ff7a9d19114ec77a9f97378d1f68c.zip | |
Add packed support to generateSizeOfMessageProc()
Diffstat (limited to 'src/protobuf/gen.nim')
| -rw-r--r-- | src/protobuf/gen.nim | 27 |
1 files changed, 18 insertions, 9 deletions
diff --git a/src/protobuf/gen.nim b/src/protobuf/gen.nim index cd47138..3ef13c2 100644 --- a/src/protobuf/gen.nim +++ b/src/protobuf/gen.nim @@ -396,15 +396,24 @@ proc generateSizeOfMessageProc(desc: NimNode): NimNode = # TODO: packed if isRepeated(field): - procBody.add quote do: - for value in `fname`: - let - sizeOfField = `sizeofproc`(value) - tagSize = sizeOfUint32(uint32(makeTag(`number`, `wiretype`))) - `resultId` = `resultId` + - sizeOfField + - sizeOfUint64(sizeOfField) + - tagSize + if isPacked(field): + procBody.add quote do: + if `hasproc`(`messageId`): + let + tagSize = sizeOfUint32(uint32(makeTag(`number`, WireType.LengthDelimited))) + dataSize = packedFieldSize(`fname`, `wiretype`) + sizeOfSize = sizeOfUint64(dataSize) + `resultId` = tagSize + dataSize + sizeOfSize + else: + procBody.add quote do: + for value in `fname`: + let + sizeOfField = `sizeofproc`(value) + tagSize = sizeOfUint32(uint32(makeTag(`number`, `wiretype`))) + `resultId` = `resultId` + + sizeOfField + + sizeOfUint64(sizeOfField) + + tagSize else: let sizeOfFieldId = ident("sizeOfField") |