NodeJS -- 简单实现一个可写流

红太狼 2022-04-24 11:54 222阅读 0赞

实现node.js里面的可写流主要在于俩个点

1：有一个缓冲区，用于对没有来得及写入的数据进行缓存

2：有一个字段用来标记当前是否为写入状态。

在代码体现为, this.buffer一个数组用来保存缓冲区里面的值，this.writing表示当前是否为写入状态true表示正在写入

思路，我们调用write方法，读入数据时对当前状态进行判断，如果writing为false,那么置为true 调用fs.write方法去写入文件里面

在此期间，读入的其他字符会保存在缓冲区里面。每一写完一次数据，我们就会去检测缓冲区里面是否存在值，如果有，那么就去写入。此外，当缓冲区里的大小大于this.highWaterMark时，那么就需要return false, 不要在读了。

来实现看看把

const fs = require("fs");
    const EventEmitter = require("events");
    
    class WriteStream extends EventEmitter {
        constructor (path, options) {
            super(path, options);
            this.path = path;
            this.flags = options.flags || "w";
            this.mode = options.mode || 0o666;
            this.highWaterMark = options.highWaterMark || 16 * 2024;
            this.encoding = options.encoding || "utf8";
            this.start = options.start || 0;
            this.autoClose = options.autoClose || true;
    
            this.writing = false; //表示是否正在写入文件
            this.length = 0;// 当前缓冲区的大小
            this.buffer = []; //缓冲区
            this.fd = null; //文件描述符
            this.pos = 0; //写入文件的偏移量
            this.open(); //打开文件
     
        }
        open () {
            fs.open(this.path, this.flags, this.mode, (err, fd) => { //首先打开文件
                if (err) { //如果出现错误，
                    if (this.autoClose) {
                        fs.close( fd);
                    }
                    this.emit("error", "open出现问题");
                }
                this.fd = fd;
                this.emit("open")
            })   
        }
        write (chunk, encoding = "utf8", cb = () => {}) { //如果当前正在写数据，那么就应该放在缓存区里面
            chunk = Buffer.isBuffer(chunk)?chunk:Buffer.from(chunk, encoding); //转成buffer
           
            if (this.writing) { //如果现在正在写，那么就需要存入缓存区里面
                this.buffer.push({
                    chunk,
                    encoding,
                    cb
                })
                this.length += chunk.length;
            } else {
                this.writing = true;// 转化状态，表示正在写
                this.length += chunk.length;
                this._write(chunk, encoding, () => this.clearBuffer());
            }
           // console.log(this.length, this.highWaterMark)
            return (this.length < this.highWaterMark);
        }
        _write (chunk, encoding, cb) {
            if (typeof this.fd !== "number") {
                return this.once("open", ()=> this._write(chunk, encoding, cb))
            }
            fs.write(this.fd, chunk, 0, chunk.length, this.pos, (err, byte) => {
                if (err) {
                    if (this.autoClose) {
                        fs.close(this.fd, ()=>{});
                    }
                    this.emit("error", "_write出现问题");
                }
                this.pos += byte; //下一次的偏移量
              
                this.length -= byte;
              //  console.log(this.pos);
                cb && cb(); 
            })
        }
    
        clearBuffer () {
       
            let data = this.buffer.shift(); //弹出第一个
            if (data) {
                this._write(data.chunk, data.encoding, () => this.clearBuffer()); //递归调用自己，
            } else {
                this.writing = false; //表面已经可以继续写了，现在空闲
                this.emit("drain", "drain"); //发布drain事件
            }
        }
    }
    
    
    module.exports = WriteStream;

验证一下

const fs = require("fs");
    const WriteStream = require("./WriteSteam");
    const ws = new WriteStream("./1.txt", {
        highWaterMark: 3
    });
    ws.on("error", (data) => {
        console.log(data);
    })
    let n=9;
    function write() {
        let flag = true;
        while (flag && n>0) {
            
            flag = ws.write(n+"");
            n--;
        }
    }
    
    ws.on("drain", (data) => {
        console.log(data);
        write();
    })
    
     
        write()

没有问题咯![20190425002148586.png][]

[20190425002148586.png]: /images/20220215/659d17cce9b94f8d833cec87e43f9b63.png

发表评论取消回复

表情：

评论列表（有 0 条评论，222人围观）

还没有评论，来说两句吧...

相关阅读

相关【nodeJs】用cheerio写一个爬虫吧

爬取Cnode社区的文章用到的模块：express cheerio superagent superagent([http://visionmedia.gi

本是古典何须时尚/ 2022年05月24日 12:40/ 0 赞/ 241 阅读

相关【nodeJs】用nodeJs实现一个简单的前后端交互的例子

实现效果图： ![这里写图片描述][70] Ajax知识补充： 1，创建XMLHttpRequest对象： var xmlHttp = new XMLH

短命女/ 2022年05月24日 12:35/ 0 赞/ 248 阅读

相关【nodeJS】一个最简单的express程序

执行npm install express 安装好express后就可以使用express框架了下面是一个最简单的express程序 var ex

桃扇骨/ 2022年05月24日 11:39/ 0 赞/ 308 阅读

相关【NodeJS】用NodeJs写一个最简单的http服务器

const http = require('http') http.createServer(function(req, res) { r

浅浅的花香味﹌/ 2022年05月10日 02:44/ 0 赞/ 306 阅读

相关 NodeJS -- 简单实现一个可写流

实现node.js里面的可写流主要在于俩个点 1：有一个缓冲区，用于对没有来得及写入的数据进行缓存 2：有一个字段用来标记当前是否为写入状态。在代码体现为, this.

红太狼/ 2022年04月24日 11:54/ 0 赞/ 223 阅读

相关如何使用nodejs写一个接口

> 小编推荐：[Fundebug][]提供JS错误监控、微信小程序错误监控、微信小游戏错误监控，Node.j错误监控和Java错误监控。真的是一个很好用的错误监控费服务，众多大

心已赠人/ 2022年02月14日 08:57/ 0 赞/ 269 阅读

相关如何使用nodejs写一个接口

> 小编推荐：[Fundebug][]提供JS错误监控、微信小程序错误监控、微信小游戏错误监控，Node.j错误监控和Java错误监控。真的是一个很好用的错误监控费服务，众多大

梦里梦外;/ 2022年02月14日 08:51/ 0 赞/ 228 阅读

相关 NodeJS -- 实现一个可读流（流动模式）

上一节，我们实现了一个可写流[可写流][Link 1] 那么我们这一节实现一个可读流的（流动模式）。nodeJs有俩种模式的可读流，一种是暂停模式，一种是流动模式。什么是

灰太狼/ 2022年02月13日 08:51/ 0 赞/ 247 阅读

相关 NodeJS -- 一个简单的静态资源服务器

我们的目标是创建一个静态服务器，我们可以指定主机，端口，和根目录，如果访问的是文件夹，那么就显示文件目录，如果是文件，那么就显示出来。思路：我们使用http模块创建一个服务

女爷i/ 2022年01月30日 03:49/ 0 赞/ 370 阅读

相关 NodeJS -- 简单的实现一个登录功能

实现目标：默认起始页面时登录页面，输入账号密码，如果没有此账号那么跳到注册页面，注册成功后进入个人页面。登录时，如果账号密码不匹配，那么跳到错误页面，三秒后跳到登录页面。

谁践踏了优雅/ 2022年01月23日 07:09/ 0 赞/ 473 阅读