311 lines
12 KiB
Java
311 lines
12 KiB
Java
|
package io.grpc.internal;
|
||
|
|
|
||
|
|
import java.io.IOException;
|
||
|
|
import java.io.OutputStream;
|
||
|
|
import java.nio.ByteBuffer;
|
||
|
|
import java.nio.InvalidMarkException;
|
||
|
|
import java.util.ArrayDeque;
|
||
|
|
import java.util.Deque;
|
||
|
|
import java.util.Iterator;
|
||
|
|
|
||
|
|
/* loaded from: classes6.dex */
|
||
|
|
public class CompositeReadableBuffer extends AbstractReadableBuffer {
|
||
|
|
private boolean marked;
|
||
|
|
private final Deque<ReadableBuffer> readableBuffers;
|
||
|
|
private int readableBytes;
|
||
|
|
private Deque<ReadableBuffer> rewindableBuffers;
|
||
|
|
private static final NoThrowReadOperation<Void> UBYTE_OP = new NoThrowReadOperation<Void>() { // from class: io.grpc.internal.CompositeReadableBuffer.1
|
||
|
|
@Override // io.grpc.internal.CompositeReadableBuffer.NoThrowReadOperation, io.grpc.internal.CompositeReadableBuffer.ReadOperation
|
||
|
|
public int read(ReadableBuffer readableBuffer, int i, Void r3, int i2) {
|
||
|
|
return readableBuffer.readUnsignedByte();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
private static final NoThrowReadOperation<Void> SKIP_OP = new NoThrowReadOperation<Void>() { // from class: io.grpc.internal.CompositeReadableBuffer.2
|
||
|
|
@Override // io.grpc.internal.CompositeReadableBuffer.NoThrowReadOperation, io.grpc.internal.CompositeReadableBuffer.ReadOperation
|
||
|
|
public int read(ReadableBuffer readableBuffer, int i, Void r3, int i2) {
|
||
|
|
readableBuffer.skipBytes(i);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
private static final NoThrowReadOperation<byte[]> BYTE_ARRAY_OP = new NoThrowReadOperation<byte[]>() { // from class: io.grpc.internal.CompositeReadableBuffer.3
|
||
|
|
@Override // io.grpc.internal.CompositeReadableBuffer.NoThrowReadOperation, io.grpc.internal.CompositeReadableBuffer.ReadOperation
|
||
|
|
public int read(ReadableBuffer readableBuffer, int i, byte[] bArr, int i2) {
|
||
|
|
readableBuffer.readBytes(bArr, i2, i);
|
||
|
|
return i2 + i;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
private static final NoThrowReadOperation<ByteBuffer> BYTE_BUF_OP = new NoThrowReadOperation<ByteBuffer>() { // from class: io.grpc.internal.CompositeReadableBuffer.4
|
||
|
|
@Override // io.grpc.internal.CompositeReadableBuffer.NoThrowReadOperation, io.grpc.internal.CompositeReadableBuffer.ReadOperation
|
||
|
|
public int read(ReadableBuffer readableBuffer, int i, ByteBuffer byteBuffer, int i2) {
|
||
|
|
int limit = byteBuffer.limit();
|
||
|
|
byteBuffer.limit(byteBuffer.position() + i);
|
||
|
|
readableBuffer.readBytes(byteBuffer);
|
||
|
|
byteBuffer.limit(limit);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
private static final ReadOperation<OutputStream> STREAM_OP = new ReadOperation<OutputStream>() { // from class: io.grpc.internal.CompositeReadableBuffer.5
|
||
|
|
@Override // io.grpc.internal.CompositeReadableBuffer.ReadOperation
|
||
|
|
public int read(ReadableBuffer readableBuffer, int i, OutputStream outputStream, int i2) throws IOException {
|
||
|
|
readableBuffer.readBytes(outputStream, i);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
|
/* loaded from: classes6.dex */
|
||
|
|
public interface NoThrowReadOperation<T> extends ReadOperation<T> {
|
||
|
|
@Override // io.grpc.internal.CompositeReadableBuffer.ReadOperation
|
||
|
|
int read(ReadableBuffer readableBuffer, int i, T t, int i2);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
|
/* loaded from: classes6.dex */
|
||
|
|
public interface ReadOperation<T> {
|
||
|
|
int read(ReadableBuffer readableBuffer, int i, T t, int i2) throws IOException;
|
||
|
|
}
|
||
|
|
|
||
|
|
public CompositeReadableBuffer(int i) {
|
||
|
|
this.readableBuffers = new ArrayDeque(i);
|
||
|
|
}
|
||
|
|
|
||
|
|
public CompositeReadableBuffer() {
|
||
|
|
this.readableBuffers = new ArrayDeque();
|
||
|
|
}
|
||
|
|
|
||
|
|
public void addBuffer(ReadableBuffer readableBuffer) {
|
||
|
|
boolean z = this.marked && this.readableBuffers.isEmpty();
|
||
|
|
enqueueBuffer(readableBuffer);
|
||
|
|
if (z) {
|
||
|
|
this.readableBuffers.peek().mark();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private void enqueueBuffer(ReadableBuffer readableBuffer) {
|
||
|
|
if (!(readableBuffer instanceof CompositeReadableBuffer)) {
|
||
|
|
this.readableBuffers.add(readableBuffer);
|
||
|
|
this.readableBytes += readableBuffer.readableBytes();
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
CompositeReadableBuffer compositeReadableBuffer = (CompositeReadableBuffer) readableBuffer;
|
||
|
|
while (!compositeReadableBuffer.readableBuffers.isEmpty()) {
|
||
|
|
this.readableBuffers.add(compositeReadableBuffer.readableBuffers.remove());
|
||
|
|
}
|
||
|
|
this.readableBytes += compositeReadableBuffer.readableBytes;
|
||
|
|
compositeReadableBuffer.readableBytes = 0;
|
||
|
|
compositeReadableBuffer.close();
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.ReadableBuffer
|
||
|
|
public int readUnsignedByte() {
|
||
|
|
return executeNoThrow(UBYTE_OP, 1, null, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.ReadableBuffer
|
||
|
|
public void skipBytes(int i) {
|
||
|
|
executeNoThrow(SKIP_OP, i, null, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.ReadableBuffer
|
||
|
|
public void readBytes(byte[] bArr, int i, int i2) {
|
||
|
|
executeNoThrow(BYTE_ARRAY_OP, i2, bArr, i);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.ReadableBuffer
|
||
|
|
public void readBytes(ByteBuffer byteBuffer) {
|
||
|
|
executeNoThrow(BYTE_BUF_OP, byteBuffer.remaining(), byteBuffer, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.ReadableBuffer
|
||
|
|
public void readBytes(OutputStream outputStream, int i) throws IOException {
|
||
|
|
execute(STREAM_OP, i, outputStream, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* JADX WARN: Multi-variable type inference failed */
|
||
|
|
/* JADX WARN: Type inference failed for: r1v0 */
|
||
|
|
/* JADX WARN: Type inference failed for: r1v1 */
|
||
|
|
/* JADX WARN: Type inference failed for: r1v2, types: [io.grpc.internal.CompositeReadableBuffer] */
|
||
|
|
/* JADX WARN: Type inference failed for: r1v5 */
|
||
|
|
/* JADX WARN: Type inference failed for: r1v9 */
|
||
|
|
@Override // io.grpc.internal.ReadableBuffer
|
||
|
|
public ReadableBuffer readBytes(int i) {
|
||
|
|
ReadableBuffer poll;
|
||
|
|
int i2;
|
||
|
|
ReadableBuffer readableBuffer;
|
||
|
|
boolean z;
|
||
|
|
if (i <= 0) {
|
||
|
|
return ReadableBuffers.empty();
|
||
|
|
}
|
||
|
|
checkReadable(i);
|
||
|
|
this.readableBytes -= i;
|
||
|
|
ReadableBuffer readableBuffer2 = null;
|
||
|
|
?? r1 = 0;
|
||
|
|
while (true) {
|
||
|
|
ReadableBuffer peek = this.readableBuffers.peek();
|
||
|
|
int readableBytes = peek.readableBytes();
|
||
|
|
if (readableBytes > i) {
|
||
|
|
readableBuffer = peek.readBytes(i);
|
||
|
|
i2 = 0;
|
||
|
|
} else {
|
||
|
|
if (this.marked) {
|
||
|
|
poll = peek.readBytes(readableBytes);
|
||
|
|
advanceBuffer();
|
||
|
|
} else {
|
||
|
|
poll = this.readableBuffers.poll();
|
||
|
|
}
|
||
|
|
ReadableBuffer readableBuffer3 = poll;
|
||
|
|
i2 = i - readableBytes;
|
||
|
|
readableBuffer = readableBuffer3;
|
||
|
|
}
|
||
|
|
if (readableBuffer2 == null) {
|
||
|
|
readableBuffer2 = readableBuffer;
|
||
|
|
z = r1;
|
||
|
|
} else {
|
||
|
|
if (r1 == 0) {
|
||
|
|
CompositeReadableBuffer compositeReadableBuffer = new CompositeReadableBuffer(i2 != 0 ? Math.min(this.readableBuffers.size() + 2, 16) : 2);
|
||
|
|
compositeReadableBuffer.addBuffer(readableBuffer2);
|
||
|
|
readableBuffer2 = compositeReadableBuffer;
|
||
|
|
r1 = readableBuffer2;
|
||
|
|
}
|
||
|
|
r1.addBuffer(readableBuffer);
|
||
|
|
z = r1;
|
||
|
|
}
|
||
|
|
if (i2 <= 0) {
|
||
|
|
return readableBuffer2;
|
||
|
|
}
|
||
|
|
i = i2;
|
||
|
|
r1 = z;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.AbstractReadableBuffer, io.grpc.internal.ReadableBuffer
|
||
|
|
public boolean markSupported() {
|
||
|
|
Iterator<ReadableBuffer> it = this.readableBuffers.iterator();
|
||
|
|
while (it.hasNext()) {
|
||
|
|
if (!it.next().markSupported()) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.AbstractReadableBuffer, io.grpc.internal.ReadableBuffer
|
||
|
|
public void mark() {
|
||
|
|
if (this.rewindableBuffers == null) {
|
||
|
|
this.rewindableBuffers = new ArrayDeque(Math.min(this.readableBuffers.size(), 16));
|
||
|
|
}
|
||
|
|
while (!this.rewindableBuffers.isEmpty()) {
|
||
|
|
this.rewindableBuffers.remove().close();
|
||
|
|
}
|
||
|
|
this.marked = true;
|
||
|
|
ReadableBuffer peek = this.readableBuffers.peek();
|
||
|
|
if (peek != null) {
|
||
|
|
peek.mark();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.AbstractReadableBuffer, io.grpc.internal.ReadableBuffer
|
||
|
|
public void reset() {
|
||
|
|
if (!this.marked) {
|
||
|
|
throw new InvalidMarkException();
|
||
|
|
}
|
||
|
|
ReadableBuffer peek = this.readableBuffers.peek();
|
||
|
|
if (peek != null) {
|
||
|
|
int readableBytes = peek.readableBytes();
|
||
|
|
peek.reset();
|
||
|
|
this.readableBytes += peek.readableBytes() - readableBytes;
|
||
|
|
}
|
||
|
|
while (true) {
|
||
|
|
ReadableBuffer pollLast = this.rewindableBuffers.pollLast();
|
||
|
|
if (pollLast == null) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
pollLast.reset();
|
||
|
|
this.readableBuffers.addFirst(pollLast);
|
||
|
|
this.readableBytes += pollLast.readableBytes();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.AbstractReadableBuffer, io.grpc.internal.ReadableBuffer
|
||
|
|
public boolean byteBufferSupported() {
|
||
|
|
Iterator<ReadableBuffer> it = this.readableBuffers.iterator();
|
||
|
|
while (it.hasNext()) {
|
||
|
|
if (!it.next().byteBufferSupported()) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.AbstractReadableBuffer, io.grpc.internal.ReadableBuffer
|
||
|
|
public ByteBuffer getByteBuffer() {
|
||
|
|
if (this.readableBuffers.isEmpty()) {
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
return this.readableBuffers.peek().getByteBuffer();
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.AbstractReadableBuffer, io.grpc.internal.ReadableBuffer, java.io.Closeable, java.lang.AutoCloseable
|
||
|
|
public void close() {
|
||
|
|
while (!this.readableBuffers.isEmpty()) {
|
||
|
|
this.readableBuffers.remove().close();
|
||
|
|
}
|
||
|
|
if (this.rewindableBuffers != null) {
|
||
|
|
while (!this.rewindableBuffers.isEmpty()) {
|
||
|
|
this.rewindableBuffers.remove().close();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private <T> int execute(ReadOperation<T> readOperation, int i, T t, int i2) throws IOException {
|
||
|
|
checkReadable(i);
|
||
|
|
if (!this.readableBuffers.isEmpty()) {
|
||
|
|
advanceBufferIfNecessary();
|
||
|
|
}
|
||
|
|
while (i > 0 && !this.readableBuffers.isEmpty()) {
|
||
|
|
ReadableBuffer peek = this.readableBuffers.peek();
|
||
|
|
int min = Math.min(i, peek.readableBytes());
|
||
|
|
i2 = readOperation.read(peek, min, t, i2);
|
||
|
|
i -= min;
|
||
|
|
this.readableBytes -= min;
|
||
|
|
advanceBufferIfNecessary();
|
||
|
|
}
|
||
|
|
if (i <= 0) {
|
||
|
|
return i2;
|
||
|
|
}
|
||
|
|
throw new AssertionError("Failed executing read operation");
|
||
|
|
}
|
||
|
|
|
||
|
|
private <T> int executeNoThrow(NoThrowReadOperation<T> noThrowReadOperation, int i, T t, int i2) {
|
||
|
|
try {
|
||
|
|
return execute(noThrowReadOperation, i, t, i2);
|
||
|
|
} catch (IOException e) {
|
||
|
|
throw new AssertionError(e);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private void advanceBufferIfNecessary() {
|
||
|
|
if (this.readableBuffers.peek().readableBytes() == 0) {
|
||
|
|
advanceBuffer();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private void advanceBuffer() {
|
||
|
|
if (this.marked) {
|
||
|
|
this.rewindableBuffers.add(this.readableBuffers.remove());
|
||
|
|
ReadableBuffer peek = this.readableBuffers.peek();
|
||
|
|
if (peek != null) {
|
||
|
|
peek.mark();
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
this.readableBuffers.remove().close();
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.internal.ReadableBuffer
|
||
|
|
public int readableBytes() {
|
||
|
|
return this.readableBytes;
|
||
|
|
}
|
||
|
|
}
|