├── README.md └── netty-websocket ├── .gitignore ├── .mvn └── wrapper │ ├── maven-wrapper.jar │ └── maven-wrapper.properties ├── mvnw ├── mvnw.cmd ├── pom.xml └── src ├── main ├── java │ └── com │ │ └── wangzai │ │ └── nettywebsocket │ │ ├── NettyWebsocketApplication.java │ │ └── Server │ │ ├── ChatRoomHandler.java │ │ ├── NettyServer.java │ │ ├── NettyServerApplication.java │ │ └── NettyWebsocketHandler.java └── resources │ ├── application.properties │ └── resources │ └── index.html └── test └── java └── com └── wangzai └── nettywebsocket └── NettyWebsocketApplicationTests.java /README.md: -------------------------------------------------------------------------------- 1 | # nettyChat 2 | -------------------------------------------------------------------------------- /netty-websocket/.gitignore: -------------------------------------------------------------------------------- 1 | /target/ 2 | !.mvn/wrapper/maven-wrapper.jar 3 | 4 | ### STS ### 5 | .apt_generated 6 | .classpath 7 | .factorypath 8 | .project 9 | .settings 10 | .springBeans 11 | .sts4-cache 12 | 13 | ### IntelliJ IDEA ### 14 | .idea 15 | *.iws 16 | *.iml 17 | *.ipr 18 | 19 | ### NetBeans ### 20 | /nbproject/private/ 21 | /build/ 22 | /nbbuild/ 23 | /dist/ 24 | /nbdist/ 25 | /.nb-gradle/ -------------------------------------------------------------------------------- /netty-websocket/.mvn/wrapper/maven-wrapper.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangZaier/nettyChat/74b4b4997d58e3e6687b6886328346ae8eb1508a/netty-websocket/.mvn/wrapper/maven-wrapper.jar -------------------------------------------------------------------------------- /netty-websocket/.mvn/wrapper/maven-wrapper.properties: -------------------------------------------------------------------------------- 1 | distributionUrl=https://repo1.maven.org/maven2/org/apache/maven/apache-maven/3.5.3/apache-maven-3.5.3-bin.zip 2 | -------------------------------------------------------------------------------- /netty-websocket/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 | -------------------------------------------------------------------------------- /netty-websocket/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 | -------------------------------------------------------------------------------- /netty-websocket/pom.xml: -------------------------------------------------------------------------------- 1 | 2 | 4 | 4.0.0 5 | 6 | com.wangzai 7 | netty-websocket 8 | 0.0.1-SNAPSHOT 9 | jar 10 | 11 | netty-websocket 12 | 13 | 14 | org.springframework.boot 15 | spring-boot-starter-parent 16 | 1.5.10.RELEASE 17 | 18 | 19 | 20 | 21 | UTF-8 22 | UTF-8 23 | 1.8 24 | 25 | 26 | 27 | 28 | org.springframework.boot 29 | spring-boot-starter 30 | 31 | 32 | 33 | org.springframework.boot 34 | spring-boot-starter-web 35 | 36 | 37 | 38 | 39 | io.netty 40 | netty-all 41 | 4.1.22.Final 42 | 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 | 63 | 64 | -------------------------------------------------------------------------------- /netty-websocket/src/main/java/com/wangzai/nettywebsocket/NettyWebsocketApplication.java: -------------------------------------------------------------------------------- 1 | package com.wangzai.nettywebsocket; 2 | 3 | import com.wangzai.nettywebsocket.Server.NettyServer; 4 | import com.wangzai.nettywebsocket.Server.NettyWebsocketHandler; 5 | import io.netty.bootstrap.ServerBootstrap; 6 | import io.netty.channel.ChannelInitializer; 7 | import io.netty.channel.ChannelOption; 8 | import io.netty.channel.nio.NioEventLoopGroup; 9 | import io.netty.channel.socket.SocketChannel; 10 | import io.netty.channel.socket.nio.NioServerSocketChannel; 11 | import io.netty.handler.codec.http.HttpObjectAggregator; 12 | import io.netty.handler.codec.http.HttpRequestDecoder; 13 | import io.netty.handler.codec.http.HttpResponseEncoder; 14 | import io.netty.handler.logging.LogLevel; 15 | import io.netty.handler.logging.LoggingHandler; 16 | import org.springframework.beans.factory.annotation.Autowired; 17 | import org.springframework.boot.CommandLineRunner; 18 | import org.springframework.boot.SpringApplication; 19 | import org.springframework.boot.autoconfigure.SpringBootApplication; 20 | import org.springframework.context.ConfigurableApplicationContext; 21 | import org.springframework.context.annotation.Bean; 22 | 23 | @SpringBootApplication 24 | public class NettyWebsocketApplication { 25 | 26 | 27 | 28 | 29 | 30 | public static void main(String[] args) throws Exception { 31 | // SpringApplication.run(NettyWebsocketApplication.class, args); 32 | 33 | ConfigurableApplicationContext context = SpringApplication.run(NettyWebsocketApplication.class, args); 34 | // ServerBootstrap serverBootstrap = context.getBean(ServerBootstrap.class); 35 | // serverBootstrap.bind(8888).sync(); 36 | NettyServer nettyServer = context.getBean(NettyServer.class); 37 | nettyServer.start(); 38 | 39 | } 40 | 41 | 42 | } 43 | -------------------------------------------------------------------------------- /netty-websocket/src/main/java/com/wangzai/nettywebsocket/Server/ChatRoomHandler.java: -------------------------------------------------------------------------------- 1 | package com.wangzai.nettywebsocket.Server; 2 | 3 | import io.netty.channel.Channel; 4 | import io.netty.channel.ChannelHandler; 5 | import io.netty.channel.ChannelHandlerContext; 6 | import io.netty.channel.SimpleChannelInboundHandler; 7 | import io.netty.channel.group.ChannelGroup; 8 | import io.netty.channel.group.DefaultChannelGroup; 9 | import io.netty.handler.codec.http.websocketx.TextWebSocketFrame; 10 | import io.netty.util.concurrent.GlobalEventExecutor; 11 | import org.springframework.stereotype.Component; 12 | 13 | @Component 14 | @ChannelHandler.Sharable 15 | public class ChatRoomHandler extends SimpleChannelInboundHandler { 16 | 17 | 18 | private static ChannelGroup channels = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE); 19 | 20 | 21 | //广播 22 | @Override 23 | protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception { 24 | 25 | //获取内容 26 | String content = msg.text(); 27 | 28 | //获取当前的Channel 29 | Channel incoming = ctx.channel(); 30 | 31 | for (Channel channel : channels) { 32 | //不是刚刚进来的 33 | if (channel != incoming ){ 34 | channel.writeAndFlush(new TextWebSocketFrame(ctx.channel().remoteAddress() + ":" + content)); 35 | }else { 36 | channel.writeAndFlush(new TextWebSocketFrame("我自己:" + content)); 37 | } 38 | } 39 | 40 | } 41 | 42 | 43 | //进入:,提示所有人,有人进来了 44 | @Override 45 | public void handlerAdded(ChannelHandlerContext ctx) throws Exception { 46 | 47 | for (Channel channel : channels) { 48 | channel.writeAndFlush(new TextWebSocketFrame(ctx.channel().remoteAddress() + "已经进入聊天室")); 49 | } 50 | 51 | channels.add(ctx.channel()); 52 | } 53 | 54 | 55 | 56 | //离开,提示 57 | @Override 58 | public void handlerRemoved(ChannelHandlerContext ctx) throws Exception { 59 | channels.remove(ctx.channel()); 60 | for (Channel channel : channels) { 61 | channel.writeAndFlush(new TextWebSocketFrame(ctx.channel().remoteAddress() + "离开了聊天室")); 62 | } 63 | 64 | } 65 | } 66 | 67 | -------------------------------------------------------------------------------- /netty-websocket/src/main/java/com/wangzai/nettywebsocket/Server/NettyServer.java: -------------------------------------------------------------------------------- 1 | package com.wangzai.nettywebsocket.Server; 2 | 3 | import io.netty.bootstrap.ServerBootstrap; 4 | import io.netty.channel.Channel; 5 | import org.springframework.beans.factory.annotation.Autowired; 6 | import org.springframework.beans.factory.annotation.Qualifier; 7 | import org.springframework.stereotype.Component; 8 | 9 | import javax.annotation.PreDestroy; 10 | 11 | @Component 12 | public class NettyServer { 13 | // 14 | @Autowired 15 | @Qualifier("serverBootstrap") 16 | private ServerBootstrap serverBootstrap; 17 | 18 | private Channel serverChannel; 19 | 20 | public void start() throws Exception { 21 | System.out.println("netty启动"); 22 | serverChannel = serverBootstrap.bind(8888).sync().channel().closeFuture().sync().channel(); 23 | } 24 | 25 | @PreDestroy 26 | public void stop() throws Exception { 27 | serverChannel.close(); 28 | serverChannel.parent().close(); 29 | } 30 | 31 | 32 | } 33 | -------------------------------------------------------------------------------- /netty-websocket/src/main/java/com/wangzai/nettywebsocket/Server/NettyServerApplication.java: -------------------------------------------------------------------------------- 1 | package com.wangzai.nettywebsocket.Server; 2 | 3 | import io.netty.bootstrap.ServerBootstrap; 4 | import io.netty.channel.ChannelInitializer; 5 | import io.netty.channel.ChannelOption; 6 | import io.netty.channel.nio.NioEventLoopGroup; 7 | import io.netty.channel.socket.SocketChannel; 8 | import io.netty.channel.socket.nio.NioServerSocketChannel; 9 | import io.netty.handler.codec.http.HttpObjectAggregator; 10 | import io.netty.handler.codec.http.HttpRequestDecoder; 11 | import io.netty.handler.codec.http.HttpResponseEncoder; 12 | import io.netty.handler.codec.http.HttpServerCodec; 13 | import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler; 14 | import io.netty.handler.logging.LogLevel; 15 | import io.netty.handler.logging.LoggingHandler; 16 | import io.netty.handler.stream.ChunkedWriteHandler; 17 | import org.springframework.beans.factory.annotation.Autowired; 18 | import org.springframework.context.annotation.Bean; 19 | import org.springframework.context.annotation.Configuration; 20 | 21 | 22 | @Configuration 23 | public class NettyServerApplication { 24 | 25 | 26 | 27 | @Autowired 28 | ChatRoomHandler chatRoomHandler; 29 | 30 | @Bean(name = "bossGroup", destroyMethod = "shutdownGracefully") 31 | public NioEventLoopGroup bossGroup() { 32 | return new NioEventLoopGroup(); 33 | } 34 | 35 | @Bean(name = "workerGroup", destroyMethod = "shutdownGracefully") 36 | public NioEventLoopGroup workerGroup() { 37 | return new NioEventLoopGroup(); 38 | } 39 | 40 | 41 | @Bean(name = "serverBootstrap") 42 | public ServerBootstrap bootstrap() { 43 | ServerBootstrap serverBootstrap = new ServerBootstrap(); 44 | serverBootstrap.group(bossGroup(), workerGroup()) 45 | .channel(NioServerSocketChannel.class) 46 | .childHandler(new ChannelInitializer() { 47 | @Override 48 | protected void initChannel(SocketChannel socketChannel) throws Exception { 49 | socketChannel.pipeline().addLast(new HttpServerCodec()); 50 | socketChannel.pipeline().addLast(new ChunkedWriteHandler()); 51 | socketChannel.pipeline().addLast(new HttpObjectAggregator(65536)); 52 | socketChannel.pipeline().addLast(new WebSocketServerProtocolHandler("/websocket"));//握手操作 53 | socketChannel.pipeline().addLast(chatRoomHandler); 54 | 55 | } 56 | }) 57 | .option(ChannelOption.SO_BACKLOG, 128) 58 | .childOption(ChannelOption.SO_KEEPALIVE, true); 59 | return serverBootstrap; 60 | } 61 | 62 | 63 | } 64 | -------------------------------------------------------------------------------- /netty-websocket/src/main/java/com/wangzai/nettywebsocket/Server/NettyWebsocketHandler.java: -------------------------------------------------------------------------------- 1 | package com.wangzai.nettywebsocket.Server; 2 | 3 | import io.netty.buffer.ByteBuf; 4 | import io.netty.buffer.Unpooled; 5 | import io.netty.channel.ChannelHandler; 6 | import io.netty.channel.ChannelHandlerContext; 7 | import io.netty.channel.ChannelInboundHandlerAdapter; 8 | import io.netty.handler.codec.http.DefaultFullHttpResponse; 9 | import io.netty.handler.codec.http.FullHttpRequest; 10 | import io.netty.handler.codec.http.HttpResponseStatus; 11 | import io.netty.handler.codec.http.HttpVersion; 12 | import io.netty.handler.codec.http.websocketx.*; 13 | import io.netty.util.CharsetUtil; 14 | import org.springframework.stereotype.Component; 15 | 16 | @Component 17 | @ChannelHandler.Sharable 18 | public class NettyWebsocketHandler extends ChannelInboundHandlerAdapter { 19 | 20 | 21 | private WebSocketServerHandshaker handshaker; 22 | 23 | @Override 24 | public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception { 25 | 26 | if (msg instanceof FullHttpRequest) { 27 | 28 | FullHttpRequest request = (FullHttpRequest) msg; 29 | 30 | //检验请求 31 | if (!request.decoderResult().isSuccess() || !"websocket".equals(request.headers().get("Upgrade"))) { 32 | //返回错误信息 33 | DefaultFullHttpResponse response = new DefaultFullHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.BAD_REQUEST); 34 | if (response.status().code() != 200) { 35 | ByteBuf byteBuf = Unpooled.copiedBuffer("请求异常", CharsetUtil.UTF_8); 36 | response.content().writeBytes(byteBuf); 37 | byteBuf.release(); 38 | } 39 | ctx.writeAndFlush(response); 40 | return; 41 | } 42 | 43 | WebSocketServerHandshakerFactory webSocketServerHandshakerFactory = new WebSocketServerHandshakerFactory("ws://localhost:8888/websocket", null, false); 44 | 45 | handshaker = webSocketServerHandshakerFactory.newHandshaker(request); 46 | 47 | if (handshaker == null) { 48 | webSocketServerHandshakerFactory.sendUnsupportedVersionResponse(ctx.channel()); 49 | } else { 50 | handshaker.handshake(ctx.channel(), request); 51 | } 52 | 53 | } else if (msg instanceof WebSocketFrame) { 54 | 55 | //判断是否是关闭链路命令 56 | if (msg instanceof CloseWebSocketFrame) { 57 | handshaker.close(ctx.channel(), (CloseWebSocketFrame) msg); 58 | } 59 | 60 | //我们只处理文本内容 61 | if(msg instanceof TextWebSocketFrame){ 62 | //获取文本 63 | String request = ((TextWebSocketFrame)msg).text(); 64 | 65 | //回写 66 | ctx.writeAndFlush(new TextWebSocketFrame("返回数据:" + request)); 67 | 68 | 69 | } 70 | } 71 | } 72 | 73 | 74 | } 75 | -------------------------------------------------------------------------------- /netty-websocket/src/main/resources/application.properties: -------------------------------------------------------------------------------- 1 | server.port=9999 -------------------------------------------------------------------------------- /netty-websocket/src/main/resources/resources/index.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | WebSocket Chat 6 | 7 | 8 | 42 |
43 |

WebSocket 聊天室：

44 | 45 |
46 | 47 | 48 | 49 |
50 |
51 |
52 | 53 | 54 | -------------------------------------------------------------------------------- /netty-websocket/src/test/java/com/wangzai/nettywebsocket/NettyWebsocketApplicationTests.java: -------------------------------------------------------------------------------- 1 | package com.wangzai.nettywebsocket; 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 NettyWebsocketApplicationTests { 11 | 12 | @Test 13 | public void contextLoads() { 14 | } 15 | 16 | } 17 | --------------------------------------------------------------------------------