// 获取mock文件的mock数据 const setMockData = (moduleName) => { const {mockData} = require(moduleName); return mockData(); }; // 中间件暴露方法 module.exports = function (options) { const {routes, root} = options; return async (req, res, next) => { let {isReq, host} = await valid.isRequestPath(routes, req); // 不是请求地址直接return掉 if (!isReq) { next(); return; } // 如果存在Mock对应的文件 let filePath = await valid.isMockFileName(root, req.path); if (filePath) { // 检验本地mock文件开关是否开启 let check = await valid.inspectMockCheck(filePath); if (check) { // 发送本地mock数据 return res.send(setMockData(filePath)) } else { // 请求结果 let body = await request(host, req, res).catch(proxyRes => { res.status(proxyRes.statusCode); }); // 发送请求的结果信息 return res.send(body); } } else { // 请求返回主体 let body = await request(host, req, res).catch(proxyRes => { res.status(proxyRes.statusCode); next(); }); if (body) { // 定义需要写入文件路径 const filePath = path.resolve(root, `mock${req.path}.js`); // 写入mock文件 writeMockFile(filePath, body); // 响应返回主体 return res.send(body); } } }; };
以下是一些校验方法，详细代码就不贴了，具体源码可查看：https://github.com/moxaIce/lo...
isRequestPath校验是否为api接口请求, 返回 Promise包含isReq布尔值，host请求域名, route请求路由的对象
isMockFileName是否存在对应的mock文件，返回Promise返回匹配路径或者空字符串
inspectMockCheck校验模拟文件请求，开关是否开起, 返回布尔值
至于request方法和writeMockFile方法看下面的小结。
以下是我自己画的一个逻辑图，有点丑见谅：
请求代理
代理的作用不用多说，都知道是解决了前端起的服务和直接请求后台的跨域问题。我这儿主要是在中间件内部通过http.request方法发起一个http请求，对于http.request方法的使用可以看这里, 里面也有比较详细的示例，我这儿贴上我写的代码：
/** * @description 请求方法 */ const url = require('url'); const http = require('http'); module.exports = function (host, req, res) { let body = ''; return new Promise((resolve, reject) => { const parse = url.parse(host); let proxy = http.request( { host: host.hostname, port: parse.port, method: req.method, path: req.path, headers: req.headers }, (proxyRes) => { // 非200字段内直接响应错误 ， 在主逻辑里处理 if (proxyRes.statusCode < 200 || proxyRes.statusCode > 300) { reject(proxyRes) } proxyRes.on('data', (chunk) => { body += chunk.toString(); }).on('end', () => { try { resolve(JSON.parse(body)); } catch (e) { // 将响应结果返回，在主文件做异常回调 reject(proxyRes) } }).on('error', (err) => { console.log(`error is`, err); }) }); proxy.on('error', (e) => { console.error(`请求报错：${e.message}`) }); proxy.end() }) };
代理的实现比较简单，主要通过外层传入host和requset, response在内部用url解析得到ip然后配置request的options, 通过监听data与end事件将得到的主体报文resolve出去，以及中间对非200段内的响应处理。
文件写入
通过中间传options传入的root， 可以得到完整的mock路径path.resolve(__dirname, mock${req.path}.js)。传入到写入mock文件方法里
module.exports = async function (filePath, body) { await dirExists(path.dirname(filePath)); fs.writeFile(filePath, echoTpl(JSON.stringify(body)), function (err) { if (err) { console.log(`写入文件失败`) } }); }
定义mockjs模板
module.exports = async function (filePath, body) { await dirExists(path.dirname(filePath)); fs.writeFile(filePath, echoTpl(JSON.stringify(body)), function (err) { if (err) { console.log(`写入文件失败`) } }); }
dirExists通过递归的方式写入文件目录
module.exports = async function (filePath, body) { await dirExists(path.dirname(filePath)); fs.writeFile(filePath, echoTpl(JSON.stringify(body)), function (err) { if (err) { console.log(`写入文件失败`) } }); }
效果演示
使用koa起一个node服务并且暴露如下路由和其中的数据,具体代码可以看这儿，我这儿只贴上了关键代码
服务端代码：
router.get('/app/home/baseInfo', user_controller.baseInfo) router.post('/app/login', user_controller.login) const login = async (ctx, next) => { ctx.body = { success: true, message: '', code: 0, data: { a: 1, b: '2' } } }; const baseInfo = async (ctx, next) => { ctx.body = { success: true, errorMsg: '', data: { avatar: 'http://aqvatarius.com/themes/taurus/html/img/example/user/dmitry_b.jpg', total: 333, completed: 30, money: '500' } }; };
client代码