package io.reactiverse.pgclient.impl;
import io.reactiverse.pgclient.Row;
import io.reactiverse.pgclient.impl.codec.ColumnDesc;
import io.reactiverse.pgclient.impl.codec.DataFormat;
import io.reactiverse.pgclient.impl.codec.DataTypeCodec;
import io.reactiverse.pgclient.impl.codec.decoder.RowDecoder;
import io.reactiverse.pgclient.impl.codec.decoder.RowDescription;
import io.netty.buffer.ByteBuf;
import java.util.function.BiConsumer;
import java.util.stream.Collector;
public class RowResultDecoder<C, R> implements RowDecoder {
private final Collector<Row, C, R> collector;
private final boolean singleton;
private final BiConsumer<C, Row> accumulator;
private RowDescription desc;
private int size;
private C container;
private Row row;
RowResultDecoder(Collector<Row, C, R> collector, boolean singleton, RowDescription desc) {
this.collector = collector;
this.singleton = singleton;
this.accumulator = collector.accumulator();
this.desc = desc;
}
public RowDescription description() {
return desc;
}
public int size() {
return size;
}
@Override
public void decodeRow(int len, ByteBuf in) {
if (container == null) {
container = collector.supplier().get();
}
if (singleton) {
if (row == null) {
row = new RowImpl(desc);
} else {
row.clear();
}
} else {
row = new RowImpl(desc);
}
Row row = new RowImpl(desc);
for (int c = 0; c < len; ++c) {
int length = in.readInt();
Object decoded = null;
if (length != -1) {
ColumnDesc columnDesc = desc.columns()[c];
if (columnDesc.getDataFormat() == DataFormat.BINARY) {
decoded = DataTypeCodec.decodeBinary(columnDesc.getDataType(), in.readerIndex(), length, in);
} else {
decoded = DataTypeCodec.decodeText(columnDesc.getDataType(), in.readerIndex(), length, in);
}
in.skipBytes(length);
}
row.addValue(decoded);
}
accumulator.accept(container, row);
size++;
}
R complete() {
if (container == null) {
container = collector.supplier().get();
}
return collector.finisher().apply(container);
}
void reset() {
container = null;
size = 0;
}
}