com.firefly.codec.http2.encode.GoAwayGenerator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of firefly-jdk7 Show documentation

There is a newer version: 4.0.3.2

package com.firefly.codec.http2.encode;

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

import com.firefly.codec.http2.frame.Flags;
import com.firefly.codec.http2.frame.Frame;
import com.firefly.codec.http2.frame.FrameType;
import com.firefly.codec.http2.frame.GoAwayFrame;
import com.firefly.utils.io.BufferUtils;

public class GoAwayGenerator extends FrameGenerator {
	public GoAwayGenerator(HeaderGenerator headerGenerator) {
		super(headerGenerator);
	}

	@Override
	public List generate(Frame frame) {
		GoAwayFrame goAwayFrame = (GoAwayFrame) frame;
		return Arrays.asList(
				generateGoAway(goAwayFrame.getLastStreamId(), goAwayFrame.getError(), goAwayFrame.getPayload()));
	}

	public ByteBuffer generateGoAway(int lastStreamId, int error, byte[] payload) {
		if (lastStreamId < 0)
			throw new IllegalArgumentException("Invalid last stream id: " + lastStreamId);

		// The last streamId + the error code.
		int fixedLength = 4 + 4;

		// Make sure we don't exceed the default frame max length.
		int maxPayloadLength = Frame.DEFAULT_MAX_LENGTH - fixedLength;
		if (payload != null && payload.length > maxPayloadLength)
			payload = Arrays.copyOfRange(payload, 0, maxPayloadLength);

		int length = fixedLength + (payload != null ? payload.length : 0);
		ByteBuffer header = generateHeader(FrameType.GO_AWAY, length, Flags.NONE, 0);

		header.putInt(lastStreamId);
		header.putInt(error);

		if (payload != null) {
			header.put(payload);
		}

		BufferUtils.flipToFlush(header, 0);
		return header;
	}
}