FinalContinuationFrameHandler.java

package sprout.server.websocket.framehandler.builtins;

import sprout.beans.annotation.Component;
import sprout.beans.annotation.Order;
import sprout.server.websocket.WebSocketFrame;
import sprout.server.websocket.WebSocketUtils;
import sprout.server.websocket.exception.WebSocketProtocolException;
import sprout.server.websocket.framehandler.AbstractFrameHandler;
import sprout.server.websocket.framehandler.FrameProcessingContext;

@Component
@Order(40)
public class FinalContinuationFrameHandler extends AbstractFrameHandler {
    @Override
    protected boolean process(FrameProcessingContext state) throws Exception {
        if (!state.isFragmented()) {
            throw new WebSocketProtocolException("Protocol Error: Received final continuation frame with no preceding message.");
        }
        if (state.getFragmentedOpcode() == 0x1) {
            WebSocketUtils.readTextToBuffer(state.getFrame().getPayloadStream(), state.getTextBuffer());
        } else {
            WebSocketUtils.readBinaryToBuffer(state.getFrame().getPayloadStream(), state.getBinaryBuffer());
        }
        return true;
    }

    @Override
    public boolean canHandle(FrameProcessingContext state) {
        WebSocketFrame frame = state.getFrame();
        return frame.isFin() && frame.getOpcode() == 0x0;
    }
}