Solon

SocketD 消息鉴权模式

本案在消息上报模式的基础上增加签权为例演示:

1、服务端

//启动服务端
public class ServerApp {
    public static void main(String[] args) {
        //启动Solon容器(SocketD bean&plugin 由solon容器管理)
        Solon.start(ServerApp.class, args, app -> app.enableSocketD(true));
    }
}

//定义服务端监听
@ServerEndpoint
public class ServerListener implements Listener {
    @Override
    public void onOpen(Session session) {
        System.out.println("有客户端链上来喽...");
    }

    @Override
    public void onMessage(Session session, Message message) throws IOException {
        if (session.getHandshaked() == false) {
            //消息是线程池分发的,确保鉴权时的消息顺序锁一下
            synchronized (session) {
                onMessage0(session, message);
            }
        } else {
            onMessage0(session, message);
        }
    }

    private void onMessage0(Session session, Message message) throws IOException {
        //如果是握手,则做鉴权处理
        if (message.flag() == MessageFlag.handshake) {
            if (Utils.isEmpty(message.header())) {
                session.close();
            } else {
                Map<String, String> headers = HeaderUtil.decodeHeaderMap(message.header());
                if ("1".equals(headers.get("token"))) {
                    //如果token是1,则设为成功握手
                    System.out.println("签权成功");
                    session.setHandshaked(true);
                } else {
                    session.close();
                }
            }

            return;
        }

        //没有握手成功之前,不能做别的事;且自动断开链接
        if (session.getHandshaked() == false) {
            System.out.println("这个客户端很坏,没签权就想发包:(");
            session.close();
        }

        //如果是心跳,则乎略
        if (message.flag() == MessageFlag.heartbeat) {
            System.out.println("服务端:我收到心跳");
            return;
        }

        //业务处理
        System.out.println("服务端:我收到:" + message.bodyAsString());
    }
}

2、客户端

//启动客户端
public class ClientApp {
    public static void main(String[] args) throws Throwable {
        //启动Solon容器(SocketD bean&plugin 由solon容器管理)
        Solon.start(ClientApp.class, args);

        //创建会话(如果后端是WebSocekt,协议头为:ws)
        Session session = SocketD.createSession("tcp://localhost:28080");

        //发送业务握手(进行签权)
        session.sendHandshake(Message.wrapHandshake("sn=1&token=1"));

        //设定30秒自动上发心跳(如果断开了,也尝试自动重链)
        session.sendHeartbeatAuto(30);

        //上报消息
        session.send("Helloworld server!");
    }
}