├── .gitignore ├── .mvn └── wrapper │ ├── MavenWrapperDownloader.java │ ├── maven-wrapper.jar │ └── maven-wrapper.properties ├── README.md ├── libs └── paramcheck.jar ├── mvnw ├── mvnw.cmd ├── pom.xml └── src ├── main ├── java │ └── com │ │ └── lanfangyi │ │ └── nettyim │ │ ├── .swp │ │ ├── NettyImApplication.java │ │ ├── base │ │ ├── ErrorInfo.java │ │ ├── IMRequest.java │ │ └── IMResponse.java │ │ ├── bean │ │ ├── BaseBean.java │ │ ├── Friend.java │ │ ├── SendTask.java │ │ └── User.java │ │ ├── config │ │ ├── NettyConfig.java │ │ └── SwaggerConfig.java │ │ ├── constants │ │ ├── ErrorCode.java │ │ ├── StatusConstant.java │ │ ├── SystemUserConstant.java │ │ └── UserTypeConstant.java │ │ ├── controller │ │ ├── FriendController.java │ │ ├── SendTaskController.java │ │ └── UserController.java │ │ ├── enums │ │ └── MsgTypeEnum.java │ │ ├── exception │ │ └── SendMsgButChannelIsNullxception.java │ │ ├── filter │ │ ├── DruidStatFilter.java │ │ └── DruidStatViewServlet.java │ │ ├── future │ │ ├── SendMsgFuture.java │ │ └── resp │ │ │ └── SendMsgFutureResp.java │ │ ├── handler │ │ ├── HeartBeatHandler.java │ │ └── WebSocketHandler.java │ │ ├── holder │ │ └── ChannelHolder.java │ │ ├── init │ │ └── NettyIMServerInitializer.java │ │ ├── listener │ │ └── AsyncSendMsgListener.java │ │ ├── mapper │ │ ├── FriendMapper.java │ │ ├── SendTaskMapper.java │ │ └── UserMapper.java │ │ ├── pojo │ │ └── Msg.java │ │ ├── pool │ │ └── FutureTaskPool.java │ │ ├── server │ │ └── NettyIMServer.java │ │ ├── service │ │ ├── SendMsgService.java │ │ ├── UserService.java │ │ └── impl │ │ │ └── UserServiceImpl.java │ │ └── utils │ │ ├── DataCheckUtil.java │ │ ├── DateUtil.java │ │ ├── IdGetUtil.java │ │ └── UUIDUtil.java └── resources │ ├── application-dev.yml │ ├── application.yml │ ├── doc │ ├── table.sql │ └── todo │ ├── log4j.properties │ ├── mapper │ ├── FriendMapper.xml │ ├── SendTaskMapper.xml │ └── UserMapper.xml │ └── templates │ ├── message.html │ └── message2.html └── test └── java └── com └── lanfangyi └── nettyim └── NettyImApplicationTests.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/lanfangyi/netty-im/9bac28da60f95681bc6d2c23c94fc21625e52f85/.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 | # netty-im 2 | 基于springboot+mybatis+druid+swagger+netty+websocket的即时通讯工程 3 | -------------------------------------------------------------------------------- /libs/paramcheck.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lanfangyi/netty-im/9bac28da60f95681bc6d2c23c94fc21625e52f85/libs/paramcheck.jar -------------------------------------------------------------------------------- /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.7.RELEASE 9 | 10 | 11 | com.lanfangyi 12 | netty-im 13 | 0.0.1-SNAPSHOT 14 | netty-im 15 | Demo project for Spring Boot 16 | 17 | 18 | 1.8 19 | 20 | 21 | 22 | 23 | 24 | org.springframework.boot 25 | spring-boot-starter-web 26 | 27 | 28 | mysql 29 | mysql-connector-java 30 | 31 | 32 | 33 | com.alibaba 34 | druid-spring-boot-starter 35 | 1.1.10 36 | 37 | 38 | 39 | org.mybatis.spring.boot 40 | mybatis-spring-boot-starter 41 | 1.3.2 42 | 43 | 44 | 45 | org.aspectj 46 | aspectjweaver 47 | 1.9.4 48 | 49 | 50 | 51 | org.projectlombok 52 | lombok 53 | 1.16.22 54 | provided 55 | 56 | 57 | 58 | 59 | 60 | io.springfox 61 | springfox-swagger2 62 | 2.7.0 63 | 64 | 65 | guava 66 | com.google.guava 67 | 68 | 69 | 70 | 71 | io.springfox 72 | springfox-swagger-ui 73 | 2.7.0 74 | 75 | 76 | 77 | org.mybatis 78 | mybatis-spring 79 | 1.3.2 80 | 81 | 82 | org.mybatis 83 | mybatis 84 | 3.4.6 85 | 86 | 87 | 88 | com.alibaba 89 | fastjson 90 | 1.1.43 91 | 92 | 93 | 94 | io.netty 95 | netty-all 96 | 4.1.21.final 97 | 98 | 99 | 100 | org.apache.commons 101 | commons-lang3 102 | 3.9 103 | 104 | 105 | 106 | 107 | tk.mybatis 108 | mapper-spring-boot-starter 109 | 2.1.5 110 | 111 | 112 | 113 | 114 | 115 | 116 | 117 | 118 | 119 | 120 | 121 | com.google.guava 122 | guava 123 | 16.0.1 124 | 125 | 126 | 127 | 128 | 129 | 130 | com.lanfangyi 131 | paramcheck 132 | 0.0.1-SNAPSHOT 133 | system 134 | ${project.basedir}/libs/paramcheck.jar 135 | 136 | 137 | 138 | 139 | 140 | 141 | org.springframework.boot 142 | spring-boot-maven-plugin 143 | 144 | 145 | org.springframework.boot 146 | spring-boot-maven-plugin 147 | 148 | true 149 | 150 | 151 | 152 | 153 | 154 | 155 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/.swp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lanfangyi/netty-im/9bac28da60f95681bc6d2c23c94fc21625e52f85/src/main/java/com/lanfangyi/nettyim/.swp -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/NettyImApplication.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim; 2 | 3 | import com.alibaba.druid.pool.DruidDataSource; 4 | import com.lanfangyi.nettyim.pool.FutureTaskPool; 5 | import org.mybatis.spring.annotation.MapperScan; 6 | import org.springframework.boot.SpringApplication; 7 | import org.springframework.boot.autoconfigure.SpringBootApplication; 8 | import org.springframework.boot.context.properties.ConfigurationProperties; 9 | import org.springframework.boot.web.servlet.ServletComponentScan; 10 | import org.springframework.context.annotation.Bean; 11 | import org.springframework.context.annotation.ComponentScan; 12 | 13 | import javax.annotation.PreDestroy; 14 | import javax.sql.DataSource; 15 | 16 | @MapperScan("com.lanfangyi.nettyim.mapper") 17 | @SpringBootApplication 18 | @ServletComponentScan 19 | @ComponentScan("com.lanfangyi.*") //不加这一行，jar包中的aop不生效 20 | public class NettyImApplication { 21 | 22 | public static void main(String[] args) { 23 | SpringApplication.run(NettyImApplication.class, args); 24 | } 25 | 26 | @Bean("druidDataSource") 27 | @ConfigurationProperties(prefix = "spring.datasource") 28 | public DataSource druidDataSource() { 29 | return new DruidDataSource(); 30 | } 31 | 32 | @PreDestroy 33 | public void shutDown() { 34 | FutureTaskPool.shutDown(); 35 | } 36 | } 37 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/base/ErrorInfo.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.base; 2 | 3 | import lombok.AllArgsConstructor; 4 | import lombok.Data; 5 | import lombok.Getter; 6 | import lombok.NoArgsConstructor; 7 | import lombok.Setter; 8 | import lombok.ToString; 9 | 10 | /** 11 | * @author lanfangyi@haodf.com 12 | * @version 1.0 13 | * @since 2019/8/14 4:27 PM 14 | */ 15 | @Data 16 | @AllArgsConstructor 17 | public class ErrorInfo { 18 | 19 | private int code ; 20 | 21 | private String msg; 22 | 23 | } 24 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/base/IMRequest.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.base; 2 | 3 | import com.lanfangyi.service.paramcheck.req.BaseRequest; 4 | import io.swagger.annotations.ApiModelProperty; 5 | import lombok.Data; 6 | import lombok.Getter; 7 | import lombok.Setter; 8 | import lombok.ToString; 9 | 10 | import java.io.Serializable; 11 | 12 | /** 13 | * @author lanfangyi@haodf.com 14 | * @version 1.0 15 | * @since 2019/8/14 3:34 PM 16 | */ 17 | @Data 18 | public class IMRequest extends BaseRequest { 19 | 20 | @ApiModelProperty(value = "自定义协议的标示", example = "myProtocol") 21 | private String protocol; 22 | 23 | } 24 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/base/IMResponse.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.base; 2 | 3 | import com.lanfangyi.service.paramcheck.resp.BaseResponse; 4 | import lombok.Data; 5 | 6 | /** 7 | * @author lanfangyi@haodf.com 8 | * @version 1.0 9 | * @since 2019/8/14 3:34 PM 10 | */ 11 | @Data 12 | public class IMResponse extends BaseResponse { 13 | 14 | public static IMResponse success(T dataBody) { 15 | IMResponse response = new IMResponse<>(); 16 | response.setDataBody(dataBody); 17 | return response; 18 | } 19 | 20 | public static IMResponse error(int code, String msg, T dataBody) { 21 | IMResponse response = new IMResponse<>(); 22 | response.setCode(code); 23 | response.setMessage(msg); 24 | response.setDataBody(dataBody); 25 | return response; 26 | } 27 | 28 | public static IMResponse error(int code, T dataBody) { 29 | IMResponse response = new IMResponse<>(); 30 | response.setCode(code); 31 | response.setDataBody(dataBody); 32 | return response; 33 | } 34 | 35 | public static IMResponse error(ErrorInfo errorInfo, T dataBody) { 36 | IMResponse response = new IMResponse<>(); 37 | response.setCode(errorInfo.getCode()); 38 | response.setMessage(errorInfo.getMsg()); 39 | response.setDataBody(dataBody); 40 | return response; 41 | } 42 | } 43 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/bean/BaseBean.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.bean; 2 | 3 | import com.lanfangyi.nettyim.utils.IdGetUtil; 4 | import lombok.Data; 5 | 6 | import javax.persistence.Column; 7 | import javax.persistence.Id; 8 | import java.io.Serializable; 9 | import java.util.Date; 10 | 11 | /** 12 | * @author lanfangyi@haodf.com 13 | * @version 1.0 14 | * @since 2019/8/20 10:30 PM 15 | */ 16 | @Data 17 | class BaseBean implements Serializable { 18 | 19 | @Id 20 | private Long id = IdGetUtil.get(); 21 | 22 | @Column(name = "ctime") 23 | private Date cTime = new Date(); 24 | 25 | @Column(name = "utime") 26 | private Date uTime = new Date(); 27 | } 28 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/bean/Friend.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.bean; 2 | 3 | import com.lanfangyi.nettyim.constants.StatusConstant; 4 | import com.lanfangyi.nettyim.utils.IdGetUtil; 5 | import lombok.Data; 6 | 7 | import javax.persistence.Column; 8 | import javax.persistence.Table; 9 | import java.util.Date; 10 | 11 | /** 12 | * @author lanfangyi@haodf.com 13 | * @version 1.0 14 | * @since 2019/9/2 5:13 PM 15 | */ 16 | @Data 17 | @Table(name = "friends") 18 | public class Friend extends BaseBean { 19 | 20 | @Column(name = "userid") 21 | private Long userId; 22 | 23 | @Column(name = "usertype") 24 | private String userType; 25 | 26 | @Column(name = "frienduserid") 27 | private Long friendUserId; 28 | 29 | @Column(name = "friendusertype") 30 | private String friendUserType; 31 | 32 | @Column(name = "status") 33 | private Integer status; 34 | 35 | public Friend(Long userId, String userType, Long friendUserId, String friendUserType) { 36 | this.setId(IdGetUtil.get()); 37 | this.userId = userId; 38 | this.userType = userType; 39 | this.friendUserId = friendUserId; 40 | this.friendUserType = friendUserType; 41 | this.status = StatusConstant.VALID; 42 | this.setCTime(new Date()); 43 | this.setUTime(new Date()); 44 | } 45 | } 46 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/bean/SendTask.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.bean; 2 | 3 | import com.lanfangyi.nettyim.constants.StatusConstant; 4 | import com.lanfangyi.nettyim.utils.IdGetUtil; 5 | import lombok.Data; 6 | import lombok.Synchronized; 7 | 8 | import javax.persistence.Column; 9 | import javax.persistence.Table; 10 | 11 | /** 12 | * @author lanfangyi@haodf.com 13 | * @version 1.0 14 | * @since 2019/9/2 5:13 PM 15 | */ 16 | @Data 17 | @Table(name = "sendtasks") 18 | public class SendTask extends BaseBean { 19 | 20 | @Column(name = "providerid") 21 | private Long providerId; 22 | 23 | @Column(name = "providertype") 24 | private String providerType; 25 | 26 | @Column(name = "receiveuserid") 27 | private Long receiveUserId; 28 | 29 | @Column(name = "receiveusertype") 30 | private String receiveUserType; 31 | 32 | /** 33 | * 1代表不需要call， 2代表需要 34 | */ 35 | @Column(name = "iscall") 36 | private Integer isCall; 37 | 38 | /** 39 | * 尝试发送的次数 40 | */ 41 | @Column(name = "trytimes") 42 | private Integer tryTimes; 43 | 44 | @Column(name = "data") 45 | private String data; 46 | 47 | @Column(name = "status") 48 | private Integer status; 49 | 50 | public SendTask(Long providerId, String providerType, Long receiveUserId, String receiveUserType, String data) { 51 | this.setId(IdGetUtil.get()); 52 | this.providerId = providerId; 53 | this.providerType = providerType; 54 | this.receiveUserId = receiveUserId; 55 | this.receiveUserType = receiveUserType; 56 | this.status = StatusConstant.VALID; 57 | this.data = data; 58 | this.isCall = 1; 59 | } 60 | 61 | @Synchronized 62 | public int increaseTryTimes() { 63 | if (this.tryTimes == null) { 64 | this.tryTimes = 0; 65 | } 66 | this.tryTimes += 1; 67 | return this.tryTimes; 68 | } 69 | 70 | public String getUserKey() { 71 | return this.receiveUserId + "_" + this.receiveUserType; 72 | } 73 | } 74 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/bean/User.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.bean; 2 | 3 | import lombok.Data; 4 | 5 | import javax.persistence.Column; 6 | import javax.persistence.Table; 7 | 8 | /** 9 | * @author lanfangyi@haodf.com 10 | * @version 1.0 11 | * @since 2019/8/20 10:21 PM 12 | */ 13 | @Data 14 | @Table(name = "users") 15 | public class User extends BaseBean { 16 | 17 | public static String DEFAULT_HEAD_IMG = "http://pic24.nipic.com/20120922/10898738_143746326185_2.jpg"; 18 | 19 | /** 20 | * 用户名 21 | */ 22 | @Column(name = "name") 23 | private String name; 24 | 25 | /** 26 | * 描述，比如联系医助 27 | */ 28 | @Column(name = "password") 29 | private String password; 30 | 31 | /** 32 | * 状态 33 | */ 34 | @Column(name = "status") 35 | private Integer status; 36 | 37 | /** 38 | * 头像 39 | */ 40 | @Column(name = "headimg") 41 | private String headImg = DEFAULT_HEAD_IMG; 42 | } 43 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/config/NettyConfig.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.config; 2 | 3 | import io.netty.channel.group.ChannelGroup; 4 | import io.netty.channel.group.DefaultChannelGroup; 5 | import io.netty.util.concurrent.GlobalEventExecutor; 6 | 7 | /** 8 | * @author lanfangyi@haodf.com 9 | * @version 1.0 10 | * @since 2019/7/30 2:24 PM 11 | */ 12 | public class NettyConfig { 13 | 14 | //存储每一个客户端接入进来的对象 15 | public static ChannelGroup group = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE); 16 | } 17 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/config/SwaggerConfig.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.config; 2 | 3 | import org.springframework.context.annotation.Bean; 4 | import org.springframework.context.annotation.Configuration; 5 | import springfox.documentation.builders.ApiInfoBuilder; 6 | import springfox.documentation.builders.PathSelectors; 7 | import springfox.documentation.builders.RequestHandlerSelectors; 8 | import springfox.documentation.service.ApiInfo; 9 | import springfox.documentation.spi.DocumentationType; 10 | import springfox.documentation.spring.web.plugins.Docket; 11 | import springfox.documentation.swagger2.annotations.EnableSwagger2; 12 | 13 | /** 14 | * @author lanfangyi@haodf.com 15 | * @version 1.0 16 | * @since 2019/8/15 6:19 PM 17 | */ 18 | @Configuration 19 | @EnableSwagger2 20 | public class SwaggerConfig { 21 | 22 | @Bean 23 | public Docket createRestApi() { 24 | return new Docket(DocumentationType.SWAGGER_2) 25 | .apiInfo(apiInfo()) 26 | .select() 27 | .apis(RequestHandlerSelectors.basePackage("com.haodf.service.centerim.service")) 28 | .paths(PathSelectors.any()) 29 | .build(); 30 | } 31 | 32 | private ApiInfo apiInfo() { 33 | return new ApiInfoBuilder().title("Spring Boot中使用Swagger构建Rest Api").version("1.0").build(); 34 | } 35 | } -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/constants/ErrorCode.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.constants; 2 | 3 | 4 | import com.lanfangyi.nettyim.base.ErrorInfo; 5 | 6 | /** 7 | * @author lanfangyi@haodf.com 8 | * @version 1.0 9 | * @since 2019/7/31 2:50 PM 10 | */ 11 | public class ErrorCode { 12 | 13 | private ErrorCode() { 14 | } 15 | 16 | public static final ErrorInfo SYSTEM_ERROR = new ErrorInfo(1001, "系统错误"); 17 | 18 | public static final ErrorInfo DATABASE_ERROR = new ErrorInfo(1002, "数据库错误"); 19 | 20 | public static final ErrorInfo USERNAME_ALREDY_EXIST = new ErrorInfo(1003, "用户名已存在"); 21 | 22 | public static final ErrorInfo PASSWORD_NOT_CURRECT = new ErrorInfo(1004, "用户名或密码错误"); 23 | 24 | } 25 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/constants/StatusConstant.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.constants; 2 | 3 | /** 4 | * @author lanfangyi@haodf.com 5 | * @version 1.0 6 | * @since 2019/7/31 2:55 PM 7 | */ 8 | public class StatusConstant { 9 | 10 | public final static int VALID = 0; 11 | 12 | public final static int INVALID = 1; 13 | } 14 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/constants/SystemUserConstant.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.constants; 2 | 3 | /** 4 | * @author lanfangyi@haodf.com 5 | * @version 1.0 6 | * @since 2019/9/3 11:32 AM 7 | */ 8 | public class SystemUserConstant { 9 | 10 | public static final long SYSTEM_USERID = 6; 11 | 12 | public static final String SYSTEM_USER_PASSWORD = "admin"; 13 | 14 | public static final String SYSTEM_USER_TYPE = UserTypeConstant.SYSTEM_USER; 15 | 16 | public static final String SYSTEM_USER_HEADIMG = ""; 17 | } 18 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/constants/UserTypeConstant.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.constants; 2 | 3 | /** 4 | * @author lanfangyi@haodf.com 5 | * @version 1.0 6 | * @since 2019/9/2 5:10 PM 7 | */ 8 | public class UserTypeConstant { 9 | 10 | public static final String SYSTEM_USER = "system_user"; 11 | 12 | public static final String NOMAL_USER = "nomal_user"; 13 | 14 | } 15 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/controller/FriendController.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.controller; 2 | 3 | import com.lanfangyi.nettyim.base.IMResponse; 4 | import com.lanfangyi.nettyim.bean.Friend; 5 | import com.lanfangyi.nettyim.bean.User; 6 | import com.lanfangyi.nettyim.constants.ErrorCode; 7 | import com.lanfangyi.nettyim.constants.StatusConstant; 8 | import com.lanfangyi.nettyim.constants.UserTypeConstant; 9 | import com.lanfangyi.nettyim.mapper.FriendMapper; 10 | import com.lanfangyi.nettyim.mapper.UserMapper; 11 | import com.lanfangyi.nettyim.utils.IdGetUtil; 12 | import com.lanfangyi.service.paramcheck.annotation.Valid; 13 | import com.lanfangyi.service.paramcheck.annotation.activeannotation.NotBlank; 14 | import com.lanfangyi.service.paramcheck.annotation.activeannotation.NotNull; 15 | import lombok.extern.slf4j.Slf4j; 16 | import org.springframework.web.bind.annotation.PostMapping; 17 | import org.springframework.web.bind.annotation.RequestMapping; 18 | import org.springframework.web.bind.annotation.RestController; 19 | 20 | import javax.annotation.Resource; 21 | import java.util.Date; 22 | 23 | /** 24 | * @author lanfangyi@haodf.com 25 | * @version 1.0 26 | * @since 2019/8/20 10:44 PM 27 | */ 28 | @RestController 29 | @RequestMapping("/friend") 30 | @Slf4j 31 | public class FriendController { 32 | 33 | @Resource 34 | private FriendMapper friendMapper; 35 | 36 | /** 37 | * 添加好友 38 | * 39 | * @param userId 自己的ID 40 | * @param userType 自己的类型 41 | * @param friendUserId 好友的ID 42 | * @param friendUserType 好友类型 43 | * @return Friend 好友的信息 44 | */ 45 | @PostMapping("/addFriend") 46 | @Valid 47 | public IMResponse addFriend( 48 | @NotNull Long userId, 49 | @NotBlank(among = {UserTypeConstant.NOMAL_USER, UserTypeConstant.SYSTEM_USER}) String userType, 50 | @NotNull Long friendUserId, 51 | @NotBlank(among = {UserTypeConstant.NOMAL_USER, UserTypeConstant.SYSTEM_USER}) String friendUserType 52 | ) { 53 | Friend friend = new Friend(userId, userType, friendUserId, friendUserType); 54 | friendMapper.insert(friend); 55 | return IMResponse.success(friend); 56 | } 57 | 58 | } 59 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/controller/SendTaskController.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.controller; 2 | 3 | import com.lanfangyi.nettyim.base.IMResponse; 4 | import com.lanfangyi.nettyim.bean.SendTask; 5 | import com.lanfangyi.nettyim.constants.ErrorCode; 6 | import com.lanfangyi.nettyim.constants.SystemUserConstant; 7 | import com.lanfangyi.nettyim.constants.UserTypeConstant; 8 | import com.lanfangyi.nettyim.holder.ChannelHolder; 9 | import com.lanfangyi.nettyim.listener.AsyncSendMsgListener; 10 | import com.lanfangyi.nettyim.mapper.SendTaskMapper; 11 | import com.lanfangyi.service.paramcheck.annotation.Valid; 12 | import com.lanfangyi.service.paramcheck.annotation.activeannotation.NotBlank; 13 | import com.lanfangyi.service.paramcheck.aop.validate.ErrorLevelEnum; 14 | import io.netty.channel.Channel; 15 | import org.springframework.web.bind.annotation.PostMapping; 16 | import org.springframework.web.bind.annotation.RequestBody; 17 | import org.springframework.web.bind.annotation.RequestMapping; 18 | import org.springframework.web.bind.annotation.RestController; 19 | 20 | import javax.annotation.Resource; 21 | import javax.validation.constraints.Min; 22 | import java.util.ArrayList; 23 | import java.util.List; 24 | import java.util.Map; 25 | import java.util.concurrent.ConcurrentHashMap; 26 | 27 | /** 28 | * 所有发消息的动作都是一个推送任务，此controller专门处理推送任务 29 | * 30 | * @author lanfangyi@haodf.com 31 | * @version 1.0 32 | * @since 2019/8/28 6:24 PM 33 | */ 34 | @RestController 35 | @RequestMapping("/sendTask") 36 | public class SendTaskController { 37 | 38 | @Resource 39 | private SendTaskMapper sendTaskMapper; 40 | 41 | @Resource 42 | private AsyncSendMsgListener asyncSendMsgListener; 43 | 44 | /** 45 | * 创建推送任务 46 | * 47 | * @param providerId 消息发送者的ID 48 | * @param providerType 消息发送者的类型 49 | * @param receiveUserId 消息接受者的ID 50 | * @param receiveUserType 消息接收者的类型 51 | * @param dataJson 要推送的数据 52 | * @return 创建成功的ID 53 | */ 54 | @PostMapping("/createSendTask") 55 | @Valid(addErrLog = true, errLogLevel = ErrorLevelEnum.WARN) 56 | public IMResponse createSendTask(@Min(1) Long providerId, 57 | @NotBlank(among = {UserTypeConstant.NOMAL_USER, UserTypeConstant.SYSTEM_USER}) String providerType, 58 | @Min(1) Long receiveUserId, 59 | @NotBlank(among = {UserTypeConstant.NOMAL_USER, UserTypeConstant.SYSTEM_USER}) String receiveUserType, 60 | @RequestBody String dataJson) { 61 | 62 | //创建推送任务 63 | SendTask sendTask = new SendTask(providerId, providerType, receiveUserId, receiveUserType, dataJson); 64 | int insert = sendTaskMapper.insert(sendTask); 65 | if (insert == 1) { 66 | asyncSendMsgListener.send(sendTask.getId()); 67 | return IMResponse.success(sendTask.getId()); 68 | } 69 | return IMResponse.error(ErrorCode.DATABASE_ERROR, null); 70 | } 71 | 72 | /** 73 | * 给所有在线用户推送消息， 只有系统管理员有这个权限 74 | * 75 | * @param systemUserId 系统管理员的userId 76 | * @param systemUserPassword 系统管理员的密码 77 | * @param dataJson 发送的数据 78 | * @return 所创建的推送任务的ID集合 79 | */ 80 | @PostMapping("/sendMsgToAllUser") 81 | @Valid(addErrLog = true, errLogLevel = ErrorLevelEnum.WARN) 82 | public IMResponse> sendMsgToAllUser(@Min(1) Long systemUserId, 83 | @NotBlank String systemUserPassword, 84 | @RequestBody String dataJson) { 85 | if (systemUserId != SystemUserConstant.SYSTEM_USERID || !systemUserPassword.equals(SystemUserConstant.SYSTEM_USER_PASSWORD)) { 86 | return IMResponse.error(ErrorCode.PASSWORD_NOT_CURRECT, null); 87 | } 88 | List sendTaskIds = new ArrayList<>(); 89 | //获取所有在线用户的信息 90 | ConcurrentHashMap map = ChannelHolder.all(); 91 | for (Map.Entry entry : map.entrySet()) { 92 | SendTask sendTask = new SendTask(SystemUserConstant.SYSTEM_USERID, SystemUserConstant.SYSTEM_USER_PASSWORD, 93 | (Long) entry.getKey(), UserTypeConstant.NOMAL_USER, dataJson); 94 | int insert = sendTaskMapper.insert(sendTask); 95 | if (insert == 1) { 96 | asyncSendMsgListener.send(sendTask.getId()); 97 | sendTaskIds.add(sendTask.getId()); 98 | } 99 | } 100 | return IMResponse.success(sendTaskIds); 101 | } 102 | 103 | } 104 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/controller/UserController.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.controller; 2 | 3 | import com.alibaba.druid.support.json.JSONUtils; 4 | import com.lanfangyi.nettyim.base.IMResponse; 5 | import com.lanfangyi.nettyim.bean.User; 6 | import com.lanfangyi.nettyim.constants.ErrorCode; 7 | import com.lanfangyi.nettyim.constants.StatusConstant; 8 | import com.lanfangyi.nettyim.mapper.UserMapper; 9 | import com.lanfangyi.nettyim.utils.IdGetUtil; 10 | import com.lanfangyi.service.paramcheck.annotation.Valid; 11 | import com.lanfangyi.service.paramcheck.annotation.activeannotation.NotBlank; 12 | import lombok.extern.slf4j.Slf4j; 13 | import org.springframework.web.bind.annotation.PostMapping; 14 | import org.springframework.web.bind.annotation.RequestMapping; 15 | import org.springframework.web.bind.annotation.RestController; 16 | 17 | 18 | import javax.annotation.Resource; 19 | import java.util.Date; 20 | 21 | /** 22 | * @author lanfangyi@haodf.com 23 | * @version 1.0 24 | * @since 2019/8/20 10:44 PM 25 | */ 26 | @RestController 27 | @RequestMapping("/user") 28 | @Slf4j 29 | public class UserController { 30 | 31 | @Resource 32 | private UserMapper userMapper; 33 | 34 | @PostMapping("/registerOrLogin") 35 | @Valid 36 | public IMResponse registerOrLogin(@NotBlank String username, @NotBlank String password) { 37 | User userByUsername = userMapper.getUserByUsername(username); 38 | if (userByUsername != null) { 39 | if (userByUsername.getPassword().equals(password)) { 40 | return IMResponse.success(userByUsername); 41 | } else { 42 | return IMResponse.error(ErrorCode.PASSWORD_NOT_CURRECT, null); 43 | } 44 | } 45 | User user = new User(); 46 | user.setName(username); 47 | user.setPassword(password); 48 | user.setStatus(StatusConstant.VALID); 49 | userMapper.insert(user); 50 | 51 | log.info("Create user. user:{}", user); 52 | return IMResponse.success(user); 53 | } 54 | 55 | public static void main(String[] args) { 56 | System.out.println(Integer.MAX_VALUE); 57 | System.out.println(2147483647); 58 | } 59 | 60 | } 61 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/enums/MsgTypeEnum.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.enums; 2 | 3 | import lombok.Getter; 4 | import lombok.ToString; 5 | 6 | /** 7 | * @author lanfangyi@haodf.com 8 | * @version 1.0 9 | * @since 2019/9/4 1:00 PM 10 | */ 11 | @Getter 12 | @ToString 13 | public enum MsgTypeEnum { 14 | 15 | KEEP_ALIVE(1, "心跳包"); 16 | 17 | private int type; 18 | 19 | private String desc; 20 | 21 | MsgTypeEnum(int type, String desc) { 22 | this.type = type; 23 | this.desc = desc; 24 | } 25 | } 26 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/exception/SendMsgButChannelIsNullxception.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.exception; 2 | 3 | /** 4 | * @author lanfangyi@haodf.com 5 | * @version 1.0 6 | * @since 2019/7/31 1:10 PM 7 | */ 8 | public class SendMsgButChannelIsNullxception extends RuntimeException { 9 | 10 | public SendMsgButChannelIsNullxception() { 11 | super("准备发送消息，但是channel为null"); 12 | } 13 | 14 | public SendMsgButChannelIsNullxception(String message) { 15 | super(message); 16 | } 17 | 18 | public SendMsgButChannelIsNullxception(Throwable cause) { 19 | super(cause); 20 | } 21 | 22 | public SendMsgButChannelIsNullxception(String message, Throwable cause) { 23 | super(message, cause); 24 | } 25 | } 26 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/filter/DruidStatFilter.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.filter; 2 | 3 | import com.alibaba.druid.support.http.WebStatFilter; 4 | 5 | import javax.servlet.annotation.WebFilter; 6 | import javax.servlet.annotation.WebInitParam; 7 | 8 | /** 9 | * @author lanfangyi@haodf.com 10 | * @version 1.0 11 | * @since 2019/8/20 10:01 PM 12 | */ 13 | @WebFilter( 14 | filterName = "druidWebStatFilter", 15 | urlPatterns = "/*", 16 | initParams = { 17 | // 忽略的资源类型 18 | @WebInitParam(name = "exclusions", value = "*.js,*.gif,*.jpg,*.bmp,*.png,*.css,*.ico,/druid/*"), 19 | }) 20 | public class DruidStatFilter extends WebStatFilter { 21 | } -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/filter/DruidStatViewServlet.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.filter; 2 | 3 | import com.alibaba.druid.support.http.StatViewServlet; 4 | 5 | import javax.servlet.annotation.WebInitParam; 6 | import javax.servlet.annotation.WebServlet; 7 | 8 | /** 9 | * @author lanfangyi@haodf.com 10 | * @version 1.0 11 | * @since 2019/8/20 10:03 PM 12 | */ 13 | @WebServlet(urlPatterns = "/druid/*", initParams={ 14 | // IP白名单 (没有配置或者为空，则允许所有访问) 15 | @WebInitParam(name="allow",value=""), 16 | // IP黑名单 (存在共同时，deny优先于allow) 17 | @WebInitParam(name="deny",value="10.4.34.4"), 18 | // 用户名 19 | @WebInitParam(name="loginUsername",value="lanfangyi@haodf.com"), 20 | // 密码 21 | @WebInitParam(name="loginPassword",value="lanfangyi@haodf.com"), 22 | // 禁用HTML页面上的“Reset All”功能 23 | @WebInitParam(name="resetEnable",value="true") 24 | }) 25 | public class DruidStatViewServlet extends StatViewServlet{ 26 | } 27 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/future/SendMsgFuture.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.future; 2 | 3 | import com.alibaba.druid.support.json.JSONUtils; 4 | import com.lanfangyi.nettyim.bean.SendTask; 5 | import com.lanfangyi.nettyim.exception.SendMsgButChannelIsNullxception; 6 | import com.lanfangyi.nettyim.future.resp.SendMsgFutureResp; 7 | import com.lanfangyi.nettyim.utils.DateUtil; 8 | import io.netty.channel.Channel; 9 | import io.netty.channel.ChannelFuture; 10 | import io.netty.channel.ChannelFutureListener; 11 | import lombok.Data; 12 | import lombok.extern.slf4j.Slf4j; 13 | import org.springframework.http.HttpStatus; 14 | 15 | import java.util.concurrent.Callable; 16 | 17 | /** 18 | * @author lanfangyi@haodf.com 19 | * @version 1.0 20 | * @since 2019/9/2 5:57 PM 21 | */ 22 | @Data 23 | @Slf4j 24 | public class SendMsgFuture implements Callable { 25 | 26 | private long sleepTime; 27 | 28 | private HttpStatus sendStatus; 29 | 30 | /** 31 | * 要发送的json数据 32 | */ 33 | private SendTask sendTask; 34 | 35 | /** 36 | * 发送消息的channel集合 37 | */ 38 | private Channel channel; 39 | 40 | 41 | public SendMsgFuture(SendTask sendTask, Channel channel) { 42 | this.sendTask = sendTask; 43 | this.channel = channel; 44 | } 45 | 46 | /** 47 | * 推送消息的Future线程 48 | * 49 | * @return SendMsgFutureResp 50 | */ 51 | @Override 52 | public SendMsgFutureResp call() { 53 | if (this.channel == null) { 54 | throw new SendMsgButChannelIsNullxception(); 55 | } 56 | 57 | ChannelFuture channelFuture = channel.writeAndFlush(JSONUtils.toJSONString(this.sendTask)); 58 | channelFuture.addListener((ChannelFutureListener) channelFuture2 -> log.info("push message success， time：", DateUtil.getNow())); 59 | 60 | channelFuture.addListener(future -> { 61 | if (future.isSuccess()) { 62 | //将sendTask的状态置完成 63 | sendStatus = HttpStatus.OK; 64 | } else { 65 | sendStatus = HttpStatus.INTERNAL_SERVER_ERROR; 66 | } 67 | }); 68 | return SendMsgFutureResp.getDefaultSendMsgFutureResp().setCode(sendStatus); 69 | } 70 | } 71 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/future/resp/SendMsgFutureResp.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.future.resp; 2 | 3 | import org.springframework.http.HttpStatus; 4 | 5 | /** 6 | * @author lanfangyi@haodf.com 7 | * @version 1.0 8 | * @since 2019/9/2 5:59 PM 9 | */ 10 | public class SendMsgFutureResp { 11 | 12 | private HttpStatus code = HttpStatus.OK; 13 | 14 | private String msg = ""; 15 | 16 | private T data; 17 | 18 | public static SendMsgFutureResp getDefaultSendMsgFutureResp() { 19 | return new SendMsgFutureResp(); 20 | } 21 | 22 | public HttpStatus getCode() { 23 | return code; 24 | } 25 | 26 | public SendMsgFutureResp setCode(HttpStatus code) { 27 | this.code = code; 28 | return this; 29 | } 30 | 31 | public String getMsg() { 32 | return msg; 33 | } 34 | 35 | public SendMsgFutureResp setMsg(String msg) { 36 | this.msg = msg; 37 | return this; 38 | } 39 | 40 | public T getData() { 41 | return data; 42 | } 43 | 44 | public SendMsgFutureResp setData(T data) { 45 | this.data = data; 46 | return this; 47 | } 48 | } 49 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/handler/HeartBeatHandler.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.handler; 2 | 3 | import com.lanfangyi.nettyim.holder.ChannelHolder; 4 | import io.netty.channel.Channel; 5 | import io.netty.channel.ChannelHandlerContext; 6 | import io.netty.channel.ChannelInboundHandlerAdapter; 7 | import io.netty.handler.timeout.IdleState; 8 | import io.netty.handler.timeout.IdleStateEvent; 9 | import lombok.extern.slf4j.Slf4j; 10 | 11 | /** 12 | * 心跳支持 13 | * 14 | * @author lanfangyi@haodf.com 15 | * @version 1.0 16 | * @since 2019/9/20 3:35 PM 17 | */ 18 | @Slf4j 19 | public class HeartBeatHandler extends ChannelInboundHandlerAdapter { 20 | 21 | @Override 22 | public void userEventTriggered(ChannelHandlerContext ctx, Object evt) { 23 | // 判断evt是否是IdleStateEvent（用于触发用户事件，包含 读空闲/写空闲/读写空闲 ） 24 | if (evt instanceof IdleStateEvent) { 25 | // 强制类型转换 26 | IdleStateEvent event = (IdleStateEvent) evt; 27 | 28 | if (event.state() == IdleState.READER_IDLE) { 29 | log.info("进入读空闲..."); 30 | } else if (event.state() == IdleState.WRITER_IDLE) { 31 | log.info("进入写空闲..."); 32 | } else if (event.state() == IdleState.ALL_IDLE) { 33 | 34 | log.info("channel关闭前，users的数量为：" + ChannelHolder.getChannelMapSize()); 35 | 36 | Channel channel = ctx.channel(); 37 | // 关闭无用的channel，以防资源浪费 38 | channel.close(); 39 | log.info("channel关闭后，users的数量为：" + ChannelHolder.getChannelMapSize()); 40 | } 41 | } 42 | } 43 | } 44 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/handler/WebSocketHandler.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.handler; 2 | 3 | import com.lanfangyi.nettyim.config.NettyConfig; 4 | import com.lanfangyi.nettyim.holder.ChannelHolder; 5 | import com.lanfangyi.nettyim.utils.DateUtil; 6 | import io.netty.buffer.ByteBuf; 7 | import io.netty.buffer.Unpooled; 8 | import io.netty.channel.ChannelFuture; 9 | import io.netty.channel.ChannelFutureListener; 10 | import io.netty.channel.ChannelHandlerContext; 11 | import io.netty.channel.SimpleChannelInboundHandler; 12 | import io.netty.handler.codec.http.DefaultFullHttpResponse; 13 | import io.netty.handler.codec.http.FullHttpRequest; 14 | import io.netty.handler.codec.http.HttpResponseStatus; 15 | import io.netty.handler.codec.http.HttpVersion; 16 | import io.netty.handler.codec.http.websocketx.*; 17 | import io.netty.util.CharsetUtil; 18 | import lombok.extern.slf4j.Slf4j; 19 | 20 | /** 21 | * @author lanfangyi@haodf.com 22 | * @version 1.0 23 | * @since 2019/7/30 2:22 PM 24 | */ 25 | @Slf4j 26 | public class WebSocketHandler extends SimpleChannelInboundHandler { 27 | 28 | private WebSocketServerHandshaker webSocketServerHandshaker; 29 | private static final String WEB_SOCKET_URL = "ws://127.0.0.1:8081/ws"; 30 | 31 | /** 32 | * 客户端与服务端创建链接的时候调用 33 | * 34 | * @param context 35 | */ 36 | @Override 37 | public void channelActive(ChannelHandlerContext context) { 38 | NettyConfig.group.add(context.channel()); 39 | log.info("客户端建立连接，时间：{}", DateUtil.getNow()); 40 | log.info("channelid:" + context.channel().id().asLongText()); 41 | //将channel存入 42 | ChannelHolder.putChannelNoUserId(context.channel().id().toString(), context.channel()); 43 | } 44 | 45 | 46 | /** 47 | * 客户端与服务端断开连接的时候调用 48 | * 49 | * @param context 50 | */ 51 | @Override 52 | public void channelInactive(ChannelHandlerContext context) { 53 | NettyConfig.group.remove(context.channel()); 54 | log.info("客户端与服务端连接断开，时间：{}", DateUtil.getNow()); 55 | ChannelHolder.removeChannelNoUserId(context.channel().id().toString()); 56 | log.info("清除channel信息成功"); 57 | } 58 | 59 | /** 60 | * 服务端接收客户端发送过来的数据结束之后调用 61 | * 62 | * @param context 63 | */ 64 | @Override 65 | public void channelReadComplete(ChannelHandlerContext context) { 66 | log.info("服务器接读取完客户端的消息"); 67 | context.flush(); 68 | } 69 | 70 | /** 71 | * 出现异常的时候调用 72 | * 73 | * @param context 74 | * @param throwable 75 | */ 76 | @Override 77 | public void exceptionCaught(ChannelHandlerContext context, Throwable throwable) { 78 | log.info("捕捉到异常"); 79 | throwable.printStackTrace(); 80 | context.close(); 81 | } 82 | 83 | /** 84 | * 服务端处理客户端websocke请求的核心方法 85 | * 86 | * @param channelHandlerContext 87 | * @param o 88 | */ 89 | @Override 90 | protected void channelRead0(ChannelHandlerContext channelHandlerContext, Object o) { 91 | log.info("服务端处理客户端websocke请求的核心方法"); 92 | //处理客户端向服务端发起的http握手请求 93 | if (o instanceof FullHttpRequest) { 94 | handHttpRequest(channelHandlerContext, (FullHttpRequest) o); 95 | } else if (o instanceof WebSocketFrame) { 96 | //处理websocket链接业务 97 | handWebSocketFrame(channelHandlerContext, (WebSocketFrame) o); 98 | } 99 | } 100 | 101 | /** 102 | * 处理客户端与服务端之间的websocket业务 103 | * 104 | * @param context 105 | * @param webSocketFrame 106 | */ 107 | private void handWebSocketFrame(ChannelHandlerContext context, WebSocketFrame webSocketFrame) { 108 | if (webSocketFrame instanceof CloseWebSocketFrame) {//判断是否是关闭websocket的指令 109 | webSocketServerHandshaker.close(context.channel(), (CloseWebSocketFrame) webSocketFrame.retain()); 110 | } 111 | if (webSocketFrame instanceof PingWebSocketFrame) {//判断是否是ping消息 112 | context.channel().write(new PongWebSocketFrame(webSocketFrame.content().retain())); 113 | return; 114 | } 115 | if (!(webSocketFrame instanceof TextWebSocketFrame)) {//判断是否是文本消息 116 | log.info("不支持文本消息"); 117 | throw new RuntimeException(this.getClass().getName()); 118 | } 119 | //返回应答消息 120 | //获取客户端向服务端发送的消息 121 | String content = ((TextWebSocketFrame) webSocketFrame).text(); 122 | log.info("服务端收到客户端的消息：" + content); 123 | TextWebSocketFrame textWebSocketFrame = new TextWebSocketFrame(context.channel().id() + ":" + content); 124 | //服务端向每个连接上来的客户端发送消息 125 | NettyConfig.group.writeAndFlush(textWebSocketFrame); 126 | } 127 | 128 | 129 | /** 130 | * 处理客户端向服务端发起http握手请求业务 131 | * 132 | * @param context 133 | * @param fullHttpRequest 134 | */ 135 | private void handHttpRequest(ChannelHandlerContext context, FullHttpRequest fullHttpRequest) { 136 | if (!fullHttpRequest.decoderResult().isSuccess() || !("websocket".equals(fullHttpRequest.headers().get("Upgrade")))) {//判断是否http握手请求 137 | sendHttpResponse(context, fullHttpRequest, new DefaultFullHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.BAD_REQUEST)); 138 | return; 139 | } 140 | WebSocketServerHandshakerFactory webSocketServerHandshakerFactory = new WebSocketServerHandshakerFactory(WEB_SOCKET_URL, "websocket", false); 141 | webSocketServerHandshaker = webSocketServerHandshakerFactory.newHandshaker(fullHttpRequest); 142 | if (webSocketServerHandshaker == null) { 143 | WebSocketServerHandshakerFactory.sendUnsupportedVersionResponse(context.channel()); 144 | } else { 145 | webSocketServerHandshaker.handshake(context.channel(), fullHttpRequest); 146 | } 147 | } 148 | 149 | @Override 150 | public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception { 151 | if (evt instanceof WebSocketServerProtocolHandler.HandshakeComplete) { 152 | //完成握手 153 | log.info("握手完成"); 154 | } else { 155 | super.userEventTriggered(ctx, evt); 156 | } 157 | } 158 | 159 | /** 160 | * 服务端向客户端发送响应消息 161 | * 162 | * @param context 163 | * @param fullHttpRequest 164 | * @param defaultFullHttpResponse 165 | */ 166 | private void sendHttpResponse(ChannelHandlerContext context, FullHttpRequest fullHttpRequest, DefaultFullHttpResponse defaultFullHttpResponse) { 167 | if (defaultFullHttpResponse.status().code() != 200) { 168 | ByteBuf buf = Unpooled.copiedBuffer(defaultFullHttpResponse.status().toString(), CharsetUtil.UTF_8); 169 | defaultFullHttpResponse.content().writeBytes(buf); 170 | buf.release(); 171 | } 172 | //服务端向客户端发送数据 173 | ChannelFuture future = context.channel().writeAndFlush(defaultFullHttpResponse); 174 | if (defaultFullHttpResponse.status().code() != 200) { 175 | future.addListener(ChannelFutureListener.CLOSE); 176 | } 177 | } 178 | 179 | } 180 | 181 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/holder/ChannelHolder.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.holder; 2 | 3 | import io.netty.channel.Channel; 4 | import io.netty.channel.group.ChannelGroup; 5 | import io.netty.channel.group.DefaultChannelGroup; 6 | import io.netty.util.concurrent.GlobalEventExecutor; 7 | 8 | import java.util.Collection; 9 | import java.util.List; 10 | import java.util.concurrent.ConcurrentHashMap; 11 | import java.util.concurrent.CopyOnWriteArraySet; 12 | 13 | /** 14 | * 此类存储channel和用户的信息 15 | * 16 | * @author lanfangyi@haodf.com 17 | * @version 1.0 18 | * @since 2019/8/21 8:36 PM 19 | */ 20 | public class ChannelHolder { 21 | 22 | /** 23 | * key是userId。 24 | */ 25 | private static ConcurrentHashMap channelMap = new ConcurrentHashMap<>(100); 26 | 27 | private static ConcurrentHashMap channelNoUserId = new ConcurrentHashMap<>(100); 28 | 29 | public static boolean putAndOverWrite(String userKey, Channel channel) { 30 | return channelMap.put(userKey, channel) != null; 31 | } 32 | 33 | /** 34 | * 根据key拿到对应的channel 35 | * 36 | * @param userKey 用户的ID 37 | */ 38 | public static Channel getChannel(String userKey) { 39 | return channelMap.get(userKey); 40 | } 41 | 42 | public static List allChannel() { 43 | return (List) channelMap.values(); 44 | } 45 | 46 | public static ConcurrentHashMap all() { 47 | return channelMap; 48 | } 49 | 50 | public static boolean putChannelNoUserId(String channelId, Channel channel) { 51 | return channelNoUserId.put(channelId, channel) != null; 52 | } 53 | 54 | public static boolean removeChannelNoUserId(String channelId) { 55 | channelNoUserId.remove(channelId); 56 | return true; 57 | } 58 | 59 | public static int getChannelMapSize() { 60 | return channelMap.size(); 61 | } 62 | } 63 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/init/NettyIMServerInitializer.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.init; 2 | 3 | import com.lanfangyi.nettyim.handler.HeartBeatHandler; 4 | import com.lanfangyi.nettyim.handler.WebSocketHandler; 5 | import io.netty.channel.ChannelInitializer; 6 | import io.netty.channel.ChannelPipeline; 7 | import io.netty.channel.socket.SocketChannel; 8 | import io.netty.handler.codec.http.HttpObjectAggregator; 9 | import io.netty.handler.codec.http.HttpServerCodec; 10 | import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler; 11 | import io.netty.handler.stream.ChunkedWriteHandler; 12 | import io.netty.handler.timeout.IdleStateHandler; 13 | 14 | 15 | /** 16 | * @author lanfangyi@haodf.com 17 | * @version 1.0 18 | * @since 2019/8/2 10:38 PM 19 | */ 20 | public class NettyIMServerInitializer extends ChannelInitializer { 21 | 22 | @Override 23 | protected void initChannel(SocketChannel socketChannel) { 24 | //获取pipeline管道 25 | ChannelPipeline pipeline = socketChannel.pipeline(); 26 | 27 | // ====================== 支持http协议 开始 ====================== 28 | //配置http编解码器，因为WebSocket是基于http协议的应用层协议 29 | pipeline.addLast(new HttpServerCodec()); 30 | 31 | //对写大数据流的支持 32 | pipeline.addLast(new ChunkedWriteHandler()); 33 | 34 | // 对httpMessage进行聚合，聚合成FullHttpRequest或FullHttpResponse 至于聚合成哪个，由请求或者响应来决定 35 | // 几乎在netty中的编程，都会使用到此handler 36 | pipeline.addLast(new HttpObjectAggregator(1024*64)); 37 | // ====================== 支持http协议 结束 ====================== 38 | 39 | 40 | 41 | // ====================== 增加心跳支持 start TODO ====================== 42 | //参考文章：http://www.52im.net/thread-2697-1-1.html 43 | // 针对客户端，如果在1分钟时没有向服务端发送读写心跳(ALL)，则主动断开 44 | // 如果是读空闲或者写空闲，不处理 45 | pipeline.addLast(new IdleStateHandler(8, 10, 12)); 46 | // 自定义的空闲状态检测 47 | pipeline.addLast(new HeartBeatHandler()); 48 | // ====================== 增加心跳支持 end ====================== 49 | 50 | 51 | // ====================== 支持WebSocket协议 开始 ====================== 52 | /** 53 | * websocket 服务器处理的协议，用于指定给客户端连接访问的路由 : /ws 54 | * 本handler会帮你处理一些繁重的复杂的事 55 | * 会帮你处理握手动作： handshaking（close, ping, pong） ping + pong = 心跳 56 | * 对于websocket来讲，都是以frames进行传输的，不同的数据类型对应的frames也不同 57 | */ 58 | pipeline.addLast(new WebSocketServerProtocolHandler("/ws")); 59 | 60 | // 自定义的handler 61 | pipeline.addLast(new WebSocketHandler()); 62 | 63 | // ====================== 支持WebSocket协议 结束 ====================== 64 | 65 | } 66 | } 67 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/listener/AsyncSendMsgListener.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.listener; 2 | 3 | import com.lanfangyi.nettyim.bean.SendTask; 4 | import com.lanfangyi.nettyim.constants.StatusConstant; 5 | import com.lanfangyi.nettyim.holder.ChannelHolder; 6 | import com.lanfangyi.nettyim.mapper.SendTaskMapper; 7 | import com.lanfangyi.nettyim.service.SendMsgService; 8 | import lombok.extern.slf4j.Slf4j; 9 | import org.springframework.context.event.EventListener; 10 | import org.springframework.scheduling.annotation.Async; 11 | import org.springframework.scheduling.annotation.EnableAsync; 12 | import org.springframework.stereotype.Component; 13 | import org.springframework.util.StringUtils; 14 | 15 | import javax.annotation.Resource; 16 | 17 | /** 18 | * @author lanfangyi@haodf.com 19 | * @version 1.0 20 | * @since 2019/9/2 5:39 PM 21 | */ 22 | @Component 23 | @EnableAsync 24 | @Slf4j 25 | public class AsyncSendMsgListener { 26 | 27 | @Resource 28 | private SendTaskMapper sendTaskMapper; 29 | 30 | @Resource 31 | private SendMsgService sendMsgService; 32 | 33 | @EventListener 34 | @Async 35 | public void send(Long sendTaskId) { 36 | SendTask sendTask = sendTaskMapper.findById(sendTaskId); 37 | if (sendTask == null) { 38 | return; 39 | } 40 | //当前就在线，当即发送 41 | //判断用户是否在线 42 | if (judgeLogin(sendTask.getUserKey())) { 43 | String data = sendTask.getData(); 44 | //推送 45 | if (!StringUtils.isEmpty(data)) { 46 | boolean b = sendMsgService.sendMsg(sendTask); 47 | if (b) { 48 | sendTaskMapper.updateStatus(sendTaskId, StatusConstant.INVALID); 49 | } 50 | } 51 | } 52 | 53 | } 54 | 55 | /** 56 | * 判断用户的在线状态 57 | * @param userKey 用户ID和用户类型组成的字符串，用户的标示 58 | */ 59 | private boolean judgeLogin(String userKey) { 60 | //判断本机 61 | return ChannelHolder.getChannel(userKey) == null; 62 | } 63 | } 64 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/mapper/FriendMapper.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.mapper; 2 | 3 | import com.lanfangyi.nettyim.bean.Friend; 4 | import com.lanfangyi.nettyim.bean.User; 5 | import org.apache.ibatis.annotations.Param; 6 | 7 | 8 | /** 9 | * @author lanfangyi@haodf.com 10 | * @version 1.0 11 | * @since 2019/8/20 10:18 PM 12 | */ 13 | public interface FriendMapper { 14 | 15 | int insert(Friend friend); 16 | } 17 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/mapper/SendTaskMapper.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.mapper; 2 | 3 | import com.lanfangyi.nettyim.bean.SendTask; 4 | import com.lanfangyi.nettyim.bean.User; 5 | import org.apache.ibatis.annotations.Param; 6 | 7 | 8 | /** 9 | * @author lanfangyi@haodf.com 10 | * @version 1.0 11 | * @since 2019/8/20 10:18 PM 12 | */ 13 | public interface SendTaskMapper { 14 | 15 | int insert(SendTask sendTask); 16 | 17 | SendTask findById(Long id); 18 | 19 | Boolean updateStatus(Long id, Integer status); 20 | } 21 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/mapper/UserMapper.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.mapper; 2 | 3 | import com.lanfangyi.nettyim.bean.User; 4 | import org.apache.ibatis.annotations.Param; 5 | 6 | 7 | /** 8 | * @author lanfangyi@haodf.com 9 | * @version 1.0 10 | * @since 2019/8/20 10:18 PM 11 | */ 12 | public interface UserMapper { 13 | 14 | int insert(User user); 15 | 16 | Long getUserIdByUsername(@Param("username") String username); 17 | 18 | User getUserByUsername(@Param("username") String username); 19 | 20 | User getUserByUsername(@Param("username") String username, @Param("password") String password); 21 | } 22 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/pojo/Msg.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.pojo; 2 | 3 | import lombok.Data; 4 | 5 | /** 6 | * @author lanfangyi@haodf.com 7 | * @version 1.0 8 | * @since 2019/9/9 7:19 PM 9 | */ 10 | @Data 11 | public class Msg { 12 | 13 | private Integer msgType; 14 | 15 | private String content; 16 | } 17 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/pool/FutureTaskPool.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.pool; 2 | 3 | import com.google.common.util.concurrent.ListeningExecutorService; 4 | import com.google.common.util.concurrent.MoreExecutors; 5 | import lombok.extern.slf4j.Slf4j; 6 | 7 | import java.util.concurrent.BlockingQueue; 8 | import java.util.concurrent.ExecutorService; 9 | import java.util.concurrent.LinkedBlockingQueue; 10 | import java.util.concurrent.ThreadPoolExecutor; 11 | import java.util.concurrent.TimeUnit; 12 | 13 | /** 14 | * @author lanfangyi@haodf.com 15 | * @version 1.0 16 | * @since 2019/9/2 5:36 PM 17 | */ 18 | @Slf4j 19 | public class FutureTaskPool { 20 | 21 | /** 22 | * LinkedBlockingQueue（）或者（int i）: 23 | * 大小不固定的BlockingQueue，若其构造时指定大小，生成的BlockingQueue有大小限制，此处不指定大小， 24 | * 其大小有Integer.MAX_VALUE来决定。其所含的对象是FIFO顺序排序的。 25 | */ 26 | private static BlockingQueue bq = new LinkedBlockingQueue<>(); 27 | 28 | 29 | /** 30 | * ThreadPoolExecutor:创建自定义线程池，池中保存的线程数为3，允许最大的线程数为6 31 | */ 32 | private static ThreadPoolExecutor tpe = new ThreadPoolExecutor(3, 6, 50, TimeUnit.MILLISECONDS, bq); 33 | 34 | public static ListeningExecutorService getExecutorService() { 35 | return MoreExecutors.listeningDecorator(FutureTaskPool.getExecutorService()); 36 | } 37 | 38 | /** 39 | * 获取线程队列的大小 40 | */ 41 | public static BlockingQueue getBlockingQueue() { 42 | return bq; 43 | } 44 | 45 | public static boolean shutDown() { 46 | log.info("销毁线程池 start"); 47 | if (tpe != null && !tpe.isShutdown()) { 48 | tpe.shutdown(); 49 | log.info("销毁线程池 success"); 50 | } 51 | log.info("销毁线程池 end"); 52 | return true; 53 | } 54 | } 55 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/server/NettyIMServer.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.server; 2 | 3 | import com.lanfangyi.nettyim.init.NettyIMServerInitializer; 4 | import io.netty.bootstrap.ServerBootstrap; 5 | import io.netty.channel.ChannelFuture; 6 | import io.netty.channel.ChannelOption; 7 | import io.netty.channel.EventLoopGroup; 8 | import io.netty.channel.nio.NioEventLoopGroup; 9 | import io.netty.channel.socket.nio.NioServerSocketChannel; 10 | import org.slf4j.Logger; 11 | import org.slf4j.LoggerFactory; 12 | import org.springframework.beans.factory.annotation.Value; 13 | import org.springframework.stereotype.Component; 14 | 15 | import javax.annotation.PostConstruct; 16 | import javax.annotation.PreDestroy; 17 | import java.net.InetSocketAddress; 18 | 19 | /** 20 | * @author lanfangyi@haodf.com 21 | * @version 1.0 22 | * @since 2019/8/2 09:18 PM 23 | */ 24 | @Component 25 | public class NettyIMServer { 26 | 27 | private final static Logger LOGGER = LoggerFactory.getLogger(NettyIMServer.class); 28 | 29 | /** 30 | * 创建两个线程池 boss线程池中的线程是client，主要负责端口的监听；work线程池中的线程主要负责读写任务，即消息的收发； 31 | */ 32 | private EventLoopGroup boss = new NioEventLoopGroup(); 33 | private EventLoopGroup work = new NioEventLoopGroup(); 34 | 35 | @Value("${cim.server.port:8081}") 36 | private int nettyPort; 37 | 38 | /** 39 | * 启动netty服务端 40 | */ 41 | @PostConstruct 42 | public void startNettyServer() throws InterruptedException { 43 | //创建服务实例 44 | ServerBootstrap bootstrap = new ServerBootstrap() 45 | //设置工作线程组 46 | .group(boss, work) 47 | //设置channel类型 48 | .channel(NioServerSocketChannel.class) 49 | //设置服务器端口 50 | .localAddress(new InetSocketAddress(nettyPort)) 51 | //保持长连接 52 | .childOption(ChannelOption.SO_KEEPALIVE, true) 53 | //设置channel处理器 54 | .childHandler(new NettyIMServerInitializer()); 55 | 56 | //异步启动服务器，future的结果就是服务器启动结果 57 | ChannelFuture future = bootstrap.bind().sync(); 58 | if (future.isSuccess()) { 59 | LOGGER.info("启动 CenterIM server 成功"); 60 | } 61 | } 62 | 63 | /** 64 | * 当守护线程退出的时候销毁 65 | */ 66 | @PreDestroy 67 | public void destroy() { 68 | //安全的销毁两个工作线程池 69 | boss.shutdownGracefully().syncUninterruptibly(); 70 | work.shutdownGracefully().syncUninterruptibly(); 71 | LOGGER.info("关闭 CenterIM server 成功"); 72 | } 73 | 74 | } 75 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/service/SendMsgService.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.service; 2 | 3 | import com.alibaba.druid.support.json.JSONUtils; 4 | import com.google.common.util.concurrent.FutureCallback; 5 | import com.google.common.util.concurrent.Futures; 6 | import com.google.common.util.concurrent.ListenableFuture; 7 | import com.lanfangyi.nettyim.bean.SendTask; 8 | import com.lanfangyi.nettyim.future.SendMsgFuture; 9 | import com.lanfangyi.nettyim.future.resp.SendMsgFutureResp; 10 | import com.lanfangyi.nettyim.holder.ChannelHolder; 11 | import com.lanfangyi.nettyim.pool.FutureTaskPool; 12 | import com.lanfangyi.nettyim.utils.DateUtil; 13 | import io.netty.channel.Channel; 14 | import lombok.extern.slf4j.Slf4j; 15 | import org.springframework.stereotype.Service; 16 | import org.springframework.util.StringUtils; 17 | 18 | /** 19 | * @author lanfangyi@haodf.com 20 | * @version 1.0 21 | * @since 2019/9/2 5:44 PM 22 | */ 23 | @Service 24 | @Slf4j 25 | public class SendMsgService { 26 | 27 | public boolean sendMsg(SendTask sendTask) { 28 | if (null == sendTask) { 29 | return false; 30 | } 31 | if (!StringUtils.isEmpty(sendTask.getData())) { 32 | //拿到所有路由信息 33 | Channel channel = ChannelHolder.getChannel(sendTask.getUserKey()); 34 | 35 | if (channel == null || !channel.isActive()) { 36 | log.error("通道不可用，无法推送消息. sendTask：{}", JSONUtils.toJSONString(sendTask)); 37 | return false; 38 | } 39 | 40 | //给尝试次数加一 41 | sendTask.increaseTryTimes(); 42 | //创建发送线程 43 | SendMsgFuture sendMsgFuture = new SendMsgFuture(sendTask, channel); 44 | //进入线程队列, 执行发送任务 45 | ListenableFuture future = FutureTaskPool.getExecutorService().submit(sendMsgFuture); 46 | 47 | Futures.addCallback(future, new FutureCallback() { 48 | @Override 49 | public void onSuccess(SendMsgFutureResp sendMsgFutureResp) { 50 | log.info("Send message success. time:{}", DateUtil.getNow()); 51 | } 52 | 53 | @Override 54 | public void onFailure(Throwable throwable) { 55 | // TODO: 2019/9/20 加入重试线程池 56 | log.error("Send message fail. time:{}, exception:{}", DateUtil.getNow(), JSONUtils.toJSONString(throwable.getCause())); 57 | } 58 | }); 59 | return true; 60 | } 61 | return false; 62 | } 63 | } 64 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/service/UserService.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.service; 2 | 3 | /** 4 | * @author lanfangyi@haodf.com 5 | * @version 1.0 6 | * @since 2019/8/21 11:06 PM 7 | */ 8 | public interface UserService { 9 | 10 | Long getUserByUsername(String username); 11 | } 12 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/service/impl/UserServiceImpl.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.service.impl; 2 | 3 | import com.lanfangyi.nettyim.mapper.UserMapper; 4 | import com.lanfangyi.nettyim.service.UserService; 5 | import org.springframework.stereotype.Service; 6 | 7 | import javax.annotation.Resource; 8 | 9 | /** 10 | * @author lanfangyi@haodf.com 11 | * @version 1.0 12 | * @since 2019/8/21 11:06 PM 13 | */ 14 | @Service("userService") 15 | public class UserServiceImpl implements UserService { 16 | 17 | @Resource 18 | private UserMapper userMapper; 19 | 20 | @Override 21 | public Long getUserByUsername(String username) { 22 | return null; 23 | } 24 | } 25 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/utils/DataCheckUtil.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.utils; 2 | 3 | import org.springframework.util.StringUtils; 4 | 5 | public class DataCheckUtil { 6 | 7 | public boolean check(String data){ 8 | boolean flg = false; 9 | if (!StringUtils.isEmpty(data)) { 10 | //校验数据是否是json格式 11 | 12 | } 13 | return flg; 14 | } 15 | } 16 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/utils/DateUtil.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.utils; 2 | 3 | import org.apache.commons.lang3.time.DateFormatUtils; 4 | 5 | import java.text.SimpleDateFormat; 6 | 7 | /** 8 | * @author lanfangyi@haodf.com 9 | * @version 1.0 10 | * @since 2019/7/31 12:22 PM 11 | */ 12 | public class DateUtil { 13 | 14 | private final static SimpleDateFormat sdfYear = new SimpleDateFormat("yyyy"); 15 | 16 | private final static SimpleDateFormat sdfDay = new SimpleDateFormat( 17 | "yyyy-MM-dd"); 18 | 19 | private final static SimpleDateFormat sdfDays = new SimpleDateFormat( 20 | "yyyyMMdd"); 21 | 22 | private final static SimpleDateFormat sdfTime = new SimpleDateFormat( 23 | "yyyy-MM-dd HH:mm:ss"); 24 | 25 | public static String getNow() { 26 | return DateFormatUtils.format(System.currentTimeMillis(), "yyyy-MM-dd HH:mm:ss"); 27 | } 28 | 29 | } 30 | 31 | 32 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/utils/IdGetUtil.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.utils; 2 | 3 | import java.util.concurrent.atomic.AtomicLong; 4 | 5 | /** 6 | * @author lanfangyi@haodf.com 7 | * @version 1.0 8 | * @since 2019/8/4 12:39 AM 9 | */ 10 | public class IdGetUtil { 11 | 12 | private static AtomicLong atomicLong = new AtomicLong(); 13 | 14 | private static long getRandomNum() { 15 | long randomNum = atomicLong.getAndIncrement(); 16 | if (randomNum == Integer.MAX_VALUE) { 17 | atomicLong.set(0); 18 | } 19 | return randomNum; 20 | } 21 | 22 | /** 23 | * 最大并发：Integer.MAX_VALUE 24 | */ 25 | public static Long get() { 26 | return Long.valueOf(System.currentTimeMillis() + "" + getRandomNum()); 27 | } 28 | } 29 | -------------------------------------------------------------------------------- /src/main/java/com/lanfangyi/nettyim/utils/UUIDUtil.java: -------------------------------------------------------------------------------- 1 | package com.lanfangyi.nettyim.utils; 2 | 3 | import java.util.ArrayList; 4 | import java.util.List; 5 | import java.util.UUID; 6 | 7 | /** 8 | * @author lanfangyi@haodf.com 9 | * @version 1.0 10 | * @since 2019/8/1 2:37 PM 11 | */ 12 | public class UUIDUtil { 13 | 14 | public static String get() { 15 | String uuid = UUID.randomUUID().toString(); 16 | //去掉“-”符号 17 | return uuid.replaceAll("-", ""); 18 | } 19 | 20 | public static List get(int num) { 21 | List uuidList = new ArrayList<>(); 22 | if (num > 0) { 23 | for (int i = 0; i < num; i++) { 24 | uuidList.add(get()); 25 | } 26 | } 27 | return uuidList; 28 | } 29 | } 30 | -------------------------------------------------------------------------------- /src/main/resources/application-dev.yml: -------------------------------------------------------------------------------- 1 | spring: 2 | #配置数据源 3 | datasource: 4 | url: jdbc:mysql://127.0.0.1:3306/centerim_avatar?autoReconnect=true&zeroDateTimeBehavior=convertToNull&useSSL=true 5 | username: root 6 | password: 176318surelan 7 | driver-class-name: com.mysql.cj.jdbc.Driver 8 | # driver-class-name: com.mysql.jdbc.Driver 9 | type: com.alibaba.druid.pool.DruidDataSource 10 | initialSize: 3 11 | minIdle: 2 12 | maxActive: 5 13 | maxWait: 60000 14 | timeBetweenEvictionRunsMillis: 60000 15 | minEvictableIdleTimeMillis: 30000 16 | validationQuery: SELECT 1 FROM DUAL 17 | testWhileIdle: true 18 | testOnBorrow: false 19 | #exceptionSorter: true 20 | testOnReturn: false 21 | poolPreparedStatements: true 22 | filters: stat #,wall,sl4j 23 | maxPoolPreparedStatementPerConnectionSize: 20 24 | connectionProperties: druid.stat.mergeSql=true;druid.stat.slowSqlMillis=500 25 | useGlobalDataSourceStat: true 26 | server: 27 | port: 8080 28 | 29 | mybatis: 30 | typeAliasesPackage: com.lanfangyi.nettyim.bean #这里配置实体对应的包路径 31 | mapperLocations: classpath:mapper/*.xml 32 | configuration: 33 | log-impl: org.apache.ibatis.logging.stdout.StdOutImpl #控制打印sql语句 34 | 35 | # druid后台地址：http://localhost:8080/druid/datasource.html 神奇，只有谷歌浏览器能打开这个页面 http://localhost:8080/druid/index.html 36 | # 账号密码均为：lanfangyi@haodf.com 37 | 38 | 39 | 40 | -------------------------------------------------------------------------------- /src/main/resources/application.yml: -------------------------------------------------------------------------------- 1 | spring: 2 | profiles: 3 | active: dev -------------------------------------------------------------------------------- /src/main/resources/doc/table.sql: -------------------------------------------------------------------------------- 1 | use centerim_avatar; 2 | 3 | -- drop table callbacks; 4 | -- CREATE TABLE `callbacks` ( 5 | -- `id` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 6 | -- `sendtaskid` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 7 | -- `systemcode` varchar(100) NOT NULL DEFAULT '', 8 | -- `systemtype` varchar(20) NOT NULL DEFAULT '', 9 | -- `method` varchar(100) NOT NULL DEFAULT '', 10 | -- `cancallagain` int(2) UNSIGNED NOT NULL DEFAULT 0, 11 | -- `methodparams` json NULL, 12 | -- `status` int(2) UNSIGNED NOT NULL DEFAULT 0, 13 | -- `ctime` timestamp NOT NULL , 14 | -- `utime` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, 15 | -- PRIMARY KEY (`id`) 16 | -- ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; 17 | 18 | drop table sendtasks; 19 | CREATE TABLE `sendtasks` ( 20 | `id` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 21 | `providerid` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 22 | `providertype` varchar(50) NOT NULL DEFAULT '', 23 | `receiveuserid` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 24 | `receiveusertype` varchar(50) NOT NULL DEFAULT '', 25 | `iscall` int(2) UNSIGNED NOT NULL DEFAULT 0, 26 | `trytimes` int(200) UNSIGNED NOT NULL DEFAULT 0, 27 | `status` int(2) UNSIGNED NOT NULL DEFAULT 0, 28 | `data` json NULL, 29 | `ctime` timestamp NOT NULL , 30 | `utime` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, 31 | PRIMARY KEY (`id`) 32 | ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; 33 | 34 | drop table logininfos; 35 | CREATE TABLE `logininfos` ( 36 | `id` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 37 | `userid` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 38 | `status` int(2) UNSIGNED NOT NULL DEFAULT 0, 39 | `islogin` int(2) UNSIGNED NOT NULL DEFAULT 0, 40 | `lastlogintime` timestamp NOT NULL , 41 | `lastlogouttime` timestamp NOT NULL , 42 | `routeurl` varchar(500) NOT NULL DEFAULT '', 43 | `ctime` timestamp NOT NULL , 44 | `utime` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, 45 | PRIMARY KEY (`id`) 46 | ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; 47 | 48 | -- drop table businesses; 49 | -- CREATE TABLE `businesses` ( 50 | -- `id` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 51 | -- `name` varchar(50) NOT NULL DEFAULT '', 52 | -- `describe` varchar(500) NOT NULL DEFAULT '', 53 | -- `ctime` timestamp NOT NULL , 54 | -- `utime` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, 55 | -- PRIMARY KEY (`id`) 56 | -- ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; 57 | 58 | drop table tasklogs; 59 | CREATE TABLE `tasklogs` ( 60 | `id` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 61 | `taskid` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 62 | `content` varchar(500) NOT NULL DEFAULT '', 63 | `ctime` timestamp NOT NULL , 64 | `utime` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, 65 | PRIMARY KEY (`id`) 66 | ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; 67 | 68 | drop table users; 69 | CREATE TABLE `users` ( 70 | `id` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 71 | `name` varchar(50) NOT NULL DEFAULT '', 72 | `usertype` varchar(50) NOT NULL DEFAULT '', 73 | `headimg` varchar(500) NOT NULL DEFAULT '', 74 | `password` varchar(50) NOT NULL DEFAULT '', 75 | `status` int(2) UNSIGNED NOT NULL DEFAULT 0, 76 | `ctime` timestamp NOT NULL , 77 | `utime` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, 78 | PRIMARY KEY (`id`) 79 | ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; 80 | 81 | drop table friends; 82 | CREATE TABLE `friends` ( 83 | `id` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 84 | `userid` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 85 | `usertype` varchar(50) NOT NULL DEFAULT '', 86 | `frienduserid` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 87 | `friendusertype` varchar(50) NOT NULL DEFAULT '', 88 | `status` int(2) UNSIGNED NOT NULL DEFAULT 0, 89 | `ctime` timestamp NOT NULL , 90 | `utime` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, 91 | PRIMARY KEY (`id`) 92 | ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; 93 | 94 | -- drop table userloginlogs; 95 | -- CREATE TABLE `userloginlogs` ( 96 | -- `id` bigint(20) UNSIGNED NOT NULL DEFAULT 0, 97 | -- `userId` varchar(50) NOT NULL DEFAULT '', 98 | -- `openId` varchar(200) NOT NULL DEFAULT '', 99 | -- `cid` varchar(200) NOT NULL DEFAULT '', 100 | -- `status` int(2) UNSIGNED NOT NULL DEFAULT 0, 101 | -- `ctime` timestamp NOT NULL , 102 | -- `utime` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, 103 | -- PRIMARY KEY (`id`) 104 | -- ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; -------------------------------------------------------------------------------- /src/main/resources/doc/todo: -------------------------------------------------------------------------------- 1 | 1、可以接入apollo，使得程序配置更为灵活。 2 | 2、增加后台功能。管理所有在线用户，拉黑屏蔽等。 3 | -------------------------------------------------------------------------------- /src/main/resources/log4j.properties: -------------------------------------------------------------------------------- 1 | #config root logger 2 | log4j.rootLogger = INFO,system.out 3 | log4j.appender.system.out=org.apache.log4j.ConsoleAppender 4 | log4j.appender.system.out.layout=org.apache.log4j.PatternLayout 5 | log4j.appender.system.out.layout.ConversionPattern=[Log] %5p[%F:%L]:%m%n 6 | 7 | #config this Project.file logger 8 | log4j.logger.thisProject.file=INFO,thisProject.file.out 9 | log4j.appender.thisProject.file.out=org.apache.log4j.DailyRollingFileAppender 10 | log4j.appender.thisProject.file.out.File=logContentFile.log 11 | log4j.appender.thisProject.file.out.layout=org.apache.log4j.PatternLayout 12 | 13 | -------------------------------------------------------------------------------- /src/main/resources/mapper/FriendMapper.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | id,userid,usertype,frienduserid,friendusertype,status,ctime,utime 6 | 7 | 8 | 9 | insert into friends() 10 | values(#{id}, #{userid}, #{usertype}, #{frienduserid},#{friendusertype},#{status},#{cTime},#{uTime}) 11 | 12 | 13 | -------------------------------------------------------------------------------- /src/main/resources/mapper/SendTaskMapper.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | id,providerid,providertype,receiveuserid,receiveusertype,iscall,srcids,status,data,ctime,utime 6 | 7 | 8 | 13 | 14 | 15 | insert into sendtasks() 16 | values(#{id}, #{providerid}, #{providertype}, #{receiveuserid},#{receiveusertype}, 17 | #{iscall},#{status},#{data},#{cTime},#{uTime}) 18 | 19 | 20 | 21 | delete from sendtasks 22 | where id in 23 | 24 | #{id} 25 | 26 | 27 | -------------------------------------------------------------------------------- /src/main/resources/mapper/UserMapper.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | id,name,password,status,headImg,ctime,utime 6 | 7 | 8 | 13 | 14 | 15 | insert into users() 16 | values(#{id}, #{name}, #{password}, #{status},#{headImg},#{cTime},#{uTime}) 17 | 18 | 19 | 20 | delete from users 21 | where id in 22 | 23 | #{id} 24 | 25 | 26 | 27 | 30 | 31 | 34 | 35 | -------------------------------------------------------------------------------- /src/main/resources/templates/message.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | Title 6 | 7 | 8 | 9 | 10 | 11 | 12 | 75 | 76 | 77 | -------------------------------------------------------------------------------- /src/main/resources/templates/message2.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | Title 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 | 菜鸟教程(runoob.com) 16 | 17 | 54 | 55 | 56 | 57 | 58 |
59 | 运行 WebSocket 60 |
61 | 62 | 63 | 64 | 65 | 66 | -------------------------------------------------------------------------------- /src/test/java/com/lanfangyi/nettyim/NettyImApplicationTests.java: -------------------------------------------------------------------------------- 1 | //package com.lanfangyi.nettyim; 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 NettyImApplicationTests { 11 | // 12 | // @Test 13 | // public void contextLoads() { 14 | // } 15 | // 16 | //} 17 | --------------------------------------------------------------------------------