├── .gitignore ├── .mvn └── wrapper │ ├── MavenWrapperDownloader.java │ ├── maven-wrapper.jar │ └── maven-wrapper.properties ├── README.md ├── mvnw ├── mvnw.cmd ├── pom.xml └── src ├── main ├── java │ └── org │ │ └── hongxi │ │ └── sample │ │ └── webflux │ │ ├── SampleApplication.java │ │ ├── controller │ │ ├── HealthController.java │ │ └── OrderController.java │ │ ├── dao │ │ ├── InMemoryOrderRepository.java │ │ └── OrderRepository.java │ │ ├── exception │ │ ├── BusinessException.java │ │ ├── DefaultExceptionHandler.java │ │ └── ExceptionHandlerConfiguration.java │ │ ├── filter │ │ ├── AccessFilter.java │ │ ├── ModifyBodyFilter.java │ │ ├── MonitorFilter.java │ │ ├── SessionAuthFilter.java │ │ └── SessionFilter.java │ │ ├── model │ │ └── Order.java │ │ ├── router │ │ ├── RouterConfiguration.java │ │ └── handler │ │ │ ├── OrderHandler.java │ │ │ └── ServerRedirectHandler.java │ │ └── support │ │ ├── Crypto.java │ │ ├── JacksonUtils.java │ │ ├── ModifiedServerHttpRequest.java │ │ ├── ModifiedServerHttpResponse.java │ │ ├── ModifiedServerWebExchange.java │ │ ├── ParamUtils.java │ │ ├── SessionContext.java │ │ ├── WebConstants.java │ │ └── WebUtils.java └── resources │ └── application.yml └── test └── java └── org └── hongxi └── sample ├── reactor ├── FluxTests.java └── ReactiveContextTest.java └── webflux └── SampleApplicationTests.java /.gitignore: -------------------------------------------------------------------------------- 1 | target/ 2 | !.mvn/wrapper/maven-wrapper.jar 3 | !**/src/main/**/target/ 4 | !**/src/test/**/target/ 5 | 6 | ### STS ### 7 | .apt_generated 8 | .classpath 9 | .factorypath 10 | .project 11 | .settings 12 | .springBeans 13 | .sts4-cache 14 | 15 | ### IntelliJ IDEA ### 16 | .idea 17 | *.iws 18 | *.iml 19 | *.ipr 20 | 21 | ### NetBeans ### 22 | /nbproject/private/ 23 | /nbbuild/ 24 | /dist/ 25 | /nbdist/ 26 | /.nb-gradle/ 27 | build/ 28 | !**/src/main/**/build/ 29 | !**/src/test/**/build/ 30 | 31 | ### VS Code ### 32 | .vscode/ 33 | -------------------------------------------------------------------------------- /.mvn/wrapper/MavenWrapperDownloader.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2007-present the original author or authors. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"); 5 | * you may not use this file except in compliance with the License. 6 | * You may obtain a copy of the License at 7 | * 8 | * https://www.apache.org/licenses/LICENSE-2.0 9 | * 10 | * Unless required by applicable law or agreed to in writing, software 11 | * distributed under the License is distributed on an "AS IS" BASIS, 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 | * See the License for the specific language governing permissions and 14 | * limitations under the License. 15 | */ 16 | import java.net.*; 17 | import java.io.*; 18 | import java.nio.channels.*; 19 | import java.util.Properties; 20 | 21 | public class MavenWrapperDownloader { 22 | 23 | private static final String WRAPPER_VERSION = "0.5.6"; 24 | /** 25 | * Default URL to download the maven-wrapper.jar from, if no 'downloadUrl' is provided. 26 | */ 27 | private static final String DEFAULT_DOWNLOAD_URL = "https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/" 28 | + WRAPPER_VERSION + "/maven-wrapper-" + WRAPPER_VERSION + ".jar"; 29 | 30 | /** 31 | * Path to the maven-wrapper.properties file, which might contain a downloadUrl property to 32 | * use instead of the default one. 33 | */ 34 | private static final String MAVEN_WRAPPER_PROPERTIES_PATH = 35 | ".mvn/wrapper/maven-wrapper.properties"; 36 | 37 | /** 38 | * Path where the maven-wrapper.jar will be saved to. 39 | */ 40 | private static final String MAVEN_WRAPPER_JAR_PATH = 41 | ".mvn/wrapper/maven-wrapper.jar"; 42 | 43 | /** 44 | * Name of the property which should be used to override the default download url for the wrapper. 45 | */ 46 | private static final String PROPERTY_NAME_WRAPPER_URL = "wrapperUrl"; 47 | 48 | public static void main(String args[]) { 49 | System.out.println("- Downloader started"); 50 | File baseDirectory = new File(args[0]); 51 | System.out.println("- Using base directory: " + baseDirectory.getAbsolutePath()); 52 | 53 | // If the maven-wrapper.properties exists, read it and check if it contains a custom 54 | // wrapperUrl parameter. 55 | File mavenWrapperPropertyFile = new File(baseDirectory, MAVEN_WRAPPER_PROPERTIES_PATH); 56 | String url = DEFAULT_DOWNLOAD_URL; 57 | if(mavenWrapperPropertyFile.exists()) { 58 | FileInputStream mavenWrapperPropertyFileInputStream = null; 59 | try { 60 | mavenWrapperPropertyFileInputStream = new FileInputStream(mavenWrapperPropertyFile); 61 | Properties mavenWrapperProperties = new Properties(); 62 | mavenWrapperProperties.load(mavenWrapperPropertyFileInputStream); 63 | url = mavenWrapperProperties.getProperty(PROPERTY_NAME_WRAPPER_URL, url); 64 | } catch (IOException e) { 65 | System.out.println("- ERROR loading '" + MAVEN_WRAPPER_PROPERTIES_PATH + "'"); 66 | } finally { 67 | try { 68 | if(mavenWrapperPropertyFileInputStream != null) { 69 | mavenWrapperPropertyFileInputStream.close(); 70 | } 71 | } catch (IOException e) { 72 | // Ignore ... 73 | } 74 | } 75 | } 76 | System.out.println("- Downloading from: " + url); 77 | 78 | File outputFile = new File(baseDirectory.getAbsolutePath(), MAVEN_WRAPPER_JAR_PATH); 79 | if(!outputFile.getParentFile().exists()) { 80 | if(!outputFile.getParentFile().mkdirs()) { 81 | System.out.println( 82 | "- ERROR creating output directory '" + outputFile.getParentFile().getAbsolutePath() + "'"); 83 | } 84 | } 85 | System.out.println("- Downloading to: " + outputFile.getAbsolutePath()); 86 | try { 87 | downloadFileFromURL(url, outputFile); 88 | System.out.println("Done"); 89 | System.exit(0); 90 | } catch (Throwable e) { 91 | System.out.println("- Error downloading"); 92 | e.printStackTrace(); 93 | System.exit(1); 94 | } 95 | } 96 | 97 | private static void downloadFileFromURL(String urlString, File destination) throws Exception { 98 | if (System.getenv("MVNW_USERNAME") != null && System.getenv("MVNW_PASSWORD") != null) { 99 | String username = System.getenv("MVNW_USERNAME"); 100 | char[] password = System.getenv("MVNW_PASSWORD").toCharArray(); 101 | Authenticator.setDefault(new Authenticator() { 102 | @Override 103 | protected PasswordAuthentication getPasswordAuthentication() { 104 | return new PasswordAuthentication(username, password); 105 | } 106 | }); 107 | } 108 | URL website = new URL(urlString); 109 | ReadableByteChannel rbc; 110 | rbc = Channels.newChannel(website.openStream()); 111 | FileOutputStream fos = new FileOutputStream(destination); 112 | fos.getChannel().transferFrom(rbc, 0, Long.MAX_VALUE); 113 | fos.close(); 114 | rbc.close(); 115 | } 116 | 117 | } 118 | -------------------------------------------------------------------------------- /.mvn/wrapper/maven-wrapper.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whatsmars/webflux-sample/e045208a1cecb7e7215fc7859ef4605242f7d6bb/.mvn/wrapper/maven-wrapper.jar -------------------------------------------------------------------------------- /.mvn/wrapper/maven-wrapper.properties: -------------------------------------------------------------------------------- 1 | distributionUrl=https://repo.maven.apache.org/maven2/org/apache/maven/apache-maven/3.8.1/apache-maven-3.8.1-bin.zip 2 | wrapperUrl=https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # WebFlux Sample 2 | ## Concurrency Model 3 | Both Spring MVC and Spring WebFlux support annotated controllers, but there is a key difference in the concurrency model and the default assumptions for blocking and threads. 4 | 5 | In Spring MVC (and servlet applications in general), it is assumed that applications can block the current thread, (for example, for remote calls). For this reason, servlet containers use a large thread pool to absorb potential blocking during request handling. 6 | 7 | In Spring WebFlux (and non-blocking servers in general), it is assumed that applications do not block. Therefore, non-blocking servers use a small, fixed-size thread pool (event loop workers) to handle requests. 8 | 9 | ``` 10 | “To scale” and “small number of threads” may sound contradictory but to never block the 11 | current thread (and rely on callbacks instead) means that you do not need extra threads, 12 | as there are no blocking calls to absorb. 13 | ``` 14 | 15 | © [hongxi.org](http://hongxi.org) | [web.hongxi.org](http://web.hongxi.org) -------------------------------------------------------------------------------- /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 | # Maven 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 | fi 118 | 119 | if [ -z "$JAVA_HOME" ]; then 120 | javaExecutable="`which javac`" 121 | if [ -n "$javaExecutable" ] && ! [ "`expr \"$javaExecutable\" : '\([^ ]*\)'`" = "no" ]; then 122 | # readlink(1) is not available as standard on Solaris 10. 123 | readLink=`which readlink` 124 | if [ ! `expr "$readLink" : '\([^ ]*\)'` = "no" ]; then 125 | if $darwin ; then 126 | javaHome="`dirname \"$javaExecutable\"`" 127 | javaExecutable="`cd \"$javaHome\" && pwd -P`/javac" 128 | else 129 | javaExecutable="`readlink -f \"$javaExecutable\"`" 130 | fi 131 | javaHome="`dirname \"$javaExecutable\"`" 132 | javaHome=`expr "$javaHome" : '\(.*\)/bin'` 133 | JAVA_HOME="$javaHome" 134 | export JAVA_HOME 135 | fi 136 | fi 137 | fi 138 | 139 | if [ -z "$JAVACMD" ] ; then 140 | if [ -n "$JAVA_HOME" ] ; then 141 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then 142 | # IBM's JDK on AIX uses strange locations for the executables 143 | JAVACMD="$JAVA_HOME/jre/sh/java" 144 | else 145 | JAVACMD="$JAVA_HOME/bin/java" 146 | fi 147 | else 148 | JAVACMD="`which java`" 149 | fi 150 | fi 151 | 152 | if [ ! -x "$JAVACMD" ] ; then 153 | echo "Error: JAVA_HOME is not defined correctly." >&2 154 | echo " We cannot execute $JAVACMD" >&2 155 | exit 1 156 | fi 157 | 158 | if [ -z "$JAVA_HOME" ] ; then 159 | echo "Warning: JAVA_HOME environment variable is not set." 160 | fi 161 | 162 | CLASSWORLDS_LAUNCHER=org.codehaus.plexus.classworlds.launcher.Launcher 163 | 164 | # traverses directory structure from process work directory to filesystem root 165 | # first directory with .mvn subdirectory is considered project base directory 166 | find_maven_basedir() { 167 | 168 | if [ -z "$1" ] 169 | then 170 | echo "Path not specified to find_maven_basedir" 171 | return 1 172 | fi 173 | 174 | basedir="$1" 175 | wdir="$1" 176 | while [ "$wdir" != '/' ] ; do 177 | if [ -d "$wdir"/.mvn ] ; then 178 | basedir=$wdir 179 | break 180 | fi 181 | # workaround for JBEAP-8937 (on Solaris 10/Sparc) 182 | if [ -d "${wdir}" ]; then 183 | wdir=`cd "$wdir/.."; pwd` 184 | fi 185 | # end of workaround 186 | done 187 | echo "${basedir}" 188 | } 189 | 190 | # concatenates all lines of a file 191 | concat_lines() { 192 | if [ -f "$1" ]; then 193 | echo "$(tr -s '\n' ' ' < "$1")" 194 | fi 195 | } 196 | 197 | BASE_DIR=`find_maven_basedir "$(pwd)"` 198 | if [ -z "$BASE_DIR" ]; then 199 | exit 1; 200 | fi 201 | 202 | ########################################################################################## 203 | # Extension to allow automatically downloading the maven-wrapper.jar from Maven-central 204 | # This allows using the maven wrapper in projects that prohibit checking in binary data. 205 | ########################################################################################## 206 | if [ -r "$BASE_DIR/.mvn/wrapper/maven-wrapper.jar" ]; then 207 | if [ "$MVNW_VERBOSE" = true ]; then 208 | echo "Found .mvn/wrapper/maven-wrapper.jar" 209 | fi 210 | else 211 | if [ "$MVNW_VERBOSE" = true ]; then 212 | echo "Couldn't find .mvn/wrapper/maven-wrapper.jar, downloading it ..." 213 | fi 214 | if [ -n "$MVNW_REPOURL" ]; then 215 | jarUrl="$MVNW_REPOURL/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar" 216 | else 217 | jarUrl="https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar" 218 | fi 219 | while IFS="=" read key value; do 220 | case "$key" in (wrapperUrl) jarUrl="$value"; break ;; 221 | esac 222 | done < "$BASE_DIR/.mvn/wrapper/maven-wrapper.properties" 223 | if [ "$MVNW_VERBOSE" = true ]; then 224 | echo "Downloading from: $jarUrl" 225 | fi 226 | wrapperJarPath="$BASE_DIR/.mvn/wrapper/maven-wrapper.jar" 227 | if $cygwin; then 228 | wrapperJarPath=`cygpath --path --windows "$wrapperJarPath"` 229 | fi 230 | 231 | if command -v wget > /dev/null; then 232 | if [ "$MVNW_VERBOSE" = true ]; then 233 | echo "Found wget ... using wget" 234 | fi 235 | if [ -z "$MVNW_USERNAME" ] || [ -z "$MVNW_PASSWORD" ]; then 236 | wget "$jarUrl" -O "$wrapperJarPath" 237 | else 238 | wget --http-user=$MVNW_USERNAME --http-password=$MVNW_PASSWORD "$jarUrl" -O "$wrapperJarPath" 239 | fi 240 | elif command -v curl > /dev/null; then 241 | if [ "$MVNW_VERBOSE" = true ]; then 242 | echo "Found curl ... using curl" 243 | fi 244 | if [ -z "$MVNW_USERNAME" ] || [ -z "$MVNW_PASSWORD" ]; then 245 | curl -o "$wrapperJarPath" "$jarUrl" -f 246 | else 247 | curl --user $MVNW_USERNAME:$MVNW_PASSWORD -o "$wrapperJarPath" "$jarUrl" -f 248 | fi 249 | 250 | else 251 | if [ "$MVNW_VERBOSE" = true ]; then 252 | echo "Falling back to using Java to download" 253 | fi 254 | javaClass="$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.java" 255 | # For Cygwin, switch paths to Windows format before running javac 256 | if $cygwin; then 257 | javaClass=`cygpath --path --windows "$javaClass"` 258 | fi 259 | if [ -e "$javaClass" ]; then 260 | if [ ! -e "$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.class" ]; then 261 | if [ "$MVNW_VERBOSE" = true ]; then 262 | echo " - Compiling MavenWrapperDownloader.java ..." 263 | fi 264 | # Compiling the Java class 265 | ("$JAVA_HOME/bin/javac" "$javaClass") 266 | fi 267 | if [ -e "$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.class" ]; then 268 | # Running the downloader 269 | if [ "$MVNW_VERBOSE" = true ]; then 270 | echo " - Running MavenWrapperDownloader.java ..." 271 | fi 272 | ("$JAVA_HOME/bin/java" -cp .mvn/wrapper MavenWrapperDownloader "$MAVEN_PROJECTBASEDIR") 273 | fi 274 | fi 275 | fi 276 | fi 277 | ########################################################################################## 278 | # End of extension 279 | ########################################################################################## 280 | 281 | export MAVEN_PROJECTBASEDIR=${MAVEN_BASEDIR:-"$BASE_DIR"} 282 | if [ "$MVNW_VERBOSE" = true ]; then 283 | echo $MAVEN_PROJECTBASEDIR 284 | fi 285 | MAVEN_OPTS="$(concat_lines "$MAVEN_PROJECTBASEDIR/.mvn/jvm.config") $MAVEN_OPTS" 286 | 287 | # For Cygwin, switch paths to Windows format before running java 288 | if $cygwin; then 289 | [ -n "$M2_HOME" ] && 290 | M2_HOME=`cygpath --path --windows "$M2_HOME"` 291 | [ -n "$JAVA_HOME" ] && 292 | JAVA_HOME=`cygpath --path --windows "$JAVA_HOME"` 293 | [ -n "$CLASSPATH" ] && 294 | CLASSPATH=`cygpath --path --windows "$CLASSPATH"` 295 | [ -n "$MAVEN_PROJECTBASEDIR" ] && 296 | MAVEN_PROJECTBASEDIR=`cygpath --path --windows "$MAVEN_PROJECTBASEDIR"` 297 | fi 298 | 299 | # Provide a "standardized" way to retrieve the CLI args that will 300 | # work with both Windows and non-Windows executions. 301 | MAVEN_CMD_LINE_ARGS="$MAVEN_CONFIG $@" 302 | export MAVEN_CMD_LINE_ARGS 303 | 304 | WRAPPER_LAUNCHER=org.apache.maven.wrapper.MavenWrapperMain 305 | 306 | exec "$JAVACMD" \ 307 | $MAVEN_OPTS \ 308 | -classpath "$MAVEN_PROJECTBASEDIR/.mvn/wrapper/maven-wrapper.jar" \ 309 | "-Dmaven.home=${M2_HOME}" "-Dmaven.multiModuleProjectDirectory=${MAVEN_PROJECTBASEDIR}" \ 310 | ${WRAPPER_LAUNCHER} $MAVEN_CONFIG "$@" 311 | -------------------------------------------------------------------------------- /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 Maven 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 keystroke 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 by 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.5.6/maven-wrapper-0.5.6.jar" 124 | 125 | FOR /F "tokens=1,2 delims==" %%A IN ("%MAVEN_PROJECTBASEDIR%\.mvn\wrapper\maven-wrapper.properties") DO ( 126 | IF "%%A"=="wrapperUrl" SET DOWNLOAD_URL=%%B 127 | ) 128 | 129 | @REM Extension to allow automatically downloading the maven-wrapper.jar from Maven-central 130 | @REM This allows using the maven wrapper in projects that prohibit checking in binary data. 131 | if exist %WRAPPER_JAR% ( 132 | if "%MVNW_VERBOSE%" == "true" ( 133 | echo Found %WRAPPER_JAR% 134 | ) 135 | ) else ( 136 | if not "%MVNW_REPOURL%" == "" ( 137 | SET DOWNLOAD_URL="%MVNW_REPOURL%/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar" 138 | ) 139 | if "%MVNW_VERBOSE%" == "true" ( 140 | echo Couldn't find %WRAPPER_JAR%, downloading it ... 141 | echo Downloading from: %DOWNLOAD_URL% 142 | ) 143 | 144 | powershell -Command "&{"^ 145 | "$webclient = new-object System.Net.WebClient;"^ 146 | "if (-not ([string]::IsNullOrEmpty('%MVNW_USERNAME%') -and [string]::IsNullOrEmpty('%MVNW_PASSWORD%'))) {"^ 147 | "$webclient.Credentials = new-object System.Net.NetworkCredential('%MVNW_USERNAME%', '%MVNW_PASSWORD%');"^ 148 | "}"^ 149 | "[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12; $webclient.DownloadFile('%DOWNLOAD_URL%', '%WRAPPER_JAR%')"^ 150 | "}" 151 | if "%MVNW_VERBOSE%" == "true" ( 152 | echo Finished downloading %WRAPPER_JAR% 153 | ) 154 | ) 155 | @REM End of extension 156 | 157 | @REM Provide a "standardized" way to retrieve the CLI args that will 158 | @REM work with both Windows and non-Windows executions. 159 | set MAVEN_CMD_LINE_ARGS=%* 160 | 161 | %MAVEN_JAVA_EXE% %JVM_CONFIG_MAVEN_PROPS% %MAVEN_OPTS% %MAVEN_DEBUG_OPTS% -classpath %WRAPPER_JAR% "-Dmaven.multiModuleProjectDirectory=%MAVEN_PROJECTBASEDIR%" %WRAPPER_LAUNCHER% %MAVEN_CONFIG% %* 162 | if ERRORLEVEL 1 goto error 163 | goto end 164 | 165 | :error 166 | set ERROR_CODE=1 167 | 168 | :end 169 | @endlocal & set ERROR_CODE=%ERROR_CODE% 170 | 171 | if not "%MAVEN_SKIP_RC%" == "" goto skipRcPost 172 | @REM check for post script, once with legacy .bat ending and once with .cmd ending 173 | if exist "%HOME%\mavenrc_post.bat" call "%HOME%\mavenrc_post.bat" 174 | if exist "%HOME%\mavenrc_post.cmd" call "%HOME%\mavenrc_post.cmd" 175 | :skipRcPost 176 | 177 | @REM pause the script if MAVEN_BATCH_PAUSE is set to 'on' 178 | if "%MAVEN_BATCH_PAUSE%" == "on" pause 179 | 180 | if "%MAVEN_TERMINATE_CMD%" == "on" exit %ERROR_CODE% 181 | 182 | exit /B %ERROR_CODE% 183 | -------------------------------------------------------------------------------- /pom.xml: -------------------------------------------------------------------------------- 1 | 2 | 4 | 4.0.0 5 | 6 | org.springframework.boot 7 | spring-boot-starter-parent 8 | 2.4.3 9 | 10 | 11 | 12 | org.hongxi 13 | webflux-sample 14 | 0.0.1-SNAPSHOT 15 | webflux-sample 16 | WebFlux Sample 17 | 18 | 19 | 11 20 | 21 | 22 | 23 | 24 | org.springframework.boot 25 | spring-boot-starter-webflux 26 | 27 | 28 | 29 | org.projectlombok 30 | lombok 31 | compile 32 | 33 | 34 | junit 35 | junit 36 | test 37 | 38 | 39 | org.springframework.boot 40 | spring-boot-starter-test 41 | test 42 | 43 | 44 | io.projectreactor 45 | reactor-test 46 | test 47 | 48 | 49 | 50 | 51 | 52 | 53 | org.springframework.boot 54 | spring-boot-maven-plugin 55 | 56 | 57 | 58 | 59 | 60 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/SampleApplication.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux; 2 | 3 | import org.springframework.boot.SpringApplication; 4 | import org.springframework.boot.autoconfigure.SpringBootApplication; 5 | import reactor.core.publisher.Hooks; 6 | 7 | @SpringBootApplication 8 | public class SampleApplication { 9 | 10 | public static void main(String[] args) { 11 | Hooks.onOperatorDebug(); 12 | SpringApplication.run(SampleApplication.class, args); 13 | } 14 | 15 | } 16 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/controller/HealthController.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.controller; 2 | 3 | import org.springframework.web.bind.annotation.GetMapping; 4 | import org.springframework.web.bind.annotation.RestController; 5 | import reactor.core.publisher.Mono; 6 | 7 | /** 8 | * Created by shenhongxi on 2021/5/3. 9 | */ 10 | @RestController 11 | public class HealthController { 12 | 13 | @GetMapping("/health") 14 | public Mono health() { 15 | return Mono.just("OK"); 16 | } 17 | } 18 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/controller/OrderController.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.controller; 2 | 3 | import lombok.extern.slf4j.Slf4j; 4 | import org.hongxi.sample.webflux.dao.OrderRepository; 5 | import org.hongxi.sample.webflux.model.Order; 6 | import org.hongxi.sample.webflux.support.WebUtils; 7 | import org.springframework.beans.factory.annotation.Autowired; 8 | import org.springframework.util.Assert; 9 | import org.springframework.web.bind.annotation.*; 10 | import org.springframework.web.server.ServerWebExchange; 11 | import reactor.core.publisher.Mono; 12 | 13 | import java.util.Map; 14 | 15 | /** 16 | * Created by shenhongxi on 2021/4/22. 17 | */ 18 | @Slf4j 19 | @RestController 20 | @RequestMapping("/order") 21 | public class OrderController { 22 | 23 | @Autowired 24 | private OrderRepository orderRepository; 25 | 26 | @PostMapping("/create") 27 | public Mono create(@RequestBody Order order) { 28 | return orderRepository.save(order); 29 | } 30 | 31 | @GetMapping("/detail/{id}") 32 | public Mono findById(@PathVariable String id) { 33 | log.info("id: {}", id); 34 | Assert.isTrue(id.length() >= 6, "id length < 6"); 35 | 36 | return orderRepository.findById(id); 37 | } 38 | 39 | /** 40 | * The Servlet API “request parameter” concept conflates query parameters, 41 | * form data, and multiparts into one. However, in WebFlux, each is accessed 42 | * individually through ServerWebExchange. While @RequestParam binds to query 43 | * parameters only, you can use data binding to apply query parameters, 44 | * form data, and multiparts to a command object. 45 | * 46 | * @param name 47 | * @param exchange 48 | * @return 49 | */ 50 | @PostMapping("/delete") 51 | public Mono deleteById(@RequestParam String name, 52 | ServerWebExchange exchange) { 53 | log.info("name: {}, userId: {}", name, WebUtils.getUserId(exchange)); 54 | return exchange.getFormData(). 55 | flatMap(data -> { 56 | log.info("start: {}", data.getFirst("start")); 57 | return orderRepository.deleteById(data.getFirst("id")); 58 | }); 59 | } 60 | 61 | @PostMapping("/create2") 62 | public Mono create(ServerWebExchange exchange) { 63 | Map params = WebUtils.getRequestParams(exchange); 64 | if (params == null || !params.containsKey("id")) { 65 | return Mono.empty(); 66 | } 67 | Order order = new Order(params.get("id").toString()); 68 | if (params.containsKey("start")) { 69 | order.setStart((Long) params.get("start")); 70 | } 71 | return orderRepository.save(order); 72 | } 73 | } 74 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/dao/InMemoryOrderRepository.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.dao; 2 | 3 | import org.hongxi.sample.webflux.model.Order; 4 | import org.springframework.stereotype.Repository; 5 | import reactor.core.publisher.Mono; 6 | 7 | import java.util.Map; 8 | import java.util.concurrent.ConcurrentHashMap; 9 | 10 | @Repository 11 | public class InMemoryOrderRepository implements OrderRepository { 12 | 13 | final Map orders; 14 | 15 | public InMemoryOrderRepository() { 16 | orders = new ConcurrentHashMap<>(); 17 | } 18 | 19 | @Override 20 | public Mono findById(String id) { 21 | return Mono.justOrEmpty(orders.get(id)); 22 | } 23 | 24 | @Override 25 | public Mono save(Order order) { 26 | orders.put(order.getId(), order); 27 | 28 | return Mono.just(order); 29 | } 30 | 31 | @Override 32 | public Mono deleteById(String id) { 33 | return Mono.justOrEmpty(orders.remove(id)); 34 | } 35 | } 36 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/dao/OrderRepository.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.dao; 2 | 3 | import org.hongxi.sample.webflux.model.Order; 4 | import reactor.core.publisher.Mono; 5 | 6 | public interface OrderRepository { 7 | 8 | Mono findById(String id); 9 | 10 | Mono save(Order order); 11 | 12 | Mono deleteById(String id); 13 | } 14 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/exception/BusinessException.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.exception; 2 | 3 | import lombok.Getter; 4 | import lombok.Setter; 5 | 6 | /** 7 | * Created by shenhongxi on 2020/8/16. 8 | */ 9 | public class BusinessException extends RuntimeException { 10 | 11 | @Getter 12 | @Setter 13 | private int code; 14 | @Getter 15 | @Setter 16 | private String msg; 17 | 18 | public BusinessException(int code, String msg) { 19 | super(msg); 20 | this.code = code; 21 | this.msg = msg; 22 | } 23 | } 24 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/exception/DefaultExceptionHandler.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.exception; 2 | 3 | import lombok.extern.slf4j.Slf4j; 4 | import org.springframework.boot.autoconfigure.web.ErrorProperties; 5 | import org.springframework.boot.autoconfigure.web.WebProperties; 6 | import org.springframework.boot.autoconfigure.web.reactive.error.DefaultErrorWebExceptionHandler; 7 | import org.springframework.boot.web.error.ErrorAttributeOptions; 8 | import org.springframework.boot.web.reactive.error.ErrorAttributes; 9 | import org.springframework.context.ApplicationContext; 10 | import org.springframework.http.HttpStatus; 11 | import org.springframework.web.reactive.function.server.ServerRequest; 12 | import org.springframework.web.server.ResponseStatusException; 13 | 14 | import java.util.HashMap; 15 | import java.util.Map; 16 | 17 | /** 18 | * Created by shenhongxi on 2021/4/22. 19 | */ 20 | @Slf4j 21 | public class DefaultExceptionHandler extends DefaultErrorWebExceptionHandler { 22 | 23 | public DefaultExceptionHandler(ErrorAttributes errorAttributes, 24 | WebProperties.Resources resources, 25 | ErrorProperties errorProperties, 26 | ApplicationContext applicationContext) { 27 | super(errorAttributes, resources, errorProperties, applicationContext); 28 | } 29 | 30 | @Override 31 | protected Map getErrorAttributes(ServerRequest request, ErrorAttributeOptions options) { 32 | Map errorAttributes = new HashMap<>(); 33 | HttpStatus status; 34 | int code; 35 | String msg; 36 | Throwable error = super.getError(request); 37 | if (error instanceof BusinessException) { 38 | status = HttpStatus.OK; 39 | code = ((BusinessException) error).getCode(); 40 | msg = error.getMessage(); 41 | options.excluding(ErrorAttributeOptions.Include.STACK_TRACE); 42 | } else if (error instanceof ResponseStatusException && 43 | ((ResponseStatusException) error).getStatus().is4xxClientError()) { 44 | status = ((ResponseStatusException) error).getStatus(); 45 | code = status.value(); 46 | if (status == HttpStatus.BAD_REQUEST) { 47 | msg = "参数错误"; 48 | } else { 49 | msg = error.getMessage(); 50 | } 51 | } else { 52 | log.error("exception handled, request:{}", request.path(), error); 53 | status = HttpStatus.INTERNAL_SERVER_ERROR; 54 | code = status.value(); 55 | msg = HttpStatus.INTERNAL_SERVER_ERROR.getReasonPhrase(); 56 | } 57 | errorAttributes.put("status", status.value()); 58 | errorAttributes.put("code", code); 59 | errorAttributes.put("msg", msg); 60 | return errorAttributes; 61 | } 62 | } 63 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/exception/ExceptionHandlerConfiguration.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.exception; 2 | 3 | import org.springframework.beans.factory.ObjectProvider; 4 | import org.springframework.boot.autoconfigure.AutoConfigureBefore; 5 | import org.springframework.boot.autoconfigure.condition.ConditionalOnClass; 6 | import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication; 7 | import org.springframework.boot.autoconfigure.web.ServerProperties; 8 | import org.springframework.boot.autoconfigure.web.WebProperties; 9 | import org.springframework.boot.autoconfigure.web.reactive.WebFluxAutoConfiguration; 10 | import org.springframework.boot.web.reactive.error.ErrorAttributes; 11 | import org.springframework.boot.web.reactive.error.ErrorWebExceptionHandler; 12 | import org.springframework.context.ApplicationContext; 13 | import org.springframework.context.annotation.Bean; 14 | import org.springframework.context.annotation.Configuration; 15 | import org.springframework.core.annotation.Order; 16 | import org.springframework.http.codec.ServerCodecConfigurer; 17 | import org.springframework.web.reactive.config.WebFluxConfigurer; 18 | import org.springframework.web.reactive.result.view.ViewResolver; 19 | 20 | import java.util.stream.Collectors; 21 | 22 | @Configuration(proxyBeanMethods = false) 23 | @ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.REACTIVE) 24 | @ConditionalOnClass(WebFluxConfigurer.class) 25 | @AutoConfigureBefore(WebFluxAutoConfiguration.class) 26 | public class ExceptionHandlerConfiguration { 27 | 28 | @Bean 29 | @Order(-2) 30 | public ErrorWebExceptionHandler errorWebExceptionHandler( 31 | ErrorAttributes errorAttributes, 32 | WebProperties.Resources resources, 33 | ServerProperties serverProperties, 34 | ApplicationContext applicationContext, 35 | ObjectProvider viewResolversProvider, 36 | ServerCodecConfigurer serverCodecConfigurer) { 37 | DefaultExceptionHandler exceptionHandler = new DefaultExceptionHandler( 38 | errorAttributes, resources, serverProperties.getError(), applicationContext); 39 | exceptionHandler.setViewResolvers(viewResolversProvider.orderedStream().collect(Collectors.toList())); 40 | exceptionHandler.setMessageWriters(serverCodecConfigurer.getWriters()); 41 | exceptionHandler.setMessageReaders(serverCodecConfigurer.getReaders()); 42 | return exceptionHandler; 43 | } 44 | } -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/filter/AccessFilter.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.filter; 2 | 3 | import lombok.extern.slf4j.Slf4j; 4 | import org.hongxi.sample.webflux.support.WebConstants; 5 | import org.hongxi.sample.webflux.support.WebUtils; 6 | import org.springframework.core.annotation.Order; 7 | import org.springframework.stereotype.Component; 8 | import org.springframework.util.AntPathMatcher; 9 | import org.springframework.web.server.ServerWebExchange; 10 | import org.springframework.web.server.WebFilter; 11 | import org.springframework.web.server.WebFilterChain; 12 | import reactor.core.publisher.Mono; 13 | 14 | import java.util.Arrays; 15 | 16 | /** 17 | * Created by shenhongxi on 2021/4/26. 18 | */ 19 | @Slf4j 20 | @Order(-4) 21 | @Component 22 | public class AccessFilter implements WebFilter { 23 | 24 | private final AntPathMatcher antPathMatcher = new AntPathMatcher(); 25 | 26 | @Override 27 | public Mono filter(ServerWebExchange exchange, WebFilterChain chain) { 28 | String path = exchange.getRequest().getPath().value(); 29 | log.info("access start from path: {}", path); 30 | exchange.getAttributes().put(WebUtils.SHOULD_NOT_FILTER_ATTR, shouldNotFilter(path)); 31 | return chain.filter(exchange) 32 | .doOnEach(signal -> onEach(exchange)); 33 | } 34 | 35 | private void onEach(ServerWebExchange exchange) { 36 | log.info("access end, now start clear some context attributes"); 37 | exchange.getAttributes().remove(WebUtils.START_TIMESTAMP_ATTR); 38 | } 39 | 40 | private boolean shouldNotFilter(String path) { 41 | if (path == null) { 42 | return false; 43 | } 44 | if (path.equals("/health") || path.startsWith("/actuator") 45 | || path.startsWith("/error")) { 46 | return true; 47 | } 48 | return path.contains(".") && Arrays.stream(WebConstants.EXCLUDE_RESOURCE_PATHS) 49 | .anyMatch(pattern -> antPathMatcher.match(pattern, path)); 50 | } 51 | } 52 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/filter/ModifyBodyFilter.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.filter; 2 | 3 | import lombok.extern.slf4j.Slf4j; 4 | import org.hongxi.sample.webflux.support.*; 5 | import org.springframework.beans.factory.annotation.Autowired; 6 | import org.springframework.core.annotation.Order; 7 | import org.springframework.http.MediaType; 8 | import org.springframework.http.codec.ServerCodecConfigurer; 9 | import org.springframework.http.server.reactive.ServerHttpRequest; 10 | import org.springframework.http.server.reactive.ServerHttpResponse; 11 | import org.springframework.stereotype.Component; 12 | import org.springframework.web.server.ServerWebExchange; 13 | import org.springframework.web.server.WebFilter; 14 | import org.springframework.web.server.WebFilterChain; 15 | import reactor.core.publisher.Mono; 16 | 17 | import java.util.Map; 18 | 19 | /** 20 | * Created by shenhongxi on 2021/4/29. 21 | */ 22 | @Slf4j 23 | @Order(-1) 24 | @Component 25 | public class ModifyBodyFilter implements WebFilter { 26 | 27 | @Autowired 28 | private ServerCodecConfigurer codecConfigurer; 29 | 30 | @Override 31 | public Mono filter(ServerWebExchange exchange, WebFilterChain chain) { 32 | if (WebUtils.shouldNotFilter(exchange)) { 33 | return chain.filter(exchange); 34 | } 35 | return ParamUtils.from(exchange) 36 | .map(params -> decorate(exchange, params)) 37 | .flatMap(chain::filter); 38 | } 39 | 40 | private ServerWebExchange decorate(ServerWebExchange exchange, Map params) { 41 | if (params.isEmpty()) { 42 | return exchange; 43 | } 44 | 45 | ServerHttpResponse serverHttpResponse = new ModifiedServerHttpResponse(exchange, codecConfigurer.getReaders()); 46 | 47 | MediaType contentType = exchange.getRequest().getHeaders().getContentType(); 48 | if (contentType != null && contentType.isCompatibleWith(MediaType.APPLICATION_JSON)) { 49 | Map decrypted = Crypto.decrypt(params); 50 | exchange.getAttributes().put(WebUtils.REQUEST_PARAMS_ATTR, decrypted); 51 | byte[] rawBody = JacksonUtils.serialize(decrypted); 52 | ServerHttpRequest serverHttpRequest = new ModifiedServerHttpRequest(exchange, rawBody); 53 | return exchange.mutate().request(serverHttpRequest).response(serverHttpResponse).build(); 54 | } 55 | 56 | ServerWebExchange serverWebExchange = new ModifiedServerWebExchange(exchange); 57 | return serverWebExchange.mutate().response(serverHttpResponse).build(); 58 | } 59 | } 60 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/filter/MonitorFilter.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.filter; 2 | 3 | import lombok.extern.slf4j.Slf4j; 4 | import org.hongxi.sample.webflux.support.WebUtils; 5 | import org.springframework.beans.factory.InitializingBean; 6 | import org.springframework.beans.factory.annotation.Autowired; 7 | import org.springframework.core.annotation.Order; 8 | import org.springframework.stereotype.Component; 9 | import org.springframework.util.CollectionUtils; 10 | import org.springframework.util.StringUtils; 11 | import org.springframework.web.method.HandlerMethod; 12 | import org.springframework.web.reactive.result.condition.PatternsRequestCondition; 13 | import org.springframework.web.reactive.result.method.RequestMappingInfo; 14 | import org.springframework.web.reactive.result.method.annotation.RequestMappingHandlerMapping; 15 | import org.springframework.web.server.ServerWebExchange; 16 | import org.springframework.web.server.WebFilter; 17 | import org.springframework.web.server.WebFilterChain; 18 | import org.springframework.web.util.pattern.PathPattern; 19 | import reactor.core.publisher.Mono; 20 | 21 | import java.util.HashMap; 22 | import java.util.Map; 23 | import java.util.Set; 24 | 25 | /** 26 | * Created by shenhongxi on 2021/4/22. 27 | */ 28 | @Slf4j 29 | @Order(-3) 30 | @Component 31 | public class MonitorFilter implements WebFilter, InitializingBean { 32 | 33 | private static final String UNKNOWN_PATH = "/unknown"; 34 | 35 | // 针对只有一个path映射到的method 36 | // key: methodSign value: pathPattern 37 | private final Map pathPatterns = new HashMap<>(128); 38 | // 多个path映射到同一个method @RequestMapping({"/hi", "/hello"}) 39 | // key: methodSign value: RequestMappingInfo 40 | private final Map requestMappings = new HashMap<>(); 41 | @Autowired 42 | private RequestMappingHandlerMapping requestMappingHandlerMapping; 43 | 44 | @Override 45 | public Mono filter(ServerWebExchange exchange, WebFilterChain chain) { 46 | if (exchange.getAttributeOrDefault(WebUtils.SHOULD_NOT_FILTER_ATTR, false)) { 47 | return chain.filter(exchange); 48 | } 49 | preHandle(exchange); 50 | return chain.filter(exchange) 51 | .doOnSuccess(signal -> postHandle(exchange, null)) 52 | .doOnError(cause -> postHandle(exchange, cause)); 53 | } 54 | 55 | private void preHandle(ServerWebExchange exchange) { 56 | log.info("preHandle"); 57 | exchange.getAttributes().put(WebUtils.START_TIMESTAMP_ATTR, System.currentTimeMillis()); 58 | String pathPattern = getPathPattern(exchange); 59 | if (pathPattern != null && !pathPattern.equals(UNKNOWN_PATH)) { 60 | exchange.getAttributes().put(WebUtils.PATH_PATTERN_ATTR, pathPattern); 61 | } 62 | // 可以记录接口维度的耗时、调用次数、异常率等数据 63 | } 64 | 65 | private void postHandle(ServerWebExchange exchange, Throwable throwable) { 66 | log.info("postHandle"); 67 | Long start = exchange.getAttribute(WebUtils.START_TIMESTAMP_ATTR); 68 | if (start != null) { 69 | long cost = System.currentTimeMillis() - start; 70 | log.info("path: {}, cost: {}, error: {}", 71 | exchange.getRequest().getPath(), cost, throwable != null); 72 | } 73 | } 74 | 75 | private void preLoadAllPaths() { 76 | Map requestMapInfoHandlerMap = requestMappingHandlerMapping.getHandlerMethods(); 77 | for (RequestMappingInfo info : requestMapInfoHandlerMap.keySet()) { 78 | PatternsRequestCondition patternsRequestCondition = info.getPatternsCondition(); 79 | Set patterns = patternsRequestCondition.getPatterns(); 80 | String methodSignal = requestMapInfoHandlerMap.get(info).getMethod().toGenericString(); 81 | if (patterns.size() == 1) { 82 | pathPatterns.put(methodSignal, patterns.iterator().next().getPatternString()); 83 | } else { 84 | requestMappings.put(methodSignal, info); 85 | } 86 | } 87 | } 88 | 89 | private String getPathPattern(ServerWebExchange exchange) { 90 | String path = exchange.getRequest().getPath().value(); 91 | if (!StringUtils.hasLength(path)) { 92 | return UNKNOWN_PATH; 93 | } 94 | if (pathPatterns.containsValue(path)) { 95 | return path; 96 | } 97 | try { 98 | Mono handlerMono = requestMappingHandlerMapping.getHandler(exchange); 99 | String handlerKey = path + "_handler"; 100 | handlerMono.subscribe(handler -> exchange.getAttributes().put(handlerKey, handler)); 101 | Object handler = exchange.getAttribute(handlerKey); 102 | if (handler == null) { 103 | return UNKNOWN_PATH; 104 | } 105 | exchange.getAttributes().remove(handlerKey); 106 | if (handler instanceof HandlerMethod) { 107 | HandlerMethod handlerMethod = (HandlerMethod) handler; 108 | String methodSign = handlerMethod.getMethod().toGenericString(); 109 | String pathPattern = pathPatterns.get(methodSign); 110 | if (StringUtils.hasLength(pathPattern)) { 111 | return pathPattern; 112 | } 113 | if (requestMappings.containsKey(methodSign)) { 114 | RequestMappingInfo mappingInfo = requestMappings.get(methodSign); 115 | PatternsRequestCondition patternsRequestCondition = mappingInfo.getPatternsCondition(); 116 | Set pathPatterns = patternsRequestCondition.getPatterns(); 117 | if (!CollectionUtils.isEmpty(pathPatterns)) { 118 | return pathPatterns.iterator().next().getPatternString(); 119 | } 120 | } 121 | } 122 | } catch (Exception e) { 123 | log.warn("get path pattern error, {}", e.getClass()); 124 | } 125 | return UNKNOWN_PATH; 126 | } 127 | 128 | @Override 129 | public void afterPropertiesSet() throws Exception { 130 | preLoadAllPaths(); 131 | } 132 | } 133 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/filter/SessionAuthFilter.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.filter; 2 | 3 | import lombok.extern.slf4j.Slf4j; 4 | import org.hongxi.sample.webflux.exception.BusinessException; 5 | import org.hongxi.sample.webflux.support.SessionContext; 6 | import org.hongxi.sample.webflux.support.WebUtils; 7 | import org.springframework.core.annotation.Order; 8 | import org.springframework.stereotype.Component; 9 | import org.springframework.util.StringUtils; 10 | import org.springframework.web.server.ServerWebExchange; 11 | import org.springframework.web.server.WebFilter; 12 | import org.springframework.web.server.WebFilterChain; 13 | import reactor.core.publisher.Mono; 14 | 15 | /** 16 | * Created by shenhongxi on 2021/4/22. 17 | */ 18 | @Slf4j 19 | @Order(0) 20 | @Component 21 | public class SessionAuthFilter implements WebFilter { 22 | 23 | @Override 24 | public Mono filter(ServerWebExchange exchange, WebFilterChain chain) { 25 | if (WebUtils.shouldNotFilter(exchange)) { 26 | return chain.filter(exchange); 27 | } 28 | 29 | SessionContext sessionContext = exchange.getAttribute(WebUtils.SESSION_CONTEXT_ATTR); 30 | if (sessionContext == null) { 31 | throw new BusinessException(403, "请先登录"); 32 | } 33 | String userId = sessionContext.getUserId(); 34 | log.info("userId: {}", userId); 35 | if (!StringUtils.hasLength(userId)) { 36 | throw new BusinessException(403, "请先登录"); 37 | } 38 | return chain.filter(exchange); 39 | } 40 | } 41 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/filter/SessionFilter.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.filter; 2 | 3 | import lombok.extern.slf4j.Slf4j; 4 | import org.hongxi.sample.webflux.support.SessionContext; 5 | import org.hongxi.sample.webflux.support.WebUtils; 6 | import org.springframework.core.annotation.Order; 7 | import org.springframework.http.HttpCookie; 8 | import org.springframework.stereotype.Component; 9 | import org.springframework.util.CollectionUtils; 10 | import org.springframework.util.StringUtils; 11 | import org.springframework.web.server.ServerWebExchange; 12 | import org.springframework.web.server.WebFilter; 13 | import org.springframework.web.server.WebFilterChain; 14 | import reactor.core.publisher.Mono; 15 | 16 | import java.util.List; 17 | 18 | /** 19 | * Created by shenhongxi on 2021/4/29. 20 | */ 21 | @Slf4j 22 | @Order(-2) 23 | @Component 24 | public class SessionFilter implements WebFilter { 25 | 26 | private static final String COOKIE_NAME = "SESSIONID"; 27 | 28 | @Override 29 | public Mono filter(ServerWebExchange exchange, WebFilterChain chain) { 30 | if (WebUtils.shouldNotFilter(exchange)) { 31 | return chain.filter(exchange); 32 | } 33 | List httpCookies = exchange.getRequest().getCookies().get(COOKIE_NAME); 34 | String userId; 35 | if (!CollectionUtils.isEmpty(httpCookies)) { 36 | String sessionId = httpCookies.get(0).getValue(); 37 | userId = getUserIdBySessionId(sessionId); 38 | 39 | } else { 40 | // just for test 41 | userId = exchange.getRequest().getQueryParams().getFirst("userId"); 42 | } 43 | if (StringUtils.hasLength(userId)) { 44 | SessionContext sessionContext = new SessionContext(); 45 | sessionContext.setUserId(userId); 46 | exchange.getAttributes().put(WebUtils.SESSION_CONTEXT_ATTR, sessionContext); 47 | } 48 | return chain.filter(exchange); 49 | } 50 | 51 | private String getUserIdBySessionId(String sessionId) { 52 | return null; 53 | } 54 | } 55 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/model/Order.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.model; 2 | 3 | import lombok.Data; 4 | 5 | @Data 6 | public class Order { 7 | 8 | private String id; 9 | 10 | private Long start; 11 | 12 | public Order() { 13 | } 14 | 15 | public Order(String id) { 16 | this.id = id; 17 | } 18 | } 19 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/router/RouterConfiguration.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.router; 2 | 3 | import org.hongxi.sample.webflux.router.handler.OrderHandler; 4 | import org.hongxi.sample.webflux.router.handler.ServerRedirectHandler; 5 | import org.springframework.context.annotation.Bean; 6 | import org.springframework.context.annotation.Configuration; 7 | import org.springframework.http.HttpMethod; 8 | import org.springframework.web.reactive.function.server.RouterFunction; 9 | import org.springframework.web.reactive.function.server.ServerResponse; 10 | 11 | import static org.springframework.http.MediaType.APPLICATION_JSON; 12 | import static org.springframework.web.reactive.function.server.RequestPredicates.*; 13 | import static org.springframework.web.reactive.function.server.RouterFunctions.nest; 14 | import static org.springframework.web.reactive.function.server.RouterFunctions.route; 15 | 16 | /** 17 | * Created by shenhongxi on 2021/4/22. 18 | */ 19 | @Configuration 20 | public class RouterConfiguration { 21 | 22 | final ServerRedirectHandler serverRedirectHandler = new ServerRedirectHandler(); 23 | 24 | @Bean 25 | public RouterFunction routes(OrderHandler orderHandler) { 26 | return 27 | nest(path("/func/order"), 28 | nest(accept(APPLICATION_JSON), 29 | route(GET("/{id}"), orderHandler::get) 30 | .andRoute(method(HttpMethod.GET), orderHandler::list) 31 | ) 32 | .andNest(contentType(APPLICATION_JSON), 33 | route(POST("/create"), orderHandler::create) 34 | ) 35 | .andNest((serverRequest) -> serverRequest.cookies() 36 | .containsKey("Redirect-Traffic"), 37 | route(all(), serverRedirectHandler) 38 | ) 39 | ); 40 | } 41 | } 42 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/router/handler/OrderHandler.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.router.handler; 2 | 3 | import lombok.extern.slf4j.Slf4j; 4 | import org.hongxi.sample.webflux.dao.OrderRepository; 5 | import org.hongxi.sample.webflux.model.Order; 6 | import org.springframework.stereotype.Service; 7 | import org.springframework.util.Assert; 8 | import org.springframework.web.reactive.function.server.ServerRequest; 9 | import org.springframework.web.reactive.function.server.ServerResponse; 10 | import reactor.core.publisher.Mono; 11 | 12 | import java.net.URI; 13 | 14 | @Slf4j 15 | @Service 16 | public class OrderHandler { 17 | 18 | final OrderRepository orderRepository; 19 | 20 | public OrderHandler(OrderRepository repository) { 21 | orderRepository = repository; 22 | } 23 | 24 | public Mono create(ServerRequest request) { 25 | return request 26 | .bodyToMono(Order.class) 27 | .flatMap(orderRepository::save) 28 | .flatMap(o -> 29 | ServerResponse.created(URI.create("/order/" + o.getId())) 30 | .build() 31 | ); 32 | } 33 | 34 | public Mono get(ServerRequest request) { 35 | String id = request.pathVariable("id"); 36 | log.info("id: {}", id); 37 | Assert.isTrue(id.length() >= 6, "id length < 6"); 38 | 39 | return orderRepository 40 | .findById(id) 41 | .flatMap(order -> 42 | ServerResponse 43 | .ok() 44 | .syncBody(order) 45 | ) 46 | .switchIfEmpty(ServerResponse.notFound().build()); 47 | } 48 | 49 | public Mono list(ServerRequest request) { 50 | return null; 51 | } 52 | } 53 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/router/handler/ServerRedirectHandler.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.router.handler; 2 | 3 | import org.springframework.core.io.buffer.DataBuffer; 4 | import org.springframework.web.reactive.function.BodyInserters; 5 | import org.springframework.web.reactive.function.client.WebClient; 6 | import org.springframework.web.reactive.function.server.HandlerFunction; 7 | import org.springframework.web.reactive.function.server.ServerRequest; 8 | import org.springframework.web.reactive.function.server.ServerResponse; 9 | import reactor.core.publisher.Mono; 10 | 11 | public class ServerRedirectHandler implements HandlerFunction { 12 | 13 | final WebClient webClient = WebClient.create(); 14 | 15 | @Override 16 | public Mono handle(ServerRequest request) { 17 | return webClient 18 | .method(request.method()) 19 | .uri(request.headers() 20 | .header("Redirect-Traffic") 21 | .get(0)) 22 | .headers((h) -> h.addAll(request.headers().asHttpHeaders())) 23 | .body(BodyInserters.fromDataBuffers( 24 | request.bodyToFlux(DataBuffer.class) 25 | )) 26 | .cookies(c -> request 27 | .cookies() 28 | .forEach((key, list) -> list.forEach(cookie -> c.add(key, cookie.getValue()))) 29 | ) 30 | .exchange() 31 | .flatMap(cr -> ServerResponse 32 | .status(cr.statusCode()) 33 | .cookies(c -> c.addAll(cr.cookies())) 34 | .headers(hh -> hh.addAll(cr.headers().asHttpHeaders())) 35 | .body(BodyInserters.fromDataBuffers( 36 | cr.bodyToFlux(DataBuffer.class) 37 | )) 38 | ); 39 | } 40 | } 41 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/support/Crypto.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.support; 2 | 3 | import org.springframework.util.MultiValueMap; 4 | import org.springframework.util.MultiValueMapAdapter; 5 | 6 | import java.util.Collections; 7 | import java.util.HashMap; 8 | import java.util.Map; 9 | 10 | /** 11 | * Created by shenhongxi on 2021/4/29. 12 | */ 13 | public class Crypto { 14 | 15 | /** 16 | * 模拟解密逻辑：添加一个请求参数 start 17 | * @param requestBody 18 | * @return 19 | */ 20 | public static Map decrypt(Map requestBody) { 21 | Map decrypted = new HashMap<>(requestBody); 22 | decrypted.put("start", System.currentTimeMillis()); 23 | return decrypted; 24 | } 25 | 26 | /** 27 | * 模拟解密逻辑：添加一个请求参数 start 28 | * @param formData 29 | * @return 30 | */ 31 | public static MultiValueMap decrypt(MultiValueMap formData) { 32 | MultiValueMap decrypted = new MultiValueMapAdapter<>(formData); 33 | decrypted.put("start", Collections.singletonList(String.valueOf(System.currentTimeMillis()))); 34 | return decrypted; 35 | } 36 | 37 | /** 38 | * 模拟加密逻辑：添加一个响应参数 end 39 | * @param responseBody 40 | * @return 41 | */ 42 | public static byte[] encrypt(byte[] responseBody) { 43 | Map result = JacksonUtils.deserialize(responseBody, Map.class); 44 | result.put("end", System.currentTimeMillis()); 45 | return JacksonUtils.serialize(result); 46 | } 47 | } 48 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/support/JacksonUtils.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.support; 2 | 3 | import com.fasterxml.jackson.core.JsonProcessingException; 4 | import com.fasterxml.jackson.databind.ObjectMapper; 5 | 6 | import java.io.IOException; 7 | import java.nio.charset.StandardCharsets; 8 | 9 | /** 10 | * Created by shenhongxi on 2021/4/30. 11 | */ 12 | public abstract class JacksonUtils { 13 | 14 | private static final ObjectMapper objectMapper = new ObjectMapper(); 15 | 16 | public static String toJson(T obj) { 17 | if (obj == null) return null; 18 | try { 19 | return objectMapper.writeValueAsString(obj); 20 | } catch (JsonProcessingException e) { 21 | throw new RuntimeException(e); 22 | } 23 | } 24 | 25 | public static T fromJson(String src, Class clazz) { 26 | try { 27 | return objectMapper.readValue(src, clazz); 28 | } catch (IOException e) { 29 | throw new RuntimeException(e); 30 | } 31 | } 32 | 33 | public static byte[] serialize(T obj) { 34 | if (obj == null) return null; 35 | try { 36 | return objectMapper.writeValueAsString(obj).getBytes(StandardCharsets.UTF_8); 37 | } catch (JsonProcessingException e) { 38 | throw new RuntimeException(e); 39 | } 40 | } 41 | 42 | public static T deserialize(byte[] src, Class clazz) { 43 | try { 44 | return objectMapper.readValue(new String(src, StandardCharsets.UTF_8), clazz); 45 | } catch (IOException e) { 46 | throw new RuntimeException(e); 47 | } 48 | } 49 | } 50 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/support/ModifiedServerHttpRequest.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.support; 2 | 3 | import org.springframework.core.io.buffer.DataBuffer; 4 | import org.springframework.http.HttpHeaders; 5 | import org.springframework.http.MediaType; 6 | import org.springframework.http.server.reactive.ServerHttpRequestDecorator; 7 | import org.springframework.util.MultiValueMap; 8 | import org.springframework.util.MultiValueMapAdapter; 9 | import org.springframework.web.server.ServerWebExchange; 10 | import reactor.core.publisher.Flux; 11 | 12 | import java.nio.charset.StandardCharsets; 13 | import java.util.HashMap; 14 | import java.util.List; 15 | import java.util.Map; 16 | 17 | /** 18 | * Created by shenhongxi on 2021/4/29. 19 | */ 20 | public class ModifiedServerHttpRequest extends ServerHttpRequestDecorator { 21 | 22 | private final ServerWebExchange exchange; 23 | 24 | private final byte[] rawBody; 25 | 26 | public ModifiedServerHttpRequest(ServerWebExchange exchange, byte[] rawBody) { 27 | super(exchange.getRequest()); 28 | this.exchange = exchange; 29 | this.rawBody = rawBody; 30 | } 31 | 32 | @Override 33 | public MultiValueMap getQueryParams() { 34 | Map> targetMap = new HashMap<>(); 35 | MultiValueMap all = new MultiValueMapAdapter<>(targetMap); 36 | all.addAll(super.getQueryParams()); 37 | Map params = JacksonUtils.deserialize(this.rawBody, Map.class); 38 | params.forEach((k, v) -> { 39 | if (v != null) { 40 | all.add(k, v.toString()); 41 | } 42 | }); 43 | return all; 44 | } 45 | 46 | @Override 47 | public Flux getBody() { 48 | return Flux.just(exchange.getResponse().bufferFactory().wrap(this.rawBody)); 49 | } 50 | 51 | @Override 52 | public HttpHeaders getHeaders() { 53 | HttpHeaders headers = HttpHeaders.writableHttpHeaders(super.getHeaders()); 54 | headers.setContentLength(this.rawBody.length); 55 | headers.setContentType(MediaType.APPLICATION_JSON); 56 | return headers; 57 | } 58 | 59 | /** 60 | * @return body json string 61 | */ 62 | public String bodyString() { 63 | return new String(rawBody, StandardCharsets.UTF_8); 64 | } 65 | } 66 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/support/ModifiedServerHttpResponse.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.support; 2 | 3 | import org.reactivestreams.Publisher; 4 | import org.springframework.core.io.buffer.DataBuffer; 5 | import org.springframework.http.HttpHeaders; 6 | import org.springframework.http.HttpStatus; 7 | import org.springframework.http.MediaType; 8 | import org.springframework.http.codec.HttpMessageReader; 9 | import org.springframework.http.server.reactive.ServerHttpResponseDecorator; 10 | import org.springframework.web.reactive.function.client.ClientResponse; 11 | import org.springframework.web.server.ServerWebExchange; 12 | import reactor.core.publisher.Flux; 13 | import reactor.core.publisher.Mono; 14 | 15 | import java.nio.charset.StandardCharsets; 16 | import java.util.List; 17 | 18 | /** 19 | * Created by shenhongxi on 2021/4/29. 20 | */ 21 | public class ModifiedServerHttpResponse extends ServerHttpResponseDecorator { 22 | 23 | private final List> messageReaders; 24 | 25 | private String body; 26 | 27 | public ModifiedServerHttpResponse(ServerWebExchange exchange, 28 | List> messageReaders) { 29 | super(exchange.getResponse()); 30 | this.messageReaders = messageReaders; 31 | } 32 | 33 | @Override 34 | public Mono writeWith(Publisher body) { 35 | // 这里只是借用 ClientResponse 这个类获取修改之前的 body 36 | // server 端最终返回的是 ServerResponse/ServerHttpResponse 37 | ClientResponse clientResponse = prepareClientResponse(body); 38 | Mono modifiedBody = clientResponse.bodyToMono(byte[].class) 39 | .map(originalBody -> { 40 | this.body = new String(originalBody, StandardCharsets.UTF_8); 41 | return Crypto.encrypt(originalBody); 42 | }).map(encrypted -> getDelegate().bufferFactory().wrap(encrypted)) 43 | .doOnNext(data -> getDelegate().getHeaders().setContentLength(data.readableByteCount())); 44 | return super.writeWith(modifiedBody); 45 | } 46 | 47 | /** 48 | * @return body json string 49 | */ 50 | public String bodyString() { 51 | return this.body; 52 | } 53 | 54 | private ClientResponse prepareClientResponse(Publisher body) { 55 | ClientResponse.Builder builder = ClientResponse.create(HttpStatus.OK, messageReaders); 56 | HttpHeaders httpHeaders = new HttpHeaders(); 57 | httpHeaders.setContentType(MediaType.APPLICATION_JSON); 58 | return builder.headers(headers -> headers.putAll(httpHeaders)).body(Flux.from(body)).build(); 59 | } 60 | } 61 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/support/ModifiedServerWebExchange.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.support; 2 | 3 | import org.springframework.util.MultiValueMap; 4 | import org.springframework.web.server.ServerWebExchange; 5 | import org.springframework.web.server.ServerWebExchangeDecorator; 6 | import reactor.core.publisher.Mono; 7 | 8 | /** 9 | * Created by shenhongxi on 2021/4/29. 10 | */ 11 | public class ModifiedServerWebExchange extends ServerWebExchangeDecorator { 12 | 13 | public ModifiedServerWebExchange(ServerWebExchange delegate) { 14 | super(delegate); 15 | } 16 | 17 | @Override 18 | public Mono> getFormData() { 19 | return super.getFormData() 20 | .map(Crypto::decrypt); 21 | } 22 | } 23 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/support/ParamUtils.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.support; 2 | 3 | import org.springframework.core.ResolvableType; 4 | import org.springframework.http.MediaType; 5 | import org.springframework.http.codec.DecoderHttpMessageReader; 6 | import org.springframework.http.codec.HttpMessageReader; 7 | import org.springframework.http.codec.json.Jackson2JsonDecoder; 8 | import org.springframework.web.bind.support.WebExchangeDataBinder; 9 | import org.springframework.web.server.ServerWebExchange; 10 | import reactor.core.publisher.Mono; 11 | 12 | import java.util.Collections; 13 | import java.util.Map; 14 | 15 | /** 16 | * Created by shenhongxi on 2021/4/29. 17 | */ 18 | public abstract class ParamUtils { 19 | 20 | @SuppressWarnings("rawtypes") 21 | private static final HttpMessageReader messageReader = 22 | new DecoderHttpMessageReader<>(new Jackson2JsonDecoder()); 23 | 24 | @SuppressWarnings("unchecked") 25 | public static Mono> from(ServerWebExchange exchange) { 26 | Map data = exchange.getAttribute(WebUtils.REQUEST_PARAMS_ATTR); 27 | if (data != null) { 28 | return Mono.just(data); 29 | } 30 | 31 | Mono> params; 32 | MediaType contentType = exchange.getRequest().getHeaders().getContentType(); 33 | if (contentType != null && contentType.isCompatibleWith(MediaType.APPLICATION_JSON)) { 34 | params = (Mono>) messageReader.readMono( 35 | ResolvableType.forType(Map.class), exchange.getRequest(), Collections.emptyMap()); 36 | } else { 37 | params = WebExchangeDataBinder.extractValuesToBind(exchange); 38 | } 39 | return params.doOnNext(e -> exchange.getAttributes().put(WebUtils.REQUEST_PARAMS_ATTR, e)); 40 | } 41 | } 42 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/support/SessionContext.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.support; 2 | 3 | import lombok.Data; 4 | 5 | /** 6 | * Created by shenhongxi on 2021/4/29. 7 | */ 8 | @Data 9 | public class SessionContext { 10 | 11 | private String userId; 12 | } 13 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/support/WebConstants.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.support; 2 | 3 | /** 4 | * Created by shenhongxi on 2021/5/3. 5 | */ 6 | public class WebConstants { 7 | 8 | public static final String[] EXCLUDE_RESOURCE_PATHS = new String[]{ 9 | "/**/*.html", 10 | "/**/*.css", 11 | "/**/*.js", 12 | "/**/*.png", 13 | "/**/*.jpg", 14 | "/**/*.jpeg", 15 | "/**/*.gif", 16 | "/**/*.ico", 17 | "/**/*.svg" 18 | }; 19 | } 20 | -------------------------------------------------------------------------------- /src/main/java/org/hongxi/sample/webflux/support/WebUtils.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux.support; 2 | 3 | import org.springframework.web.server.ServerWebExchange; 4 | 5 | import java.util.Map; 6 | 7 | /** 8 | * Created by shenhongxi on 2021/4/29. 9 | */ 10 | public abstract class WebUtils { 11 | 12 | /** 13 | * a convenient way to acquire request params 14 | * 15 | * @see ParamUtils 16 | * 17 | * @param exchange 18 | * @return 19 | */ 20 | public static Map getRequestParams(ServerWebExchange exchange) { 21 | return exchange.getAttribute(REQUEST_PARAMS_ATTR); 22 | } 23 | 24 | public static SessionContext getSessionContext(ServerWebExchange exchange) { 25 | return exchange.getAttribute(SESSION_CONTEXT_ATTR); 26 | } 27 | 28 | public static String getUserId(ServerWebExchange exchange) { 29 | SessionContext sessionContext = getSessionContext(exchange); 30 | return sessionContext != null ? sessionContext.getUserId() : null; 31 | } 32 | 33 | public static boolean shouldNotFilter(ServerWebExchange exchange) { 34 | return exchange.getAttributeOrDefault(WebUtils.SHOULD_NOT_FILTER_ATTR, 35 | exchange.getAttribute(WebUtils.PATH_PATTERN_ATTR) == null); 36 | } 37 | 38 | public static final String START_TIMESTAMP_ATTR = qualify("startTimestamp"); 39 | public static final String PATH_PATTERN_ATTR = qualify("pathPattern"); 40 | public static final String SHOULD_NOT_FILTER_ATTR = qualify("shouldNotFilter"); 41 | public static final String SESSION_CONTEXT_ATTR = qualify("sessionContext"); 42 | public static final String REQUEST_PARAMS_ATTR = qualify("requestParams"); 43 | 44 | private static String qualify(String attr) { 45 | return WebUtils.class.getName() + "." + attr; 46 | } 47 | } 48 | -------------------------------------------------------------------------------- /src/main/resources/application.yml: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/test/java/org/hongxi/sample/reactor/FluxTests.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.reactor; 2 | 3 | import lombok.AllArgsConstructor; 4 | import lombok.NoArgsConstructor; 5 | import org.junit.Test; 6 | import org.reactivestreams.Publisher; 7 | import org.reactivestreams.Subscriber; 8 | import org.reactivestreams.Subscription; 9 | import reactor.core.Disposable; 10 | import reactor.core.publisher.BaseSubscriber; 11 | import reactor.core.publisher.ConnectableFlux; 12 | import reactor.core.publisher.Flux; 13 | import reactor.core.publisher.Mono; 14 | import reactor.test.StepVerifier; 15 | import reactor.util.function.Tuple2; 16 | import reactor.util.function.Tuples; 17 | 18 | import java.time.Duration; 19 | import java.time.Instant; 20 | import java.util.*; 21 | import java.util.function.Function; 22 | import java.util.stream.IntStream; 23 | 24 | /** 25 | * Created by shenhongxi on 2020/10/6. 26 | */ 27 | public class FluxTests { 28 | 29 | private final Random random = new Random(); 30 | 31 | @Test 32 | public void createFlux() { 33 | Flux stream1 = Flux.just("Hello", "world"); 34 | Flux stream2 = Flux.fromArray(new Integer[] {1, 2, 3}); 35 | Flux stream3 = Flux.fromIterable(Arrays.asList(9, 8, 7)); 36 | Flux stream4 = Flux.range(2010, 9); 37 | 38 | Flux empty = Flux.empty(); 39 | Flux never = Flux.never(); 40 | } 41 | 42 | @Test 43 | public void createMono() { 44 | Mono stream5 = Mono.just("One"); 45 | Mono stream6 = Mono.justOrEmpty(null); 46 | Mono stream7 = Mono.justOrEmpty(Optional.empty()); 47 | 48 | Mono stream8 = Mono.fromCallable(this::httpRequest); 49 | 50 | Mono error = Mono.error(new RuntimeException("Unknown id")); 51 | } 52 | 53 | private String httpRequest() { 54 | return "OK"; 55 | } 56 | 57 | @Test 58 | public void subscribeExample() { 59 | Flux.range(1, 5).subscribe(System.out::println); 60 | 61 | System.out.println("---"); 62 | 63 | Flux.just("A", "B", "C") 64 | .subscribe( 65 | data -> System.out.println("onNext: " + data), 66 | err -> { /* ignored */ } 67 | ); 68 | 69 | System.out.println("---"); 70 | 71 | Flux.just("A", "B", "C") 72 | .subscribe( 73 | data -> System.out.println("onNext: " + data), 74 | err -> { /* ignored */ }, 75 | () -> System.out.println("onComplete") 76 | ); 77 | 78 | System.out.println("---"); 79 | 80 | Flux.range(1, 100) 81 | .subscribe( 82 | data -> System.out.println("onNext: " + data), 83 | err -> { /* ignored */ }, 84 | () -> System.out.println("onComplete"), 85 | subscription -> { 86 | subscription.request(4); 87 | subscription.cancel(); 88 | } 89 | ); 90 | } 91 | 92 | @Test 93 | public void customSubscriberExample() { 94 | Subscriber subscriber = new Subscriber<>() { 95 | volatile Subscription subscription; 96 | 97 | @Override 98 | public void onSubscribe(Subscription subscription) { 99 | this.subscription = subscription; 100 | System.out.println("initial request for 1 element"); 101 | subscription.request(1); 102 | } 103 | 104 | @Override 105 | public void onNext(String s) { 106 | System.out.println("onNext: " + s); 107 | System.out.println("requesting 1 more element"); 108 | subscription.request(1); 109 | } 110 | 111 | @Override 112 | public void onError(Throwable throwable) { 113 | System.out.println("onError: " + throwable.getMessage()); 114 | } 115 | 116 | @Override 117 | public void onComplete() { 118 | System.out.println("onComplete"); 119 | } 120 | }; 121 | 122 | Flux stream = Flux.just("Hello", "world", "!"); 123 | stream.subscribe(subscriber); 124 | } 125 | 126 | @Test 127 | public void baseSubscriberExample() { 128 | Flux stream = Flux.just("Hello", "world", "!"); 129 | stream.subscribe(new BaseSubscriber<>() { 130 | @Override 131 | protected void hookOnSubscribe(Subscription subscription) { 132 | System.out.println("initial request for 1 element"); 133 | subscription.request(1); 134 | } 135 | 136 | @Override 137 | protected void hookOnNext(String value) { 138 | System.out.println("onNext: " + value); 139 | System.out.println("requesting 1 more element"); 140 | request(1); 141 | } 142 | }); 143 | } 144 | 145 | @Test 146 | public void disposableExample() throws InterruptedException { 147 | Disposable disposable = Flux.interval(Duration.ofMillis(50)) 148 | .subscribe(data -> System.out.println("onNext: " + data)); 149 | Thread.sleep(200); 150 | disposable.dispose(); 151 | } 152 | 153 | @Test 154 | public void mapExample() { 155 | Flux.just(1, 2, 3) 156 | .map(i -> i * 2) 157 | .subscribe(System.out::println); 158 | } 159 | 160 | @Test 161 | public void indexExample() { 162 | Flux.range(2018, 5) 163 | .timestamp() 164 | .index() 165 | .subscribe(e -> System.out.println( 166 | String.format("index: %d, ts: %s, value: %s", 167 | e.getT1(), 168 | Instant.ofEpochMilli(e.getT2().getT1()), 169 | e.getT2().getT2()) 170 | )); 171 | } 172 | 173 | @Test 174 | public void filterExample() { 175 | Flux.just(1, 2, 3) 176 | .filter(i -> i % 2 == 0) 177 | .subscribe(System.out::println); 178 | 179 | System.out.println("---"); 180 | 181 | Flux.just(1, 2, 3) 182 | .take(2) 183 | .subscribe(System.out::println); 184 | 185 | System.out.println("---"); 186 | 187 | Flux.just(1, 2, 3) 188 | .takeLast(1) 189 | .subscribe(System.out::println); 190 | 191 | System.out.println("---"); 192 | 193 | Flux.just(1, 2, 3) 194 | .takeUntil(i -> i % 2 == 0) 195 | .subscribe(System.out::println); 196 | 197 | System.out.println("---"); 198 | 199 | Flux.just(1, 2, 3) 200 | .elementAt(1) 201 | .subscribe(System.out::println); 202 | } 203 | 204 | @Test 205 | public void shouldCreateDefer() { 206 | Mono userMono = requestUserData(null); 207 | StepVerifier.create(userMono) 208 | .expectNextCount(0) 209 | .expectErrorMessage("Invalid user session") 210 | .verify(); 211 | } 212 | 213 | private Mono requestUserData(String sessionId) { 214 | return Mono.defer(() -> 215 | isValidSession(sessionId) 216 | ? Mono.fromCallable(() -> requestUser(sessionId)) 217 | : Mono.error(new RuntimeException("Invalid user session"))); 218 | } 219 | 220 | private boolean isValidSession(String sessionId) { 221 | return sessionId != null; 222 | } 223 | 224 | private User requestUser(String sessionId) { 225 | return new User(); 226 | } 227 | 228 | @Test 229 | public void startStopStreamProcessing() throws Exception { 230 | Mono startCommand = Mono.delay(Duration.ofSeconds(1)); 231 | Mono stopCommand = Mono.delay(Duration.ofSeconds(3)); 232 | Flux streamOfData = Flux.interval(Duration.ofMillis(100)); 233 | 234 | streamOfData 235 | .skipUntilOther(startCommand) 236 | .takeUntilOther(stopCommand) 237 | .subscribe(System.out::println); 238 | 239 | Thread.sleep(4000); 240 | } 241 | 242 | @Test 243 | public void collectExample() { 244 | Flux.just(1, 6, 2, 8, 3, 1, 5, 1) 245 | .collectSortedList(Comparator.reverseOrder()) 246 | .subscribe(System.out::println); 247 | } 248 | 249 | @Test 250 | public void distinctExample() { 251 | Flux.just(1, 2, 2, 3, 3, 3, 1, 1, 4) 252 | .distinct() 253 | .subscribe(System.out::println); 254 | 255 | System.out.println(); 256 | 257 | Flux.just(1, 2, 2, 3, 3, 3, 1, 1, 4) 258 | .distinctUntilChanged() 259 | .subscribe(System.out::println); 260 | } 261 | 262 | @Test 263 | public void anyExample() { 264 | Flux.just(3, 5, 7, 9, 11, 15, 16, 17) 265 | .count() 266 | .subscribe(System.out::println); 267 | 268 | Flux.just(3, 5, 7, 9, 11, 15, 16, 17) 269 | .any(e -> e % 2 == 0) 270 | .subscribe(System.out::println); 271 | 272 | Flux.just(3, 5, 7, 9, 11, 15, 16, 17) 273 | .all(e -> e % 2 == 0) 274 | .subscribe(System.out::println); 275 | 276 | Flux.just(3, 5, 7, 9, 11, 15, 16, 17) 277 | .hasElement(7) 278 | .subscribe(System.out::println); 279 | 280 | Flux.just(3, 5, 7, 9, 11, 15, 16, 17) 281 | .hasElements() 282 | .subscribe(System.out::println); 283 | } 284 | 285 | @Test 286 | public void sortExample() { 287 | Flux.just(1, 4, 2, 3) 288 | .sort() 289 | .subscribe(System.out::println); 290 | 291 | Flux.just(1, 4, 2, 3) 292 | .sort(Comparator.reverseOrder()) 293 | .subscribe(System.out::println); 294 | } 295 | 296 | @Test 297 | public void reduceExample() { 298 | Flux.range(1, 5) 299 | .reduce(0, Integer::sum) 300 | .subscribe(System.out::println); 301 | } 302 | 303 | @Test 304 | public void scanExample() { 305 | Flux.range(1, 5) 306 | .scan(0, Integer::sum) 307 | .subscribe(System.out::println); 308 | } 309 | 310 | @Test 311 | public void thenExample() { 312 | Flux.just(1, 2, 3) 313 | .thenMany(Flux.just(4, 5)) 314 | .subscribe(System.out::println); 315 | } 316 | 317 | @Test 318 | public void concatExample() { 319 | Flux.concat( 320 | Flux.range(1, 3), 321 | Flux.range(4, 2), 322 | Flux.range(6, 5) 323 | ).subscribe(System.out::println); 324 | } 325 | 326 | @Test 327 | public void mergeExample() { 328 | Flux.merge( 329 | Flux.range(1, 3), 330 | Flux.range(4, 2), 331 | Flux.range(6, 5) 332 | ).subscribe(System.out::println); 333 | } 334 | 335 | @Test 336 | public void zipExample() { 337 | Flux.zip( 338 | Flux.range(1, 3), 339 | Flux.range(4, 2), 340 | Flux.range(6, 5) 341 | ).subscribe(System.out::println); 342 | } 343 | 344 | @Test 345 | public void bufferExample() { 346 | Flux.range(1, 13) 347 | .buffer(4) 348 | .subscribe(System.out::println); 349 | } 350 | 351 | @Test 352 | public void windowExample() { 353 | Flux> fluxFlux = Flux.range(101, 20) 354 | .windowUntil(this::isPrime, true); 355 | 356 | fluxFlux.subscribe(window -> window 357 | .collectList() 358 | .subscribe(System.out::println)); 359 | } 360 | 361 | private boolean isPrime(int number) { 362 | return number > 2 363 | && IntStream.rangeClosed(2, (int) Math.sqrt(number)) 364 | .noneMatch(n -> (number % n == 0)); 365 | } 366 | 367 | @Test 368 | public void groupByExample() { 369 | Flux.range(1, 7) 370 | .groupBy(e -> e % 2 == 0 ? "Even" : "Odd") 371 | .subscribe(groupFlux -> groupFlux 372 | .scan( 373 | new LinkedList<>(), 374 | (list, elem) -> { 375 | if (list.size() > 1) { 376 | list.remove(0); 377 | } 378 | list.add(elem); 379 | return list; 380 | }) 381 | .filter(arr -> !arr.isEmpty()) 382 | .subscribe(data -> 383 | System.out.println(groupFlux.key() + ": " + data))); 384 | } 385 | 386 | @Test 387 | public void flatMapExample() throws InterruptedException { 388 | Flux.just("user-1", "user-2", "user-3") 389 | .flatMap(u -> requestBooks(u) 390 | .map(b -> u + "/" + b)) 391 | .subscribe(System.out::println); 392 | 393 | Thread.sleep(1000); 394 | } 395 | 396 | private Flux requestBooks(String user) { 397 | return Flux.range(1, random.nextInt(3) + 1) 398 | .delayElements(Duration.ofMillis(3)) 399 | .map(i -> "book-" + i); 400 | } 401 | 402 | @Test 403 | public void sampleExample() throws InterruptedException { 404 | Flux.range(1, 100) 405 | .delayElements(Duration.ofMillis(1)) 406 | .sample(Duration.ofMillis(20)) 407 | .subscribe(System.out::println); 408 | 409 | Thread.sleep(1000); 410 | } 411 | 412 | @Test 413 | public void doOnExample() { 414 | Flux.just(1, 2, 3) 415 | .concatWith(Flux.error(new RuntimeException("Conn error"))) 416 | .doOnEach(System.out::println) 417 | .subscribe(); 418 | } 419 | 420 | @Test 421 | public void doOnExample2() { 422 | Mono.error(new RuntimeException("test exception")) 423 | .doOnSuccess(signal -> System.out.println("success")) 424 | .doOnError(cause -> System.out.println("error: " + cause.getMessage())) 425 | .subscribe(); 426 | } 427 | 428 | @Test 429 | public void signalProcessing() { 430 | Flux.range(1, 3) 431 | .doOnNext(e -> System.out.println("data : " + e)) 432 | .materialize() 433 | .doOnNext(e -> System.out.println("signal: " + e)) 434 | .dematerialize() 435 | .collectList() 436 | .subscribe(r -> System.out.println("result: " + r)); 437 | } 438 | 439 | @Test 440 | public void signalProcessingWithLog() { 441 | Flux.range(1, 3) 442 | .log("FluxEvents") 443 | .subscribe(e -> {}, e -> {}, () -> {}, s -> s.request(2)); 444 | } 445 | 446 | @Test 447 | public void usingPushOperator() throws InterruptedException { 448 | Flux.push(emitter -> IntStream 449 | .range(2000, 100000) 450 | .forEach(emitter::next)) 451 | .delayElements(Duration.ofMillis(1)) 452 | .subscribe(System.out::println); 453 | 454 | Thread.sleep(1000); 455 | } 456 | 457 | @Test 458 | public void usingCreateOperator() throws InterruptedException { 459 | Flux.create(emitter -> { 460 | emitter.onDispose(() -> System.out.println("Disposed")); 461 | // push events to emitter 462 | }).subscribe(System.out::println); 463 | 464 | Thread.sleep(1000); 465 | } 466 | 467 | @Test 468 | public void usingGenerate() throws InterruptedException { 469 | Flux.generate( 470 | () -> Tuples.of(0L, 1L), 471 | (state, sink) -> { 472 | System.out.println("generated value: " + state.getT2()); 473 | sink.next(state.getT2()); 474 | long newValue = state.getT1() + state.getT2(); 475 | return Tuples.of(state.getT2(), newValue); 476 | }) 477 | .take(7) 478 | .subscribe(System.out::println); 479 | 480 | Thread.sleep(100); 481 | } 482 | 483 | @Test 484 | public void tryWithResources() { 485 | try (Connection conn = Connection.newConnection()) { 486 | conn.getData().forEach( 487 | data -> System.out.println("Received data: " + data) 488 | ); 489 | } catch (Exception e) { 490 | System.out.println("Error: " + e.getMessage()); 491 | } 492 | } 493 | 494 | @Test 495 | public void usingOperator() { 496 | Flux ioRequestResults = Flux.using( 497 | Connection::newConnection, 498 | connection -> Flux.fromIterable(connection.getData()), 499 | Connection::close 500 | ); 501 | 502 | ioRequestResults 503 | .subscribe( 504 | data -> System.out.println("Received data: " + data), 505 | e -> System.out.println("Error: " + e.getMessage()), 506 | () -> System.out.println("Stream finished")); 507 | } 508 | 509 | @Test 510 | public void usingWhenExample() throws InterruptedException { 511 | // Flux.usingWhen( 512 | // Transaction.beginTransaction(), 513 | // transaction -> transaction.insertRows(Flux.just("A", "B")), 514 | // Transaction::commit, 515 | // Transaction::rollback 516 | // ).subscribe( 517 | // d -> System.out.println("onNext: " + d), 518 | // e -> System.out.println("onError: " + e.getMessage()), 519 | // () -> System.out.println("onComplete") 520 | // ); 521 | // 522 | // Thread.sleep(1000); 523 | } 524 | 525 | private Flux recommendedBooks(String userId) { 526 | return Flux.defer(() -> { 527 | if (random.nextInt(10) < 7) { 528 | return Flux.error(new RuntimeException("Conn error")) 529 | .delaySequence(Duration.ofMillis(100)); 530 | } else { 531 | return Flux.just("Blue Mars", "The Expanse") 532 | .delayElements(Duration.ofMillis(50)); 533 | } 534 | }).doOnSubscribe(s -> System.out.println("Request for " + userId)); 535 | } 536 | 537 | @Test 538 | public void handlingErrors() throws InterruptedException { 539 | // Flux.just("user-1") 540 | // .flatMap(user -> 541 | // recommendedBooks(user) 542 | // .retryBackoff(5, Duration.ofMillis(100)) 543 | // .timeout(Duration.ofSeconds(3)) 544 | // .onErrorResume(e -> Flux.just("The Martian")) 545 | // ) 546 | // .subscribe( 547 | // b -> System.out.println("onNext: " + b), 548 | // e -> System.err.println("onError: " + e.getMessage()), 549 | // () -> System.out.println("onComplete") 550 | // ); 551 | // 552 | // Thread.sleep(5000); 553 | } 554 | 555 | @Test 556 | public void coldPublisher() { 557 | Flux coldPublisher = Flux.defer(() -> { 558 | System.out.println("Generating new items"); 559 | return Flux.just(UUID.randomUUID().toString()); 560 | }); 561 | 562 | System.out.println("No data was generated so far"); 563 | coldPublisher.subscribe(e -> System.out.println("onNext: " + e)); 564 | coldPublisher.subscribe(e -> System.out.println("onNext: " + e)); 565 | System.out.println("Data was generated twice for two subscribers"); 566 | } 567 | 568 | @Test 569 | public void connectExample() { 570 | Flux source = Flux.range(0, 3) 571 | .doOnSubscribe(s -> 572 | System.out.println("new subscription for the cold publisher")); 573 | 574 | ConnectableFlux conn = source.publish(); 575 | 576 | conn.subscribe(e -> System.out.println("[Subscriber 1] onNext: " + e)); 577 | conn.subscribe(e -> System.out.println("[Subscriber 2] onNext: " + e)); 578 | 579 | System.out.println("all subscribers are ready, connecting"); 580 | conn.connect(); 581 | } 582 | 583 | @Test 584 | public void cachingExample() throws InterruptedException { 585 | Flux source = Flux.range(0, 2) 586 | .doOnSubscribe(s -> 587 | System.out.println("new subscription for the cold publisher")); 588 | 589 | Flux cachedSource = source.cache(Duration.ofSeconds(1)); 590 | 591 | cachedSource.subscribe(e -> System.out.println("[S 1] onNext: " + e)); 592 | cachedSource.subscribe(e -> System.out.println("[S 2] onNext: " + e)); 593 | 594 | Thread.sleep(1200); 595 | 596 | cachedSource.subscribe(e -> System.out.println("[S 3] onNext: " + e)); 597 | } 598 | 599 | @Test 600 | public void shareExample() throws InterruptedException { 601 | Flux source = Flux.range(0, 5) 602 | .delayElements(Duration.ofMillis(100)) 603 | .doOnSubscribe(s -> 604 | System.out.println("new subscription for the cold publisher")); 605 | 606 | Flux sharedSource = source.share(); 607 | 608 | sharedSource.subscribe(e -> System.out.println("[S 1] onNext: " + e)); 609 | Thread.sleep(400); 610 | sharedSource.subscribe(e -> System.out.println("[S 2] onNext: " + e)); 611 | 612 | Thread.sleep(1000); 613 | } 614 | 615 | @Test 616 | public void elapsedExample() throws InterruptedException { 617 | Flux.range(0, 5) 618 | .delayElements(Duration.ofMillis(100)) 619 | .elapsed() 620 | .subscribe(e -> System.out.println( 621 | String.format("Elapsed %d ms: %d", e.getT1(), e.getT2()))); 622 | 623 | Thread.sleep(1000); 624 | } 625 | 626 | @Test 627 | public void transformExample() { 628 | Function, Flux> logUserInfo = 629 | stream -> stream 630 | .index() 631 | .doOnNext(tp -> 632 | System.out.println( 633 | String.format("[%d] User: %s", tp.getT1(), tp.getT2()))) 634 | .map(Tuple2::getT2); 635 | 636 | Flux.range(1000, 3) 637 | .map(i -> "user-" + i) 638 | .transform(logUserInfo) 639 | .subscribe(e -> System.out.println("onNext: " + e)); 640 | } 641 | 642 | @Test 643 | public void composeExample() { 644 | // Function, Flux> logUserInfo = (stream) -> { 645 | // if (random.nextBoolean()) { 646 | // return stream 647 | // .doOnNext(e -> System.out.println("[path A] User: " + e)); 648 | // } else { 649 | // return stream 650 | // .doOnNext(e -> System.out.println("[path B] User: " + e)); 651 | // } 652 | // }; 653 | // 654 | // Flux publisher = Flux.just("1", "2") 655 | // .compose(logUserInfo); 656 | // 657 | // publisher.subscribe(); 658 | // publisher.subscribe(); 659 | } 660 | 661 | static class Connection implements AutoCloseable { 662 | private final Random rnd = new Random(); 663 | 664 | static Connection newConnection() { 665 | System.out.println("IO Connection created"); 666 | return new Connection(); 667 | } 668 | 669 | public Iterable getData() { 670 | if (rnd.nextInt(10) < 3) { 671 | throw new RuntimeException("Communication error"); 672 | } 673 | return Arrays.asList("Some", "data"); 674 | } 675 | 676 | @Override 677 | public void close() { 678 | System.out.println("IO Connection closed"); 679 | } 680 | } 681 | 682 | static class Transaction { 683 | private static final Random random = new Random(); 684 | private final int id; 685 | 686 | public Transaction(int id) { 687 | this.id = id; 688 | System.out.println(String.format("[T: {%d}] created", id)); 689 | } 690 | 691 | public static Mono beginTransaction() { 692 | return Mono.defer(() -> 693 | Mono.just(new Transaction(random.nextInt(1000)))); 694 | } 695 | 696 | public Flux insertRows(Publisher rows) { 697 | return Flux.from(rows) 698 | .delayElements(Duration.ofMillis(100)) 699 | .flatMap(row -> { 700 | if (random.nextInt(10) < 2) { 701 | return Mono.error(new RuntimeException("Error on: " + row)); 702 | } else { 703 | return Mono.just(row); 704 | } 705 | }); 706 | } 707 | 708 | public Mono commit() { 709 | return Mono.defer(() -> { 710 | System.out.println(String.format("[T: {%d}] commit", id)); 711 | if (random.nextBoolean()) { 712 | return Mono.empty(); 713 | } else { 714 | return Mono.error(new RuntimeException("Conflict")); 715 | } 716 | }); 717 | } 718 | 719 | public Mono rollback() { 720 | return Mono.defer(() -> { 721 | System.out.println(String.format("[T: {%d}] rollback", id)); 722 | if (random.nextBoolean()) { 723 | return Mono.empty(); 724 | } else { 725 | return Mono.error(new RuntimeException("Conn error")); 726 | } 727 | }); 728 | } 729 | } 730 | 731 | @Test 732 | public void concatMapExample() { 733 | List users = new ArrayList<>(); 734 | users.add(new User(null, "lily")); 735 | users.add(new User("2", "lucy")); 736 | Flux.fromIterable(users) 737 | .concatMap(user -> { 738 | if (user.id == null) { 739 | return Mono.empty(); 740 | } else { 741 | return Mono.just(user.id); 742 | } 743 | }) 744 | .next() 745 | .flatMap(Mono::just) 746 | .subscribe(System.out::println); 747 | } 748 | 749 | @AllArgsConstructor 750 | @NoArgsConstructor 751 | static class User { 752 | public String id, name; 753 | } 754 | } 755 | -------------------------------------------------------------------------------- /src/test/java/org/hongxi/sample/reactor/ReactiveContextTest.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.reactor; 2 | 3 | import org.junit.Test; 4 | import reactor.core.publisher.Mono; 5 | import reactor.util.context.Context; 6 | 7 | public class ReactiveContextTest { 8 | 9 | @Test 10 | public void showcaseContext() { 11 | printCurrentContext("top") 12 | .subscriberContext(Context.of("top", "context")) 13 | .flatMap(__ -> printCurrentContext("middle")) 14 | .subscriberContext(Context.of("middle", "context")) 15 | .flatMap(__ -> printCurrentContext("bottom")) 16 | .subscriberContext(Context.of("bottom", "context")) 17 | .flatMap(__ -> printCurrentContext("initial")) 18 | .block(); 19 | } 20 | 21 | void print(String id, Context context) { 22 | System.out.println(id + " {"); 23 | System.out.print(" "); 24 | System.out.println(context); 25 | System.out.println("}"); 26 | System.out.println(); 27 | } 28 | 29 | Mono printCurrentContext(String id) { 30 | return Mono 31 | .subscriberContext() 32 | .doOnNext(context -> print(id, context)); 33 | } 34 | } 35 | -------------------------------------------------------------------------------- /src/test/java/org/hongxi/sample/webflux/SampleApplicationTests.java: -------------------------------------------------------------------------------- 1 | package org.hongxi.sample.webflux; 2 | 3 | import org.hamcrest.MatcherAssert; 4 | import org.hamcrest.Matchers; 5 | import org.hongxi.sample.webflux.model.Order; 6 | import org.junit.jupiter.api.Test; 7 | import org.junit.runner.RunWith; 8 | import org.springframework.beans.factory.annotation.Autowired; 9 | import org.springframework.boot.test.autoconfigure.web.reactive.AutoConfigureWebTestClient; 10 | import org.springframework.boot.test.context.SpringBootTest; 11 | import org.springframework.http.MediaType; 12 | import org.springframework.test.context.junit4.SpringRunner; 13 | import org.springframework.test.web.reactive.server.WebTestClient; 14 | import org.springframework.web.reactive.function.BodyInserters; 15 | 16 | @RunWith(SpringRunner.class) 17 | @SpringBootTest 18 | @AutoConfigureWebTestClient 19 | class SampleApplicationTests { 20 | 21 | @Autowired 22 | WebTestClient webTestClient; 23 | 24 | @Test 25 | public void shouldBeAbleToPostOrder() { 26 | Order order = new Order("123456"); 27 | webTestClient 28 | .post() 29 | .uri("/order/create?userId=123") 30 | .contentType(MediaType.APPLICATION_JSON) 31 | .body(BodyInserters.fromValue(order)) 32 | .exchange() 33 | .expectStatus().isOk(); 34 | } 35 | 36 | @Test 37 | public void shouldBeNotFound() { 38 | webTestClient 39 | .get() 40 | .uri("/order/list?userId=123") 41 | .exchange() 42 | .expectStatus().isNotFound(); 43 | } 44 | 45 | @Test 46 | public void shouldBeAbleToFindOrder() { 47 | shouldBeAbleToPostOrder(); 48 | webTestClient 49 | .get() 50 | .uri("/order/detail/123456?userId=123") 51 | .accept(MediaType.APPLICATION_JSON) 52 | .exchange() 53 | .expectStatus().isOk() 54 | .expectBody(Order.class) 55 | .consumeWith(e -> MatcherAssert.assertThat(e.getResponseBody(), Matchers.hasProperty("id", Matchers.equalTo("123456")))); 56 | } 57 | 58 | @Test 59 | public void shouldBeAbleToDeleteOrder() { 60 | shouldBeAbleToPostOrder(); 61 | webTestClient 62 | .post() 63 | .uri("/order/delete?userId=123&name=lily") 64 | .contentType(MediaType.APPLICATION_FORM_URLENCODED) 65 | .body(BodyInserters.fromFormData("id", "123456")) 66 | .accept(MediaType.APPLICATION_JSON) 67 | .exchange() 68 | .expectStatus().isOk() 69 | .expectBody(Order.class) 70 | .consumeWith(e -> MatcherAssert.assertThat(e.getResponseBody(), Matchers.hasProperty("id", Matchers.equalTo("123456")))); 71 | } 72 | 73 | } 74 | --------------------------------------------------------------------------------