Spring Boot 入门之监听器-AOP-拦截器-过滤器

一、前言

上一篇《Spring Boot 入门之基础篇(一)》介绍了 Spring Boot 的环境搭建以及项目启动打包等基础内容,本篇继续深入介绍 Spring Boot 与 Web 开发相关的知识。

二、整合模板引擎

由于 jsp 不被 SpringBoot 推荐使用,所以模板引擎主要介绍 Freemarker 和 Thymeleaf。

2.1 整合 Freemarker

2.1.1 添加 Freemarker 依赖

 
  1. <dependency>
  2. <groupId>org.springframework.boot</groupId>
  3. <artifactId>spring-boot-starter-freemarker</artifactId>
  4. </dependency>

2.1.2 添加 Freemarker 模板配置

在 application.properties 中添加如下内容:

 
  1. spring.freemarker.allow-request-override=false
  2. spring.freemarker.cache=true
  3. spring.freemarker.check-template-location=true
  4. spring.freemarker.charset=UTF-8
  5. spring.freemarker.content-type=text/html
  6. spring.freemarker.expose-request-attributes=false
  7. spring.freemarker.expose-session-attributes=false
  8. spring.freemarker.expose-spring-macro-helpers=false
  9. spring.freemarker.prefix=
  10. spring.freemarker.suffix=.ftl

上述配置都是默认值。

2.1.3 Freemarker 案例演示

在 controller 包中创建 FreemarkerController:

 
  1. @Controller
  2. @RequestMapping("freemarker")
  3. public class FreemarkerController {
  4.  
  5. @RequestMapping("hello")
  6. public String hello(Map<String,Object> map) {
  7.  
  8. map.put("msg", "Hello Freemarker");
  9. return "hello";
  10. }
  11. }

在 templates 目录中创建名为 hello.ftl 文件,内容如下:

 
  1. <!DOCTYPE html>
  2. <html lang="zh">
  3. <head>
  4. <meta charset="UTF-8">
  5. <title>Document</title>
  6. <link href="/css/index.css" rel="stylesheet"/>
  7. </head>
  8. <body>
  9. <div class="container">
  10. <h2>${msg}</h2>
  11. </div>
  12. </body>
  13. </html>

结果如下:

image

2.2 整合 Thymeleaf

2.2.1 添加 Thymeleaf 依赖

在 pom.xml 文件中添加:

 
  1. <dependency>
  2. <groupId>org.springframework.boot</groupId>
  3. <artifactId>spring-boot-starter-thymeleaf</artifactId>
  4. </dependency>

2.2.2 添加 Thymeleaf 模板配置

在 application.properties 中添加如下内容:

 
  1. spring.thymeleaf.cache=true
  2. spring.thymeleaf.prefix=classpath:/templates/
  3. spring.thymeleaf.suffix=.html
  4. spring.thymeleaf.mode=HTML5
  5. spring.thymeleaf.encoding=UTF-8
  6. spring.thymeleaf.content-type=text/html

上述配置都是默认值。

2.2.3 Thymeleaf 案例演示

在 controller 包中创建 ThymeleafController:

 
  1. @Controller
  2. @RequestMapping("thymeleaf")
  3. public class ThymeleafController {
  4.  
  5. @RequestMapping("hello")
  6. public String hello(Map<String,Object> map) {
  7. map.put("msg", "Hello Thymeleaf");
  8. return "hello";
  9. }
  10. }

在 template 目录下创建名为 hello.html 的文件,内容如下:

 
  1. <!DOCTYPE html>
  2. <html lang="zh">
  3. <head>
  4. <meta charset="UTF-8">
  5. <title>Document</title>
  6. <link href="/css/index.css" rel="stylesheet"/>
  7. </head>
  8. <body>
  9. <div class="container">
  10. <h2 th:text="${msg}"></h2>
  11. </div>
  12. </body>
  13. </html>

结果如下:

image

三、整合 Fastjson

3.1 添加依赖

 
  1. <dependency>
  2. <groupId>com.alibaba</groupId>
  3. <artifactId>fastjson</artifactId>
  4. <version>1.2.35</version>
  5. </dependency>

3.2 整合 Fastjson

创建一个配置管理类 WebConfig ,如下:

 
  1. @Configuration
  2. public class WebConfig {
  3.  
  4. @Bean
  5. public HttpMessageConverters fastJsonHttpMessageConverters() {
  6. FastJsonHttpMessageConverter fastJsonHttpMessageConverter = new FastJsonHttpMessageConverter();
  7.  
  8. FastJsonConfig fastJsonConfig = new FastJsonConfig();
  9. fastJsonConfig.setSerializerFeatures(SerializerFeature.PrettyFormat);
  10.  
  11. fastJsonHttpMessageConverter.setFastJsonConfig(fastJsonConfig);
  12.  
  13. HttpMessageConverter<?> converter = fastJsonHttpMessageConverter;
  14.  
  15. return new HttpMessageConverters(converter);
  16.  
  17. }
  18. }

3.3 演示案例:

创建一个实体类 User:

 
  1. public class User {
  2.  
  3. private Integer id;
  4.  
  5. private String username;
  6.  
  7. private String password;
  8.  
  9. private Date birthday;
  10.  
  11. }

getter 和 setter 此处省略。

创建控制器类 FastjsonController :

 
  1. @Controller
  2. @RequestMapping("fastjson")
  3. public class FastJsonController {
  4.  
  5. @RequestMapping("/test")
  6. @ResponseBody
  7. public User test() {
  8. User user = new User();
  9.  
  10. user.setId(1);
  11. user.setUsername("jack");
  12. user.setPassword("jack123");
  13. user.setBirthday(new Date());
  14.  
  15. return user;
  16. }
  17. }

打开浏览器,访问 http://localhost:8080/fastjson/test,结果如下图:

image

此时,还不能看出 Fastjson 是否正常工作,我们在 User 类中使用 Fastjson 的注解,如下内容:

 
  1. @JSONField(format="yyyy-MM-dd")
  2. private Date birthday;

再次访问 http://localhost:8080/fastjson/test,结果如下图:

image

日期格式与我们修改的内容格式一致,说明 Fastjson 整合成功。

四、自定义 Servlet

4.1 编写 Servlet

 
  1. public class ServletTest extends HttpServlet {
  2.  
  3. @Override
  4. protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
  5. doPost(req, resp);
  6. }
  7.  
  8. @Override
  9. protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
  10. resp.setContentType("text/html;charset=utf-8");
  11. resp.getWriter().write("自定义 Servlet");
  12. }
  13.  
  14. }

4.2 注册 Servlet

将 Servelt 注册成 Bean。在上文创建的 WebConfig 类中添加如下代码:

 
  1. @Bean
  2. public ServletRegistrationBean servletRegistrationBean() {
  3. return new ServletRegistrationBean(new ServletTest(),"/servletTest");
  4. }

结果如下:

image

五、自定义过滤器/第三方过滤器

5.1 编写过滤器

 
  1. public class TimeFilter implements Filter {
  2.  
  3. @Override
  4. public void init(FilterConfig filterConfig) throws ServletException {
  5. System.out.println("=======初始化过滤器=========");
  6. }
  7.  
  8. @Override
  9. public void doFilter(ServletRequest request, ServletResponse response, FilterChain filterChain)
  10. throws IOException, ServletException {
  11.  
  12. long start = System.currentTimeMillis();
  13.  
  14. filterChain.doFilter(request, response);
  15.  
  16. System.out.println("filter 耗时:" + (System.currentTimeMillis() - start));
  17.  
  18. }
  19.  
  20. @Override
  21. public void destroy() {
  22. System.out.println("=======销毁过滤器=========");
  23. }
  24.  
  25. }

5.2 注册过滤器

要是该过滤器生效,有两种方式:

1) 使用 @Component 注解

2) 添加到过滤器链中,此方式适用于使用第三方的过滤器。将过滤器写到 WebConfig 类中,如下:

 
  1. @Bean
  2. public FilterRegistrationBean timeFilter() {
  3. FilterRegistrationBean registrationBean = new FilterRegistrationBean();
  4.  
  5. TimeFilter timeFilter = new TimeFilter();
  6. registrationBean.setFilter(timeFilter);
  7.  
  8. List<String> urls = new ArrayList<>();
  9. urls.add("/*");
  10. registrationBean.setUrlPatterns(urls);
  11.  
  12. return registrationBean;
  13. }

结果如下:

image

六、自定义监听器

6.1 编写监听器

 
  1. public class ListenerTest implements ServletContextListener {
  2.  
  3. @Override
  4. public void contextInitialized(ServletContextEvent sce) {
  5. System.out.println("监听器初始化...");
  6. }
  7.  
  8. @Override
  9. public void contextDestroyed(ServletContextEvent sce) {
  10.  
  11. }
  12.  
  13. }

6.2 注册监听器

注册监听器为 Bean,在 WebConfig 配置类中添加如下代码:

 
  1. @Bean
  2. public ServletListenerRegistrationBean<ListenerTest> servletListenerRegistrationBean() {
  3. return new ServletListenerRegistrationBean<ListenerTest>(new ListenerTest());
  4. }

当启动容器时,结果如下:

image

针对自定义 Servlet、Filter 和 Listener 的配置,还有另一种方式:

 
  1. @SpringBootApplication
  2. public class SpringbootWebApplication implements ServletContextInitializer {
  3.  
  4. @Override
  5. public void onStartup(ServletContext servletContext) throws ServletException {
  6. // 配置 Servlet
  7. servletContext.addServlet("servletTest",new ServletTest())
  8. .addMapping("/servletTest");
  9. // 配置过滤器
  10. servletContext.addFilter("timeFilter",new TimeFilter())
  11. .addMappingForUrlPatterns(EnumSet.of(DispatcherType.REQUEST),true,"/*");
  12. // 配置监听器
  13. servletContext.addListener(new ListenerTest());
  14. }
  15.  
  16. public static void main(String[] args) {
  17. SpringApplication.run(SpringbootWebApplication.class, args);
  18. }
  19. }

七、自定义拦截器

7.1 编写拦截器

使用 @Component 让 Spring 管理其生命周期:

 
  1. @Component
  2. public class TimeInterceptor implements HandlerInterceptor {
  3.  
  4. @Override
  5. public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
  6.  
  7. System.out.println("========preHandle=========");
  8. System.out.println(((HandlerMethod)handler).getBean().getClass().getName());
  9. System.out.println(((HandlerMethod)handler).getMethod().getName());
  10.  
  11. request.setAttribute("startTime", System.currentTimeMillis());
  12.  
  13. return true;
  14. }
  15.  
  16. @Override
  17. public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView)
  18. throws Exception {
  19.  
  20. System.out.println("========postHandle=========");
  21. Long start = (Long) request.getAttribute("startTime");
  22. System.out.println("耗时:"+(System.currentTimeMillis() - start));
  23. }
  24.  
  25. @Override
  26. public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception exception)
  27. throws Exception {
  28.  
  29. System.out.println("========afterCompletion=========");
  30. Long start = (Long) request.getAttribute("startTime");
  31. System.out.println("耗时:"+(System.currentTimeMillis() - start));
  32.  
  33. System.out.println(exception);
  34. }
  35.  
  36. }

7.2 注册拦截器

编写拦截器后,我们还需要将其注册到拦截器链中,如下配置:

 
  1. @Configuration
  2. public class WebConfig extends WebMvcConfigurerAdapter{
  3.  
  4. @Autowired
  5. private TimeInterceptor timeInterceptor;
  6.  
  7.  
  8. @Override
  9. public void addInterceptors(InterceptorRegistry registry) {
  10. registry.addInterceptor(timeInterceptor);
  11. }
  12.  
  13. }

请求一个 controller ,结果如下:

image

八、配置 AOP 切面

8.1 添加依赖

 
  1. <dependency>
  2. <groupId>org.springframework.boot</groupId>
  3. <artifactId>spring-boot-starter-aop</artifactId>
  4. </dependency>

8.2 编写切面类

使用 @Component,@Aspect 标记到切面类上:

 
  1. @Aspect
  2. @Component
  3. public class TimeAspect {
  4.  
  5. @Around("execution(* com.light.springboot.controller.FastJsonController..*(..))")
  6. public Object method(ProceedingJoinPoint pjp) throws Throwable {
  7.  
  8. System.out.println("=====Aspect处理=======");
  9. Object[] args = pjp.getArgs();
  10. for (Object arg : args) {
  11. System.out.println("参数为:" + arg);
  12. }
  13.  
  14. long start = System.currentTimeMillis();
  15.  
  16. Object object = pjp.proceed();
  17.  
  18. System.out.println("Aspect 耗时:" + (System.currentTimeMillis() - start));
  19.  
  20. return object;
  21. }
  22. }

请求 FastJsonController 控制器的方法,结果如下:

image

九、错误处理

9.1 友好页面

先演示非友好页面,修改 FastJsonController 类中的 test 方法:

 
  1. @RestController
  2. @RequestMapping("fastjson")
  3. public class FastJsonController {
  4.  
  5. @RequestMapping("/test")
  6. public User test() {
  7. User user = new User();
  8.  
  9. user.setId(1);
  10. user.setUsername("jack");
  11. user.setPassword("jack123");
  12. user.setBirthday(new Date());
  13.  
  14. // 模拟异常
  15. int i = 1/0;
  16.  
  17. return user;
  18. }
  19. }

浏览器请求:http://localhost:8080/fastjson/test,结果如下:

image

当系统报错时,返回到页面的内容通常是一些杂乱的代码段,这种显示对用户来说不友好,因此我们需要自定义一个友好的提示系统异常的页面。

在 src/main/resources 下创建 /public/error,在该目录下再创建一个名为 5xx.html 文件,该页面的内容就是当系统报错时返回给用户浏览的内容:

 
  1. <!DOCTYPE html>
  2. <html lang="zh">
  3. <head>
  4. <meta charset="UTF-8">
  5. <title>系统错误</title>
  6. <link href="/css/index.css" rel="stylesheet"/>
  7. </head>
  8. <body>
  9. <div class="container">
  10. <h2>系统内部错误</h2>
  11. </div>
  12. </body>
  13. </html>

路径时固定的,Spring Boot 会在系统报错时将返回视图指向该目录下的文件。

如下图:

image

上边处理的 5xx 状态码的问题,接下来解决 404 状态码的问题。

当出现 404 的情况时,用户浏览的页面也不够友好,因此我们也需要自定义一个友好的页面给用户展示。

在 /public/error 目录下再创建一个名为 404.html 的文件:

 
  1. <!DOCTYPE html>
  2. <html lang="zh">
  3. <head>
  4. <meta charset="UTF-8">
  5. <title>访问异常</title>
  6. <link href="/css/index.css" rel="stylesheet"/>
  7. </head>
  8. <body>
  9. <div class="container">
  10. <h2>找不到页面</h2>
  11. </div>
  12. </body>
  13. </html>

我们请求一个不存在的资源,如:http://localhost:8080/fastjson/test2,结果如下图:

image

9.2 全局异常捕获

如果项目前后端是通过 JSON 进行数据通信,则当出现异常时可以常用如下方式处理异常信息。

编写一个类充当全局异常的处理类,需要使用 @ControllerAdvice 和 @ExceptionHandler 注解:

 
  1. @ControllerAdvice
  2. public class GlobalDefaultExceptionHandler {
  3.  
  4. /**
  5. * 处理 Exception 类型的异常
  6. * @param e
  7. * @return
  8. */
  9. @ExceptionHandler(Exception.class)
  10. @ResponseBody
  11. public Map<String,Object> defaultExceptionHandler(Exception e) {
  12.  
  13. Map<String,Object> map = new HashMap<String,Object>();
  14. map.put("code", 500);
  15. map.put("msg", e.getMessage());
  16. return map;
  17. }
  18. }

其中,方法名为任意名,入参一般使用 Exception 异常类,方法返回值可自定义。

启动项目,访问 http://localhost:8080/fastjson/test,结果如下图:

image

我们还可以自定义异常,在全局异常的处理类中捕获和判断,从而对不同的异常做出不同的处理。

十、文件上传和下载

10.1 添加依赖

 
  1. <!-- 工具 -->
  2. <dependency>
  3. <groupId>commons-io</groupId>
  4. <artifactId>commons-io</artifactId>
  5. <version>2.4</version>
  6. </dependency>

10.2 实现

编写一个实体类,用于封装返回信息:

 
  1. public class FileInfo {
  2.  
  3. private String path;
  4.  
  5. public FileInfo(String path) {
  6. this.path = path;
  7. }
  8.  
  9. public String getPath() {
  10. return path;
  11. }
  12.  
  13. public void setPath(String path) {
  14. this.path = path;
  15. }
  16.  
  17. }

编写 Controller,用于处理文件上传下载:

 
  1. @RestController
  2. @RequestMapping("/file")
  3. public class FileController {
  4.  
  5. private String path = "d:\\";
  6.  
  7. @PostMapping
  8. public FileInfo upload(MultipartFile file) throws Exception {
  9.  
  10. System.out.println(file.getName());
  11. System.out.println(file.getOriginalFilename());
  12. System.out.println(file.getSize());
  13.  
  14. File localFile = new File(path, file.getOriginalFilename());
  15.  
  16. file.transferTo(localFile);
  17.  
  18. return new FileInfo(localFile.getAbsolutePath());
  19. }
  20.  
  21. @GetMapping("/{id}")
  22. public void download(@PathVariable String id, HttpServletRequest request, HttpServletResponse response) {
  23. try (InputStream inputStream = new FileInputStream(new File(path, id + ".jpg"));
  24. OutputStream outputStream = response.getOutputStream();) {
  25.  
  26. response.setContentType("application/x-download");
  27. response.addHeader("Content-Disposition", "attachment;filename=" + id + ".jpg");
  28.  
  29. IOUtils.copy(inputStream, outputStream);
  30. } catch (Exception e) {
  31. e.printStackTrace();
  32. }
  33. }
  34. }

基本上都是在学习 javaweb 时用到的 API。

文件上传测试结果如下图:

image

十一、CORS 支持

前端页面:

 
  1. <!DOCTYPE html>
  2. <html lang="en">
  3.  
  4. <head>
  5. <meta charset="UTF-8">
  6. <title>跨域测试</title>
  7. </head>
  8.  
  9. <body>
  10. <button id="test">测试</button>
  11. <script type="text/javascript" src="jquery-1.12.3.min.js"></script>
  12. <script type="text/javascript">
  13. $(function() {
  14. $("#test").on("click", function() {
  15. $.ajax({
  16. "url": "http://localhost:8080/fastjson/test",
  17. "type": "get",
  18. "dataType": "json",
  19. "success": function(data) {
  20. console.log(data);
  21. }
  22. })
  23. });
  24. });
  25. </script>
  26. </body>
  27.  
  28. </html>

通过 http 容器启动前端页面代码,笔者使用 Sublime Text 的插件启动的,测试结果如下:

image

从图中可知,前端服务器启动端口为 8088 与后端服务器 8080 不同源,因此出现跨域的问题。

现在开始解决跨域问题,可以两种维度控制客户端请求。

粗粒度控制

方式一

 
  1. @Configuration
  2. public class WebConfig {
  3.  
  4. @Bean
  5. public WebMvcConfigurer corsConfigurer() {
  6. return new WebMvcConfigurerAdapter() {
  7. @Override
  8. public void addCorsMappings(CorsRegistry registry) {
  9. registry.addMapping("/fastjson/**")
  10. .allowedOrigins("http://localhost:8088");// 允许 8088 端口访问
  11. }
  12. };
  13. }
  14. }

方式二

 
  1. @Configuration
  2. public class WebConfig extends WebMvcConfigurerAdapter{
  3.  
  4. @Override
  5. public void addCorsMappings(CorsRegistry registry) {
  6. registry.addMapping("/fastjson/**")
  7. .allowedOrigins("http://localhost:8088");// 允许 8088 端口访问
  8. }
  9. }

配置后,重新发送请求,结果如下:

image

细粒度控制

在 FastJsonController 类中的方法上添加 @CrossOrigin(origins="xx") 注解:

 
  1. @RequestMapping("/test")
  2. @CrossOrigin(origins="http://localhost:8088")
  3. public User test() {
  4. User user = new User();
  5.  
  6. user.setId(1);
  7. user.setUsername("jack");
  8. user.setPassword("jack123");
  9. user.setBirthday(new Date());
  10.  
  11. return user;
  12. }

在使用该注解时,需要注意 @RequestMapping 使用的请求方式类型,即 GET 或 POST。

十二、整合 WebSocket

12.1 添加依赖

 
  1. <dependency>
  2. <groupId>org.springframework.boot</groupId>
  3. <artifactId>spring-boot-starter-websocket</artifactId>
  4. </dependency>

12.2 实现方式

方式一:

该方式只适用于通过 jar 包直接运行项目的情况。

WebSocket 配置类:

 
  1. @Configuration
  2. public class WebSocketConfig {
  3.  
  4. @Bean
  5. public ServerEndpointExporter serverEndpointExporter() {
  6. return new ServerEndpointExporter();
  7. }
  8.  
  9. }

WebSocket 处理类:

 
  1. @ServerEndpoint(value = "/webSocketServer/{userName}")
  2. @Component
  3. public class WebSocketServer {
  4.  
  5. private static final Set<WebSocketServer> connections = new CopyOnWriteArraySet<>();
  6.  
  7. private String nickname;
  8. private Session session;
  9.  
  10. private static String getDatetime(Date date) {
  11. SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
  12. return format.format(date);
  13. }
  14.  
  15. @OnOpen
  16. public void start(@PathParam("userName") String userName, Session session) {
  17. this.nickname = userName;
  18. this.session = session;
  19. connections.add(this);
  20. String message = String.format("* %s %s", nickname, "加入聊天!");
  21. broadcast(message);
  22. }
  23.  
  24. @OnClose
  25. public void end() {
  26. connections.remove(this);
  27. String message = String.format("* %s %s", nickname, "退出聊天!");
  28. broadcast(message);
  29. }
  30.  
  31. @OnMessage
  32. public void pushMsg(String message) {
  33. broadcast("【" + this.nickname + "】" + getDatetime(new Date()) + " : " + message);
  34. }
  35.  
  36. @OnError
  37. public void onError(Throwable t) throws Throwable {
  38.  
  39. }
  40.  
  41. private static void broadcast(String msg) {
  42. // 广播形式发送消息
  43. for (WebSocketServer client : connections) {
  44. try {
  45. synchronized (client) {
  46. client.session.getBasicRemote().sendText(msg);
  47. }
  48. } catch (IOException e) {
  49. connections.remove(client);
  50. try {
  51. client.session.close();
  52. } catch (IOException e1) {
  53. e.printStackTrace();
  54. }
  55. String message = String.format("* %s %s", client.nickname, "断开连接");
  56. broadcast(message);
  57. }
  58. }
  59. }
  60. }

前端页面:

 
  1. <!DOCTYPE html>
  2. <html>
  3.  
  4. <head lang="zh">
  5. <meta charset="UTF-8">
  6. <link rel="stylesheet" href="css/bootstrap.min.css">
  7. <link rel="stylesheet" href="css/bootstrap-theme.min.css">
  8. <script src="js/jquery-1.12.3.min.js"></script>
  9. <script src="js/bootstrap.js"></script>
  10. <style type="text/css">
  11. #msg {
  12. height: 400px;
  13. overflow-y: auto;
  14. }
  15.  
  16. #userName {
  17. width: 200px;
  18. }
  19.  
  20. #logout {
  21. display: none;
  22. }
  23. </style>
  24. <title>webSocket测试</title>
  25. </head>
  26.  
  27. <body>
  28. <div class="container">
  29. <div class="page-header" id="tou">webSocket及时聊天Demo程序</div>
  30. <p class="text-right" id="logout">
  31. <button class="btn btn-danger" id="logout-btn">退出</button>
  32. </p>
  33. <div class="well" id="msg"></div>
  34. <div class="col-lg">
  35. <div class="input-group">
  36. <input type="text" class="form-control" placeholder="发送信息..." id="message"> <span class="input-group-btn">
  37. <button class="btn btn-default" type="button" id="send"
  38. disabled="disabled">发送</button>
  39. </span>
  40. </div>
  41. <div class="input-group">
  42. <input id="userName" type="text" class="form-control" name="userName" placeholder="输入您的用户名" />
  43. <button class="btn btn-default" type="button" id="connection-btn">建立连接</button>
  44. </div>
  45. <!-- /input-group -->
  46. </div>
  47. <!-- /.col-lg-6 -->
  48. </div>
  49. <!-- /.row -->
  50. </div>
  51. <script type="text/javascript">
  52. $(function() {
  53. var websocket;
  54. $("#connection-btn").bind("click", function() {
  55. var userName = $("#userName").val();
  56. if (userName == null || userName == "") {
  57. alert("请输入您的用户名");
  58. return;
  59. }
  60. connection(userName);
  61. });
  62.  
  63. function connection(userName) {
  64. var host = window.location.host;
  65. if ('WebSocket' in window) {
  66. websocket = new WebSocket("ws://" + host +
  67. "/webSocketServer/" + userName);
  68. } else if ('MozWebSocket' in window) {
  69. websocket = new MozWebSocket("ws://" + host +
  70. "/webSocketServer/" + userName);
  71. }
  72. websocket.onopen = function(evnt) {
  73. $("#tou").html("链接服务器成功!")
  74. $("#send").prop("disabled", "");
  75. $("#connection-btn").prop("disabled", "disabled");
  76. $("#logout").show();
  77. };
  78. websocket.onmessage = function(evnt) {
  79. $("#msg").html($("#msg").html() + "<br/>" + evnt.data);
  80. };
  81. websocket.onerror = function(evnt) {
  82. $("#tou").html("报错!")
  83. };
  84. websocket.onclose = function(evnt) {
  85. $("#tou").html("与服务器断开了链接!");
  86. $("#send").prop("disabled", "disabled");
  87. $("#connection-btn").prop("disabled", "");
  88. $("#logout").hide();
  89. }
  90. }
  91.  
  92. function send() {
  93. if (websocket != null) {
  94. var $message = $("#message");
  95. var data = $message.val();
  96. if (data == null || data == "") {
  97. return;
  98. }
  99. websocket.send(data);
  100. $message.val("");
  101. } else {
  102. alert('未与服务器链接.');
  103. }
  104. }
  105.  
  106. $('#send').bind('click', function() {
  107. send();
  108. });
  109.  
  110. $(document).on("keypress", function(event) {
  111. if (event.keyCode == "13") {
  112. send();
  113. }
  114. });
  115.  
  116. $("#logout-btn").on("click", function() {
  117. websocket.close(); //关闭TCP连接
  118. });
  119. });
  120. </script>
  121. </body>
  122.  
  123. </html>

演示图如下:

image

如果使用该方式实现 WebSocket 功能并打包成 war 运行会报错:

 
  1. javax.websocket.DeploymentException: Multiple Endpoints may not be deployed to the same path

方式二:

该方式适用于 jar 包方式运行和 war 方式运行。

WebSocket 配置类:

 
  1. @Configuration
  2. @EnableWebSocket
  3. public class WebSocketConfig implements WebSocketConfigurer {
  4. @Override
  5. public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
  6. registry.addHandler(webSocketServer(), "/webSocketServer/*");
  7. }
  8.  
  9. @Bean
  10. public WebSocketHandler webSocketServer() {
  11. return new WebSocketServer();
  12. }
  13. }

WebSocket 处理类:

 
  1. public class WebSocketServer extends TextWebSocketHandler {
  2.  
  3. private static final Map<WebSocketSession, String> connections = new ConcurrentHashMap<>();
  4.  
  5. private static String getDatetime(Date date) {
  6. SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
  7. return format.format(date);
  8. }
  9.  
  10. /**
  11. * 建立连接
  12. */
  13. @Override
  14. public void afterConnectionEstablished(WebSocketSession session) throws Exception {
  15.  
  16. String uri = session.getUri().toString();
  17. String userName = uri.substring(uri.lastIndexOf("/") + 1);
  18.  
  19. String nickname = URLDecoder.decode(userName, "utf-8");
  20.  
  21. connections.put(session, nickname);
  22. String message = String.format("* %s %s", nickname, "加入聊天!");
  23.  
  24. broadcast(new TextMessage(message));
  25. }
  26.  
  27. /**
  28. * 断开连接
  29. */
  30. @Override
  31. public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
  32. String nickname = connections.remove(session);
  33. String message = String.format("* %s %s", nickname, "退出聊天!");
  34.  
  35. broadcast(new TextMessage(message));
  36. }
  37.  
  38. /**
  39. * 处理消息
  40. */
  41. @Override
  42. protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
  43. String msg = "【" + connections.get(session) + "】" + getDatetime(new Date()) + " : " + message.getPayload();
  44.  
  45. broadcast(new TextMessage(msg));
  46. }
  47.  
  48. private static void broadcast(TextMessage msg) {
  49. // 广播形式发送消息
  50. for (WebSocketSession session : connections.keySet()) {
  51. try {
  52. synchronized (session) {
  53. session.sendMessage(msg);
  54. }
  55. } catch (Exception e) {
  56. connections.remove(session);
  57. try {
  58. session.close();
  59. } catch (Exception e2) {
  60. e2.printStackTrace();
  61. }
  62. String message = String.format("* %s %s", connections.get(session), "断开连接");
  63. broadcast(new TextMessage(message));
  64. }
  65. }
  66. }
  67. }

运行结果与上图一致。

十三、整合 Swagger2

13.1 添加依赖

 
  1. <dependency>
  2. <groupId>io.springfox</groupId>
  3. <artifactId>springfox-swagger2</artifactId>
  4. <version>2.7.0</version>
  5. </dependency>
  6. <dependency>
  7. <groupId>io.springfox</groupId>
  8. <artifactId>springfox-swagger-ui</artifactId>
  9. <version>2.7.0</version>
  10. </dependency>

13.2 配置

重新创建一个配置类,如下:

 
  1. @Configuration
  2. @EnableSwagger2
  3. public class Swagger2Configuration {
  4.  
  5. @Bean
  6. public Docket accessToken() {
  7. return new Docket(DocumentationType.SWAGGER_2)
  8. .groupName("api")// 定义组
  9. .select() // 选择那些路径和 api 会生成 document
  10. .apis(RequestHandlerSelectors.basePackage("com.light.springboot.controller")) // 拦截的包路径
  11. .paths(PathSelectors.regex("/*/.*"))// 拦截的接口路径
  12. .build() // 创建
  13. .apiInfo(apiInfo()); // 配置说明
  14. }
  15.  
  16. private ApiInfo apiInfo() {
  17. return new ApiInfoBuilder()//
  18. .title("Spring Boot 之 Web 篇")// 标题
  19. .description("spring boot Web 相关内容")// 描述
  20. .termsOfServiceUrl("http://www.extlight.com")//
  21. .contact(new Contact("moonlightL", "http://www.extlight.com", "[email protected]"))// 联系
  22. .version("1.0")// 版本
  23. .build();
  24. }
  25. }

为了能更好的说明接口信息,我们还可以在 Controller 类上使用 Swagger2 相关注解说明信息。

我们以 FastJsonController 为例:

 
  1. @Api(value = "FastJson测试", tags = { "测试接口" })
  2. @RestController
  3. @RequestMapping("fastjson")
  4. public class FastJsonController {
  5.  
  6. @ApiOperation("获取用户信息")
  7. @ApiImplicitParam(name = "name", value = "用户名", dataType = "string", paramType = "query")
  8. @GetMapping("/test/{name}")
  9. public User test(@PathVariable("name") String name) {
  10. User user = new User();
  11.  
  12. user.setId(1);
  13. user.setUsername(name);
  14. user.setPassword("jack123");
  15. user.setBirthday(new Date());
  16.  
  17. return user;
  18. }
  19. }

注意,上边的方法是用 @GetMapping 注解,如果只是使用 @RequestMapping 注解,不配置 method 属性,那么 API 文档会生成 7 种请求方式。

启动项目,打开浏览器访问 http://localhost:8080/swagger-ui.html。结果如下图:

image

猜你喜欢

转载自blog.csdn.net/aa1215018028/article/details/81233381
今日推荐