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

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 协议 - 消息鉴权模式
Solon rpc 之 SocketD 协议 - RPC鉴权模式

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!");
    }
}

附:示例源码

posted @ 2021-01-12 09:59  带刺的坐椅  阅读(225)  评论(0编辑  收藏  举报