com.litongjava.tio.http.server.util.SseEmitter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tio-http-server Show documentation

tio http server

There is a newer version: 3.7.3.v20241201-RELEASE

package com.litongjava.tio.http.server.util;

import java.nio.charset.StandardCharsets;

import com.litongjava.tio.core.ChannelContext;
import com.litongjava.tio.core.Tio;
import com.litongjava.tio.http.common.encoder.ChunkEncoder;
import com.litongjava.tio.http.common.sse.ChunkedPacket;

public class SseEmitter {

  public static String LFLF = "\n\n";

  public static void pushSSEChunk(ChannelContext channelContext, String string) {
    String text = "data:" + string + LFLF;
    byte[] bytes = text.getBytes();
    ChunkedPacket ssePacket = new ChunkedPacket(ChunkEncoder.encodeChunk(bytes));
    Tio.bSend(channelContext, ssePacket);
  }

  public static void pushSSEChunk(ChannelContext channelContext, String event, String data) {
    StringBuilder sb = new StringBuilder();
    if (event != null) {
      sb.append("event:").append(event).append("\n");
    }
    sb.append("data:").append(data).append(LFLF);
    byte[] bytes = sb.toString().getBytes(StandardCharsets.UTF_8);
    ChunkedPacket ssePacket = new ChunkedPacket(ChunkEncoder.encodeChunk(bytes));
    Tio.bSend(channelContext, ssePacket);
  }

  public static void pushChunk(ChannelContext channelContext, byte[] bytes) {
    ChunkedPacket ssePacket = new ChunkedPacket(ChunkEncoder.encodeChunk(bytes));
    Tio.bSend(channelContext, ssePacket);
  }

  public static void closeSeeConnection(ChannelContext channelContext) {
    try {
      // 给客户端足够的时间接受消息
      Thread.sleep(1000);
      Tio.remove(channelContext, "remove");
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
  }

  public static void closeChunkConnection(ChannelContext channelContext) {
    // 关闭连接
    byte[] zeroChunk = ChunkEncoder.encodeChunk(new byte[0]);
    ChunkedPacket endPacket = new ChunkedPacket(zeroChunk);
    Tio.bSend(channelContext, endPacket);

    try {
      // 给客户端足够的时间接受消息
      Thread.sleep(1000);
      Tio.remove(channelContext, "remove");
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
  }

}