├── .gitignore ├── .mvn └── wrapper │ ├── MavenWrapperDownloader.java │ ├── maven-wrapper.jar │ └── maven-wrapper.properties ├── README.md ├── mvnw ├── mvnw.cmd ├── pom.xml └── src ├── main ├── java │ └── com │ │ └── example │ │ └── websocket │ │ ├── WebSocketApplication.java │ │ ├── commonutils │ │ └── DateUtil.java │ │ ├── webconfig │ │ └── WebMvcConfig.java │ │ ├── websocket │ │ ├── controller │ │ │ └── WebSocketController.java │ │ ├── server │ │ │ ├── WebSocketServer.java │ │ │ └── WebSocketServer2.java │ │ └── task │ │ │ └── ProductExpireTask.java │ │ └── websocketconfig │ │ ├── MyEndpointConfigure.java │ │ └── WebSocketConfig.java └── resources │ ├── application.properties │ ├── image │ ├── 微信图片_20190702135004.png │ ├── 微信图片_20190702135058.png │ ├── 微信图片_20190702150014.png │ └── 微信图片_20190702150146.png │ └── static │ ├── client.html │ ├── index.html │ └── websocket.html └── test └── java └── com └── example └── websocket └── WebsocketApplicationTests.java /.gitignore: -------------------------------------------------------------------------------- 1 | HELP.md 2 | target/ 3 | !.mvn/wrapper/maven-wrapper.jar 4 | !**/src/main/** 5 | !**/src/test/** 6 | 7 | ### STS ### 8 | .apt_generated 9 | .classpath 10 | .factorypath 11 | .project 12 | .settings 13 | .springBeans 14 | .sts4-cache 15 | 16 | ### IntelliJ IDEA ### 17 | .idea 18 | *.iws 19 | *.iml 20 | *.ipr 21 | 22 | ### NetBeans ### 23 | /nbproject/private/ 24 | /nbbuild/ 25 | /dist/ 26 | /nbdist/ 27 | /.nb-gradle/ 28 | build/ 29 | 30 | ### VS Code ### 31 | .vscode/ 32 | -------------------------------------------------------------------------------- /.mvn/wrapper/MavenWrapperDownloader.java: -------------------------------------------------------------------------------- 1 | /* 2 | Licensed to the Apache Software Foundation (ASF) under one 3 | or more contributor license agreements. See the NOTICE file 4 | distributed with this work for additional information 5 | regarding copyright ownership. The ASF licenses this file 6 | to you under the Apache License, Version 2.0 (the 7 | "License"); you may not use this file except in compliance 8 | with the License. You may obtain a copy of the License at 9 | 10 | https://www.apache.org/licenses/LICENSE-2.0 11 | 12 | Unless required by applicable law or agreed to in writing, 13 | software distributed under the License is distributed on an 14 | "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 15 | KIND, either express or implied. See the License for the 16 | specific language governing permissions and limitations 17 | under the License. 18 | */ 19 | 20 | import java.io.File; 21 | import java.io.FileInputStream; 22 | import java.io.FileOutputStream; 23 | import java.io.IOException; 24 | import java.net.URL; 25 | import java.nio.channels.Channels; 26 | import java.nio.channels.ReadableByteChannel; 27 | import java.util.Properties; 28 | 29 | public class MavenWrapperDownloader { 30 | 31 | /** 32 | * Default URL to download the maven-wrapper.jar from, if no 'downloadUrl' is provided. 33 | */ 34 | private static final String DEFAULT_DOWNLOAD_URL = 35 | "https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.4.2/maven-wrapper-0.4.2.jar"; 36 | 37 | /** 38 | * Path to the maven-wrapper.properties file, which might contain a downloadUrl property to 39 | * use instead of the default one. 40 | */ 41 | private static final String MAVEN_WRAPPER_PROPERTIES_PATH = 42 | ".mvn/wrapper/maven-wrapper.properties"; 43 | 44 | /** 45 | * Path where the maven-wrapper.jar will be saved to. 46 | */ 47 | private static final String MAVEN_WRAPPER_JAR_PATH = 48 | ".mvn/wrapper/maven-wrapper.jar"; 49 | 50 | /** 51 | * Name of the property which should be used to override the default download url for the wrapper. 52 | */ 53 | private static final String PROPERTY_NAME_WRAPPER_URL = "wrapperUrl"; 54 | 55 | public static void main(String args[]) { 56 | System.out.println("- Downloader started"); 57 | File baseDirectory = new File(args[0]); 58 | System.out.println("- Using base directory: " + baseDirectory.getAbsolutePath()); 59 | 60 | // If the maven-wrapper.properties exists, read it and check if it contains a custom 61 | // wrapperUrl parameter. 62 | File mavenWrapperPropertyFile = new File(baseDirectory, MAVEN_WRAPPER_PROPERTIES_PATH); 63 | String url = DEFAULT_DOWNLOAD_URL; 64 | if(mavenWrapperPropertyFile.exists()) { 65 | FileInputStream mavenWrapperPropertyFileInputStream = null; 66 | try { 67 | mavenWrapperPropertyFileInputStream = new FileInputStream(mavenWrapperPropertyFile); 68 | Properties mavenWrapperProperties = new Properties(); 69 | mavenWrapperProperties.load(mavenWrapperPropertyFileInputStream); 70 | url = mavenWrapperProperties.getProperty(PROPERTY_NAME_WRAPPER_URL, url); 71 | } catch (IOException e) { 72 | System.out.println("- ERROR loading '" + MAVEN_WRAPPER_PROPERTIES_PATH + "'"); 73 | } finally { 74 | try { 75 | if(mavenWrapperPropertyFileInputStream != null) { 76 | mavenWrapperPropertyFileInputStream.close(); 77 | } 78 | } catch (IOException e) { 79 | // Ignore ... 80 | } 81 | } 82 | } 83 | System.out.println("- Downloading from: : " + url); 84 | 85 | File outputFile = new File(baseDirectory.getAbsolutePath(), MAVEN_WRAPPER_JAR_PATH); 86 | if(!outputFile.getParentFile().exists()) { 87 | if(!outputFile.getParentFile().mkdirs()) { 88 | System.out.println( 89 | "- ERROR creating output direcrory '" + outputFile.getParentFile().getAbsolutePath() + "'"); 90 | } 91 | } 92 | System.out.println("- Downloading to: " + outputFile.getAbsolutePath()); 93 | try { 94 | downloadFileFromURL(url, outputFile); 95 | System.out.println("Done"); 96 | System.exit(0); 97 | } catch (Throwable e) { 98 | System.out.println("- Error downloading"); 99 | e.printStackTrace(); 100 | System.exit(1); 101 | } 102 | } 103 | 104 | private static void downloadFileFromURL(String urlString, File destination) throws Exception { 105 | URL website = new URL(urlString); 106 | ReadableByteChannel rbc; 107 | rbc = Channels.newChannel(website.openStream()); 108 | FileOutputStream fos = new FileOutputStream(destination); 109 | fos.getChannel().transferFrom(rbc, 0, Long.MAX_VALUE); 110 | fos.close(); 111 | rbc.close(); 112 | } 113 | 114 | } 115 | -------------------------------------------------------------------------------- /.mvn/wrapper/maven-wrapper.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/17661977890/websocket/9a264519a4ad360c06b0eed1ee240813437a085e/.mvn/wrapper/maven-wrapper.jar -------------------------------------------------------------------------------- /.mvn/wrapper/maven-wrapper.properties: -------------------------------------------------------------------------------- 1 | distributionUrl=https://repo.maven.apache.org/maven2/org/apache/maven/apache-maven/3.6.0/apache-maven-3.6.0-bin.zip 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # websocket 2 | springboot 整合websocket实现消息推送（主动推送，具体用户推送，群发，定时推送） 3 | 4 | * Spring Boot 开发私有即时通信系统（WebSocket）：https://www.jianshu.com/p/0f498adb3820 项目地址：https://gitee.com/anoyi/anyim 5 | 6 | * 使用WebSocket构建交互式Web应用程序 7 | 本指南将引导您完成创建“hello world”应用程序的过程，该应用程序在浏览器和服务器之间来回发送消息。WebSocket是一个非常薄，轻量级的TCP层。它使得非常适合使用“子协议”来嵌入消息。在本指南中，我们将深入研究并使用Spring的STOMP消息来创建交互式Web应用程序。 8 | https://spring.io/guides/gs/messaging-stomp-websocket/ 9 | 10 | ## 原理图解（与传统http请求差异） 11 | 12 | ![image](http://dl2.iteye.com/upload/attachment/0130/1491/f14c9efc-07f6-3fcd-bdf7-faf2ebf0f6ae.png) 13 | 14 | 15 | ##使用websocket有两种方式： 16 | * 1是使用sockjs， 17 | * 2是使用h5的标准。使用Html5标准自然更方便简单，所以记录的是配合h5的使用方法（有两个案例，两个websocketserver类，分别对应不同的html页面 ） 18 | 19 | * 案例2 是针对不同用户来测试webscoket的几个方法 20 | 21 | ## pom 22 | * 核心是@ServerEndpoint这个注解。这个注解是Javaee标准里的注解，tomcat7以上已经对其进行了实现， 23 | 如果是用传统方法使用tomcat发布项目，只要在pom文件中引入javaee标准即可使用。 24 | 25 | 26 | javax 27 | javaee-api 28 | 7.0 29 | provided 30 | 31 | 32 | * 但使用springboot的内置tomcat时，就不需要引入javaee-api了，spring-boot已经包含了。使用springboot的websocket功能首先引入springboot组件 33 | 34 | 35 | org.springframework.boot 36 | spring-boot-starter-websocket 37 | 1.3.5.RELEASE 38 | 39 | 40 | * 顺便说一句，springboot的高级组件会自动引用基础的组件，像spring-boot-starter-websocket就引入了spring-boot-starter-web和spring-boot-starter，所以不要重复引入。 41 | 42 | 43 | ## 配置类（WebSocketConfig） 44 | * 使用@ServerEndpoint创立websocket endpoint [配置端点类以及具体实现：WebSocketServer] 45 | 首先要注入ServerEndpointExporter，这个bean会自动注册使用了@ServerEndpoint注解声明的Websocket endpoint。 46 | 要注意，如果使用独立的servlet容器，而不是直接使用springboot的内置容器，就不要注入ServerEndpointExporter，因为它将由容器自己提供和管理 47 | 48 | ## 测试 案例1 49 | * 1、启动项目 50 | * 2、浏览器输入http://localhost:8080/websocket.html 51 | * 3、可以打开多个页面，后台控制台会答应当前在线人数的变化 52 | 53 | ## 测试 案例2 54 | * 1、访问 http://localhost:8080/index.html 和 http://localhost:8080/client.html 分别打开两个页面并连接到websocket， 55 | * 2、http://localhost:8080/socket?userName=lily&message=helloworld 给lily发送消息， 56 | * 3、http://localhost:8080/webSocket/socket/all?message=hello,大家好 群发消息，给在线的所有人 57 | ![image](https://github.com/17661977890/websocket/blob/master/src/main/resources/image/%E5%BE%AE%E4%BF%A1%E5%9B%BE%E7%89%87_20190702135004.png) 58 | ![image](https://github.com/17661977890/websocket/blob/master/src/main/resources/image/%E5%BE%AE%E4%BF%A1%E5%9B%BE%E7%89%87_20190702135058.png) 59 | 60 | * 定时任务发送消息： 61 | ![image](https://github.com/17661977890/websocket/blob/master/src/main/resources/image/%E5%BE%AE%E4%BF%A1%E5%9B%BE%E7%89%87_20190702150146.png) 62 | -------------------------------------------------------------------------------- /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 | # https://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 Mingw, 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 | ########################################################################################## 204 | # Extension to allow automatically downloading the maven-wrapper.jar from Maven-central 205 | # This allows using the maven wrapper in projects that prohibit checking in binary data. 206 | ########################################################################################## 207 | if [ -r "$BASE_DIR/.mvn/wrapper/maven-wrapper.jar" ]; then 208 | if [ "$MVNW_VERBOSE" = true ]; then 209 | echo "Found .mvn/wrapper/maven-wrapper.jar" 210 | fi 211 | else 212 | if [ "$MVNW_VERBOSE" = true ]; then 213 | echo "Couldn't find .mvn/wrapper/maven-wrapper.jar, downloading it ..." 214 | fi 215 | jarUrl="https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.4.2/maven-wrapper-0.4.2.jar" 216 | while IFS="=" read key value; do 217 | case "$key" in (wrapperUrl) jarUrl="$value"; break ;; 218 | esac 219 | done < "$BASE_DIR/.mvn/wrapper/maven-wrapper.properties" 220 | if [ "$MVNW_VERBOSE" = true ]; then 221 | echo "Downloading from: $jarUrl" 222 | fi 223 | wrapperJarPath="$BASE_DIR/.mvn/wrapper/maven-wrapper.jar" 224 | 225 | if command -v wget > /dev/null; then 226 | if [ "$MVNW_VERBOSE" = true ]; then 227 | echo "Found wget ... using wget" 228 | fi 229 | wget "$jarUrl" -O "$wrapperJarPath" 230 | elif command -v curl > /dev/null; then 231 | if [ "$MVNW_VERBOSE" = true ]; then 232 | echo "Found curl ... using curl" 233 | fi 234 | curl -o "$wrapperJarPath" "$jarUrl" 235 | else 236 | if [ "$MVNW_VERBOSE" = true ]; then 237 | echo "Falling back to using Java to download" 238 | fi 239 | javaClass="$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.java" 240 | if [ -e "$javaClass" ]; then 241 | if [ ! -e "$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.class" ]; then 242 | if [ "$MVNW_VERBOSE" = true ]; then 243 | echo " - Compiling MavenWrapperDownloader.java ..." 244 | fi 245 | # Compiling the Java class 246 | ("$JAVA_HOME/bin/javac" "$javaClass") 247 | fi 248 | if [ -e "$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.class" ]; then 249 | # Running the downloader 250 | if [ "$MVNW_VERBOSE" = true ]; then 251 | echo " - Running MavenWrapperDownloader.java ..." 252 | fi 253 | ("$JAVA_HOME/bin/java" -cp .mvn/wrapper MavenWrapperDownloader "$MAVEN_PROJECTBASEDIR") 254 | fi 255 | fi 256 | fi 257 | fi 258 | ########################################################################################## 259 | # End of extension 260 | ########################################################################################## 261 | 262 | export MAVEN_PROJECTBASEDIR=${MAVEN_BASEDIR:-"$BASE_DIR"} 263 | if [ "$MVNW_VERBOSE" = true ]; then 264 | echo $MAVEN_PROJECTBASEDIR 265 | fi 266 | MAVEN_OPTS="$(concat_lines "$MAVEN_PROJECTBASEDIR/.mvn/jvm.config") $MAVEN_OPTS" 267 | 268 | # For Cygwin, switch paths to Windows format before running java 269 | if $cygwin; then 270 | [ -n "$M2_HOME" ] && 271 | M2_HOME=`cygpath --path --windows "$M2_HOME"` 272 | [ -n "$JAVA_HOME" ] && 273 | JAVA_HOME=`cygpath --path --windows "$JAVA_HOME"` 274 | [ -n "$CLASSPATH" ] && 275 | CLASSPATH=`cygpath --path --windows "$CLASSPATH"` 276 | [ -n "$MAVEN_PROJECTBASEDIR" ] && 277 | MAVEN_PROJECTBASEDIR=`cygpath --path --windows "$MAVEN_PROJECTBASEDIR"` 278 | fi 279 | 280 | WRAPPER_LAUNCHER=org.apache.maven.wrapper.MavenWrapperMain 281 | 282 | exec "$JAVACMD" \ 283 | $MAVEN_OPTS \ 284 | -classpath "$MAVEN_PROJECTBASEDIR/.mvn/wrapper/maven-wrapper.jar" \ 285 | "-Dmaven.home=${M2_HOME}" "-Dmaven.multiModuleProjectDirectory=${MAVEN_PROJECTBASEDIR}" \ 286 | ${WRAPPER_LAUNCHER} $MAVEN_CONFIG "$@" 287 | -------------------------------------------------------------------------------- /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 https://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 set title of command window 39 | title %0 40 | @REM enable echoing my setting MAVEN_BATCH_ECHO to 'on' 41 | @if "%MAVEN_BATCH_ECHO%" == "on" echo %MAVEN_BATCH_ECHO% 42 | 43 | @REM set %HOME% to equivalent of $HOME 44 | if "%HOME%" == "" (set "HOME=%HOMEDRIVE%%HOMEPATH%") 45 | 46 | @REM Execute a user defined script before this one 47 | if not "%MAVEN_SKIP_RC%" == "" goto skipRcPre 48 | @REM check for pre script, once with legacy .bat ending and once with .cmd ending 49 | if exist "%HOME%\mavenrc_pre.bat" call "%HOME%\mavenrc_pre.bat" 50 | if exist "%HOME%\mavenrc_pre.cmd" call "%HOME%\mavenrc_pre.cmd" 51 | :skipRcPre 52 | 53 | @setlocal 54 | 55 | set ERROR_CODE=0 56 | 57 | @REM To isolate internal variables from possible post scripts, we use another setlocal 58 | @setlocal 59 | 60 | @REM ==== START VALIDATION ==== 61 | if not "%JAVA_HOME%" == "" goto OkJHome 62 | 63 | echo. 64 | echo Error: JAVA_HOME not found in your environment. >&2 65 | echo Please set the JAVA_HOME variable in your environment to match the >&2 66 | echo location of your Java installation. >&2 67 | echo. 68 | goto error 69 | 70 | :OkJHome 71 | if exist "%JAVA_HOME%\bin\java.exe" goto init 72 | 73 | echo. 74 | echo Error: JAVA_HOME is set to an invalid directory. >&2 75 | echo JAVA_HOME = "%JAVA_HOME%" >&2 76 | echo Please set the JAVA_HOME variable in your environment to match the >&2 77 | echo location of your Java installation. >&2 78 | echo. 79 | goto error 80 | 81 | @REM ==== END VALIDATION ==== 82 | 83 | :init 84 | 85 | @REM Find the project base dir, i.e. the directory that contains the folder ".mvn". 86 | @REM Fallback to current working directory if not found. 87 | 88 | set MAVEN_PROJECTBASEDIR=%MAVEN_BASEDIR% 89 | IF NOT "%MAVEN_PROJECTBASEDIR%"=="" goto endDetectBaseDir 90 | 91 | set EXEC_DIR=%CD% 92 | set WDIR=%EXEC_DIR% 93 | :findBaseDir 94 | IF EXIST "%WDIR%"\.mvn goto baseDirFound 95 | cd .. 96 | IF "%WDIR%"=="%CD%" goto baseDirNotFound 97 | set WDIR=%CD% 98 | goto findBaseDir 99 | 100 | :baseDirFound 101 | set MAVEN_PROJECTBASEDIR=%WDIR% 102 | cd "%EXEC_DIR%" 103 | goto endDetectBaseDir 104 | 105 | :baseDirNotFound 106 | set MAVEN_PROJECTBASEDIR=%EXEC_DIR% 107 | cd "%EXEC_DIR%" 108 | 109 | :endDetectBaseDir 110 | 111 | IF NOT EXIST "%MAVEN_PROJECTBASEDIR%\.mvn\jvm.config" goto endReadAdditionalConfig 112 | 113 | @setlocal EnableExtensions EnableDelayedExpansion 114 | for /F "usebackq delims=" %%a in ("%MAVEN_PROJECTBASEDIR%\.mvn\jvm.config") do set JVM_CONFIG_MAVEN_PROPS=!JVM_CONFIG_MAVEN_PROPS! %%a 115 | @endlocal & set JVM_CONFIG_MAVEN_PROPS=%JVM_CONFIG_MAVEN_PROPS% 116 | 117 | :endReadAdditionalConfig 118 | 119 | SET MAVEN_JAVA_EXE="%JAVA_HOME%\bin\java.exe" 120 | set WRAPPER_JAR="%MAVEN_PROJECTBASEDIR%\.mvn\wrapper\maven-wrapper.jar" 121 | set WRAPPER_LAUNCHER=org.apache.maven.wrapper.MavenWrapperMain 122 | 123 | set DOWNLOAD_URL="https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.4.2/maven-wrapper-0.4.2.jar" 124 | FOR /F "tokens=1,2 delims==" %%A IN (%MAVEN_PROJECTBASEDIR%\.mvn\wrapper\maven-wrapper.properties) DO ( 125 | IF "%%A"=="wrapperUrl" SET DOWNLOAD_URL=%%B 126 | ) 127 | 128 | @REM Extension to allow automatically downloading the maven-wrapper.jar from Maven-central 129 | @REM This allows using the maven wrapper in projects that prohibit checking in binary data. 130 | if exist %WRAPPER_JAR% ( 131 | echo Found %WRAPPER_JAR% 132 | ) else ( 133 | echo Couldn't find %WRAPPER_JAR%, downloading it ... 134 | echo Downloading from: %DOWNLOAD_URL% 135 | powershell -Command "(New-Object Net.WebClient).DownloadFile('%DOWNLOAD_URL%', '%WRAPPER_JAR%')" 136 | echo Finished downloading %WRAPPER_JAR% 137 | ) 138 | @REM End of extension 139 | 140 | %MAVEN_JAVA_EXE% %JVM_CONFIG_MAVEN_PROPS% %MAVEN_OPTS% %MAVEN_DEBUG_OPTS% -classpath %WRAPPER_JAR% "-Dmaven.multiModuleProjectDirectory=%MAVEN_PROJECTBASEDIR%" %WRAPPER_LAUNCHER% %MAVEN_CONFIG% %* 141 | if ERRORLEVEL 1 goto error 142 | goto end 143 | 144 | :error 145 | set ERROR_CODE=1 146 | 147 | :end 148 | @endlocal & set ERROR_CODE=%ERROR_CODE% 149 | 150 | if not "%MAVEN_SKIP_RC%" == "" goto skipRcPost 151 | @REM check for post script, once with legacy .bat ending and once with .cmd ending 152 | if exist "%HOME%\mavenrc_post.bat" call "%HOME%\mavenrc_post.bat" 153 | if exist "%HOME%\mavenrc_post.cmd" call "%HOME%\mavenrc_post.cmd" 154 | :skipRcPost 155 | 156 | @REM pause the script if MAVEN_BATCH_PAUSE is set to 'on' 157 | if "%MAVEN_BATCH_PAUSE%" == "on" pause 158 | 159 | if "%MAVEN_TERMINATE_CMD%" == "on" exit %ERROR_CODE% 160 | 161 | exit /B %ERROR_CODE% 162 | -------------------------------------------------------------------------------- /pom.xml: -------------------------------------------------------------------------------- 1 | 2 | 4 | 4.0.0 5 | 6 | org.springframework.boot 7 | spring-boot-starter-parent 8 | 2.1.6.RELEASE 9 | 10 | 11 | com.example 12 | websocket 13 | 0.0.1-SNAPSHOT 14 | websocket 15 | Demo project for Spring Boot 16 | 17 | 18 | 1.8 19 | 20 | 21 | 22 | 23 | org.springframework.boot 24 | spring-boot-starter 25 | 26 | 27 | 28 | org.projectlombok 29 | lombok 30 | true 31 | 32 | 33 | org.springframework.boot 34 | spring-boot-starter-test 35 | test 36 | 37 | 38 | org.apache.commons 39 | commons-lang3 40 | 3.8.1 41 | 42 | 43 | 44 | org.springframework.boot 45 | spring-boot-starter-websocket 46 | 47 | 48 | 49 | 50 | 51 | 52 | org.springframework.boot 53 | spring-boot-maven-plugin 54 | 55 | 56 | 57 | 58 | 59 | -------------------------------------------------------------------------------- /src/main/java/com/example/websocket/WebSocketApplication.java: -------------------------------------------------------------------------------- 1 | package com.example.websocket; 2 | 3 | import org.springframework.boot.SpringApplication; 4 | import org.springframework.boot.autoconfigure.SpringBootApplication; 5 | import org.springframework.scheduling.annotation.EnableScheduling; 6 | 7 | /** 8 | * @EnableScheduling 注解：开启定时任务 9 | */ 10 | @EnableScheduling 11 | @SpringBootApplication 12 | public class WebSocketApplication { 13 | 14 | public static void main(String[] args) { 15 | SpringApplication.run(WebSocketApplication.class, args); 16 | } 17 | 18 | } 19 | -------------------------------------------------------------------------------- /src/main/java/com/example/websocket/commonutils/DateUtil.java: -------------------------------------------------------------------------------- 1 | package com.example.websocket.commonutils; 2 | 3 | import java.text.ParsePosition; 4 | import java.text.SimpleDateFormat; 5 | import java.util.Calendar; 6 | import java.util.Date; 7 | 8 | import org.apache.commons.lang3.StringUtils; 9 | 10 | /** 11 | * @Description 时间处理工具类 12 | * @author bin 13 | */ 14 | public class DateUtil { 15 | 16 | private static final SimpleDateFormat dayOfDateFormat = new SimpleDateFormat("yyyy-MM-dd"); 17 | 18 | private static final SimpleDateFormat secondOfDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); 19 | 20 | /** 21 | * @return 当天时间加一天，返回"yyyy-MM-dd"格式 22 | */ 23 | public static String addOneDay() { 24 | Calendar calendar = Calendar.getInstance(); 25 | calendar.add(Calendar.DAY_OF_MONTH, 1); 26 | return dayOfDateFormat.format(calendar.getTime()); 27 | } 28 | 29 | /** 30 | * @return 当天时间加一月，返回"yyyy-MM-dd"格式 31 | */ 32 | public static String addOneMonth() { 33 | Calendar calendar = Calendar.getInstance(); 34 | calendar.add(Calendar.MONTH, 1); 35 | return dayOfDateFormat.format(calendar.getTime()); 36 | } 37 | 38 | /** 39 | * @param dayNumber 加的天数 40 | * @return 返回当天时间添加几天之后的时间, 返回"yyyy-MM-dd"格式 41 | */ 42 | public static String addFewDays(int dayNumber) { 43 | Calendar calendar = Calendar.getInstance(); 44 | calendar.add(Calendar.DAY_OF_MONTH, dayNumber); 45 | return dayOfDateFormat.format(calendar.getTime()); 46 | } 47 | 48 | /** 49 | * @param dateString 需要转换成时间格式的日期字符串 50 | * @return 返回字符串转换成的时间 51 | */ 52 | public static Date stringToDate(String dateString) { 53 | ParsePosition parsePosition = new ParsePosition(0); 54 | if (dateString.contains(" ")) { 55 | return secondOfDateFormat.parse(dateString, parsePosition); 56 | } else { 57 | return dayOfDateFormat.parse(dateString, parsePosition); 58 | } 59 | } 60 | 61 | /** 62 | * @param date 需要转换成字符串格式的日期 63 | * @return 返回"yyyy-MM-dd"格式的转换后的字符串 64 | */ 65 | public static String dateToShotString(Date date) { 66 | return dayOfDateFormat.format(date); 67 | } 68 | 69 | /** 70 | * @param date 需要转换成字符串格式的日期 71 | * @return 返回"yyyy-MM-dd HH:mm:ss"格式的转换后的字符串 72 | */ 73 | public static String dateToLongString(Date date) { 74 | return secondOfDateFormat.format(date); 75 | } 76 | 77 | /** 78 | * @param dateString 需要获取0点的时间字符串,如果获取当天0点,传null即可 79 | * @return 返回"yyyy-MM-dd HH:mm:ss"格式的某天0点字符串 80 | */ 81 | public static String getZeroTime(String dateString) { 82 | if (StringUtils.isBlank(dateString)) { 83 | Calendar calendar = Calendar.getInstance(); 84 | calendar.set(Calendar.HOUR_OF_DAY, 0); 85 | calendar.set(Calendar.MINUTE, 0); 86 | calendar.set(Calendar.SECOND, 0); 87 | return secondOfDateFormat.format(calendar.getTime()); 88 | } else { 89 | Date date = stringToDate(dateString); 90 | return dateToLongString(date); 91 | } 92 | } 93 | } -------------------------------------------------------------------------------- /src/main/java/com/example/websocket/webconfig/WebMvcConfig.java: -------------------------------------------------------------------------------- 1 | package com.example.websocket.webconfig; 2 | 3 | import org.springframework.context.annotation.Configuration; 4 | import org.springframework.web.servlet.config.annotation.ViewControllerRegistry; 5 | import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; 6 | 7 | /** 8 | * 在SpringBoot2.0及Spring 5.0 WebMvcConfigurerAdapter已被废弃，目前找到解决方案就有 9 | * 1 直接实现WebMvcConfigurer （官方推荐） 10 | * 2 直接继承WebMvcConfigurationSupport 11 | * @ https://blog.csdn.net/lenkvin/article/details/79482205 12 | */ 13 | @Configuration 14 | public class WebMvcConfig implements WebMvcConfigurer { 15 | 16 | /** 17 | * 为各个页面提供路径映射 18 | * @param registry 19 | */ 20 | @Override 21 | public void addViewControllers(ViewControllerRegistry registry) { 22 | registry.addViewController("/client").setViewName("client"); 23 | registry.addViewController("/index").setViewName("index"); 24 | } 25 | } -------------------------------------------------------------------------------- /src/main/java/com/example/websocket/websocket/controller/WebSocketController.java: -------------------------------------------------------------------------------- 1 | package com.example.websocket.websocket.controller; 2 | 3 | import com.example.websocket.websocket.server.WebSocketServer; 4 | import com.example.websocket.websocket.server.WebSocketServer2; 5 | import org.springframework.beans.factory.annotation.Autowired; 6 | import org.springframework.web.bind.annotation.*; 7 | 8 | import javax.websocket.Session; 9 | import java.io.IOException; 10 | import java.util.HashMap; 11 | import java.util.Map; 12 | 13 | /** 14 | * @Author 彬 15 | * @Date 2019/7/2 16 | */ 17 | @RequestMapping("/webSocket") 18 | @RestController 19 | public class WebSocketController { 20 | @Autowired 21 | private WebSocketServer webSocketServer; 22 | @Autowired 23 | private WebSocketServer2 webSocketServer2; 24 | 25 | /** 26 | * ================================================案例一============================================ 27 | * @param param 28 | * @return 29 | */ 30 | @RequestMapping(value="/pushVideoListToWeb",method= RequestMethod.POST,consumes = "application/json") 31 | public Map pushVideoListToWeb(@RequestBody Map param) { 32 | Map result =new HashMap(); 33 | try { 34 | WebSocketServer.sendInfo("有新客户呼入"); 35 | result.put("operationResult", true); 36 | }catch (IOException e) { 37 | result.put("operationResult", true); 38 | } 39 | return result; 40 | } 41 | 42 | 43 | /** 44 | * ==================================================案例2================================================ 45 | */ 46 | 47 | /** 48 | * 给指定用户推送消息 49 | * @param userName 用户名 50 | * @param message 消息 51 | * @throws IOException 52 | */ 53 | @RequestMapping(value = "/socket", method = RequestMethod.GET) 54 | public void testSocket1(@RequestParam String userName, @RequestParam String message){ 55 | webSocketServer2.sendInfo(userName, message); 56 | } 57 | 58 | 59 | /** 60 | * 给所有用户推送消息 61 | * @param message 消息 62 | * @throws IOException 63 | */ 64 | @RequestMapping(value = "/socket/all", method = RequestMethod.GET) 65 | public void testSocket2(@RequestParam String message){ 66 | try { 67 | webSocketServer2.onMessage(message); 68 | } catch (IOException e) { 69 | e.printStackTrace(); 70 | } 71 | } 72 | } 73 | -------------------------------------------------------------------------------- /src/main/java/com/example/websocket/websocket/server/WebSocketServer.java: -------------------------------------------------------------------------------- 1 | package com.example.websocket.websocket.server; 2 | 3 | import java.io.IOException; 4 | import java.util.HashMap; 5 | import java.util.Map; 6 | import java.util.concurrent.CopyOnWriteArraySet; 7 | 8 | import javax.websocket.OnClose; 9 | import javax.websocket.OnError; 10 | import javax.websocket.OnMessage; 11 | import javax.websocket.OnOpen; 12 | import javax.websocket.Session; 13 | import javax.websocket.server.PathParam; 14 | import javax.websocket.server.ServerEndpoint; 15 | 16 | import lombok.extern.slf4j.Slf4j; 17 | 18 | import org.springframework.stereotype.Component; 19 | 20 | /** 21 | * 案例1 22 | */ 23 | @Slf4j 24 | //@ServerEndpoint("/websocket/{name}") 25 | @ServerEndpoint(value = "/websocket") 26 | @Component 27 | public class WebSocketServer { 28 | //静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。 29 | private static int onlineCount = 0; 30 | //concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。 31 | private static CopyOnWriteArraySet webSocketSet = new CopyOnWriteArraySet(); 32 | 33 | //与某个客户端的连接会话，需要通过它来给客户端发送数据 34 | private Session session; 35 | 36 | 37 | /** 38 | * 连接建立成功调用的方法 39 | */ 40 | @OnOpen 41 | public void onOpen(Session session) { 42 | this.session = session; 43 | webSocketSet.add(this); //加入set中 44 | addOnlineCount(); //在线数加1 45 | log.info("有新连接加入！当前在线人数为" + getOnlineCount()); 46 | try { 47 | //这里可以写推送的业务逻辑代码，这里暂时使用主动推送 48 | sendMessage("连接成功"); 49 | } catch (IOException e) { 50 | log.error("websocket IO异常"); 51 | } 52 | } 53 | 54 | /** 55 | * 连接关闭调用的方法 56 | */ 57 | @OnClose 58 | public void onClose() { 59 | webSocketSet.remove(this); //从set中删除 60 | subOnlineCount(); //在线数减1 61 | log.info("有一连接关闭！当前在线人数为" + getOnlineCount()); 62 | } 63 | 64 | /** 65 | * 收到客户端消息后调用的方法 66 | * @param message 客户端发送过来的消息 67 | */ 68 | @OnMessage 69 | public void onMessage(String message, Session session) { 70 | log.info("来自客户端的消息:" + message); 71 | //群发消息 72 | for (WebSocketServer item : webSocketSet) { 73 | try { 74 | item.sendMessage(message); 75 | } catch (IOException e) { 76 | e.printStackTrace(); 77 | } 78 | } 79 | } 80 | 81 | /** 82 | * 发生错误时调用 83 | * @param session 84 | * @param error 85 | */ 86 | @OnError 87 | public void onError(Session session, Throwable error) { 88 | log.error("发生错误"); 89 | error.printStackTrace(); 90 | } 91 | 92 | /** 93 | * 发送消息的方法： 94 | * 使用这个方法，可以实现服务器主动推送。 95 | * @param message 96 | * @throws IOException 97 | */ 98 | public void sendMessage(String message) throws IOException { 99 | this.session.getBasicRemote().sendText(message); 100 | } 101 | 102 | /** 103 | * 群发自定义消息 104 | * */ 105 | public static void sendInfo(String message) throws IOException { 106 | log.info(message); 107 | for (WebSocketServer item : webSocketSet) { 108 | try { 109 | item.sendMessage(message); 110 | } catch (IOException e) { 111 | continue; 112 | } 113 | } 114 | } 115 | 116 | public static synchronized int getOnlineCount() { 117 | return onlineCount; 118 | } 119 | 120 | public static synchronized void addOnlineCount() { 121 | WebSocketServer.onlineCount++; 122 | } 123 | 124 | public static synchronized void subOnlineCount() { 125 | WebSocketServer.onlineCount--; 126 | } 127 | } -------------------------------------------------------------------------------- /src/main/java/com/example/websocket/websocket/server/WebSocketServer2.java: -------------------------------------------------------------------------------- 1 | package com.example.websocket.websocket.server; 2 | import org.springframework.stereotype.Component; 3 | 4 | import javax.websocket.*; 5 | import javax.websocket.server.PathParam; 6 | import javax.websocket.server.ServerEndpoint; 7 | import java.io.IOException; 8 | import java.util.HashMap; 9 | import java.util.Map; 10 | import java.util.concurrent.atomic.AtomicInteger; 11 | 12 | /** 13 | * 案例2 14 | * @author bin 15 | * @project: springboot-socket 16 | * @package: com.sailing.websocket.common 17 | * @Description: WebSocket服务端代码，包含接收消息，推送消息等接口 18 | * @date 2018/12/200948 19 | */ 20 | @Component 21 | @ServerEndpoint(value = "/socket/{name}") 22 | public class WebSocketServer2 { 23 | 24 | //静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。 25 | private static AtomicInteger online = new AtomicInteger(); 26 | 27 | //concurrent包的线程安全Set，用来存放每个客户端对应的WebSocketServer对象。 28 | private static Map sessionPools = new HashMap<>(); 29 | 30 | /** 31 | * 发送消息方法 32 | * @param session 客户端与socket建立的会话 33 | * @param message 消息 34 | * @throws IOException 35 | */ 36 | public void sendMessage(Session session, String message) throws IOException{ 37 | if(session != null){ 38 | session.getBasicRemote().sendText(message); 39 | } 40 | } 41 | 42 | 43 | /** 44 | * 连接建立成功调用 45 | * @param session 客户端与socket建立的会话 46 | * @param userName 客户端的userName 47 | */ 48 | @OnOpen 49 | public void onOpen(Session session, @PathParam(value = "name") String userName){ 50 | sessionPools.put(userName, session); 51 | addOnlineCount(); 52 | System.out.println(userName + "加入webSocket！当前人数为" + online); 53 | try { 54 | sendMessage(session, "欢迎" + userName + "加入连接！"); 55 | } catch (IOException e) { 56 | e.printStackTrace(); 57 | } 58 | } 59 | 60 | /** 61 | * 关闭连接时调用 62 | * @param userName 关闭连接的客户端的姓名 63 | */ 64 | @OnClose 65 | public void onClose(@PathParam(value = "name") String userName){ 66 | sessionPools.remove(userName); 67 | subOnlineCount(); 68 | System.out.println(userName + "断开webSocket连接！当前人数为" + online); 69 | } 70 | 71 | /** 72 | * 收到客户端消息时触发（群发） 73 | * @param message 74 | * @throws IOException 75 | */ 76 | @OnMessage 77 | public void onMessage(String message) throws IOException{ 78 | for (Session session: sessionPools.values()) { 79 | try { 80 | sendMessage(session, message); 81 | } catch(Exception e){ 82 | e.printStackTrace(); 83 | continue; 84 | } 85 | } 86 | } 87 | 88 | /** 89 | * 发生错误时候 90 | * @param session 91 | * @param throwable 92 | */ 93 | @OnError 94 | public void onError(Session session, Throwable throwable){ 95 | System.out.println("发生错误"); 96 | throwable.printStackTrace(); 97 | } 98 | 99 | /** 100 | * 给指定用户发送消息 101 | * @param userName 用户名 102 | * @param message 消息 103 | * @throws IOException 104 | */ 105 | public void sendInfo(String userName, String message){ 106 | Session session = sessionPools.get(userName); 107 | try { 108 | sendMessage(session, message); 109 | }catch (Exception e){ 110 | e.printStackTrace(); 111 | } 112 | } 113 | 114 | /** 115 | * 根据用户名获取session会话对象 116 | * @param userName 117 | * @return 118 | */ 119 | public Session getSession(String userName){ 120 | Session session = sessionPools.get(userName); 121 | return session; 122 | } 123 | 124 | public static void addOnlineCount(){ 125 | online.incrementAndGet(); 126 | } 127 | 128 | public static void subOnlineCount() { 129 | online.decrementAndGet(); 130 | } 131 | } 132 | -------------------------------------------------------------------------------- /src/main/java/com/example/websocket/websocket/task/ProductExpireTask.java: -------------------------------------------------------------------------------- 1 | package com.example.websocket.websocket.task; 2 | 3 | 4 | import com.example.websocket.commonutils.DateUtil; 5 | import com.example.websocket.websocket.server.WebSocketServer2; 6 | import org.springframework.beans.factory.annotation.Autowired; 7 | import org.springframework.scheduling.annotation.Scheduled; 8 | import org.springframework.stereotype.Component; 9 | import org.springframework.web.socket.TextMessage; 10 | 11 | import javax.websocket.Session; 12 | import java.io.IOException; 13 | import java.util.HashMap; 14 | import java.util.Map; 15 | 16 | 17 | @Component 18 | public class ProductExpireTask { 19 | @Autowired 20 | private WebSocketServer2 webSocketServer2; 21 | 22 | //每天早上0点执行 23 | @Scheduled(cron = "0 0 0 1/1 * ?") 24 | public void productExpire() { 25 | //距离到期还有一个月提醒 26 | String oneMonthExpireDate = DateUtil.addOneMonth(); 27 | dealExpireProduct(oneMonthExpireDate); 28 | 29 | //距离到期还有一天提醒 30 | String oneDayExpireDate = DateUtil.addOneDay(); 31 | dealExpireProduct(oneDayExpireDate); 32 | 33 | //距离到期还有一周提醒 34 | String oneWeekExpireDate = DateUtil.addFewDays(7); 35 | dealExpireProduct(oneWeekExpireDate); 36 | } 37 | 38 | private void dealExpireProduct(String expireDate) { 39 | //todo 推送逻辑代码---根据具体业务场景来写 40 | } 41 | 42 | /** 43 | * 模拟定时任务服务端主动给某个用户发送信息 44 | * 每5秒执行一次 45 | */ 46 | @Scheduled(cron = "*/5 * * * * ?") 47 | public void sendMessage() { 48 | String message = "你好"; 49 | Session session = webSocketServer2.getSession("bin"); 50 | if(session!=null){ 51 | try { 52 | webSocketServer2.sendMessage(session, String.valueOf(new TextMessage(message))); 53 | } catch (IOException e) { 54 | e.printStackTrace(); 55 | } 56 | } 57 | } 58 | } -------------------------------------------------------------------------------- /src/main/java/com/example/websocket/websocketconfig/MyEndpointConfigure.java: -------------------------------------------------------------------------------- 1 | package com.example.websocket.websocketconfig; 2 | 3 | import org.springframework.beans.BeansException; 4 | import org.springframework.beans.factory.BeanFactory; 5 | import org.springframework.context.ApplicationContext; 6 | import org.springframework.context.ApplicationContextAware; 7 | 8 | import javax.websocket.server.ServerEndpointConfig; 9 | 10 | public class MyEndpointConfigure extends ServerEndpointConfig.Configurator implements ApplicationContextAware { 11 | private static volatile BeanFactory context; 12 | 13 | @Override 14 | public T getEndpointInstance(Class clazz) throws InstantiationException { 15 | return context.getBean(clazz); 16 | } 17 | 18 | @Override 19 | public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { 20 | MyEndpointConfigure.context = applicationContext; 21 | } 22 | } -------------------------------------------------------------------------------- /src/main/java/com/example/websocket/websocketconfig/WebSocketConfig.java: -------------------------------------------------------------------------------- 1 | package com.example.websocket.websocketconfig; 2 | 3 | import org.springframework.context.annotation.Bean; 4 | import org.springframework.context.annotation.Configuration; 5 | import org.springframework.web.socket.server.standard.ServerEndpointExporter; 6 | 7 | /** 8 | * websocket配置类中 9 | */ 10 | @Configuration 11 | public class WebSocketConfig { 12 | /** 13 | * 第一个配置是因为使用springboot内置容器，自己开发时需要配置，如果有独立的容器需要将其注释掉， 14 | * 也就意味着，如果将项目打成WAR包，部署到服务器，使用Tomcat启动时，需要注释掉ServerEndpointExporter配置； 15 | * @return 16 | */ 17 | @Bean 18 | public ServerEndpointExporter serverEndpointExporter() { 19 | return new ServerEndpointExporter(); 20 | } 21 | 22 | /** 23 | * MyEndpointConfigure配置是因为我的需求需要，需要在websocket类中注入service层或者dao层的接口， 24 | * MyEndpointConfigure配置就是为了解决websocket无法注入的问题，如果没有需要可以不用配置 25 | * @return 26 | */ 27 | @Bean 28 | public MyEndpointConfigure newConfigure() { 29 | return new MyEndpointConfigure(); 30 | } 31 | } -------------------------------------------------------------------------------- /src/main/resources/application.properties: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/main/resources/image/微信图片_20190702135004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/17661977890/websocket/9a264519a4ad360c06b0eed1ee240813437a085e/src/main/resources/image/微信图片_20190702135004.png -------------------------------------------------------------------------------- /src/main/resources/image/微信图片_20190702135058.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/17661977890/websocket/9a264519a4ad360c06b0eed1ee240813437a085e/src/main/resources/image/微信图片_20190702135058.png -------------------------------------------------------------------------------- /src/main/resources/image/微信图片_20190702150014.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/17661977890/websocket/9a264519a4ad360c06b0eed1ee240813437a085e/src/main/resources/image/微信图片_20190702150014.png -------------------------------------------------------------------------------- /src/main/resources/image/微信图片_20190702150146.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/17661977890/websocket/9a264519a4ad360c06b0eed1ee240813437a085e/src/main/resources/image/微信图片_20190702150146.png -------------------------------------------------------------------------------- /src/main/resources/static/client.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | myWebSocket 6 | 7 | 8 | Welcome
9 | 10 | 11 | 12 |
13 | 14 |
15 | 16 | 17 | 18 | 75 | -------------------------------------------------------------------------------- /src/main/resources/static/index.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | Title 6 | 7 | 8 | Welcome
9 | 10 |
11 |
12 | 13 | 14 | 65 | -------------------------------------------------------------------------------- /src/main/resources/static/websocket.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | myWebSocket 6 | 7 | 8 | Welcome
9 | 10 | 11 | 12 |
13 | 14 |
15 | 16 | 17 | 18 | 75 | -------------------------------------------------------------------------------- /src/test/java/com/example/websocket/WebsocketApplicationTests.java: -------------------------------------------------------------------------------- 1 | package com.example.websocket; 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 WebsocketApplicationTests { 11 | 12 | @Test 13 | public void contextLoads() { 14 | } 15 | 16 | } 17 | --------------------------------------------------------------------------------