智慧水务管理系统 - 精河县供水工程综合管理平台

encoder.js 4.2KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
  1. "use strict";
  2. module.exports = encoder;
  3. var Enum = require("./enum"),
  4. types = require("./types"),
  5. util = require("./util");
  6. /**
  7. * Generates a partial message type encoder.
  8. * @param {Codegen} gen Codegen instance
  9. * @param {Field} field Reflected field
  10. * @param {number} fieldIndex Field index
  11. * @param {string} ref Variable reference
  12. * @returns {Codegen} Codegen instance
  13. * @ignore
  14. */
  15. function genTypePartial(gen, field, fieldIndex, ref) {
  16. return field.delimited
  17. ? gen("types[%i].encode(%s,w.uint32(%i),q+1).uint32(%i)", fieldIndex, ref, (field.id << 3 | 3) >>> 0, (field.id << 3 | 4) >>> 0)
  18. : gen("types[%i].encode(%s,w.uint32(%i).fork(),q+1).ldelim()", fieldIndex, ref, (field.id << 3 | 2) >>> 0);
  19. }
  20. /**
  21. * Generates an encoder specific to the specified message type.
  22. * @param {Type} mtype Message type
  23. * @returns {Codegen} Codegen instance
  24. */
  25. function encoder(mtype) {
  26. /* eslint-disable no-unexpected-multiline, block-scoped-var, no-redeclare */
  27. var gen = util.codegen(["m", "w", "q"])
  28. ("if(!w)")
  29. ("w=Writer.create()")
  30. ("if(q===undefined)q=0")
  31. ("if(q>util.recursionLimit)")
  32. ("throw Error(\"max depth exceeded\")");
  33. var i, ref;
  34. // "when a message is serialized its known fields should be written sequentially by field number"
  35. var fields = /* initializes */ mtype.fieldsArray.slice().sort(util.compareFieldsById);
  36. for (var i = 0; i < fields.length; ++i) {
  37. var field = fields[i].resolve(),
  38. index = mtype._fieldsArray.indexOf(field),
  39. type = field.resolvedType instanceof Enum ? "int32" : field.type,
  40. wireType = types.basic[type];
  41. ref = "m" + util.safeProp(field.name);
  42. // Map fields
  43. if (field.map) {
  44. gen
  45. ("if(%s!=null&&Object.hasOwnProperty.call(m,%j)){", ref, field.name) // !== undefined && !== null
  46. ("for(var ks=Object.keys(%s),i=0;i<ks.length;++i){", ref);
  47. if (field.keyType === "bool") gen
  48. ("w.uint32(%i).fork().uint32(%i).bool(util.boolFromKey(ks[i]))", (field.id << 3 | 2) >>> 0, 8 | types.mapKey[field.keyType]);
  49. else if (types.long[field.keyType] !== undefined) gen
  50. ("w.uint32(%i).fork().uint32(%i).%s(util.longFromKey(ks[i],%j))", (field.id << 3 | 2) >>> 0, 8 | types.mapKey[field.keyType], field.keyType, field.keyType === "uint64" || field.keyType === "fixed64");
  51. else gen
  52. ("w.uint32(%i).fork().uint32(%i).%s(ks[i])", (field.id << 3 | 2) >>> 0, 8 | types.mapKey[field.keyType], field.keyType);
  53. if (wireType === undefined) gen
  54. ("types[%i].encode(%s[ks[i]],w.uint32(18).fork(),q+1).ldelim().ldelim()", index, ref); // can't be groups
  55. else gen
  56. (".uint32(%i).%s(%s[ks[i]]).ldelim()", 16 | wireType, type, ref);
  57. gen
  58. ("}")
  59. ("}");
  60. // Repeated fields
  61. } else if (field.repeated) { gen
  62. ("if(%s!=null&&%s.length){", ref, ref); // !== undefined && !== null
  63. // Packed repeated
  64. if (field.packed && types.packed[type] !== undefined) { gen
  65. ("w.uint32(%i).fork()", (field.id << 3 | 2) >>> 0)
  66. ("for(var i=0;i<%s.length;++i)", ref)
  67. ("w.%s(%s[i])", type, ref)
  68. ("w.ldelim()");
  69. // Non-packed
  70. } else { gen
  71. ("for(var i=0;i<%s.length;++i)", ref);
  72. if (wireType === undefined)
  73. genTypePartial(gen, field, index, ref + "[i]");
  74. else gen
  75. ("w.uint32(%i).%s(%s[i])", (field.id << 3 | wireType) >>> 0, type, ref);
  76. } gen
  77. ("}");
  78. // Non-repeated
  79. } else {
  80. if (!field.required) gen
  81. ("if(%s!=null&&Object.hasOwnProperty.call(m,%j))", ref, field.name); // !== undefined && !== null
  82. if (wireType === undefined)
  83. genTypePartial(gen, field, index, ref);
  84. else gen
  85. ("w.uint32(%i).%s(%s)", (field.id << 3 | wireType) >>> 0, type, ref);
  86. }
  87. }
  88. return gen
  89. ("if(m.$unknowns!=null&&Object.hasOwnProperty.call(m,\"$unknowns\"))")
  90. ("for(var i=0;i<m.$unknowns.length;++i)")
  91. ("w.raw(m.$unknowns[i])")
  92. ("return w");
  93. /* eslint-enable no-unexpected-multiline, block-scoped-var, no-redeclare */
  94. }