Accurate implementation of H3 `Request.beginNanoTime()` (#10911)
#9900: Accurate implementation of H3 Request.beginNanoTime() Signed-off-by: Ludovic Orban <lorban@bitronix.be>
This commit is contained in:
parent
382638b84d
commit
b9bd3f2e83
|
@ -44,22 +44,19 @@ public class MessageParser
|
|||
private final BooleanSupplier isLast;
|
||||
private BodyParser unknownBodyParser;
|
||||
private State state = State.HEADER;
|
||||
protected boolean dataMode;
|
||||
private boolean dataMode;
|
||||
private long beginNanoTime;
|
||||
private boolean beginNanoTimeStored;
|
||||
|
||||
public MessageParser(ParserListener listener, QpackDecoder decoder, long streamId, BooleanSupplier isLast)
|
||||
{
|
||||
this.listener = listener;
|
||||
this.decoder = decoder;
|
||||
decoder.setBeginNanoTimeSupplier(this::getBeginNanoTime);
|
||||
this.streamId = streamId;
|
||||
this.isLast = isLast;
|
||||
}
|
||||
|
||||
public long getBeginNanoTime()
|
||||
{
|
||||
return beginNanoTime;
|
||||
}
|
||||
|
||||
public void init(UnaryOperator<ParserListener> wrapper)
|
||||
{
|
||||
ParserListener listener = wrapper.apply(this.listener);
|
||||
|
@ -73,6 +70,21 @@ public class MessageParser
|
|||
{
|
||||
headerParser.reset();
|
||||
state = State.HEADER;
|
||||
beginNanoTimeStored = false;
|
||||
}
|
||||
|
||||
private void storeBeginNanoTime()
|
||||
{
|
||||
if (!beginNanoTimeStored)
|
||||
{
|
||||
beginNanoTime = NanoTime.now();
|
||||
beginNanoTimeStored = true;
|
||||
}
|
||||
}
|
||||
|
||||
private long getBeginNanoTime()
|
||||
{
|
||||
return beginNanoTime;
|
||||
}
|
||||
|
||||
public ParserListener getListener()
|
||||
|
@ -108,6 +120,7 @@ public class MessageParser
|
|||
{
|
||||
case HEADER ->
|
||||
{
|
||||
storeBeginNanoTime();
|
||||
if (headerParser.parse(buffer))
|
||||
{
|
||||
state = State.BODY;
|
||||
|
@ -124,7 +137,6 @@ public class MessageParser
|
|||
{
|
||||
BodyParser bodyParser = null;
|
||||
long frameType = headerParser.getFrameType();
|
||||
beginNanoTime = NanoTime.now(); // TODO #9900 check beginNanoTime's accuracy
|
||||
if (frameType >= 0 && frameType < bodyParsers.length)
|
||||
bodyParser = bodyParsers[(int)frameType];
|
||||
|
||||
|
|
|
@ -23,8 +23,10 @@ import org.eclipse.jetty.http3.frames.DataFrame;
|
|||
import org.eclipse.jetty.http3.generator.MessageGenerator;
|
||||
import org.eclipse.jetty.http3.parser.MessageParser;
|
||||
import org.eclipse.jetty.http3.parser.ParserListener;
|
||||
import org.eclipse.jetty.http3.qpack.QpackDecoder;
|
||||
import org.eclipse.jetty.io.ByteBufferPool;
|
||||
import org.eclipse.jetty.util.BufferUtil;
|
||||
import org.eclipse.jetty.util.NanoTime;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertArrayEquals;
|
||||
|
@ -58,6 +60,8 @@ public class DataGenerateParseTest
|
|||
new MessageGenerator(bufferPool, null, true).generate(accumulator, 0, input, null);
|
||||
|
||||
List<DataFrame> frames = new ArrayList<>();
|
||||
QpackDecoder decoder = new QpackDecoder(instructions -> {});
|
||||
decoder.setBeginNanoTimeSupplier(NanoTime::now);
|
||||
MessageParser parser = new MessageParser(new ParserListener()
|
||||
{
|
||||
@Override
|
||||
|
@ -65,7 +69,7 @@ public class DataGenerateParseTest
|
|||
{
|
||||
frames.add(frame);
|
||||
}
|
||||
}, null, 13, () -> true);
|
||||
}, decoder, 13, () -> true);
|
||||
parser.init(UnaryOperator.identity());
|
||||
for (ByteBuffer buffer : accumulator.getByteBuffers())
|
||||
{
|
||||
|
|
|
@ -30,6 +30,7 @@ import org.eclipse.jetty.http3.parser.ParserListener;
|
|||
import org.eclipse.jetty.http3.qpack.QpackDecoder;
|
||||
import org.eclipse.jetty.http3.qpack.QpackEncoder;
|
||||
import org.eclipse.jetty.io.ByteBufferPool;
|
||||
import org.eclipse.jetty.util.NanoTime;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertEquals;
|
||||
|
@ -54,6 +55,7 @@ public class HeadersGenerateParseTest
|
|||
|
||||
QpackDecoder decoder = new QpackDecoder(instructions -> {});
|
||||
decoder.setMaxHeadersSize(4 * 1024);
|
||||
decoder.setBeginNanoTimeSupplier(NanoTime::now);
|
||||
List<HeadersFrame> frames = new ArrayList<>();
|
||||
MessageParser parser = new MessageParser(new ParserListener()
|
||||
{
|
||||
|
|
|
@ -21,6 +21,7 @@ import java.util.Iterator;
|
|||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.atomic.AtomicInteger;
|
||||
import java.util.function.LongSupplier;
|
||||
|
||||
import org.eclipse.jetty.http.HttpField;
|
||||
import org.eclipse.jetty.http.MetaData;
|
||||
|
@ -59,6 +60,7 @@ public class QpackDecoder implements Dumpable
|
|||
private int _maxHeadersSize;
|
||||
private int _maxBlockedStreams;
|
||||
private int _maxTableCapacity;
|
||||
private LongSupplier _beginNanoTimeSupplier;
|
||||
|
||||
private static class MetaDataNotification
|
||||
{
|
||||
|
@ -96,6 +98,11 @@ public class QpackDecoder implements Dumpable
|
|||
return _maxHeadersSize;
|
||||
}
|
||||
|
||||
public void setBeginNanoTimeSupplier(LongSupplier beginNanoTimeSupplier)
|
||||
{
|
||||
_beginNanoTimeSupplier = beginNanoTimeSupplier;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param maxHeadersSize The maximum allowed size of a headers block, expressed as total of all name and value characters, plus 32 per field
|
||||
*/
|
||||
|
@ -173,7 +180,7 @@ public class QpackDecoder implements Dumpable
|
|||
{
|
||||
// Parse the buffer into an Encoded Field Section.
|
||||
int base = signBit ? requiredInsertCount - deltaBase - 1 : requiredInsertCount + deltaBase;
|
||||
EncodedFieldSection encodedFieldSection = new EncodedFieldSection(streamId, handler, requiredInsertCount, base, buffer);
|
||||
EncodedFieldSection encodedFieldSection = new EncodedFieldSection(streamId, handler, requiredInsertCount, base, buffer, _beginNanoTimeSupplier.getAsLong());
|
||||
|
||||
// Decode it straight away if we can, otherwise add it to the list of EncodedFieldSections.
|
||||
if (requiredInsertCount <= insertCount)
|
||||
|
|
|
@ -41,6 +41,7 @@ public class MetaDataBuilder
|
|||
private QpackException.StreamException _streamException;
|
||||
private boolean _request;
|
||||
private boolean _response;
|
||||
private long _beginNanoTime = Long.MIN_VALUE;
|
||||
|
||||
/**
|
||||
* @param maxHeadersSize The maximum size of the headers, expressed as total name and value characters.
|
||||
|
@ -60,6 +61,13 @@ public class MetaDataBuilder
|
|||
return _maxSize;
|
||||
}
|
||||
|
||||
public void setBeginNanoTime(long beginNanoTime)
|
||||
{
|
||||
if (beginNanoTime == Long.MIN_VALUE)
|
||||
beginNanoTime++;
|
||||
_beginNanoTime = beginNanoTime;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the size.
|
||||
*
|
||||
|
@ -247,11 +255,13 @@ public class MetaDataBuilder
|
|||
if (_path == null)
|
||||
throw new QpackException.StreamException(H3_GENERAL_PROTOCOL_ERROR, "No Path");
|
||||
}
|
||||
long nanoTime = _beginNanoTime == Long.MIN_VALUE ? NanoTime.now() : _beginNanoTime;
|
||||
_beginNanoTime = Long.MIN_VALUE;
|
||||
if (isConnect)
|
||||
return new MetaData.ConnectRequest(NanoTime.now(), _scheme, _authority, _path, fields, _protocol); // TODO #9900 make beginNanoTime accurate
|
||||
return new MetaData.ConnectRequest(nanoTime, _scheme, _authority, _path, fields, _protocol);
|
||||
else
|
||||
return new MetaData.Request(
|
||||
NanoTime.now(), // TODO #9900 make beginNanoTime accurate
|
||||
nanoTime,
|
||||
_method,
|
||||
_scheme.asString(),
|
||||
_authority,
|
||||
|
|
|
@ -44,13 +44,15 @@ public class EncodedFieldSection
|
|||
private final int _requiredInsertCount;
|
||||
private final int _base;
|
||||
private final QpackDecoder.Handler _handler;
|
||||
private final long _beginNanoTime;
|
||||
|
||||
public EncodedFieldSection(long streamId, QpackDecoder.Handler handler, int requiredInsertCount, int base, ByteBuffer content) throws QpackException
|
||||
public EncodedFieldSection(long streamId, QpackDecoder.Handler handler, int requiredInsertCount, int base, ByteBuffer content, long beginNanoTime) throws QpackException
|
||||
{
|
||||
_streamId = streamId;
|
||||
_requiredInsertCount = requiredInsertCount;
|
||||
_base = base;
|
||||
_handler = handler;
|
||||
_beginNanoTime = beginNanoTime;
|
||||
|
||||
try
|
||||
{
|
||||
|
@ -104,6 +106,7 @@ public class EncodedFieldSection
|
|||
HttpField decodedField = encodedField.decode(context);
|
||||
metaDataBuilder.emit(decodedField);
|
||||
}
|
||||
metaDataBuilder.setBeginNanoTime(_beginNanoTime);
|
||||
return metaDataBuilder.build();
|
||||
}
|
||||
|
||||
|
|
|
@ -24,6 +24,7 @@ import org.eclipse.jetty.http3.qpack.internal.instruction.LiteralNameEntryInstru
|
|||
import org.eclipse.jetty.http3.qpack.internal.instruction.SectionAcknowledgmentInstruction;
|
||||
import org.eclipse.jetty.http3.qpack.internal.instruction.SetCapacityInstruction;
|
||||
import org.eclipse.jetty.util.BufferUtil;
|
||||
import org.eclipse.jetty.util.NanoTime;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
|
@ -53,6 +54,7 @@ public class BlockedStreamsTest
|
|||
_decoderHandler = new TestDecoderHandler();
|
||||
_encoder = new QpackEncoder(_encoderHandler);
|
||||
_decoder = new QpackDecoder(_decoderHandler);
|
||||
_decoder.setBeginNanoTimeSupplier(NanoTime::now);
|
||||
}
|
||||
|
||||
@Test
|
||||
|
|
|
@ -28,6 +28,7 @@ import org.eclipse.jetty.http3.qpack.internal.instruction.SetCapacityInstruction
|
|||
import org.eclipse.jetty.http3.qpack.internal.parser.DecoderInstructionParser;
|
||||
import org.eclipse.jetty.http3.qpack.internal.parser.EncoderInstructionParser;
|
||||
import org.eclipse.jetty.util.BufferUtil;
|
||||
import org.eclipse.jetty.util.NanoTime;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Disabled;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
@ -65,6 +66,7 @@ public class EncodeDecodeTest
|
|||
}
|
||||
};
|
||||
_decoder = new QpackDecoder(_decoderHandler);
|
||||
_decoder.setBeginNanoTimeSupplier(NanoTime::now);
|
||||
|
||||
_encoderInstructionParser = new EncoderInstructionParser(new EncoderParserDebugHandler(_encoder));
|
||||
_decoderInstructionParser = new DecoderInstructionParser(new DecoderParserDebugHandler(_decoder));
|
||||
|
|
|
@ -20,6 +20,7 @@ import org.eclipse.jetty.http.HttpField;
|
|||
import org.eclipse.jetty.http.HttpFields;
|
||||
import org.eclipse.jetty.http.HttpVersion;
|
||||
import org.eclipse.jetty.http.MetaData;
|
||||
import org.eclipse.jetty.util.NanoTime;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
|
@ -38,6 +39,7 @@ public class EvictionTest
|
|||
public void before()
|
||||
{
|
||||
_decoder = new QpackDecoder(_decoderHandler);
|
||||
_decoder.setBeginNanoTimeSupplier(NanoTime::now);
|
||||
_decoder.setMaxHeadersSize(1024);
|
||||
_decoder.setMaxTableCapacity(4 * 1024);
|
||||
|
||||
|
|
|
@ -18,6 +18,7 @@ import java.nio.ByteBuffer;
|
|||
import org.eclipse.jetty.http3.qpack.QpackException.SessionException;
|
||||
import org.eclipse.jetty.http3.qpack.internal.instruction.SectionAcknowledgmentInstruction;
|
||||
import org.eclipse.jetty.util.BufferUtil;
|
||||
import org.eclipse.jetty.util.NanoTime;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
|
@ -48,6 +49,7 @@ public class SectionAcknowledgmentTest
|
|||
_decoderHandler = new TestDecoderHandler();
|
||||
_encoder = new QpackEncoder(_encoderHandler);
|
||||
_decoder = new QpackDecoder(_decoderHandler);
|
||||
_decoder.setBeginNanoTimeSupplier(NanoTime::now);
|
||||
}
|
||||
|
||||
@Test
|
||||
|
|
Loading…
Reference in New Issue