代碼地址 https://github.com/MoceptWeb/egg-schedule
看之前確定至少了解了egg的文檔了, 不然看下面是浪費時間哈
egg定時任務
很多情況我們需要去定時(或者就一次)執(zhí)行某些腳本, 這個時候就可以利用egg的schedule功能了
怎樣寫
schedule
- {app_root}/app/schedule.js
// 通過 schedule 屬性來設置定時任務的執(zhí)行間隔等配置
static get schedule() {
return {
interval: '5m', // 5 min間隔
type: 'worker', // worker 類型:每臺機器上只有一個 worker 會執(zhí)行這個定時任務
};
}
async subscribe() {
this.ctx.logger.info('schedule updateOa2User begin')
const res = await this.ctx.service.portal.oa.updateOa2User();
this.ctx.logger.info('schedule updateOa2User finish')
}
啟動時候自定決定啟動哪些定時任務
在get schedule() 中配置immediate:配置了該參數(shù)為 true 時,這個定時任務會在應用啟動并 ready 后立刻執(zhí)行一次這個定時任務
app.beforeStart(async () => {
// await app.runSchedule('updateOa2User');updatePortal2User
await app.runSchedule('updatePortal2User');
});
service
- {app_root}/app/service/portal/oa.js
sqlserver 連接
// helper.js 統(tǒng)一連接一個sqlserver connection
const mssql = require('mssql')
let pool = await mssql.connect(config)
// app.js 連接錯誤的日志
mssql.on('error', err => {
console.log('database err', err)
app.logger.error(err);
// ... error handler
})
異步中的多個全部異步完成的操作
- 例如對一個數(shù)據(jù)進行一個個的異步操作, 但是最后獲取全部異步后的結果
切忌不要在異步中做同步操作, 不然執(zhí)行順序很意外?。?!
async updateUser(oaUser) {
const self = this;
let promiseAll = [];
// 設置promise對象,同步發(fā)請求
oaUser.forEach(oa => {
promiseAll.push(Promise.resolve(self.updateUserByName(oa)));
});
await Promise.all(promiseAll);
}
- {app_root}/app/service/portal/portal.js
異步中的多個全部異步完成的操作
如果需要一步步獲取全部數(shù)據(jù)(這里是同步)后才要進行全部異步其他操作, 那么需要先將同步操作封裝成promise支持異步操作
async readLine(target) {
let array = [];
const self = this;
const file = await new Promise(function (resolve, reject) {
lineReader.eachLine(target, function(line, last) {
// line.split(/\s/)[0]
let data = line.split(/\s/)
if(data[2]) {
array.push({
user_id: self.ctx.helper.trim(data[0], '"'),
mail: self.ctx.helper.trim(data[1], '"'),
user_center_id: self.ctx.helper.trim(data[2], '"'),
})
}
if(last) {
resolve(array)
}
})
})
return file;
}
// 使用
const fileData = await this.readLine(target);
let promiseAll = [];
fileData.forEach(portal => {
promiseAll.push(Promise.resolve(self.updateUserByPortalId(portal)));
});
await Promise.all(promiseAll);
mysql
- sql的占位符
const res = await conn.query('update t_user set mail = ? where user_id = ?', [oaUser.email, user.user_id]);
- sql的事務
async sqlBeginTransaction() {
const conn = await this.app.mysql.beginTransaction();
try {
const res = await conn.query('update t_user set mail = ? where user_id = ?', [oaUser.email, user.user_id]);
if(res && res.affectedRows === 1) {
await conn.commit();
} else if(res.affectedRows > 1){
// this.ctx.logger.error('')
await conn.rollback()
} else {
await conn.rollback()
}
} catch (err) {
// error, rollback
await conn.rollback(); // rollback call won't throw err
throw err;
}
}
extend
- {app_root}/app/extend/helper.js
統(tǒng)一封裝各種數(shù)據(jù)的連接和error信息
test
- {app_root}/app/test/service/portal/portal.test.js
待完善
config
謹記更環(huán)境用的配置是不一樣的
單元測試用的是 config.unittest.js
為什么這樣寫
TODO 分析源碼
拓展和優(yōu)化
- 將service中的純數(shù)據(jù)庫封裝到model
- 基于redis的任務消息隊列的優(yōu)化
不需要人工去進行開啟任務 - 編寫egg-mssql連接sqlserver