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);
        
            //设置加强协议,使用加密协议(客户端也要一同设置)
            SocketD.setProtocol(new MessageProtocolSecret() {
                @Override
                public byte[] encrypt(byte[] bytes) throws Exception {
                    return EncryptUtils.aesEncrypt(bytes,"pLft36Ok5zfmP6zI");
                }

                @Override
                public byte[] decrypt(byte[] bytes) throws Exception {
                    return EncryptUtils.aesDecrypt(bytes,"pLft36Ok5zfmP6zI");
                }
            });
        });
    }
}

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

    @Override
    public void onMessage(Session session, Message message) {
        //收到消息,做业务处理
        if(message.flag() != MessageFlag.heartbeat){
            System.out.println("服务端:我收到心跳");
        }else {
            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, app->{
            //设置加强协议,使用加密协议(服务端也要一同设置)
            SocketD.setProtocol(new MessageProtocolSecret() {
                @Override
                public byte[] encrypt(byte[] bytes) throws Exception {
                    return EncryptUtils.aesEncrypt(bytes,"pLft36Ok5zfmP6zI");
                }

                @Override
                public byte[] decrypt(byte[] bytes) throws Exception {
                    return EncryptUtils.aesDecrypt(bytes,"pLft36Ok5zfmP6zI");
                }
            });
        });
        
        //创建会话(如果后端是WebSocekt,协议头为:ws)
        Session session = SocketD.createSession("tcp://localhost:28080");

        //设定30秒自动上发心跳(如果断开了,也尝试自动重链)
        session.sendHeartbeatAuto(30);
        
        //上报消息
        session.send("Helloworld server!");
    }
}