89 lines
3.1 KiB
JavaScript
89 lines
3.1 KiB
JavaScript
'use strict';
|
|
const mongoose = require('mongoose');
|
|
const Fund = require('./src/modules/fund/fund.model');
|
|
const Action = require('./src/modules/action/action.model');
|
|
const PrettyStream = require('bunyan-pretty-colors');
|
|
const bunyan = require('bunyan');
|
|
const path = require('path');
|
|
const fs = require('fs');
|
|
|
|
const prettyStdOut = new PrettyStream();
|
|
|
|
if (!fs.existsSync('clean')) {
|
|
fs.mkdirSync('clean');
|
|
}
|
|
const logger = function (option) {
|
|
option = option || {};
|
|
return bunyan.createLogger({
|
|
name: "log",
|
|
src: false,
|
|
streams: [
|
|
{
|
|
level: 'info',
|
|
stream: prettyStdOut
|
|
},
|
|
{
|
|
level: 'info',
|
|
type: 'rotating-file',
|
|
path: path.join(option.root, option.file || 'info.log'),
|
|
period: '1d', // daily rotation
|
|
count: 365 // keep 30 days
|
|
}
|
|
]
|
|
});
|
|
};
|
|
const filename = 'clean' + new Date().getFullYear() + new Date().getMonth() + new Date().getDate();
|
|
const mylog = logger(({ root: 'clean', file: filename }));
|
|
|
|
const probe = {
|
|
currentIndex: 0
|
|
};
|
|
async function clean () {
|
|
let allFund = await Fund.find({ fundSum: { $gt: 0 } });
|
|
// 批量更新，每次十个
|
|
for (let currentIndex = 0; currentIndex < allFund.length; currentIndex++) {
|
|
const userFund = allFund[currentIndex];
|
|
const fundList = userFund.fundList;
|
|
const boughtAmountSum = fundList.reduce((sum, fund) => sum + +fund.amount, 0);
|
|
const releasedAmout = (await Action.find({ type: '1', userId: userFund.userId })).reduce((sum, action) => sum + +action.amount, 0);
|
|
const rest = boughtAmountSum - releasedAmout;
|
|
mylog.info('No. ', currentIndex, '用户' + userFund.userId, '剩余 ', rest);
|
|
const vic = userFund.asset.vic || 0;
|
|
await Fund.findByIdAndUpdate(userFund.id, { vicFundSum: vic, 'asset.vic': 0, fundList: [], 'option.restNeetToRelease': rest, snapshot: { asset: userFund.asset } });
|
|
probe.currentIndex = currentIndex;
|
|
}
|
|
process.exit();
|
|
}
|
|
|
|
const start = async (job, done) => {
|
|
const SysConfig = require('./configSec');
|
|
mongoose.set('useCreateIndex', true);
|
|
const { DB_USER_NAME, DB_PASSWD, DB_HOST, DB_PORT, DB_NAME } = SysConfig;
|
|
const db = mongoose.connection;
|
|
db.on('error', mylog.error);
|
|
db.once('open', function() {
|
|
mylog.info('数据库连接成功...');
|
|
});
|
|
mylog.info('开始连接数据库...');
|
|
await mongoose.connect(`mongodb://${DB_USER_NAME}:${DB_PASSWD}@${DB_HOST}:${DB_PORT}/${DB_NAME}`, {
|
|
useNewUrlParser: true,
|
|
bufferMaxEntries: 0,
|
|
autoReconnect: true,
|
|
useFindAndModify: false,
|
|
useUnifiedTopology: true
|
|
});
|
|
const beforeExit = async (msg) => {
|
|
mylog.info('意外退出', msg);
|
|
mylog.info('进行到第'+ probe.currentIndex + '个用户');
|
|
process.exit(0);
|
|
};
|
|
process.on('SIGINT', beforeExit);
|
|
process.on('uncaughtException', beforeExit);
|
|
process.on('unhandledRejection', beforeExit);
|
|
typeof process.send === 'function' && process.send('子进程开始执行mine');
|
|
clean(job, done);
|
|
typeof process.send === 'function' && process.send('子进程mine执行完毕');
|
|
};
|
|
|
|
start();
|