Solon

SocketD Rpc鉴权模式

本案在RPC调用模式的基础上增加签权为例演示:

1、接口定义

Rpc 模式借用了 Nami 做客户端定义(Nami 是 Solon 伴生框架,定位为 Rpc 通用客户端)

@NamiClient("demo:/demoe/rpc")
public interface HelloService {
    @Handshake
    boolean auth(String sn, String token);

    String hello(String name);
}

2、服务端

//启动服务端
public class ServerApp {
    public static void main(String[] args) {
        //启动Solon容器(SocketD bean&plugin 由solon容器管理)
        Solon.start(ServerApp.class, args, app -> app.enableSocketD(true))
                .socket("**", (session, message) -> {
                    if (message.flag() != MessageFlag.handshake) {
                        if (session.getHandshaked() == false) {
                            System.out.println("这个客户端很坏,没签权就想发包:(");
                            session.close();
                        }
                    }
                });

        //.socket(..) 可替代 @ServerEndpoint 使用
    }
}

//定义远程服务组件
@Mapping(value = "/demoe/rpc", method = MethodType.SOCKET)
@Component(remoting = true)
public class HelloServiceImpl implements HelloService {
    @Override
    public boolean auth(String sn, String token) {
        Session session = (Session) Context.current().request();

        if ("1".equals(token)) {
            session.setHandshaked(true);
            System.out.println("签权成功!");
            return true;
        }else{
            session.setHandshaked(false);
            return false;
        }
    }

    public String hello(String name) {
        return "name=" + name;
    }
}

3、客户端

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

        //[客户端] 调用 [服务端] 的 rpc
        //
        HelloService rpc = SocketD.create("tcp://localhost:28080", HelloService.class);

        if (rpc.auth("1", "1")) {
            System.out.println("RPC result: " + rpc.hello("noear"));
        }
    }
}