在thinkjs中实现Gzip压缩和静态文件缓存

thinkjs在作者 welefen的努力下即将支持 目前需要实现Gzip压缩需要在github上下载最新版的thinkjs

修改core/View.js的 31-71行附近

 /**
 * 输出模版文件内容
 * @param {[type]} templateFile [description]
 * @param {[type]} charset [description]
 * @param {[type]} contentType [description]
 * @param {[type]} content [description]
 * @return {[type]} [description]
 */
 display: function(templateFile, charset, contentType){
 var self = this;
 return tag('view_init', this.http).then(function(){
 return self.fetch(templateFile);
 }).then(function(content){
 return self.render(content, charset, contentType).then(function(){
 return tag('view_end', self.http, content);
 });
 }).then(function(){
 return self.http.end();
 }).catch(function(err){
 console.error(err.stack);
 return self.http.end();
 });
 },
 /**
 * 渲染模版
 * @param {[type]} content [description]
 * @param {[type]} charset [description]
 * @param {[type]} contentType [description]
 * @return {[type]} [description]
 */
 render: function(content, charset, contentType){
 if (!this.http.cthIsSend) {
 charset = charset || C('encoding');
 contentType = contentType || C('tpl_content_type');
 this.http.setHeader('Content-Type', contentType + '; charset=' + charset);
 }
 if (C('show_exec_time')) {
 this.http.sendTime('Exec-Time');
 }
 return this.http.echo(content || '', charset || C('encoding'));
 },

修改 core/Controller.js 的340行左右

 /**
 * 结束输出,输出完成时一定要调用这个方法
 * @param {[type]} obj [description]
 * @return {[type]} [description]
 */
 end: function(obj, encoding) {
 var self=this;
 if (obj !== undefined) {
 self.echo(obj, encoding).then(function(){
 self.http.end();
 });
 }else{
 self.http.end();
 }
 },

这些相信不久welefen会修改的。 接下来实现tag行为 GzipContentBehavior.js. 我注释掉了deflate压缩去掉注释就可以实现deflate压缩,把文件放到App/Lib/Behavior下面

var zlib = require('zlib');
/**
 * Gzip压缩
 */
module.exports = Behavior(function () {
 return {
 run: function (content) {
 var self = this;
 var deferred = getDefer();
 // if (self.http.headers["accept-encoding"].indexOf("deflate") != -1) {
 // self.deflate(content, self.http, deferred);
 // } else if (self.http.headers["accept-encoding"].indexOf("gzip") != -1) {
 // self.gzip(content, self.http, deferred);
 // } else {
 // return content;
 // }
 if (self.http.headers["accept-encoding"].indexOf("gzip") != -1) {
 self.gzip(content, self.http, deferred);
 } else {
 return content;
 }
 return deferred.promise;
 },
 gzip: function (content, http, deferred) { //检测是否是手机访问
 zlib.gzip(content, function (err, data) {
 if (!err) {
 http.res.setHeader("Content-Encoding", "gzip");
 deferred.resolve(data);
 } else {
 deferred.resolve(content);
 }
 });
 },
 deflate: function (content, http, deferred) { //检测是否是手机访问
 zlib.deflate(content, function (err, data) {
 if (!err) {
 http.res.setHeader("Content-Encoding", "deflate");
 deferred.resolve(data);
 } else {
 deferred.resolve(content);
 }
 });
 },
 }
})

然后在App/Conf/tag.js里面配置一下

//行为
module.exports = {
 resource_check: ['CheckResource'],
 content_write:['GzipContent']
}

看到了CheckResource,它是实现静态文件gzip压缩和缓存的.它使用mime库所以需要npm install mime一下

var fs = require('fs');
var mime = require('mime');
var path = require("path");
var zlib = require('zlib');
/**
 * 静态资源请求
 * @return {[type]} [description]
 */
module.exports = Behavior(function () {
 'use strict';
 return {
 options: {
 'url_resource_on': false
 },
 run: function () {
 if (!RESOURCE_PATH || !this.options.url_resource_on || !this.http.pathname) {
 return false;
 }
 var pathname = this.http.pathname;
 if (pathname.indexOf('/') === 0) {
 pathname = pathname.substr(1);
 }
 var reg = C('url_resource_reg');
 //通过正则判断是否是静态资源请求
 if (!reg.test(pathname)) {
 return false;
 }
 var file = RESOURCE_PATH + '/' + pathname;
 var res = this.http.res;
 var req = this.http.req;
 fs.stat(file, function (err, stats) {
 if (err) {
 res.statusCode = 404;
 res.end();
 } else {
 var contentType = mime.lookup(file);
 var exp_maxage = C('expires_maxAge') || 30 * 24 * 3600;
 var expires = new Date();
 var lastModified = stats.mtime.toUTCString();
 var ifModifiedSince = "If-Modified-Since".toLowerCase();
 var acceptEncoding = "Accept-Encoding".toLowerCase();
 expires.setTime(expires.getTime() + exp_maxage * 1000);
 res.setHeader('Content-Type', contentType || "text/plain");
 res.setHeader("Cache-Control", "max-age=" + exp_maxage);
 res.setHeader("Expires", expires.toUTCString());
 res.setHeader("Last-Modified", lastModified);
 if (req.headers[ifModifiedSince] && lastModified == req.headers[ifModifiedSince]) {
 res.writeHead(304, "Not Modified");
 res.end();
 } else {
 var fileStream = fs.createReadStream(file);
 if ((req.headers[acceptEncoding] || '').indexOf('gzip') != -1&&contentType.match(/(javascript|css)/)) {
 res.setHeader('Content-Encoding', 'gzip');
 var gzip = fileStream.pipe(zlib.createGzip());
 gzip.pipe(res);
 gzip.on('end', function () {
 res.end();
 });
 }else {
 fileStream.pipe(res);
 fileStream.on('end', function () {
 res.end();
 });
 }
 }
 }
 });
 //返回一个pendding promise, 不让后续执行
 return getDefer().promise;
 }
 };
});

至此我们就完成了thinkjs的gzip和浏览器缓存的功能。 刚刚接触thinkjs不久,如果代码中有的地方不对。还望海涵。