prep_encoder.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  1. package protocol
  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 len(in) > math.MaxInt16 {
  35. pe.err = EncodingError("String too long")
  36. } else {
  37. pe.length += len(in)
  38. }
  39. }
  40. func (pe *prepEncoder) putBytes(in []byte) {
  41. pe.length += 4
  42. if in == nil {
  43. return
  44. }
  45. if len(in) > math.MaxInt32 {
  46. pe.err = EncodingError("Bytes too long.")
  47. } else {
  48. pe.length += len(in)
  49. }
  50. }
  51. func (pe *prepEncoder) putRaw(in []byte) {
  52. pe.length += len(in)
  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. }