com.litongjava.tio.websocket.common.WesocketServerEncoder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tio-websocket-common Show documentation
There is a newer version: 3.7.3.v20241201-RELEASE
package com.litongjava.tio.websocket.common;

import java.nio.ByteBuffer;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.litongjava.tio.core.ChannelContext;
import com.litongjava.tio.core.TioConfig;
import com.litongjava.tio.core.utils.ByteBufferUtils;

/**
 * 参考了baseio: https://gitee.com/generallycloud/baseio
 * com.generallycloud.nio.codec.http11.WebSocketProtocolEncoder
 * @author tanyaowu
 *
 */
public class WesocketServerEncoder {
  public static final int MAX_HEADER_LENGTH = 20480;

  private static void checkLength(byte[] bytes, int length, int offset) {
    if (bytes == null) {
      throw new IllegalArgumentException("null");
    }

    if (offset < 0) {
      throw new IllegalArgumentException("invalidate offset " + offset);
    }

    if (bytes.length - offset < length) {
      throw new IllegalArgumentException("invalidate length " + bytes.length);
    }
  }

  public static ByteBuffer encode(WebsocketResponse wsResponse, TioConfig tioConfig, ChannelContext channelContext) {
    byte[] wsBody = wsResponse.getBody();//就是ws的body，不包括ws的头
    byte[][] wsBodies = wsResponse.getBodys();
    int wsBodyLength = 0;
    if (wsBody != null) {
      wsBodyLength += wsBody.length;
    } else if (wsBodies != null) {
      for (int i = 0; i < wsBodies.length; i++) {
        byte[] bs = wsBodies[i];
        wsBodyLength += bs.length;
      }
    }

    byte header0 = (byte) (0x8f & (wsResponse.getWsOpcode().getCode() | 0xf0));
    ByteBuffer buf = null;
    if (wsBodyLength < 126) {
      buf = ByteBuffer.allocate(2 + wsBodyLength);
      buf.put(header0);
      buf.put((byte) wsBodyLength);
    } else if (wsBodyLength < (1 << 16) - 1) {
      buf = ByteBuffer.allocate(4 + wsBodyLength);
      buf.put(header0);
      buf.put((byte) 126);
      ByteBufferUtils.writeUB2WithBigEdian(buf, wsBodyLength);
    } else {
      buf = ByteBuffer.allocate(10 + wsBodyLength);
      buf.put(header0);
      buf.put((byte) 127);

      //			buf.put(new byte[] { 0, 0, 0, 0 });
      buf.position(buf.position() + 4);

      ByteBufferUtils.writeUB4WithBigEdian(buf, wsBodyLength);
    }

    if (wsBody != null && wsBody.length > 0) {
      buf.put(wsBody);
    } else if (wsBodies != null) {
      for (int i = 0; i < wsBodies.length; i++) {
        byte[] bs = wsBodies[i];
        buf.put(bs);
      }
    }

    return buf;
  }

  public static void int2Byte(byte[] bytes, int value, int offset) {
    checkLength(bytes, 4, offset);

    bytes[offset + 3] = (byte) (value & 0xff);
    bytes[offset + 2] = (byte) (value >> 8 * 1 & 0xff);
    bytes[offset + 1] = (byte) (value >> 8 * 2 & 0xff);
    bytes[offset + 0] = (byte) (value >> 8 * 3);
  }

  /**
   *
   * @author tanyaowu
   * 2017年2月22日 下午4:06:42
   *
   */
  public WesocketServerEncoder() {

  }
}