com.payneteasy.websocket.WebSocketFrameBuilder 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 static com.payneteasy.websocket.WebSocketFrame.OpCode.BINARY_FRAME;
import static com.payneteasy.websocket.WebSocketFrame.OpCode.CONNECTION_CLOSE;
import static com.payneteasy.websocket.WebSocketFrame.OpCode.TEXT_FRAME;
import static com.payneteasy.websocket.WebSocketUtil.applyMask;
import static com.payneteasy.websocket.WebSocketUtil.createMask;

/**
 *
 */
public class WebSocketFrameBuilder {


    public static WebSocketFrame createTextFrame(String aText) {
        byte[] mask = createMask();
        return createTextFrame(aText, mask);
    }

    public static WebSocketFrame createBinaryFrameWithMask(byte[] aBuffer) {
        byte[] mask = createMask();
        applyMask(mask, aBuffer);

        return new WebSocketFrame.Builder()
                .opCode(BINARY_FRAME)
                .fin(true)
                .maskedPayload(true)
                .maskingKey_4(mask)
                .payloadLength(aBuffer.length)
                .applicationData(aBuffer)
                .build();
    }

    public static WebSocketFrame createTextFrame(String aText, byte[] mask) {

        byte[] data = aText.getBytes(WebSocketUtil.UTF_8);

        applyMask(mask, data);

        return new WebSocketFrame.Builder()
                .opCode(TEXT_FRAME)
                .fin(true)
                .maskedPayload(true)
                .maskingKey_4(mask)
                .payloadLength(data.length)
                .applicationData(data)
                .build();
    }

    public static WebSocketFrame createConnectionClose(String aReason) {
        byte[] data = WebSocketUtil.toBytes(aReason);
        byte[] mask = createMask();
        applyMask(mask, data);

        return new WebSocketFrame.Builder()
                .opCode(CONNECTION_CLOSE)
                .fin(true)
                .maskedPayload(true)
                .maskingKey_4(mask)
                .applicationData(data)
                .build();
    }
}