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