@aliyunmq/mq-http-sdk/lib/helper.js

Aliyun Message Queue(MQ) Nodejs Http SDK

'use strict';

const xml2js = require('xml2js');

exports.parseXML = function (input) {
  return new Promise((resolve, reject) => {
    xml2js.parseString(input, (err, obj) => {
      if (err) {
        return reject(err);
      }
      resolve(obj);
    });
  });
};

exports.extract = function extract (arr) {
  if (arr && arr.length === 1 && typeof arr[0] === 'string') {
    return arr[0];
  }

  arr.forEach((item) => {
    Object.keys(item).forEach((key) => {
      item[key] = extract(item[key]);
    });
  });

  return arr;
};

function format (params) {
  if (typeof params === 'string') {
    return `<![CDATA[${params}]]>`;
  }

  var xml = '';
  Object.keys(params).forEach((key) => {
    const value = params[key];
    if (typeof value === 'object') {
      xml += `<${key}>${format(value)}</${key}>`;
    } else {
      xml += `<${key}><![CDATA[${value}]]></${key}>`;
    }
  });
  return xml;
}

exports.toXMLBuffer = function (entityType, params, subType) {
  var xml = '<?xml version="1.0" encoding="UTF-8"?>';
  xml +=    `<${entityType} xmlns="http://mq.aliyuncs.com/doc/v1/">`;
  if (Array.isArray(params)) {
    params.forEach((item) => {
      xml +=  `<${subType}>`;
      xml += format(item);
      xml +=  `</${subType}>`;
    });
  } else {
    xml +=    format(params);
  }
  xml +=    `</${entityType}>`;
  return Buffer.from(xml, 'utf8');
};

exports.getCanonicalizedMQHeaders = function (headers) {
  return Object.keys(headers)
    .filter((key) => key.startsWith('x-mq-'))
    .sort()
    .map((key) => `${key}:${headers[key]}\n`)
    .join('');
};

exports.processMsgProperties = function (msg) {
  var props = {};
  if (msg.Properties) {
    var props = {};
    var kvArray = msg.Properties.split('|');
    for (var i=0; i < kvArray.length; i++) {
      if (kvArray[i] == '') {
        continue;
      }
      var kAndV = kvArray[i].split(':');
      if (kAndV.length != 2 || kAndV[0] == '' || kAndV[1] == '') {
        continue;
      }
      props[kAndV[0]] = kAndV[1];
    }
    if (props['KEYS']) {
      msg.MessageKey = props['KEYS'];
      delete props['KEYS'];
    }
    if (props['__STARTDELIVERTIME']) {
      msg.StartDeliverTime = parseInt(props['__STARTDELIVERTIME']);
      delete props['__STARTDELIVERTIME'];
    }
    if (props['__TransCheckT']) {
      msg.StartDeliverTime = parseInt(props['__TransCheckT']);
      delete props['__TransCheckT'];
    }
    if (props['__SHARDINGKEY']) {
      msg.ShardingKey = props['__SHARDINGKEY'];
      delete props['__SHARDINGKEY'];
    }
  }
  msg.Properties = props;
};