使用Node.js原生API写一个Web服务器

本文可运行代码示例已经上传GitHub，大家可以拿下来玩玩：https://github.com/dennis-jiang/Front-End-Knowledges/tree/master/Examples/Node.js/HttpServer

Hello World

要搭建一个简单的Web服务器，使用原生的http模块就够了，一个简单的Hello World程序几行代码就够了： 

const http = require('http')  
const port = 3000  
const server = http.createServer((req, res) => {  
  res.statusCode = 200  
  res.setHeader('Content-Type', 'text/plain')  
  res.end('Hello World')  
})  
server.listen(port, () => {  
  console.log(`Server is running on http://127.0.0.1:${port}/`)  
}) 

这个例子就很简单，直接用http.createServer创建了一个服务器，这个服务器也没啥逻辑，只是在访问的时候返回Hello World。服务器创建后，使用server.listen运行在3000端口就行。

这个例子确实简单，但是他貌似除了输出一个Hello World之外，啥也干不了，离我们一般使用的Web服务器还差了很远，主要是差了这几块：

1.  不支持HTTP动词，比如GET，POST等
2.  不支持路由
3.  没有静态资源托管
4.  不能持久化数据

前面三点是一个Web服务器必备的基础功能，第四点是否需要要看情况，毕竟目前很多Node的Web服务器只是作为一个中间层，真正跟数据库打交道做持久化的还是各种微服务，但是我们也应该知道持久化怎么做。

所以下面我们来写一个真正能用的Web服务器，也就是说把前面缺的几点都补上。

处理路由和HTTP动词

前面我们的那个Hello World也不是完全不能用，因为代码位置还是得在http.createServer里面，我们就在里面添加路由的功能。为了跟后面的静态资源做区分，我们的API请求都以/api开头。要做路由匹配也不难，最简单的就是直接用if条件判断就行。为了能拿到请求地址，我们需要使用url模块来解析传过来的地址。而Http动词直接可以用req.method拿到。所以http.createServer改造如下： 

const url = require('url');  
const server = http.createServer((req, res) => {  
  // 获取url的各个部分  
  // url.parse可以将req.url解析成一个对象  
  // 里面包含有pathname和querystring等  
  const urlurlObject = url.parse(req.url);  
  const { pathname } = urlObject;  
  // api开头的是API请求  
  if (pathname.startsWith('/api')) {  
    // 再判断路由  
    if (pathname === '/api/users') {  
      // 获取HTTP动词  
      const method = req.method;  
      if (method === 'GET') {  
        // 写一个假数据  
        const resData = [  
          {  
            id: 1,  
            name: '小明',  
            age: 18  
          },  
          {  
            id: 2,  
            name: '小红',  
            age: 19  
          }  
        ];  
        res.setHeader('Content-Type', 'application/json')  
        res.end(JSON.stringify(resData));  
        return;  
      }  
    }  
  }  
}); 

现在我们访问/api/users就可以拿到用户列表了：

支持静态文件

上面说了API请求是以/api开头，也就是说不是以这个开头的可以认为都是静态文件，不同文件有不同的Content-Type，我们这个例子里面暂时只支持一种.jpg吧。其实就是给我们的if (pathname.startsWith('/api'))加一个else就行。返回静态文件需要：

1.  使用fs模块读取文件。
2.  返回文件的时候根据不同的文件类型设置不同的Content-Type。

所以我们这个else就长这个样子： 

// ... 省略前后代码 ...  
else {  
  // 使用path模块获取文件后缀名  
  const extName = path.extname(pathname);  
  if (extName === '.jpg') {  
    // 使用fs模块读取文件  
    fs.readFile(pathname, (err, data) => {  
      res.setHeader('Content-Type', 'image/jpeg');  
      res.write(data);  
      res.end();  
    })  
  }  
} 

然后我们在同级目录下放一个图片试一下：

数据持久化

数据持久化的方式有好几种，一般都是存数据库，少数情况下也有存文件的。存数据库比较麻烦，还需要创建和连接数据库，我们这里不好demo，我们这里演示一个存文件的例子。一般POST请求是用来存新数据的，我们在前面的基础上再添加一个POST /api/users来新增一条数据，只需要在前面的if (method === 'GET')后面加一个POST的判断就行： 

// ... 省略其他代码 ...  
else if (method === 'POST') {  
  // 注意数据传过来可能有多个chunk  
  // 我们需要拼接这些chunk  
  let postData = '';  
  req.on('data', chunk => {  
    postDatapostData = postData + chunk;  
  })  
  req.on('end', () => {  
    // 数据传完后往db.txt插入内容  
    fs.appendFile(path.join(__dirname, 'db.txt'), postData, () => {  
      res.end(postData);  // 数据写完后将数据再次返回  
    });  
  })  
} 

然后我们测试一下这个API:

再去看看文件里面写进去没有：

总结

到这里我们就完成了一个具有基本功能的web服务器，代码不复杂，但是对于帮我们理解Node web服务器的原理很有帮助。但是上述代码还有个很大的问题就是：代码很丑！所有代码都写在一堆，而且HTTP动词和路由匹配全部是使用if条件判断，如果有几百个API，再配合十来个动词，那代码简直就是个灾难！所以我们应该将路由处理，HTTP动词，静态文件，数据持久化这些功能全部抽离出来，让整个应用变得更优雅，更好扩展。这就是Express和Koa这些框架存在的意义，下一篇文章我们就去Express的源码看看他是怎么解决这个问题的，点个关注不迷路~