com.payneteasy.websocket.WebSocketFrameEncoder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of client Show documentation

There is a newer version: 1.0-8

package com.payneteasy.websocket;

import java.io.IOException;
import java.io.OutputStream;

/**
 *
 */
public class WebSocketFrameEncoder {

    public void encode(WebSocketFrame aFrame, OutputStream aOutput) throws IOException {
        int length = aFrame.payloadLength;

        // writes op-code
        byte finBit = (byte) (aFrame.fin ? 0x80 : 0x00);
        aOutput.write(finBit | aFrame.opCode.code);

        byte maskBit = (byte) (aFrame.maskedPayload ? 0x80 : 0x00);

        // writes length
        if(length <=125) {
            byte len1 = (byte) (length | maskBit);
            aOutput.write(len1);
        } else if (length <= 0xFFFF) {
            aOutput.write( (byte) (maskBit | 126) );
            aOutput.write( length >>> 8 & 0xFF );
            aOutput.write( length & 0xFF );
        } else {
            throw new IllegalStateException("Unsupported payload length "+ length);
        }

        // writes mask
        if(aFrame.maskedPayload) {
            aOutput.write( aFrame.maskingKey_4 );
        }

        if( aFrame.applicationData != null) {
            aOutput.write( aFrame.applicationData );
        }

    }
}