@ckstack/ck-push-retry
Version:
ckstack push retry server module
161 lines (160 loc) • 9.1 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
const events = __importStar(require("events"));
const JSON5 = __importStar(require("json5"));
const dk_lib_rmq_1 = require("@dkpkg/dk-lib-rmq");
const ck_lib_sequelize_1 = require("@ckstack/ck-lib-sequelize");
const ck_lib_models_1 = require("@ckstack/ck-lib-models");
const dk_lib_1 = require("@dkpkg/dk-lib");
const MyValues_1 = require("../MyValues");
const MyConfig_1 = require("../MyConfig");
class RetryDelayRmq {
static retryMessage(lhd, msgData) {
return __awaiter(this, void 0, void 0, function* () {
dk_lib_1.CkLogger.infoLog2('start resend message(rmq)', lhd);
let pushMsg = msgData;
let isHasCondition = false;
const selCond = {};
let userId = '';
switch (pushMsg === null || pushMsg === void 0 ? void 0 : pushMsg.pushVendor) {
case dk_lib_1.CkValue.PUSH_VENDOR_TYPE.FCM:
pushMsg = pushMsg;
if (pushMsg && pushMsg.targetUserIds && pushMsg.targetUserIds.length > 0) {
[userId] = pushMsg.targetUserIds;
}
break;
default:
pushMsg = pushMsg;
if (pushMsg && pushMsg.targetUserId) {
userId = pushMsg.targetUserId;
}
break;
}
if (userId) {
if (!selCond.$or) {
selCond.$or = [];
}
selCond.$or.push({
$and: {
yyyymm: pushMsg.yyyymm,
user_id: userId,
broadcast_id: pushMsg.broadcastId,
},
});
isHasCondition = true;
}
if (!isHasCondition) {
dk_lib_1.CkLogger.errLog2('end resend message(rmq). not found yyyymm, userId(targetUserId or targetUserIds), broadcastId value on rmq message data', lhd);
return;
}
const pushMsgDatas = yield ck_lib_sequelize_1.DatabaseService.getDatasByFilter(lhd, ck_lib_models_1.TbFcmHistoryTmp, selCond, ['rid', 'yyyymm', 'retry_index', 'user_id', 'arrival_flag', 'broadcast_id'], undefined, [['rid', 'ASC'], ['retry_index', 'DESC']]);
if (typeof pushMsgDatas === 'string') {
dk_lib_1.CkLogger.errLog2(`end resend message(rmq). can't find push message history tmp data. cond [${JSON5.stringify(selCond)}]`, lhd);
return;
}
const pushMsgArrivalMap = {};
if (pushMsgDatas.length > 0) {
for (const row of pushMsgDatas) {
const rowKey = dk_lib_1.CkCrypto.md5Hash(`${row.yyyymm}_${row.user_id}_${row.broadcast_id}`);
if (pushMsgArrivalMap[`${rowKey}`] === undefined) {
pushMsgArrivalMap[`${rowKey}`] = row.arrival_flag;
}
else if (pushMsgArrivalMap[`${rowKey}`] !== dk_lib_1.CkValue.PUSH_ALLIVAL.YES) {
pushMsgArrivalMap[`${rowKey}`] = row.arrival_flag;
}
}
}
const pushArrivalDatas = yield ck_lib_sequelize_1.DatabaseService.getDatasByFilter(lhd, ck_lib_models_1.TbFcmArrivalTmp, selCond, ['rid', 'yyyymm', 'retry_index', 'user_id', 'broadcast_id'], undefined, [['rid', 'ASC'], ['retry_index', 'DESC']]);
if (typeof pushArrivalDatas === 'string') {
dk_lib_1.CkLogger.errLog2(`end resend message(rmq). can't find push arrival tmp data. cond [${JSON5.stringify(selCond)}]`, lhd);
return;
}
const pushArrivalTmpMap = {};
if (pushArrivalDatas.length > 0) {
for (const row of pushArrivalDatas) {
const rowKey = dk_lib_1.CkCrypto.md5Hash(`${row.yyyymm}_${row.user_id}_${row.broadcast_id}`);
if (pushArrivalTmpMap[`${rowKey}`] === undefined) {
pushArrivalTmpMap[`${rowKey}`] = dk_lib_1.CkValue.PUSH_ALLIVAL.YES;
}
}
}
if (userId && pushMsg.currentRetryCount !== undefined && pushMsg.maxRetryCount !== undefined) {
const msgKey = dk_lib_1.CkCrypto.md5Hash(`${pushMsg.yyyymm}_${userId}_${pushMsg.broadcastId}`);
if (pushMsg.maxRetryCount > pushMsg.currentRetryCount) {
const arrivalFlag = pushMsgArrivalMap[`${msgKey}`];
const arrivalTmpFlag = pushArrivalTmpMap[`${msgKey}`];
dk_lib_1.CkLogger.infoLog2(`is arrival [${arrivalFlag === dk_lib_1.CkValue.PUSH_ALLIVAL.YES}], arrival tmp [${arrivalTmpFlag}]. currentRetryCount [${pushMsg.currentRetryCount}], maxRetryCount [${pushMsg.maxRetryCount}]`, `${pushMsg.lhd || lhd}`);
if (arrivalFlag !== dk_lib_1.CkValue.PUSH_ALLIVAL.YES && arrivalTmpFlag !== dk_lib_1.CkValue.PUSH_ALLIVAL.YES) {
pushMsg.currentRetryCount += 1;
const mqRet = yield dk_lib_rmq_1.CkRabbitMQ.sendMessage(pushMsg.lhd || lhd, MyConfig_1.conf.HTF_PUSH_QUEUE, pushMsg, true);
dk_lib_1.CkLogger.infoLog2(`re-send push. push message yyyymm [${pushMsg.yyyymm}], user id [${userId}], broadcast id [${pushMsg.broadcastId}], rmq result [${mqRet}]`, `${pushMsg.lhd || lhd}`);
}
else {
dk_lib_1.CkLogger.infoLog2(`push message arrived. push message yyyymm [${pushMsg.yyyymm}], user id [${userId}], broadcast id [${pushMsg.broadcastId}]`, `${pushMsg.lhd || lhd}`);
}
}
else {
dk_lib_1.CkLogger.infoLog2(`stop re-send push. retry count is limited. retry count [${pushMsg.currentRetryCount}], limit retry count [${pushMsg.maxRetryCount}]`, `${pushMsg.lhd || lhd}`);
}
}
else {
dk_lib_1.CkLogger.warnLog2(`can't check re-send push condition. boardcast_id [${pushMsg.broadcastId}], currentRetryCount [${pushMsg.currentRetryCount}], pushMsg.maxRetryCount [${pushMsg.maxRetryCount}], user id [${userId}]`, `${pushMsg.lhd || lhd}`);
}
dk_lib_1.CkLogger.infoLog2('end resend message(rmq)', lhd);
});
}
static onMessage(queueName, msg) {
if (RetryDelayRmq.IS_NEED_EVENT_BIND) {
RetryDelayRmq.TASK_EVENT.on(MyValues_1.CRON_EVENT_TYPE.BY_RMQ, RetryDelayRmq.retryMessage);
RetryDelayRmq.IS_NEED_EVENT_BIND = true;
}
let lhd = `${MyValues_1.PROC_LHD.IP_LHD_HTTP}`;
const content = msg.content.toString();
dk_lib_1.CkLogger.debugLog(`recv rabbitmq. msg [${content}], queue [${queueName}]`, lhd);
let msgData;
try {
msgData = JSON5.parse(content);
}
catch (err) {
dk_lib_1.CkLogger.errLog2('invalid rabbitmq message. it is not json', lhd);
return;
}
lhd = msgData.lhd || lhd;
RetryDelayRmq.TASK_EVENT.emit(MyValues_1.CRON_EVENT_TYPE.BY_RMQ, lhd, msgData);
}
}
RetryDelayRmq.TASK_EVENT = new events.EventEmitter();
RetryDelayRmq.IS_NEED_EVENT_BIND = true;
exports.default = RetryDelayRmq;