Esempio n. 1
0
  @Override
  public String readUtf8Line(boolean throwOnEof) throws IOException {
    long start = 0;
    long newline;
    while ((newline = buffer.indexOf((byte) '\n', start)) == -1) {
      start = buffer.size;
      if (source.read(buffer, Segment.SIZE) == -1) {
        if (throwOnEof) throw new EOFException();
        return buffer.size != 0 ? readUtf8((int) buffer.size) : null;
      }
    }

    if (newline > 0 && buffer.getByte(newline - 1) == '\r') {
      // Read everything until '\r\n', then skip the '\r\n'.
      String result = readUtf8((int) (newline - 1));
      skip(2);
      return result;

    } else {
      // Read everything until '\n', then skip the '\n'.
      String result = readUtf8((int) (newline));
      skip(1);
      return result;
    }
  }
Esempio n. 2
0
 @Test
 public void sourceFromInputStreamBounds() throws Exception {
   Source source = Okio.source(new ByteArrayInputStream(new byte[100]));
   try {
     source.read(new Buffer(), -1);
     fail();
   } catch (IllegalArgumentException expected) {
   }
 }
Esempio n. 3
0
 @Override
 public long seek(byte b) throws IOException {
   long start = 0;
   long index;
   while ((index = buffer.indexOf(b, start)) == -1) {
     start = buffer.size;
     if (source.read(buffer, Segment.SIZE) == -1) throw new EOFException();
   }
   return index;
 }
Esempio n. 4
0
 @Override
 public void skip(long byteCount) throws IOException {
   while (byteCount > 0) {
     if (buffer.size == 0 && source.read(buffer, Segment.SIZE) == -1) {
       throw new EOFException();
     }
     long toSkip = Math.min(byteCount, buffer.size());
     buffer.skip(toSkip);
     byteCount -= toSkip;
   }
 }
Esempio n. 5
0
  @Override
  public long read(OkBuffer sink, long byteCount) throws IOException {
    if (byteCount < 0) throw new IllegalArgumentException("byteCount < 0: " + byteCount);
    if (closed) throw new IllegalStateException("closed");

    if (buffer.size == 0) {
      long read = source.read(buffer, Segment.SIZE);
      if (read == -1) return -1;
    }

    long toRead = Math.min(byteCount, buffer.size);
    return buffer.read(sink, toRead);
  }
Esempio n. 6
0
  @Test
  public void sourceFromInputStream() throws Exception {
    InputStream in =
        new ByteArrayInputStream(("a" + repeat('b', Segment.SIZE * 2) + "c").getBytes(UTF_8));

    // Source: ab...bc
    Source source = Okio.source(in);
    Buffer sink = new Buffer();

    // Source: b...bc. Sink: abb.
    assertEquals(3, source.read(sink, 3));
    assertEquals("abb", sink.readUtf8(3));

    // Source: b...bc. Sink: b...b.
    assertEquals(Segment.SIZE, source.read(sink, 20000));
    assertEquals(repeat('b', Segment.SIZE), sink.readUtf8(sink.size()));

    // Source: b...bc. Sink: b...bc.
    assertEquals(Segment.SIZE - 1, source.read(sink, 20000));
    assertEquals(repeat('b', Segment.SIZE - 2) + "c", sink.readUtf8(sink.size()));

    // Source and sink are empty.
    assertEquals(-1, source.read(sink, 1));
  }
Esempio n. 7
0
 @Override
 public void require(long byteCount) throws IOException {
   while (buffer.size < byteCount) {
     if (source.read(buffer, Segment.SIZE) == -1) throw new EOFException();
   }
 }
Esempio n. 8
0
 @Override
 public boolean exhausted() throws IOException {
   return buffer.exhausted() && source.read(buffer, Segment.SIZE) == -1;
 }