okio解析

使用okhttp的时候,看过源码会知道,里面是使用了okio对流进行处理,那么okio到底是什么?
okio 同样也是square公司推出的io处理利器,它相对传统的io操作有一些优势。

优势

1、更低的CPU消耗和内存开销。okio内部使用了segment(片)来存放数据,segment里面放的也是byte[],但在外面有一个segmentPool来对segment进行回收复用。避免了频繁创建segment而导致的内存开销。
2、使用更简单。传统的io流,采用了装饰模式的思想。装饰模式可以增强类的功能,而不通过继承的方式。如果需要读取一个整形数据,我们会使用DataInputStream进行包装,或者如果我们需要用到缓存,会用到BufferInputStream。但okio不需要这么麻烦,okio的Buffer已经提供了一系列的方法,比如readInt(),readUtf8()等等。
3、okio中的Timeout提供了超时的处理。

实例分析

okio中最重要的两个接口分别是Sink和Source
Source:这个接口相当于InputStream,是用来读数据。数据来源可以是磁盘、网络、内存等
Sink:这个接口相当于OutputStream,是用于往外写数据。
InputStream和OutputStream可以以内存为参照物,加载到内存的输入流、而从内存中输出到别的地方,称为输出流
从一个最简单的例子入手去了解okio的内部逻辑

        File file=new File("/mnt/sdcard/需求.txt");
        InputStream in=new FileInputStream(file);
        BufferedSource source = Okio.buffer(Okio.source(in));  //创建BufferedSource
        String s = source.readUtf8();  //以UTF-8读
        System.out.println(s);     //打印

        BufferedSink sink=Okio.buffer(Okio.sink(new File("/mnt/sdcard/需求copy.txt")));
        sink.writeUtf8(s);
        source.close();
        sink.close();

这个例子是从设备磁盘中读取txt文件的内容 并打印出来。再将内容写到另一个txt文件中。
按顺序先看一下Okio.source(in)方法

  public static Source source(InputStream in) {
    return source(in, new Timeout());
  }
  private static Source source(final InputStream in, final Timeout timeout) {
    if (in == null) throw new IllegalArgumentException("in == null");
    if (timeout == null) throw new IllegalArgumentException("timeout == null");

    return new Source() {
      @Override public long read(Buffer sink, long byteCount) throws IOException {
        if (byteCount < 0) throw new IllegalArgumentException("byteCount < 0: " + byteCount);
        if (byteCount == 0) return 0;
        try {
          /检查读取是否超时 如果超时了直接抛出异常
          timeout.throwIfReached(); 
          //获取一个可写的segment
          Segment tail = sink.writableSegment(1);
          int maxToCopy = (int) Math.min(byteCount, Segment.SIZE - tail.limit);
          int bytesRead = in.read(tail.data, tail.limit, maxToCopy);
          if (bytesRead == -1) return -1;
          tail.limit += bytesRead;
          sink.size += bytesRead;
          return bytesRead;
        } catch (AssertionError e) {
          if (isAndroidGetsocknameError(e)) throw new IOException(e);
          throw e;
        }
      }

      @Override public void close() throws IOException {
        in.close();
      }

      @Override public Timeout timeout() {
        return timeout;
      }

      @Override public String toString() {
        return "source(" + in + ")";
      }
    };
  }

这个方法其实是构建了一个Source的实例对象,并实现了read函数,内部通过Buffer对象sink的writableSegment函数返回一个可写的segment,并调用inputStream的read函数,将数据写入到segment中。那么这个sink是从哪里来的,我们接着往下分析。

Okio.buffer(Okio.source(in));

进到Okio的 buffer(Source source)函数

  public static BufferedSource buffer(Source source) {
    return new RealBufferedSource(source);
  }

这个方法返回一个RealBufferedSource对象,RealBufferedSource类实现了BufferedSource接口,而BufferedSource又实现了Source接口。RealBufferedSource内部又有一个Source对象和一个Buffer对象。这里其实有点装饰模式的味道了,通过RealBufferedSource增强了read方法。


image.png

现在我们已经做好准备工作了,已经有了一个RealBufferSource对象了,接下来就是调用该对象的readxxx()方法了,我们这里用的是readUtf8(),具体看一下。

  @Override public String readUtf8() throws IOException {
    buffer.writeAll(source);
    return buffer.readUtf8();
  }

首先先看Buffer的writeAll方法

  @Override public long writeAll(Source source) throws IOException {
    if (source == null) throw new IllegalArgumentException("source == null");
    long totalBytesRead = 0;
    for (long readCount; (readCount = source.read(this, Segment.SIZE)) != -1; ) {
      totalBytesRead += readCount;
    }
    return totalBytesRead;
  }

重点看一下for循环里面的代码,其中的source对象就是前面构造出来的那个对象,不断调用它的read方法,将数据写入当前Buffer中。从前面的代码分析中可以知道,实际上,buffer对象通过调用writableSegment函数返回一个segment,将数据写到这个segment中。

Segment和Buffer

那么segment到底是什么,Buffer对象又是什么,之间有什么联系?


okiopng.png

okio使用segment来存储流数据,segment之间是双向链表的数据结构,buffer对象持有该链表的表头 head,通过head 可以快速定位到其它segment。Segment内部使用了byte[]数组,整体采用了链表和数组结合的模式,可以说在数据查找和修改移除做了相对平衡的选择。

注意:Segment中的split方法,从字面上了解是将一个segment对象 一分为二,用链表再串起来。但每个segment引用的是同一个byte[],只不过是在数组上的不同区域,即不同的pos和limit。所以这个时候需要区分该segment是属于owner还是share,这两者互斥。如果是share,那对该segment处理便有很多限制,因为会影响其它引用同一个byte[]的segment。

回到例子中,在将数据读到buffer中后,调用了readUtf8方法,将字节流转换utf8字符串。

  @Override public String readString(long byteCount, Charset charset) throws EOFException {
    checkOffsetAndCount(size, 0, byteCount);
    if (charset == null) throw new IllegalArgumentException("charset == null");
    if (byteCount > Integer.MAX_VALUE) {
      throw new IllegalArgumentException("byteCount > Integer.MAX_VALUE: " + byteCount);
    }
    if (byteCount == 0) return "";

    Segment s = head;
    if (s.pos + byteCount > s.limit) {
      // If the string spans multiple segments, delegate to readBytes().
      return new String(readByteArray(byteCount), charset);
    }

    String result = new String(s.data, s.pos, (int) byteCount, charset);
    s.pos += byteCount;
    size -= byteCount;

    if (s.pos == s.limit) {
      head = s.pop();
      SegmentPool.recycle(s);
    }

    return result;
  }

将数据读出来 通过newString处理。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 196,200评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,526评论 2 373
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,321评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,601评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,446评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,345评论 1 273
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,753评论 3 387
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,405评论 0 255
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,712评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,743评论 2 314
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,529评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,369评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,770评论 3 300
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,026评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,301评论 1 251
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,732评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,927评论 2 336

推荐阅读更多精彩内容