进化！将nodejs代码重构为ES6风格

进化！将nodejs代码重构为ES6风格

作为零后端基础直接去做node后端的小白，曾经纠结了很长一段时间去理解node的异步以及代码风格，在2016年最终使用了传统的es5和callback作为编码风格和处理异步的方法。对于callback的回调地狱，则选择async.js来处理。但是面对复杂的逻辑，缩进依然让轻微代码洁癖者感到恶心。node8的出现突然警醒了我，不能再这么堕落了！虽然稳定版本还在node6，但是依然可以利用promise或async/await重构代码！于是开始了简化代码的尝试。

注：笔者特别菜，如在观看此文时有何指正，请不吝赐教

callback

线上一段恶心的代码，不用细读，就是恶心一下

这里用了async.waterfall、async.each处理异步，request和自己简单封装的postgresql操作数据库。

router.get('/account',function (req,res,next) {
var userid=req.query.userid;

var date=moment().format('YYYY-MM-DD');
async.waterfall([
function (callback) {
postgresql.getYsClient(function (getClient,done) {
var client=getClient;
var sql=util.format("SELECT ta.id,ta.appid,ta.account_type as type,ta.name,ta.account_status " +
"FROM tb_third_account ta,tb_user_and_third_account uata " +
"WHERE ta.id=uata.account_id and ta.account_type=110102 and uata.user_id=%s",userid);
client.query(sql,function (err,results) {
done();
if(err){
return callback(err);
}
callback(null,results.rows);
})
})
},function (list,callback) {
async.each(list,function (item,cb) {
var url=urls_api.getFeedAccountReport(item.appid,date,date);
var options={
method:'GET',
url:url,
json:true
}
request(options,function (error,response,results) {
if(!error&&response.statusCode===200){

<
4000
span class="hljs-keyword">if(results.result==='failed'||results.data.length===0){
console.log(options)
return cb()
}

item.view=results.data[0].kpis[0];
item.click=results.data[0].kpis[1];
item.cost=results.data[0].kpis[2];
item.cpc=results.data[0].kpis[3];
item.ctr=results.data[0].kpis[4];
item.cpm=results.data[0].kpis[5];

cb();
}else {
cb();
}
})
},function (err) {
callback(null,list);
})
}
],function (err,list) {
res_util.back_json(res,list,'success',null);
})
});

现在我自己看这段代码都感觉不忍直视。

如果要重构成Promise风格的代码，首先要把我对sql的封装和request改成Promise函数。那么先从sql的封装开始。

co+yield

我在/modules/postgresql 里是这么写的

首先我创建了一个连接池pool_msdb这里写法就不多说了，pg文档里就有，client是获取的链接，done方法是要在查询结束后将链接放回连接池，如果不done掉的话，这条链接将一直被占用。多次使用这个方法，连接池将无链接可用。

var getClient = function(callback){

pool_msdb.connect(function (err,client,done) {
if(err){
return console.error('error fetching client from pool', err);
}

callback(client,done);
});
}

这样封装的话，外部使用的话就要这么写

postgresql.getYsClient(function (getClient,done) {
var client=getClient;
var sql='';
client.query(sql,function (err,results) {
done();
if(err){
return callback(err);
}
callback(null,results.rows);
})
})

好麻烦，将封装改写成

var msdbClientExecute=function (sql,param) {
return new Promise(function (resolve,reject) {
pool_msdb.connect(function (err,client,done) {
if(err){
return console.error('error fetching client from pool', err);
}
client.query(sql,param,function (error,results) {
done();
if(error){
console.log(sql)
return reject(error);
}
resolve(results.rows);
})
});
})
}

client.query(sql,param,callback)

这种写法比拼sql语句好在可以防止注入式攻击。也是pg文档里的，可见我当时写代码多么随意。。。

这个方法直接传入sql语句和参数，返回一个promise，promise成功的返回值就是查询结果。这里自己去拿一个链接，查询后直接done掉。避免了写代码时忘了done掉非常尴尬。

那么在调用时，就可以这么写

let sql="select * from table where id=$1"

let list =yield postgresql.ysClientExecute(sql,[userid]);

爽不爽！？？？

还有个request，request并不支持返回一个promise，那么我们在github上随随便便找一个star比较高的轮子，request-promise

const request_p=require('request-promise');

写起来炒鸡简单

var results=yield request_p(options);

那么下一步，就是用yield和co配合搞搞事情。

router.get('/account',function (req,res,next) {
let userid=req.query.userid;

let date=moment().format('YYYY-MM-DD');

co(function* () {

let sql="selext * from table where userid=$1";

let list =yield (postgresql.ysClientExecute(sql,[userid]));

for(let item of list){

let url=urls_api.getFeedAccountReport(item.appid,date,date);
let options={
method:'GET',
url:url,
json:true
}

results=yield (request_p(options));

if(results.data&&results.data.length>0&&results.data[0].kpis){

item.view=results.data[0].kpis[0];
item.click=results.data[0].kpis[1];
item.cost=results.data[0].kpis[2];
item.cpc=results.data[0].kpis[3];
item.ctr=results.data[0].kpis[4];
item.cpm=results.data[0].kpis[5];
}
}

res_util.back_json(res,list,'success',null);

})

});

这里用co作为执行器，在co这个代码块里，yield后面跟一个Pomise对象。yield会同步执行这个Promise。

async/await

co+yield虽然很爽。但是并不是我的追求。

不是说官方亲儿子是async/await吗。

我的版本还停留在node6，查了一下可以用bable去支持，但是我看着感觉麻烦（懒癌，该改）。有一个我觉得比较简单的方法，就是asyncawait这个库，star挺多看来用的人不少。引入这个库后，async/await的操作和原生唯一的不同在于需要将后面的函数加个括号。我觉得既能快速体验async/await，以后改成原生也好改，当然要试试了。

const async1=require('asyncawait/async');
const await=require('asyncawait/await');

那么代码就变成

router.get('/account',function (req,res,next) {
let {userid}=req.query;//es6写法，解构赋值

let date=moment().format('YYYY-MM-DD');

(async1(function () {

let sql="selec * from tabel where userid=$1";

let list =await (postgresql.ysClientExecute(sql,[userid]));

for(let item of list){

let url=urls_api.getFeedAccountReport(item.appid,date,date);
let options={
method:'GET',
url:url,
json:true
}

let results='';

results=await (request_p(options));

if(results.data&&results.data.length>0&&results.data[0].kpis){

item.view=results.data[0].kpis[0];
item.click=results.data[0].kpis[1];
item.cost=results.data[0].kpis[2];
item.cpc=results.data[0].kpis[3];
item.ctr=results.data[0].kpis[4];
item.cpm=results.data[0].kpis[5];
}
}

res_util.back_json(res,list,'success',null);

}))().catch(e=>{
res_util.back_json(res,null,'failed',e);
})

});

这个catch简直是神来之笔，

简直了

原来觉得写try/catch好麻烦，一行缩进啊！现在完美解决

还有谁！