1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 package org.springframework.web.socket;
18
19 import java.io.File;
20 import java.io.IOException;
21 import java.util.EnumSet;
22 import javax.servlet.DispatcherType;
23 import javax.servlet.Filter;
24
25 import org.apache.catalina.Context;
26 import org.apache.catalina.connector.Connector;
27 import org.apache.catalina.startup.Tomcat;
28 import org.apache.coyote.http11.Http11NioProtocol;
29 import org.apache.tomcat.util.descriptor.web.FilterDef;
30 import org.apache.tomcat.util.descriptor.web.FilterMap;
31 import org.apache.tomcat.websocket.server.WsContextListener;
32
33 import org.springframework.util.Assert;
34 import org.springframework.util.SocketUtils;
35 import org.springframework.web.context.WebApplicationContext;
36 import org.springframework.web.servlet.DispatcherServlet;
37
38
39
40
41
42
43 public class TomcatWebSocketTestServer implements WebSocketTestServer {
44
45 private Tomcat tomcatServer;
46
47 private int port = -1;
48
49 private Context context;
50
51
52 @Override
53 public void setup() {
54 this.port = SocketUtils.findAvailableTcpPort();
55
56 Connector connector = new Connector(Http11NioProtocol.class.getName());
57 connector.setPort(this.port);
58
59 File baseDir = createTempDir("tomcat");
60 String baseDirPath = baseDir.getAbsolutePath();
61
62 this.tomcatServer = new Tomcat();
63 this.tomcatServer.setBaseDir(baseDirPath);
64 this.tomcatServer.setPort(this.port);
65 this.tomcatServer.getService().addConnector(connector);
66 this.tomcatServer.setConnector(connector);
67 }
68
69 private File createTempDir(String prefix) {
70 try {
71 File tempFolder = File.createTempFile(prefix + ".", "." + getPort());
72 tempFolder.delete();
73 tempFolder.mkdir();
74 tempFolder.deleteOnExit();
75 return tempFolder;
76 }
77 catch (IOException ex) {
78 throw new RuntimeException("Unable to create temp directory", ex);
79 }
80 }
81
82 @Override
83 public int getPort() {
84 return this.port;
85 }
86
87 @Override
88 public void deployConfig(WebApplicationContext wac, Filter... filters) {
89 Assert.state(this.port != -1, "setup() was never called.");
90 this.context = this.tomcatServer.addContext("", System.getProperty("java.io.tmpdir"));
91 this.context.addApplicationListener(WsContextListener.class.getName());
92 Tomcat.addServlet(this.context, "dispatcherServlet", new DispatcherServlet(wac)).setAsyncSupported(true);
93 this.context.addServletMapping("/", "dispatcherServlet");
94 for (Filter filter : filters) {
95 FilterDef filterDef = new FilterDef();
96 filterDef.setFilterName(filter.getClass().getName());
97 filterDef.setFilter(filter);
98 filterDef.setAsyncSupported("true");
99 this.context.addFilterDef(filterDef);
100 FilterMap filterMap = new FilterMap();
101 filterMap.setFilterName(filter.getClass().getName());
102 filterMap.addURLPattern("/*");
103 filterMap.setDispatcher("REQUEST,FORWARD,INCLUDE,ASYNC");
104 this.context.addFilterMap(filterMap);
105 }
106 }
107
108 private EnumSet<DispatcherType> getDispatcherTypes() {
109 return EnumSet.of(DispatcherType.REQUEST, DispatcherType.FORWARD, DispatcherType.INCLUDE, DispatcherType.ASYNC);
110 }
111
112 @Override
113 public void undeployConfig() {
114 if (this.context != null) {
115 this.context.removeServletMapping("/");
116 this.tomcatServer.getHost().removeChild(this.context);
117 }
118 }
119
120 @Override
121 public void start() throws Exception {
122 this.tomcatServer.start();
123 }
124
125 @Override
126 public void stop() throws Exception {
127 this.tomcatServer.stop();
128 }
129
130 }