prep_encoder.go 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. package kafka
  2. import "math"
  3. type prepEncoder struct {
  4. length int
  5. err error
  6. }
  7. // primitives
  8. func (pe *prepEncoder) putInt8(in int8) {
  9. pe.length += 1
  10. }
  11. func (pe *prepEncoder) putInt16(in int16) {
  12. pe.length += 2
  13. }
  14. func (pe *prepEncoder) putInt32(in int32) {
  15. pe.length += 4
  16. }
  17. func (pe *prepEncoder) putInt64(in int64) {
  18. pe.length += 8
  19. }
  20. // arrays
  21. func (pe *prepEncoder) putInt32Array(in []int32) {
  22. pe.length += 4
  23. pe.length += 4 * len(in)
  24. }
  25. func (pe *prepEncoder) putArrayCount(in int) {
  26. pe.length += 4
  27. }
  28. // misc
  29. func (pe *prepEncoder) putError(in KError) {
  30. pe.length += 2
  31. }
  32. func (pe *prepEncoder) putString(in *string) {
  33. pe.length += 2
  34. if in == nil {
  35. return
  36. }
  37. if len(*in) > math.MaxInt16 {
  38. pe.err = EncodingError{"String too long"}
  39. } else {
  40. pe.length += len(*in)
  41. }
  42. }
  43. func (pe *prepEncoder) putBytes(in *[]byte) {
  44. pe.length += 4
  45. if in == nil {
  46. return
  47. }
  48. if len(*in) > math.MaxInt32 {
  49. pe.err = EncodingError{"Bytes too long"}
  50. } else {
  51. pe.length += len(*in)
  52. }
  53. }
  54. // stackable
  55. func (pe *prepEncoder) push(in pushEncoder) {
  56. pe.length += in.reserveLength()
  57. }
  58. func (pe *prepEncoder) pushLength32() {
  59. pe.length += 4
  60. }
  61. func (pe *prepEncoder) pushCRC32() {
  62. pe.length += 4
  63. }
  64. func (pe *prepEncoder) pop() {
  65. }