引用tomcat中的两个jar:websocket-api.jar 、tomcat7-websocket.jar
java后台:
package com.inspur.dataetl.monitor;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;
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.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import com.inspur.data.common.utils.ConfUtil;
/**
* @ServerEndpoint 注解是一个类层次的注解,它的功能主要是将目前的类定义成一个websocket服务器端,
* 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端
*/
@ServerEndpoint("/websocket")
public class LogMonitor {
// 静态变量,用来记录当前在线连接数。
private static int onlineCount = 0;
// 记录线程与session之间的对应关系
private static Map<String, Process> sessionProcessRelate=new HashMap<String,Process>();
private Process process;
private InputStream inputStream;
/**
* 连接建立成功调用的方法
*
* @param session
*session为与某个客户端的连接会话,需要通过它来给客户端发送数据
*/
@OnOpen
public void onOpen(Session session) {
addOnlineCount(); // 在线数加1
LOG.info("有新连接加入!当前在线人数为" + getOnlineCount());
}
/**
* 连接关闭调用的方法
*/
@OnClose
public void onClose(Session session) {
subOnlineCount(); // 在线数减1
LOG.info("有连接关闭!当前在线人数为" + getOnlineCount());
//删除该session对应的process
if (sessionProcessRelate.get(session.getId())!= null){
sessionProcessRelate.get(session.getId()).destroy();
sessionProcessRelate.remove(session.getId());
}
}
/**
* 收到客户端消息后调用的方法
*
* @param message
* 客户端发送过来的消息
* @param session
* 可选的参数
*/
@OnMessage
public void onMessage(String message, Session session) {
// 执行tail -f命令
// 如果非root启动liunux注意权限问题
// ConfUtil.getConfValue("logDir")为获取日志文件路径,message为前端传入的需要监控的文件名
String logDir = ConfUtil.getConfValue("logDir");logDir = "tail -f " + logDir + message;
try {
process = Runtime.getRuntime().exec(logDir);
sessionProcessRelate.put(session.getId(), process);
} catch (Exception e) {
LOG.error("读取日志出错", e);
return;
}
inputStream = process.getInputStream();
// 启动新的线程,防止InputStream阻塞处理WebSocket的线程
TailLogThread thread = new TailLogThread(inputStream, session);
thread.start();
}
/**
* 发生错误时调用
*
* @param session
* @param error
*/
@OnError
public void onError(Session session, Throwable error) {
LOG.error("websocket服务器端出现错误", error);
error.printStackTrace();
}
public static synchronized int getOnlineCount() {
return onlineCount;
}
public static synchronized void addOnlineCount() {
LogMonitor.onlineCount++;
}
public static synchronized void subOnlineCount() {
LogMonitor.onlineCount--;
}
}
新线程:
package com.inspur.dataetl.monitor;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import javax.websocket.Session;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
public class TailLogThread extends Thread {
public BufferedReader reader;
private Session session;
private static final Log log = LogFactory.getLog(TailLogThread.class);
public TailLogThread(InputStream in, Session session) {
try {
this.reader = new BufferedReader(new InputStreamReader(in,"UTF-8"));
} catch (UnsupportedEncodingException e) {
log.error("读取日志文件转码失败"+e.getMessage(),e);
}
this.session = session;
}
@Override
public void run() {
String line;
try {
while ((line = reader.readLine()) != null) {
// 将实时日志通过WebSocket发送给客户端
if (session.isOpen()) {
session.getBasicRemote().sendText(line);
}
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
前端jsp:
<%@ page language="java" pageEncoding="UTF-8" %>
<!DOCTYPE html>
<html>
<body>
<div class="dmp-tool operation-froup">
<button id="open" class="btn btn-info btn-sm"
οnclick="openLogMonitor()">开始监控运行过程</button>
<button id="close" style="display: none;" class="btn btn-info btn-sm"
οnclick="closeLogMonitor()">停止监控运行过程</button>
<button class="btn btn-info btn-sm" οnclick="downLog()">下载运行过程明细</button>
</div>
<div id="message"></div>
</body>
<script type="text/javascript">
/* 监控处理逻辑:
开始与停止日志监控都会与后台websocket保持连同
,停止监控时只是 在页面不打印 */
var logName = "/"+"<%= request.getParameter("logName") %>"+ ".log";
var websocket = null;
//websocket 服务器地址
var wsServer = getWsServer();
if ('WebSocket' in window) {
websocket = new WebSocket(wsServer);
} else {
alert('当前浏览器不支持 websocket,请使用支持html5的浏览器');
}
//开始监控与停止监控标志
var isMonitor = true;
//连接发生错误的回调方法
websocket.onerror = function() {
console.log("WebSocket连接发生错误");
};
//接收到消息的回调方法
websocket.onmessage = function(event) {
if (isMonitor) {
setMessageInnerHTML(event.data);
}
}
//监听窗口关闭事件,当窗口关闭时,主动去关闭websocket连接,防止连接还没断开就关闭窗口,server端会抛异常。
window.onbeforeunload = function() {
websocket.close();
}
//将消息打印到页
function setMessageInnerHTML(innerHTML) {
document.getElementById('message').innerHTML += innerHTML + '<br/>';
}
//停止日志监控
function closeLogMonitor() {
$("#open").css('display', '');
$("#close").css('display', 'none');
isMonitor = false;
}
//发送消息
function openLogMonitor() {
$("#open").css('display', 'none');
$("#close").css('display', '');
//第一次开启时向服务器发送消息
if (isMonitor) {
websocket.send(logName);
}
isMonitor = true;
}
//下载日志明细
function downLog() {
location.href=getRootPath()+"/task/TaskManage.do?method=DownLog&logName="+logName;
}
//websocket 服务器地址
function getWsServer() {
var host = window.location.host + '/', pathName = window.location.pathname
.split('/');
var projectName = pathName[1];
return ("ws:" + host + projectName + "/websocket");
}
</script>
</html>