├── .gitignore
├── README.md
├── image
└── agent.png
├── mvnw
├── mvnw.cmd
├── pom.xml
└── src
├── main
├── java
│ └── cn
│ │ └── teamstack
│ │ └── logger
│ │ ├── LogAgentApplication.java
│ │ ├── config
│ │ ├── CorsFilter.java
│ │ └── WebSocketConfiguration.java
│ │ ├── model
│ │ ├── QueryCmd.java
│ │ └── SSH.java
│ │ └── websocket
│ │ ├── ProcessThread.java
│ │ └── WebSocketHandler.java
└── resources
│ ├── application.properties
│ └── logback.xml
└── test
└── java
└── cn
└── teamstack
└── logger
└── LogDashboardAgentApplicationTests.java
/.gitignore:
--------------------------------------------------------------------------------
1 | # Compiled class file
2 | *.class
3 |
4 | # Log file
5 | *.log
6 |
7 | # BlueJ files
8 | *.ctxt
9 |
10 | # Mobile Tools for Java (J2ME)
11 | .mtj.tmp/
12 |
13 | # Package Files #
14 | *.jar
15 | *.war
16 | *.ear
17 | *.zip
18 | *.tar.gz
19 | *.rar
20 |
21 | # virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml
22 | hs_err_pid*
23 | ### IntelliJ IDEA ###
24 | .idea
25 | *.iws
26 | *.iml
27 | *.ipr
28 | .mvn/
29 | log/
--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
1 | # TailLog-Agent
2 |
3 | 配合TailLog(实时日志查看监控工具)使用的代理服务程序(Java版)
4 |
5 | 
6 |
7 | #### 了解TaiLog——实时日志查看监控工具
8 |
9 | TailLog首页
10 |
11 | TailLog码云
12 |
13 |
14 | ### 一.功能作用:
15 |
16 | 1. 该程序主要实现两个功能:
17 |
18 | * 通过ssh协议访问日志服务器并读取日志
19 |
20 | * 提供WebSocket服务,输出日志信息
21 |
22 | 2.其基本结构如下:
23 |
24 | PC <----> 代理服务器 <----> 日志服务器
25 |
26 | TailLog <----> Agent <----> 日志文件
27 |
28 | 3.主要技术栈
29 |
30 | * Spring Boot
31 |
32 | * WebSocket
33 |
34 | * JSch
35 |
36 | ### 二.部署
37 |
38 | * Step-1: 下载最新发布jar包
39 |
40 | taillog-agent-1.0.0.jar
41 |
42 | * Step-2: 在运行目录下,添加配置文件
43 |
44 | config.properties
45 |
46 | ````
47 | ##服务端口
48 | server.port=10091
49 |
50 | ##被代理服务器的ssh配置(日志文件服务器)
51 | ssh.server.log.ip=192.168.1.5
52 | ssh.server.log.port=22
53 | ssh.server.log.username=root
54 | ssh.server.log.password=123456
55 | ````
56 |
57 | * Step-3: 运行
58 |
59 | ````
60 | java -jar taillog-agent-1.0.0.jar --spring.config.location=file:./config.properties
61 | ````
62 |
63 |
64 | ### 三.使用源码打包
65 |
66 | 配置内容和方式同上
67 |
68 | ````
69 | $ mvn package
70 | $ mvn install
71 | ````
72 |
73 | ### 四.接口说明
74 |
75 | 开发人员可以根据以下接口说明使用其他语言自行实现代理程序
76 |
77 | * WebSocket访问地址:
78 | ````
79 | ws://{ip}:{port}/websocket
80 | ````
81 |
82 | 默认端口:10091
83 |
84 |
85 | * 连接参数:
86 | ````
87 | {
88 | "ssh": {
89 | "ip":"192.168.5.55",
90 | "port":"22",
91 | "username":"root",
92 | "password":"123456"
93 | },
94 | "content": "tail -f /home/project/app.log"
95 | }
96 | ````
97 | 参数说明:
98 |
99 | ssh:[选填] 被代理的服务器IP地址。客户端可以通过配置该信息指定访问的服务器。未填则使用代理配置的服务器信息
100 |
101 | content:[必填] 待执行的linux命令。客户端将发送"tail..."等命令获取日志内容
102 |
103 |
104 |
105 |
--------------------------------------------------------------------------------
/image/agent.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/djmpink/TailLog-Agent/32b51377c4c6eb03e5633bde2aff7a1c416ab699/image/agent.png
--------------------------------------------------------------------------------
/mvnw:
--------------------------------------------------------------------------------
1 | #!/bin/sh
2 | # ----------------------------------------------------------------------------
3 | # Licensed to the Apache Software Foundation (ASF) under one
4 | # or more contributor license agreements. See the NOTICE file
5 | # distributed with this work for additional information
6 | # regarding copyright ownership. The ASF licenses this file
7 | # to you under the Apache License, Version 2.0 (the
8 | # "License"); you may not use this file except in compliance
9 | # with the License. You may obtain a copy of the License at
10 | #
11 | # http://www.apache.org/licenses/LICENSE-2.0
12 | #
13 | # Unless required by applicable law or agreed to in writing,
14 | # software distributed under the License is distributed on an
15 | # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16 | # KIND, either express or implied. See the License for the
17 | # specific language governing permissions and limitations
18 | # under the License.
19 | # ----------------------------------------------------------------------------
20 |
21 | # ----------------------------------------------------------------------------
22 | # Maven2 Start Up Batch script
23 | #
24 | # Required ENV vars:
25 | # ------------------
26 | # JAVA_HOME - location of a JDK home dir
27 | #
28 | # Optional ENV vars
29 | # -----------------
30 | # M2_HOME - location of maven2's installed home dir
31 | # MAVEN_OPTS - parameters passed to the Java VM when running Maven
32 | # e.g. to debug Maven itself, use
33 | # set MAVEN_OPTS=-Xdebug -Xrunjdwp:transport=dt_socket,server=y,suspend=y,address=8000
34 | # MAVEN_SKIP_RC - flag to disable loading of mavenrc files
35 | # ----------------------------------------------------------------------------
36 |
37 | if [ -z "$MAVEN_SKIP_RC" ] ; then
38 |
39 | if [ -f /etc/mavenrc ] ; then
40 | . /etc/mavenrc
41 | fi
42 |
43 | if [ -f "$HOME/.mavenrc" ] ; then
44 | . "$HOME/.mavenrc"
45 | fi
46 |
47 | fi
48 |
49 | # OS specific support. $var _must_ be set to either true or false.
50 | cygwin=false;
51 | darwin=false;
52 | mingw=false
53 | case "`uname`" in
54 | CYGWIN*) cygwin=true ;;
55 | MINGW*) mingw=true;;
56 | Darwin*) darwin=true
57 | # Use /usr/libexec/java_home if available, otherwise fall back to /Library/Java/Home
58 | # See https://developer.apple.com/library/mac/qa/qa1170/_index.html
59 | if [ -z "$JAVA_HOME" ]; then
60 | if [ -x "/usr/libexec/java_home" ]; then
61 | export JAVA_HOME="`/usr/libexec/java_home`"
62 | else
63 | export JAVA_HOME="/Library/Java/Home"
64 | fi
65 | fi
66 | ;;
67 | esac
68 |
69 | if [ -z "$JAVA_HOME" ] ; then
70 | if [ -r /etc/gentoo-release ] ; then
71 | JAVA_HOME=`java-config --jre-home`
72 | fi
73 | fi
74 |
75 | if [ -z "$M2_HOME" ] ; then
76 | ## resolve links - $0 may be a link to maven's home
77 | PRG="$0"
78 |
79 | # need this for relative symlinks
80 | while [ -h "$PRG" ] ; do
81 | ls=`ls -ld "$PRG"`
82 | link=`expr "$ls" : '.*-> \(.*\)$'`
83 | if expr "$link" : '/.*' > /dev/null; then
84 | PRG="$link"
85 | else
86 | PRG="`dirname "$PRG"`/$link"
87 | fi
88 | done
89 |
90 | saveddir=`pwd`
91 |
92 | M2_HOME=`dirname "$PRG"`/..
93 |
94 | # make it fully qualified
95 | M2_HOME=`cd "$M2_HOME" && pwd`
96 |
97 | cd "$saveddir"
98 | # echo Using m2 at $M2_HOME
99 | fi
100 |
101 | # For Cygwin, ensure paths are in UNIX format before anything is touched
102 | if $cygwin ; then
103 | [ -n "$M2_HOME" ] &&
104 | M2_HOME=`cygpath --unix "$M2_HOME"`
105 | [ -n "$JAVA_HOME" ] &&
106 | JAVA_HOME=`cygpath --unix "$JAVA_HOME"`
107 | [ -n "$CLASSPATH" ] &&
108 | CLASSPATH=`cygpath --path --unix "$CLASSPATH"`
109 | fi
110 |
111 | # For Migwn, ensure paths are in UNIX format before anything is touched
112 | if $mingw ; then
113 | [ -n "$M2_HOME" ] &&
114 | M2_HOME="`(cd "$M2_HOME"; pwd)`"
115 | [ -n "$JAVA_HOME" ] &&
116 | JAVA_HOME="`(cd "$JAVA_HOME"; pwd)`"
117 | # TODO classpath?
118 | fi
119 |
120 | if [ -z "$JAVA_HOME" ]; then
121 | javaExecutable="`which javac`"
122 | if [ -n "$javaExecutable" ] && ! [ "`expr \"$javaExecutable\" : '\([^ ]*\)'`" = "no" ]; then
123 | # readlink(1) is not available as standard on Solaris 10.
124 | readLink=`which readlink`
125 | if [ ! `expr "$readLink" : '\([^ ]*\)'` = "no" ]; then
126 | if $darwin ; then
127 | javaHome="`dirname \"$javaExecutable\"`"
128 | javaExecutable="`cd \"$javaHome\" && pwd -P`/javac"
129 | else
130 | javaExecutable="`readlink -f \"$javaExecutable\"`"
131 | fi
132 | javaHome="`dirname \"$javaExecutable\"`"
133 | javaHome=`expr "$javaHome" : '\(.*\)/bin'`
134 | JAVA_HOME="$javaHome"
135 | export JAVA_HOME
136 | fi
137 | fi
138 | fi
139 |
140 | if [ -z "$JAVACMD" ] ; then
141 | if [ -n "$JAVA_HOME" ] ; then
142 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
143 | # IBM's JDK on AIX uses strange locations for the executables
144 | JAVACMD="$JAVA_HOME/jre/sh/java"
145 | else
146 | JAVACMD="$JAVA_HOME/bin/java"
147 | fi
148 | else
149 | JAVACMD="`which java`"
150 | fi
151 | fi
152 |
153 | if [ ! -x "$JAVACMD" ] ; then
154 | echo "Error: JAVA_HOME is not defined correctly." >&2
155 | echo " We cannot execute $JAVACMD" >&2
156 | exit 1
157 | fi
158 |
159 | if [ -z "$JAVA_HOME" ] ; then
160 | echo "Warning: JAVA_HOME environment variable is not set."
161 | fi
162 |
163 | CLASSWORLDS_LAUNCHER=org.codehaus.plexus.classworlds.launcher.Launcher
164 |
165 | # traverses directory structure from process work directory to filesystem root
166 | # first directory with .mvn subdirectory is considered project base directory
167 | find_maven_basedir() {
168 |
169 | if [ -z "$1" ]
170 | then
171 | echo "Path not specified to find_maven_basedir"
172 | return 1
173 | fi
174 |
175 | basedir="$1"
176 | wdir="$1"
177 | while [ "$wdir" != '/' ] ; do
178 | if [ -d "$wdir"/.mvn ] ; then
179 | basedir=$wdir
180 | break
181 | fi
182 | # workaround for JBEAP-8937 (on Solaris 10/Sparc)
183 | if [ -d "${wdir}" ]; then
184 | wdir=`cd "$wdir/.."; pwd`
185 | fi
186 | # end of workaround
187 | done
188 | echo "${basedir}"
189 | }
190 |
191 | # concatenates all lines of a file
192 | concat_lines() {
193 | if [ -f "$1" ]; then
194 | echo "$(tr -s '\n' ' ' < "$1")"
195 | fi
196 | }
197 |
198 | BASE_DIR=`find_maven_basedir "$(pwd)"`
199 | if [ -z "$BASE_DIR" ]; then
200 | exit 1;
201 | fi
202 |
203 | export MAVEN_PROJECTBASEDIR=${MAVEN_BASEDIR:-"$BASE_DIR"}
204 | echo $MAVEN_PROJECTBASEDIR
205 | MAVEN_OPTS="$(concat_lines "$MAVEN_PROJECTBASEDIR/.mvn/jvm.config") $MAVEN_OPTS"
206 |
207 | # For Cygwin, switch paths to Windows format before running java
208 | if $cygwin; then
209 | [ -n "$M2_HOME" ] &&
210 | M2_HOME=`cygpath --path --windows "$M2_HOME"`
211 | [ -n "$JAVA_HOME" ] &&
212 | JAVA_HOME=`cygpath --path --windows "$JAVA_HOME"`
213 | [ -n "$CLASSPATH" ] &&
214 | CLASSPATH=`cygpath --path --windows "$CLASSPATH"`
215 | [ -n "$MAVEN_PROJECTBASEDIR" ] &&
216 | MAVEN_PROJECTBASEDIR=`cygpath --path --windows "$MAVEN_PROJECTBASEDIR"`
217 | fi
218 |
219 | WRAPPER_LAUNCHER=org.apache.maven.wrapper.MavenWrapperMain
220 |
221 | exec "$JAVACMD" \
222 | $MAVEN_OPTS \
223 | -classpath "$MAVEN_PROJECTBASEDIR/.mvn/wrapper/maven-wrapper.jar" \
224 | "-Dmaven.home=${M2_HOME}" "-Dmaven.multiModuleProjectDirectory=${MAVEN_PROJECTBASEDIR}" \
225 | ${WRAPPER_LAUNCHER} $MAVEN_CONFIG "$@"
226 |
--------------------------------------------------------------------------------
/mvnw.cmd:
--------------------------------------------------------------------------------
1 | @REM ----------------------------------------------------------------------------
2 | @REM Licensed to the Apache Software Foundation (ASF) under one
3 | @REM or more contributor license agreements. See the NOTICE file
4 | @REM distributed with this work for additional information
5 | @REM regarding copyright ownership. The ASF licenses this file
6 | @REM to you under the Apache License, Version 2.0 (the
7 | @REM "License"); you may not use this file except in compliance
8 | @REM with the License. You may obtain a copy of the License at
9 | @REM
10 | @REM http://www.apache.org/licenses/LICENSE-2.0
11 | @REM
12 | @REM Unless required by applicable law or agreed to in writing,
13 | @REM software distributed under the License is distributed on an
14 | @REM "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 | @REM KIND, either express or implied. See the License for the
16 | @REM specific language governing permissions and limitations
17 | @REM under the License.
18 | @REM ----------------------------------------------------------------------------
19 |
20 | @REM ----------------------------------------------------------------------------
21 | @REM Maven2 Start Up Batch script
22 | @REM
23 | @REM Required ENV vars:
24 | @REM JAVA_HOME - location of a JDK home dir
25 | @REM
26 | @REM Optional ENV vars
27 | @REM M2_HOME - location of maven2's installed home dir
28 | @REM MAVEN_BATCH_ECHO - set to 'on' to enable the echoing of the batch commands
29 | @REM MAVEN_BATCH_PAUSE - set to 'on' to wait for a key stroke before ending
30 | @REM MAVEN_OPTS - parameters passed to the Java VM when running Maven
31 | @REM e.g. to debug Maven itself, use
32 | @REM set MAVEN_OPTS=-Xdebug -Xrunjdwp:transport=dt_socket,server=y,suspend=y,address=8000
33 | @REM MAVEN_SKIP_RC - flag to disable loading of mavenrc files
34 | @REM ----------------------------------------------------------------------------
35 |
36 | @REM Begin all REM lines with '@' in case MAVEN_BATCH_ECHO is 'on'
37 | @echo off
38 | @REM enable echoing my setting MAVEN_BATCH_ECHO to 'on'
39 | @if "%MAVEN_BATCH_ECHO%" == "on" echo %MAVEN_BATCH_ECHO%
40 |
41 | @REM set %HOME% to equivalent of $HOME
42 | if "%HOME%" == "" (set "HOME=%HOMEDRIVE%%HOMEPATH%")
43 |
44 | @REM Execute a user defined script before this one
45 | if not "%MAVEN_SKIP_RC%" == "" goto skipRcPre
46 | @REM check for pre script, once with legacy .bat ending and once with .cmd ending
47 | if exist "%HOME%\mavenrc_pre.bat" call "%HOME%\mavenrc_pre.bat"
48 | if exist "%HOME%\mavenrc_pre.cmd" call "%HOME%\mavenrc_pre.cmd"
49 | :skipRcPre
50 |
51 | @setlocal
52 |
53 | set ERROR_CODE=0
54 |
55 | @REM To isolate internal variables from possible post scripts, we use another setlocal
56 | @setlocal
57 |
58 | @REM ==== START VALIDATION ====
59 | if not "%JAVA_HOME%" == "" goto OkJHome
60 |
61 | echo.
62 | echo Error: JAVA_HOME not found in your environment. >&2
63 | echo Please set the JAVA_HOME variable in your environment to match the >&2
64 | echo location of your Java installation. >&2
65 | echo.
66 | goto error
67 |
68 | :OkJHome
69 | if exist "%JAVA_HOME%\bin\java.exe" goto init
70 |
71 | echo.
72 | echo Error: JAVA_HOME is set to an invalid directory. >&2
73 | echo JAVA_HOME = "%JAVA_HOME%" >&2
74 | echo Please set the JAVA_HOME variable in your environment to match the >&2
75 | echo location of your Java installation. >&2
76 | echo.
77 | goto error
78 |
79 | @REM ==== END VALIDATION ====
80 |
81 | :init
82 |
83 | @REM Find the project base dir, i.e. the directory that contains the folder ".mvn".
84 | @REM Fallback to current working directory if not found.
85 |
86 | set MAVEN_PROJECTBASEDIR=%MAVEN_BASEDIR%
87 | IF NOT "%MAVEN_PROJECTBASEDIR%"=="" goto endDetectBaseDir
88 |
89 | set EXEC_DIR=%CD%
90 | set WDIR=%EXEC_DIR%
91 | :findBaseDir
92 | IF EXIST "%WDIR%"\.mvn goto baseDirFound
93 | cd ..
94 | IF "%WDIR%"=="%CD%" goto baseDirNotFound
95 | set WDIR=%CD%
96 | goto findBaseDir
97 |
98 | :baseDirFound
99 | set MAVEN_PROJECTBASEDIR=%WDIR%
100 | cd "%EXEC_DIR%"
101 | goto endDetectBaseDir
102 |
103 | :baseDirNotFound
104 | set MAVEN_PROJECTBASEDIR=%EXEC_DIR%
105 | cd "%EXEC_DIR%"
106 |
107 | :endDetectBaseDir
108 |
109 | IF NOT EXIST "%MAVEN_PROJECTBASEDIR%\.mvn\jvm.config" goto endReadAdditionalConfig
110 |
111 | @setlocal EnableExtensions EnableDelayedExpansion
112 | for /F "usebackq delims=" %%a in ("%MAVEN_PROJECTBASEDIR%\.mvn\jvm.config") do set JVM_CONFIG_MAVEN_PROPS=!JVM_CONFIG_MAVEN_PROPS! %%a
113 | @endlocal & set JVM_CONFIG_MAVEN_PROPS=%JVM_CONFIG_MAVEN_PROPS%
114 |
115 | :endReadAdditionalConfig
116 |
117 | SET MAVEN_JAVA_EXE="%JAVA_HOME%\bin\java.exe"
118 |
119 | set WRAPPER_JAR="%MAVEN_PROJECTBASEDIR%\.mvn\wrapper\maven-wrapper.jar"
120 | set WRAPPER_LAUNCHER=org.apache.maven.wrapper.MavenWrapperMain
121 |
122 | %MAVEN_JAVA_EXE% %JVM_CONFIG_MAVEN_PROPS% %MAVEN_OPTS% %MAVEN_DEBUG_OPTS% -classpath %WRAPPER_JAR% "-Dmaven.multiModuleProjectDirectory=%MAVEN_PROJECTBASEDIR%" %WRAPPER_LAUNCHER% %MAVEN_CONFIG% %*
123 | if ERRORLEVEL 1 goto error
124 | goto end
125 |
126 | :error
127 | set ERROR_CODE=1
128 |
129 | :end
130 | @endlocal & set ERROR_CODE=%ERROR_CODE%
131 |
132 | if not "%MAVEN_SKIP_RC%" == "" goto skipRcPost
133 | @REM check for post script, once with legacy .bat ending and once with .cmd ending
134 | if exist "%HOME%\mavenrc_post.bat" call "%HOME%\mavenrc_post.bat"
135 | if exist "%HOME%\mavenrc_post.cmd" call "%HOME%\mavenrc_post.cmd"
136 | :skipRcPost
137 |
138 | @REM pause the script if MAVEN_BATCH_PAUSE is set to 'on'
139 | if "%MAVEN_BATCH_PAUSE%" == "on" pause
140 |
141 | if "%MAVEN_TERMINATE_CMD%" == "on" exit %ERROR_CODE%
142 |
143 | exit /B %ERROR_CODE%
144 |
--------------------------------------------------------------------------------
/pom.xml:
--------------------------------------------------------------------------------
1 |
2 |
4 | 4.0.0
5 |
6 | cn.teamstack.logger
7 | taillog-agent
8 | 1.0.0
9 | jar
10 |
11 | taillog-agent
12 | Demo project for Spring Boot
13 |
14 |
15 | org.springframework.boot
16 | spring-boot-starter-parent
17 | 1.5.7.RELEASE
18 |
19 |
20 |
21 |
22 | UTF-8
23 | UTF-8
24 | 1.8
25 |
26 |
27 |
28 |
29 | org.springframework.boot
30 | spring-boot-starter-websocket
31 |
32 |
33 |
34 | com.jcraft
35 | jsch
36 | 0.1.54
37 |
38 |
39 |
40 | com.alibaba
41 | fastjson
42 | 1.2.12
43 |
44 |
45 | org.springframework.boot
46 | spring-boot-starter-test
47 | test
48 |
49 |
50 |
51 |
52 |
53 |
54 | org.springframework.boot
55 | spring-boot-maven-plugin
56 |
57 |
58 |
59 |
60 |
61 |
62 |
--------------------------------------------------------------------------------
/src/main/java/cn/teamstack/logger/LogAgentApplication.java:
--------------------------------------------------------------------------------
1 | package cn.teamstack.logger;
2 |
3 | import org.springframework.boot.SpringApplication;
4 | import org.springframework.boot.autoconfigure.SpringBootApplication;
5 | import org.springframework.web.socket.config.annotation.EnableWebSocket;
6 |
7 | /**
8 | * @author zhouli
9 | */
10 | @SpringBootApplication
11 | @EnableWebSocket
12 | public class LogAgentApplication {
13 | public static void main(String[] args) {
14 | SpringApplication.run(LogAgentApplication.class, args);
15 | }
16 | }
17 |
--------------------------------------------------------------------------------
/src/main/java/cn/teamstack/logger/config/CorsFilter.java:
--------------------------------------------------------------------------------
1 | package cn.teamstack.logger.config;
2 |
3 | import org.slf4j.Logger;
4 | import org.slf4j.LoggerFactory;
5 | import org.springframework.stereotype.Component;
6 |
7 | import javax.servlet.Filter;
8 | import javax.servlet.FilterChain;
9 | import javax.servlet.FilterConfig;
10 | import javax.servlet.ServletException;
11 | import javax.servlet.ServletRequest;
12 | import javax.servlet.ServletResponse;
13 | import javax.servlet.http.HttpServletRequest;
14 | import javax.servlet.http.HttpServletResponse;
15 | import java.io.IOException;
16 |
17 | /**
18 | * 设置跨域
19 | *
20 | * @author zhouli
21 | * @date 2017/8/13
22 | * 解决websocket跨域问题
23 | */
24 | @Component
25 | public class CorsFilter implements Filter {
26 |
27 | @Override
28 | public void doFilter(ServletRequest req, ServletResponse res, FilterChain chain) throws IOException, ServletException {
29 | HttpServletResponse response = (HttpServletResponse) res;
30 |
31 | HttpServletRequest reqs = (HttpServletRequest) req;
32 |
33 | response.setHeader("Access-Control-Allow-Origin", reqs.getHeader("Origin"));
34 | response.setHeader("Access-Control-Allow-Credentials", "true");
35 | response.setHeader("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE");
36 | response.setHeader("Access-Control-Max-Age", "3600");
37 | response.setHeader("Access-Control-Allow-Headers", "x-requested-with");
38 | chain.doFilter(req, res);
39 | }
40 |
41 | @Override
42 | public void init(FilterConfig filterConfig) {
43 | }
44 |
45 | @Override
46 | public void destroy() {
47 | }
48 | }
49 |
--------------------------------------------------------------------------------
/src/main/java/cn/teamstack/logger/config/WebSocketConfiguration.java:
--------------------------------------------------------------------------------
1 | package cn.teamstack.logger.config;
2 |
3 | import cn.teamstack.logger.websocket.WebSocketHandler;
4 | import org.springframework.context.annotation.Bean;
5 | import org.springframework.context.annotation.Configuration;
6 | import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
7 | import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
8 |
9 | /**
10 | * WebSocket配置
11 | * @author zhouli
12 | * @date 2017/8/14
13 | */
14 | @Configuration
15 | public class WebSocketConfiguration implements WebSocketConfigurer {
16 |
17 | @Override
18 | public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
19 | registry.addHandler(webSocketHandler(), "/websocket").setAllowedOrigins("*");
20 | }
21 |
22 | @Bean
23 | public org.springframework.web.socket.WebSocketHandler webSocketHandler() {
24 | return new WebSocketHandler();
25 | }
26 |
27 |
28 | }
29 |
--------------------------------------------------------------------------------
/src/main/java/cn/teamstack/logger/model/QueryCmd.java:
--------------------------------------------------------------------------------
1 | package cn.teamstack.logger.model;
2 |
3 | import com.alibaba.fastjson.JSONObject;
4 |
5 | /**
6 | * @author zhouli
7 | * @date 2017/8/18
8 | */
9 | public class QueryCmd {
10 |
11 | /**
12 | * 自定义参数
13 | * Json格式
14 | * TODO 预留
15 | */
16 | public JSONObject params;
17 |
18 | /**
19 | * 命令
20 | */
21 | public String content;
22 |
23 | /**
24 | * ssh配置信息(选填)
25 | */
26 | public SSH ssh;
27 | }
28 |
--------------------------------------------------------------------------------
/src/main/java/cn/teamstack/logger/model/SSH.java:
--------------------------------------------------------------------------------
1 | package cn.teamstack.logger.model;
2 |
3 | /**
4 | * @author zhouli
5 | * @date 2017/9/20
6 | */
7 | public class SSH {
8 | public String ip;
9 | public String port;
10 | public String username;
11 | public String password;
12 | }
13 |
--------------------------------------------------------------------------------
/src/main/java/cn/teamstack/logger/websocket/ProcessThread.java:
--------------------------------------------------------------------------------
1 | package cn.teamstack.logger.websocket;
2 |
3 | import com.jcraft.jsch.Channel;
4 | import com.jcraft.jsch.Session;
5 | import org.springframework.web.socket.TextMessage;
6 | import org.springframework.web.socket.WebSocketSession;
7 |
8 | import java.io.BufferedReader;
9 | import java.io.IOException;
10 | import java.io.InputStream;
11 | import java.io.InputStreamReader;
12 |
13 | /**
14 | * @author zhouli
15 | * @date 2017/8/14
16 | */
17 | public class ProcessThread extends Thread {
18 | private BufferedReader reader;
19 | private WebSocketSession webSocketSession;
20 | private Channel channel;
21 | private Session session;
22 |
23 | ProcessThread(InputStream in, WebSocketSession webSocketSession, Session session, Channel channel) {
24 | this.reader = new BufferedReader(new InputStreamReader(in));
25 | this.webSocketSession = webSocketSession;
26 | this.session = session;
27 | this.channel = channel;
28 | }
29 |
30 | @Override
31 | public void run() {
32 | String line;
33 | try {
34 | while ((line = reader.readLine()) != null) {
35 | webSocketSession.sendMessage(new TextMessage(line.getBytes()));
36 | }
37 | } catch (IOException e) {
38 | e.printStackTrace();
39 | channel.disconnect();
40 | session.disconnect();
41 | }
42 | }
43 | }
44 |
--------------------------------------------------------------------------------
/src/main/java/cn/teamstack/logger/websocket/WebSocketHandler.java:
--------------------------------------------------------------------------------
1 | package cn.teamstack.logger.websocket;
2 |
3 | import cn.teamstack.logger.model.QueryCmd;
4 | import com.alibaba.fastjson.JSON;
5 | import com.jcraft.jsch.Channel;
6 | import com.jcraft.jsch.ChannelExec;
7 | import com.jcraft.jsch.JSch;
8 | import com.jcraft.jsch.JSchException;
9 | import com.jcraft.jsch.Session;
10 | import com.jcraft.jsch.UIKeyboardInteractive;
11 | import com.jcraft.jsch.UserInfo;
12 | import org.slf4j.Logger;
13 | import org.slf4j.LoggerFactory;
14 | import org.springframework.beans.factory.annotation.Value;
15 | import org.springframework.util.StringUtils;
16 | import org.springframework.web.socket.CloseStatus;
17 | import org.springframework.web.socket.TextMessage;
18 | import org.springframework.web.socket.WebSocketSession;
19 | import org.springframework.web.socket.handler.TextWebSocketHandler;
20 |
21 | import javax.swing.*;
22 | import java.io.IOException;
23 | import java.io.InputStream;
24 |
25 | /**
26 | * @author zhouli
27 | * @date 2017/9/18
28 | */
29 | public class WebSocketHandler extends TextWebSocketHandler {
30 | private static final Logger logger = LoggerFactory.getLogger(TextWebSocketHandler.class);
31 |
32 | @Value("${ssh.server.log.ip}")
33 | private String sshIP;
34 | @Value("${ssh.server.log.port}")
35 | private String sshPort;
36 | @Value("${ssh.server.log.username}")
37 | private String sshUsername;
38 | @Value("${ssh.server.log.password}")
39 | private String sshPassword;
40 |
41 | @Override
42 | public void handleTextMessage(WebSocketSession webSocketSession, TextMessage textMessage) {
43 | String message = textMessage.getPayload();
44 | QueryCmd cmd = JSON.parseObject(message, QueryCmd.class);
45 |
46 | JSch jsch = new JSch();
47 | try {
48 | Session session;
49 | if (cmd.ssh != null && !StringUtils.isEmpty(cmd.ssh.ip)) {
50 | //可以优先使用配置的ssh配置信息
51 | session = jsch.getSession(cmd.ssh.username, cmd.ssh.ip, Integer.parseInt(cmd.ssh.port));
52 | session.setPassword(cmd.ssh.password);
53 | } else {
54 | //使用代理程序默认配置的信息
55 | logger.info(sshIP);
56 | logger.info(sshPort);
57 | session = jsch.getSession(sshUsername, sshIP, Integer.parseInt(sshPort));
58 | session.setPassword(sshPassword);
59 | }
60 |
61 | UserInfo ui = new AbstractMyUserInfo() {
62 | @Override
63 | public void showMessage(String message) {
64 | JOptionPane.showMessageDialog(null, message);
65 | }
66 |
67 | @Override
68 | public boolean promptYesNo(String message) {
69 | return true;
70 | }
71 | };
72 |
73 | session.setUserInfo(ui);
74 | // making a connection with timeout.
75 | session.connect(30000);
76 | Channel channel = session.openChannel("exec");
77 | ((ChannelExec) channel).setCommand(cmd.content);
78 | channel.setInputStream(null);
79 | ((ChannelExec) channel).setErrStream(System.err);
80 | InputStream inputStream = channel.getInputStream();
81 | channel.connect();
82 |
83 | ProcessThread thread = new ProcessThread(inputStream, webSocketSession, session, channel);
84 | thread.start();
85 |
86 | } catch (JSchException | IOException e) {
87 | e.printStackTrace();
88 | }
89 | }
90 |
91 | @Override
92 | public void afterConnectionEstablished(WebSocketSession session) {
93 | logger.debug("Opened new session in instance " + this);
94 | }
95 |
96 | @Override
97 | public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
98 |
99 | }
100 |
101 | @Override
102 | public void handleTransportError(WebSocketSession session, Throwable exception)
103 | throws Exception {
104 | session.close(CloseStatus.SERVER_ERROR);
105 | }
106 |
107 | public static abstract class AbstractMyUserInfo
108 | implements UserInfo, UIKeyboardInteractive {
109 | @Override
110 | public String getPassword() {
111 | return null;
112 | }
113 |
114 | @Override
115 | public boolean promptYesNo(String str) {
116 | return false;
117 | }
118 |
119 | @Override
120 | public String getPassphrase() {
121 | return null;
122 | }
123 |
124 | @Override
125 | public boolean promptPassphrase(String message) {
126 | return false;
127 | }
128 |
129 | @Override
130 | public boolean promptPassword(String message) {
131 | return false;
132 | }
133 |
134 | @Override
135 | public void showMessage(String message) {
136 | }
137 |
138 | @Override
139 | public String[] promptKeyboardInteractive(String destination,
140 | String name,
141 | String instruction,
142 | String[] prompt,
143 | boolean[] echo) {
144 | return null;
145 | }
146 | }
147 | }
148 |
--------------------------------------------------------------------------------
/src/main/resources/application.properties:
--------------------------------------------------------------------------------
1 | #服务端口
2 | server.port=10091
3 | #log
4 | #logging.file=./log/process.log
5 | #logging.config=classpath:logback.xml
6 | #默认ssh配置 服务日志(被代理)文件所在地址
7 | ssh.server.log.ip=127.0.0.1
8 | ssh.server.log.port=22
9 | ssh.server.log.username=root
10 | ssh.server.log.password=123456
--------------------------------------------------------------------------------
/src/main/resources/logback.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
7 |
8 |
9 |
10 |
--------------------------------------------------------------------------------
/src/test/java/cn/teamstack/logger/LogDashboardAgentApplicationTests.java:
--------------------------------------------------------------------------------
1 | package cn.teamstack.logger;
2 |
3 | import org.junit.Test;
4 | import org.junit.runner.RunWith;
5 | import org.springframework.boot.test.context.SpringBootTest;
6 | import org.springframework.test.context.junit4.SpringRunner;
7 |
8 | @RunWith(SpringRunner.class)
9 | @SpringBootTest
10 | public class LogDashboardAgentApplicationTests {
11 |
12 | @Test
13 | public void contextLoads() {
14 | }
15 |
16 | }
17 |
--------------------------------------------------------------------------------