me.hekr.iotos.softgateway.sample.TcpServerSample Maven / Gradle / Ivy

Go to download

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

There is a newer version: 3.5.30

package me.hekr.iotos.softgateway.sample;

import java.nio.charset.StandardCharsets;
import java.util.concurrent.CountDownLatch;
import me.hekr.iotos.softgateway.network.common.DecodePacket;
import me.hekr.iotos.softgateway.network.common.coder.PacketCoder;
import me.hekr.iotos.softgateway.network.tcp.TcpServer;
import me.hekr.iotos.softgateway.network.tcp.TcpServerConnectionContext;
import me.hekr.iotos.softgateway.network.tcp.listener.TcpMessageListener;

/**
 * tcp 服务端
 *
 * @author iotos
 */
public class TcpServerSample {
  public static void main(String[] args) throws InterruptedException {
    TcpServer server = new TcpServer<>();
    server.bind(4201);
    server.setMessageListener(new MyMessageListener(server));
    server.setPackageCoder(new MyPacketCoder());
    server.setHeartbeatTimeout(3000000);
    server.start();
    new CountDownLatch(1).await();
    //  server.close()
  }

  static class MyPacketCoder implements PacketCoder {
    @Override
    public byte[] encode(String s) {
      return s.getBytes(StandardCharsets.UTF_8);
    }

    @Override
    public DecodePacket decode(byte[] bytes) {
      char c = (char) bytes[0];
      return DecodePacket.wrap("" + c, 1);
    }
    ;
  }

  static class MyMessageListener implements TcpMessageListener {

    private final TcpServer server;

    public MyMessageListener(TcpServer server) {
      this.server = server;
    }

    @Override
    public void onMessage(TcpServerConnectionContext ctx) {
      System.out.println("收到来自: " + ctx.getAddress() + " 的消息，" + ctx.getMessage());
      server.writeAndFlush(ctx, "回复:" + ctx.getMessage());
    }
  }
}