BatchImpl.js

/**
 * Created by Administrator on 2015/5/13.
 */
var GtReq = require('./getui/GtReq');
var GtConfig = require('./GtConfig');
var utils = require('./getui/utils');
class BatchImpl {
    constructor(appKey, push) {
        this.batchId = utils.uuid();
        this.appKey = appKey;
        this.push = push;
        this.seqId = 0;
        this.innerMsgList = new Array();
        this.lastPostData = null;
    }
    getBatchId() {
        return this.batchId;
    };

    add(message, target) {
        if (this.seqId >= 5000) {
            throw new Error("Can not add over 5000 message once! Please call submit() first.");
        } else {
            var json = utils.createPostParams(message, target, null, this.appKey);
            var item = new GtReq.SingleBatchItem({
                'seqId': this.seqId,
                'data': JSON.stringify(json)
            });
            this.innerMsgList.push(item);
            this.seqId += 1;
        }
        return this.seqId;
    };

    submit(callback) {
        var requestId = utils.uuid();

        var data = {
            'requestId': requestId,
            'appkey': this.appKey,
            'action': 'pushMessageToSingleBatchAction',
            'serialize': 'pb',
            'async': GtConfig.isPushSingleBatchAsync()
        };

        try {
            var request = new GtReq.SingleBatchRequest({
                'batchId': this.batchId,
                'batchItem': this.innerMsgList
            });
            data.singleDatas = request.toBase64();
            this.push.httpPostJson(this.push._host, data, true, callback);
        } finally {
            this.seqId = 0;
            this.innerMsgList = [];
            this.lastPostData = data;
        }

    };

    retry(callback) {
        this.push.httpPostJson(this.push._host, this.lastPostData, true, callback);
    };
}
module.exports = BatchImpl;