packet_decoder.go 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. package sarama
  2. // PacketDecoder is the interface providing helpers for reading with Kafka's encoding rules.
  3. // Types implementing Decoder only need to worry about calling methods like GetString,
  4. // not about how a string is represented in Kafka.
  5. type packetDecoder interface {
  6. // Primitives
  7. getInt8() (int8, error)
  8. getInt16() (int16, error)
  9. getInt32() (int32, error)
  10. getInt64() (int64, error)
  11. getVarint() (int64, error)
  12. getArrayLength() (int, error)
  13. getBool() (bool, error)
  14. // Collections
  15. getBytes() ([]byte, error)
  16. getVarintBytes() ([]byte, error)
  17. getRawBytes(length int) ([]byte, error)
  18. getString() (string, error)
  19. getNullableString() (*string, error)
  20. getInt32Array() ([]int32, error)
  21. getInt64Array() ([]int64, error)
  22. getStringArray() ([]string, error)
  23. // Subsets
  24. remaining() int
  25. getSubset(length int) (packetDecoder, error)
  26. peek(offset, length int) (packetDecoder, error) // similar to getSubset, but it doesn't advance the offset
  27. peekInt8(offset int) (int8, error) // similar to peek, but just one byte
  28. // Stacks, see PushDecoder
  29. push(in pushDecoder) error
  30. pop() error
  31. }
  32. // PushDecoder is the interface for decoding fields like CRCs and lengths where the validity
  33. // of the field depends on what is after it in the packet. Start them with PacketDecoder.Push() where
  34. // the actual value is located in the packet, then PacketDecoder.Pop() them when all the bytes they
  35. // depend upon have been decoded.
  36. type pushDecoder interface {
  37. // Saves the offset into the input buffer as the location to actually read the calculated value when able.
  38. saveOffset(in int)
  39. // Returns the length of data to reserve for the input of this encoder (eg 4 bytes for a CRC32).
  40. reserveLength() int
  41. // Indicates that all required data is now available to calculate and check the field.
  42. // SaveOffset is guaranteed to have been called first. The implementation should read ReserveLength() bytes
  43. // of data from the saved offset, and verify it based on the data between the saved offset and curOffset.
  44. check(curOffset int, buf []byte) error
  45. }
  46. // dynamicPushDecoder extends the interface of pushDecoder for uses cases where the length of the
  47. // fields itself is unknown until its value was decoded (for instance varint encoded length
  48. // fields).
  49. // During push, dynamicPushDecoder.decode() method will be called instead of reserveLength()
  50. type dynamicPushDecoder interface {
  51. pushDecoder
  52. decoder
  53. }