nodeJS静态服务器

引入依赖模块

let config = require('./config');
let chalk = require('chalk');//粉笔模块(命令行变色)
let http = require('http');//http模块
let fs = require('fs');//文件模块
let zlib = require('zlib');
let handlebars = require('handlebars');//模板引擎
let url = require('url');//url模块解析请求url
let mime = require('mime');//mime模块通过文件名称/路径拿到内容类型
let path = require('path');
let {promisify,inspect} = require('util');
let stat = promisify(fs.stat);
let readdir = promisify(fs.readdir);
process.env.DEBUG = 'static:*';
let debug = require('debug')('static:app');

编译模板函数

function list() {
    //编译模板，得到渲染方法，传入数据
    let tmpl = fs.readFileSync(path.resolve(__dirname,'src/template','list.html'),'utf8');
    return handlebars.compile(tmpl);//返回一个函数
}

此函数主要用于编译模板，返回一个文件list列表用于返回给客户端。

start方法启动服务

start(){
    //启动服务
    let server = http.createServer();
    server.on('request',this.request.bind(this));
    server.listen(this.config.port,()=>{
        let url = `${this.config.host}:${this.config.port}`;
        debug(`start at ${chalk.green(url)}`);
    })
}

this.config.port可以自己定义端口，debug用于在控制台打印信息。

请求处理方法

async request(req,res){
    //接收请求
    let { pathname } = url.parse(req.url);
    let filepath = path.join(this.config.root,pathname);
    try{
        let statObj = await stat(filepath);
        if(statObj.isDirectory()){//判断是否为目录
            //是目录的话使用handlebars模板引擎渲染目录
            let files = await readdir(filepath);//字符串数组
            files = files.map((file)=>{
                return {
                    name:file,
                    url:path.join(pathname,file)
                }
            });
            let html = this.list({
                title:pathname,
                files:files
            });
            res.setHeader('Content-Type','text/html');
            this.sendFile(req,res,html,statObj,'html');
        }else{
            this.sendFile(req,res,filepath,statObj,'');
        }
    }catch (e){
        debug(inspect(e));
        this.sendError(req,res)
    }
}

这个是主要处理逻辑，首先通过解析客户端的请求url,通过path模块来拿到文件路径，第一个参数是config中配置的静态文件根目录；然后来判断当前请求的是文件还是目录，这里使用一个try catch代码块来捕获异常，如果请求的是个目录的话，遍历这个目录下的所有文件并且将文件名和文件路径传给list编译模板，然后交给发送方法去处理。

发送处理方法

sendFile(req,res,filepath,statObj,type){
    //发送文件
    let expires = new Date(Date.now() + 60 * 1000);
    //缓存设置
    res.setHeader('Expires', expires.toUTCString());
    res.setHeader('Cache-Control', 'max-age=600');
    res.setHeader('Accept-Range', 'bytes');//断点续传
    if(type == 'html'){
        res.end(filepath);
    }else {
        res.setHeader('Content-Type',mime.getType(filepath));
        fs.createReadStream(filepath).pipe(res);
    }
}

发送方法里设置的缓存策略为强制缓存，并且在发送方法里调用zlibFn方法去看客户端请求是否使用压缩，并在最后将内容pipe到res里返回给客户端。

zlib压缩方法

zlibFn(req,res){
    let acceptEncoding = request.headers['accept-encoding'];//压缩
    if (!acceptEncoding) {
        acceptEncoding = '';
    }
    if (acceptEncoding.match(/\bdeflate\b/)) {
        res.setHeader('Content-Encoding','deflate');
        return zlib.createDeflate();
    } else if (acceptEncoding.match(/\bgzip\b/)) {
        res.setHeader('Content-Encoding','gzip');
        return zlib.createGzip();
    } else {
        return null;
    }
}

这里主要是通过请求头中的accept-encoding来判断是否压缩，以何种方式压缩，压缩后返回一个流对象。

初写node静态服务器（简易版）全部代码请见Github