ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-05-09 23:51:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

add tomcat support

Browse Source
This commit is contained in:
Looly 2024-11-24 19:27:09 +08:00
parent 6ed0b11626
commit 1b80ed49e8
1 changed files with 24 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
hutool-http/src/main/java/org/dromara/hutool/http/server/engine/tomcat/TomcatEngine.java
View File

@ -18,14 +18,19 @@ package org.dromara.hutool.http.server.engine.tomcat;
import org.apache.catalina.Context; import org.apache.catalina.Context;
import org.apache.catalina.LifecycleException; import org.apache.catalina.LifecycleException;
import org.apache.catalina.connector.Connector;
import org.apache.catalina.connector.Request; import org.apache.catalina.connector.Request;
import org.apache.catalina.connector.Response; import org.apache.catalina.connector.Response;
import org.apache.catalina.startup.Tomcat; import org.apache.catalina.startup.Tomcat;
import org.apache.catalina.valves.ValveBase; import org.apache.catalina.valves.ValveBase;
import org.dromara.hutool.core.lang.Assert; import org.dromara.hutool.core.lang.Assert;
import org.dromara.hutool.core.text.StrUtil;
import org.dromara.hutool.http.HttpException; import org.dromara.hutool.http.HttpException;
import org.dromara.hutool.http.server.engine.AbstractServerEngine; import org.dromara.hutool.http.server.engine.AbstractServerEngine;
import javax.servlet.ServletException;
import java.io.IOException;
/** /**
* Tomcat引擎实现 * Tomcat引擎实现
* *
@ -55,8 +60,6 @@ public class TomcatEngine extends AbstractServerEngine {
} catch (final LifecycleException e) { } catch (final LifecycleException e) {
throw new HttpException(e); throw new HttpException(e);
} }
// 阻塞
tomcat.getServer().await();
} }
@Override @Override
@ -83,29 +86,41 @@ public class TomcatEngine extends AbstractServerEngine {
} }
final Tomcat tomcat = new Tomcat(); final Tomcat tomcat = new Tomcat();
tomcat.setHostname(this.config.getHost()); tomcat.setHostname(config.getHost());
tomcat.setPort(this.config.getPort()); tomcat.setBaseDir(config.getRoot());
tomcat.setBaseDir(this.config.getRoot());
initConnector(tomcat);
initContext(tomcat); initContext(tomcat);
// TODO 配置支持HTTPS
this.tomcat = tomcat; this.tomcat = tomcat;
} }
/**
* 初始化Connector
*
* @param tomcat Tomcat
*/
private void initConnector(final Tomcat tomcat) {
final Connector connector = new Connector();
connector.setPort(config.getPort());
tomcat.setConnector(connector);
}
/** /**
* 初始化Context * 初始化Context
* *
* @param tomcat Tomcat * @param tomcat Tomcat
*/ */
private void initContext(final Tomcat tomcat) { private void initContext(final Tomcat tomcat) {
final Context context = tomcat.addContext("/", this.config.getRoot()); final Context context = tomcat.addContext(StrUtil.EMPTY, null);
context.getPipeline().addValve(new ValveBase() { context.getPipeline().addValve(new ValveBase() {
@Override @Override
public void invoke(final Request request, final Response response) { public void invoke(final Request request, final Response response) throws IOException, ServletException {
handler.handle(new TomcatRequest(request), new TomcatResponse(response)); handler.handle(new TomcatRequest(request), new TomcatResponse(response));
getNext().invoke(request, response);
} }
// 此处拦截所有请求，无需调用getNext
}); });
} }
} }