Go语言io pipe源码分析详情

作者:zwf193071 时间:2024-01-31 00:21:46 

pipe.go分析:

  • 这个文件使用到了errors包,也是用到了sync库.

  • 文件说明:pipe是一个适配器,用于连接Reader和Writer.

1.结构分析

对外暴露的是一个构造函数和构造的两个对象. 两个对象分别暴露了方法,同时这两个对象还有一个共同的底层对象. 实际上,这两个对象暴露的方法是直接调用底层对象的, 那么核心还是在底层对象上,只是通过两个对象和一个构造方法将底层对象的细节隐藏了.

2.pipe sruct分析

pipe的方法不多,新的写法却不少.   

type atomicError struct{ v atomic.Value }

    func (a *atomicError) Store(err error) {
      a.v.Store(struct{ error }{err})
    }
    func (a *atomicError) Load() error {
      err, _ := a.v.Load().(struct{ error })
      return err.error
    }

atomicError提供了error的原子读写. 

  type pipe struct {
      wrMu sync.Mutex // Serializes Write operations
      wrCh chan []byte
      rdCh chan int

      once sync.Once // Protects closing done
      done chan struct{}
      rerr atomicError
      werr atomicError
    }

可以看到pipe结构体中主要分两块:

  • 读写信道

    • 两个无缓冲信道

    • 一个互斥量(保护暴露的写函数)

  • 结束标识

    • once保证done的关闭只执行一次

    • done标志整个读写的结束

    • 剩下两个用于存储读写错误

    • PipeReader/PipeWriter的分析

3.PipeReader对外暴露的是读/关闭

    type PipeReader struct {
      p *pipe
    }

    func (r *PipeReader) Read(data []byte) (n int, err error) {
      return r.p.Read(data)
    }

    func (r *PipeReader) Close() error {
      return r.CloseWithError(nil)
    }

    func (r *PipeReader) CloseWithError(err error) error {
      return r.p.CloseRead(err)
    }

PipeWriter对外暴露的是写/关闭 

   type PipeWriter struct {
       p *pipe
     }

    func (w *PipeWriter) Write(data []byte) (n int, err error) {
      return w.p.Write(data)
    }

    func (w *PipeWriter) Close() error {
      return w.CloseWithError(nil)
    }

    func (w *PipeWriter) CloseWithError(err error) error {
      return w.p.CloseWrite(err)
    }

他们的方法集都是指针接收者.具体方法的实现是通过pipe的方法完成的. pipe的方法更加明确:读/获取读错误/结束读写并设置读错误; 写/获取写错误/结束读写并设置写错误.思路相当明确.

下面主要分析pipe的读写 

  func (p *pipe) Read(b []byte) (n int, err error) {
      select {
      case <-p.done:
        return 0, p.readCloseError()
      default:
      }

      select {
      case bw := <-p.wrCh:
        nr := copy(b, bw)
        p.rdCh <- nr
        return nr, nil
      case <-p.done:
        return 0, p.readCloseError()
      }
    }

    func (p *pipe) Write(b []byte) (n int, err error) {
      select {
      case <-p.done:
        return 0, p.writeCloseError()
      default:
        p.wrMu.Lock()
        defer p.wrMu.Unlock()
      }

      for once := true; once || len(b) > 0; once = false {
        select {
        case p.wrCh <- b:
          nw := <-p.rdCh
          b = b[nw:]
          n += nw
        case <-p.done:
          return n, p.writeCloseError()
        }
      }
      return n, nil
    }

读写都是利用两个阶段的select来完成,第一个阶段的select是判断读写有没有结束, 第二阶段处理实际的读写.

Read

  • 每次将读的数量写到读信道

Write

  • 先将缓冲写到写信道,再从读信道中获取读字节数,最后调整缓冲

  • 如果缓冲太大,一次读没读完,就将写的过程多来几遍,知道缓冲全部写完

4.写法

PipeWriter/PipeReader对外暴露的关闭,其实只可以保留一个CloseWithError, 但是为了方便客户(调用者),还是拆成两个,其实可以做测试比较一下. 性能测试发现拆成两个或写成一个可选参函数,性能上差别不大, 那这种写法的主要作用是让暴露的方法更加清晰易懂.

pipe.Write中,for循环带有once参数,可以保证循环至少来一次, 算是do while的一种实现.

5.总结

不管是PipeReader/PipeWriter,还是pipe,都对Reader/Writer有(部分)实现.

另外还有一些细节没有说道:读写错误和EOF.

反思:本次阅读是先理代码后看文档,才发现关于error部分没有留心到, 后面还是先文档后代码,这样效率会高一点.

来源:https://juejin.cn/post/7060406951619854367

标签:Go,io,pipe,源码,分析
0
投稿

猜你喜欢

  • 你需要知道的CSS3 动画技术[译]

    2009-12-30 17:02:00
  • 详解vue-router 2.0 常用基础知识点之router.push()

    2024-04-09 10:49:35
  • 跨浏览器让javascript文件携带图片数据

    2011-03-31 17:12:00
  • 如何通过Django使用本地css/js文件

    2022-04-28 22:04:33
  • python中yield函数的用法详解

    2022-03-14 13:57:13
  • 一文教你如何使用Python绘制瀑布图

    2023-07-10 18:29:28
  • Python中的Matplotlib模块入门教程

    2023-08-14 23:05:16
  • Python网络爬虫之获取网络数据

    2022-10-09 12:35:54
  • 玩转python selenium鼠标键盘操作(ActionChains)

    2023-03-06 05:21:43
  • vue中的mixins混入使用方法

    2023-07-02 17:01:47
  • 将tensorflow.Variable中的某些元素取出组成一个新的矩阵示例

    2022-01-17 23:49:50
  • asp读取xml实例代码

    2011-03-08 11:13:00
  • python2 与 python3 实现共存的方法

    2023-06-13 23:56:29
  • 不唐突的JavaScript的七条准则[翻译]

    2008-12-09 13:33:00
  • sql server数据库最大Id冲突问题解决方法之一

    2012-01-05 19:28:42
  • python批量修改交换机密码的示例

    2023-06-29 07:52:42
  • 利用J2ME与ASP建立数据库连接

    2010-04-03 20:53:00
  • MySQL数据库事务transaction示例讲解教程

    2024-01-27 06:43:04
  • Python使用回溯法子集树模板解决迷宫问题示例

    2021-07-09 14:42:43
  • 简单好用的PHP分页类

    2023-11-22 09:32:39
  • asp之家 网络编程 m.aspxhome.com