Solon rpc 之 SocketD 协议 – 消息鉴权模式

  • 2021 年 1 月 12 日
  • 筆記

Solon rpc 之 SocketD 协议系列
Solon rpc 之 SocketD 协议 – 概述
Solon rpc 之 SocketD 协议 – 消息上报模式
Solon rpc 之 SocketD 协议 – 消息应答模式
Solon rpc 之 SocketD 协议 – 消息订阅模式
Solon rpc 之 SocketD 协议 – RPC调用模式
Solon rpc 之 SocketD 协议 – 单链接双向RPC模式
Solon rpc 之 SocketD 协议 – 消息加密模式
Solon rpc 之 SocketD 协议 – 消息鉴权模式

SocketD 是一种二进制的点对点通信协议，是一种新的网络通信第七层协议。旨在用于分布式应用程序中。从这个意义上讲，SocketD可以是RSocket等其他类似协议的替代方案。它的消息协议规范具有异步，背压的双向，多路复用，断线重连，基于消息等特性。暂时只提供Java实现，目前做为Solon rpc的sockte通道协议。

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

服务端

//启动服务端
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());
    }
}

客户端

//启动客户端
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!");
    }
}

附：示例源码

VirMach 便宜 VPS

QNews

QNews