com.webpieces.http2parser.impl.GoAwayMarshaller Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of http2-parser Show documentation

A re-usable asynchronous http2 parser

There is a newer version: 2.1.1

package com.webpieces.http2parser.impl;

import com.webpieces.http2parser.api.dto.Http2ErrorCode;
import com.webpieces.http2parser.api.dto.Http2Frame;
import com.webpieces.http2parser.api.dto.Http2GoAway;
import org.webpieces.data.api.BufferPool;
import org.webpieces.data.api.DataWrapper;
import org.webpieces.data.api.DataWrapperGenerator;

import java.nio.ByteBuffer;
import java.util.List;
import java.util.Optional;

public class GoAwayMarshaller extends FrameMarshallerImpl {
    GoAwayMarshaller(BufferPool bufferPool, DataWrapperGenerator dataGen) {
        super(bufferPool, dataGen);
    }

    public byte marshalFlags(Http2Frame frame) {
        return 0x0;
    }

    public DataWrapper marshalPayload(Http2Frame frame) {
        Http2GoAway castFrame = (Http2GoAway) frame;

        ByteBuffer prelude = bufferPool.nextBuffer(8);
        prelude.putInt(castFrame.getLastStreamId()).putInt(castFrame.getErrorCode().getCode());
        prelude.flip();

        return dataGen.chainDataWrappers(
                dataGen.wrapByteBuffer(prelude),
                castFrame.getDebugData()
        );
    }

    public void unmarshalFlagsAndPayload(Http2Frame frame, byte flagsByte, Optional maybePayload) {
        Http2GoAway castFrame = (Http2GoAway) frame;

        maybePayload.ifPresent(payload ->
                {
                    List split = dataGen.split(payload, 8);
                    ByteBuffer preludeBytes = bufferPool.createWithDataWrapper(split.get(0));

                    castFrame.setLastStreamId(preludeBytes.getInt());
                    castFrame.setErrorCode(Http2ErrorCode.fromInteger(preludeBytes.getInt()));

                    castFrame.setDebugData(split.get(1));

                    bufferPool.releaseBuffer(preludeBytes);
                }
        );
    }
}