package com.test;
|
|
/*
|
* Copyright © reserved by roomdis.com, service for tgn company whose important business is rural e-commerce.
|
*/
|
import java.io.IOException;
|
import java.net.DatagramPacket;
|
import java.net.DatagramSocket;
|
import java.net.InetAddress;
|
import java.net.SocketException;
|
import java.net.UnknownHostException;
|
import java.nio.ByteBuffer;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
import javax.websocket.OnClose;
|
import javax.websocket.OnError;
|
import javax.websocket.OnMessage;
|
import javax.websocket.OnOpen;
|
import javax.websocket.Session;
|
import javax.websocket.server.ServerEndpoint;
|
|
import org.apache.log4j.Logger;
|
import org.springframework.web.context.ContextLoader;
|
|
import com.google.gson.Gson;
|
|
|
/**
|
* @author shihuc
|
* @date 2017年8月22日 下午2:20:18
|
*/
|
@ServerEndpoint("/websocket")
|
public class WebsocketService {
|
|
private static Logger logger = Logger.getLogger(WebsocketService.class);
|
|
//private HttpSendService httpSendService;
|
|
private String videoRecServerHost = "10.90.7.10";
|
|
private int videoRecServerPort = 7667;
|
|
/*
|
* 当存在多个客户端访问时，为了保证会话继续保持，将连接缓存。
|
*/
|
private static Map<String, WebsocketService> webSocketMap = new ConcurrentHashMap<String, WebsocketService>();
|
private Session session;
|
|
private static final WebsocketService instance = new WebsocketService();
|
|
public static final WebsocketService getInstance() {
|
return instance;
|
}
|
|
@OnMessage
|
public void onTextMessage(String message, Session session) throws IOException, InterruptedException {
|
|
// Print the client message for testing purposes
|
logger.info("Received: " + message);
|
//TODO: 调用接口将消息发送给客户端后台服务系统
|
Gson gson = new Gson();
|
//KefuMessage kfMsg = gson.fromJson(message, KefuMessage.class);
|
//httpSendService = ContextLoader.getCurrentWebApplicationContext().getBean(HttpSendService.class);
|
}
|
|
/**
|
* 主要用来接受二进制数据。
|
*
|
* @author shihuc
|
* @param message
|
* @param session
|
* @throws IOException
|
* @throws InterruptedException
|
*/
|
@OnMessage
|
public void onBinaryMessage(ByteBuffer message, Session session, boolean last) throws IOException, InterruptedException {
|
byte [] sentBuf = message.array();
|
|
logger.info("Binary Received: " + sentBuf.length + ", last: " + last);
|
|
//下面的代码逻辑，是用UDP协议发送视频流数据到视频处理服务器做后续逻辑处理
|
//sendToVideoRecognizer(sentBuf);
|
}
|
|
/**
|
* @author shihuc
|
* @param sentBuf
|
* @throws SocketException
|
* @throws UnknownHostException
|
* @throws IOException
|
*/
|
private void sendToVideoRecognizer(byte[] sentBuf) throws SocketException, UnknownHostException, IOException {
|
DatagramSocket client = new DatagramSocket();
|
InetAddress addr = InetAddress.getByName(videoRecServerHost);
|
DatagramPacket sendPacket = new DatagramPacket(sentBuf, sentBuf.length, addr, videoRecServerPort);
|
client.send(sendPacket);
|
client.close();
|
}
|
|
// @OnOpen
|
// public void onOpen(Session session){
|
// this.session = session;
|
// String staffId = session.getQueryString();
|
// webSocketMap.put(staffId, this);
|
// logger.info(staffId + " client opened");
|
// }
|
|
@OnOpen
|
public void onOpen(Session session){
|
logger.info("client opened: " + session.toString());
|
}
|
|
@OnClose
|
public void onClose() {
|
logger.info("client onclose");
|
}
|
|
@OnError
|
public void onError(Session session, Throwable error){
|
logger.info("connection onError");
|
logger.info(error.getCause());
|
}
|
|
public boolean sendMessage(String message, String staffId) throws IOException{
|
WebsocketService client = webSocketMap.get(staffId);
|
if (client == null) {
|
return false;
|
}
|
boolean result=false;
|
try {
|
client.session.getBasicRemote().sendText(message);
|
result=true;
|
} catch (IOException e) {
|
try {
|
client.session.close();
|
} catch (IOException e1) {
|
e1.printStackTrace();
|
}
|
}
|
return result;
|
}
|
}
|
