fizzgate/fizz-gateway-node
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Incubate fizz api dispatch web server

Browse Source
This commit is contained in:
hongqiaowei
2021-10-12 20:54:09 +08:00
parent 0190ac2c31
commit 9373370ade
4 changed files with 252 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

168
fizz-core/src/main/java/we/incubator/FizzApiDispatchHttpHandler.java Normal file
View File

@@ -0,0 +1,168 @@
/*
* Copyright (C) 2020 the original author or authors.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
package we.incubator;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.NestedExceptionUtils;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.codec.ServerCodecConfigurer;
import org.springframework.http.server.reactive.HttpHandler;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.util.MultiValueMap;
import org.springframework.util.StringUtils;
import org.springframework.web.server.ServerWebExchange;
import org.springframework.web.server.adapter.DefaultServerWebExchange;
import org.springframework.web.server.adapter.ForwardedHeaderTransformer;
import org.springframework.web.server.i18n.LocaleContextResolver;
import org.springframework.web.server.session.WebSessionManager;
import reactor.core.publisher.Mono;
import we.Fizz;
import we.plugin.auth.ApiConfigService;
import we.spring.web.server.ext.FizzServerWebExchangeDecorator;
import we.util.JacksonUtils;
import we.util.WebUtils;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
/**
* @author hongqiaowei
*/
public class FizzApiDispatchHttpHandler implements HttpHandler {
private static final String disconnected_client_log_category = "DisconnectedClient";
private static final Logger log = LoggerFactory.getLogger(FizzApiDispatchHttpHandler.class);
private static final Logger lostClientLog = LoggerFactory.getLogger(disconnected_client_log_category);
private static final Set<String> disconnected_client_exceptions = new HashSet<>(Arrays.asList("AbortedException", "ClientAbortException", "EOFException", "EofException"));
private WebSessionManager sessionManager;
private ServerCodecConfigurer serverCodecConfigurer;
private LocaleContextResolver localeContextResolver;
private ForwardedHeaderTransformer forwardedHeaderTransformer;
private boolean enableLoggingRequestDetails = false;
public FizzApiDispatchHttpHandler(WebSessionManager sessionManager, ServerCodecConfigurer codecConfigurer,
LocaleContextResolver localeContextResolver, ForwardedHeaderTransformer forwardedHeaderTransformer) {
this.sessionManager = sessionManager;
this.serverCodecConfigurer = codecConfigurer;
this.localeContextResolver = localeContextResolver;
this.forwardedHeaderTransformer = forwardedHeaderTransformer;
}
@Override
public Mono<Void> handle(ServerHttpRequest request, ServerHttpResponse response) {
if (forwardedHeaderTransformer != null) {
try {
request = forwardedHeaderTransformer.apply(request);
} catch (Throwable t) {
if (log.isDebugEnabled()) {
log.debug("Failed to apply forwarded headers to " + formatRequest(request), t);
}
response.setStatusCode(HttpStatus.BAD_REQUEST);
return response.setComplete();
}
}
DefaultServerWebExchange exchange = new DefaultServerWebExchange(request, response, sessionManager, serverCodecConfigurer, localeContextResolver);
// XXX
String clientReqPath = WebUtils.getClientReqPath(exchange);
log.info("client request path: {}", clientReqPath);
Mono<MultiValueMap<String, String>> formData = exchange.getFormData().defaultIfEmpty(FizzServerWebExchangeDecorator.EMPTY_FORM_DATA).flatMap(
dat -> {
log.info("form data: " + JacksonUtils.writeValueAsString(dat));
return Mono.just(dat);
}
);
try {
ApiConfigService apiConfigService = Fizz.context.getBean(ApiConfigService.class);
String apiConfigs = JacksonUtils.writeValueAsString(apiConfigService.serviceConfigMap);
return
formData.then(
response.writeWith(Mono.just(response.bufferFactory().wrap(apiConfigs.getBytes())))
)
.doOnSuccess (v -> logResponse(exchange))
.onErrorResume(t -> handleUnresolvedError(exchange, t))
.then(Mono.defer(response::setComplete));
} catch (Throwable t) {
throw t;
// TODO: response error
}
}
private void logResponse(ServerWebExchange exchange) {
WebUtils.traceDebug(log, traceOn -> {
HttpStatus status = exchange.getResponse().getStatusCode();
return exchange.getLogPrefix() + "Completed " + (status != null ? status : "200 OK") + (traceOn ? ", headers=" + formatHeaders(exchange.getResponse().getHeaders()) : "");
});
}
private String formatHeaders(HttpHeaders responseHeaders) {
return this.enableLoggingRequestDetails ?
responseHeaders.toString() : responseHeaders.isEmpty() ? "{}" : "{masked}";
}
private String formatRequest(ServerHttpRequest request) {
String rawQuery = request.getURI().getRawQuery();
String query = StringUtils.hasText(rawQuery) ? "?" + rawQuery : "";
return "HTTP " + request.getMethod() + " \"" + request.getPath() + query + "\"";
}
private Mono<Void> handleUnresolvedError(ServerWebExchange exchange, Throwable t) {
ServerHttpRequest request = exchange.getRequest();
ServerHttpResponse response = exchange.getResponse();
String logPrefix = exchange.getLogPrefix();
if (response.setStatusCode(HttpStatus.INTERNAL_SERVER_ERROR)) {
log.error(logPrefix + "500 Server Error for " + formatRequest(request), t);
return Mono.empty();
} else if (isDisconnectedClientError(t)) {
if (lostClientLog.isTraceEnabled()) {
lostClientLog.trace(logPrefix + "Client went away", t);
} else if (lostClientLog.isDebugEnabled()) {
lostClientLog.debug(logPrefix + "Client went away: " + t + " (stacktrace at TRACE level for '" + disconnected_client_log_category + "')");
}
return Mono.empty();
} else {
// After the response is committed, propagate errors to the server...
log.error(logPrefix + "Error [" + t + "] for " + formatRequest(request) + ", but ServerHttpResponse already committed (" + response.getStatusCode() + ")");
return Mono.error(t);
}
}
private boolean isDisconnectedClientError(Throwable t) {
String message = NestedExceptionUtils.getMostSpecificCause(t).getMessage();
if (message != null) {
String text = message.toLowerCase();
if (text.contains("broken pipe") || text.contains("connection reset by peer")) {
return true;
}
}
return disconnected_client_exceptions.contains(t.getClass().getSimpleName());
}
}

69
fizz-core/src/main/java/we/incubator/FizzApiDispatchWebServer.java Normal file
View File

@@ -0,0 +1,69 @@
/*
* Copyright (C) 2020 the original author or authors.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
package we.incubator;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.web.embedded.netty.NettyReactiveWebServerFactory;
import org.springframework.boot.web.server.WebServer;
import org.springframework.http.server.reactive.HttpHandler;
import org.springframework.web.server.adapter.HttpWebHandlerAdapter;
import org.springframework.web.server.session.DefaultWebSessionManager;
import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import javax.annotation.Resource;
/**
* @author hongqiaowei
*/
//@Configuration
//@AutoConfigureAfter({HttpHandlerAutoConfiguration.class})
public class FizzApiDispatchWebServer {
private static final Logger log = LoggerFactory.getLogger(FizzApiDispatchWebServer.class);
@Resource
private HttpHandler httpHandler;
private WebServer server;
private int port = 8601;
@PostConstruct
public void start() {
HttpWebHandlerAdapter adapter = (HttpWebHandlerAdapter) httpHandler;
NettyReactiveWebServerFactory factory = new NettyReactiveWebServerFactory(port);
server = factory.getWebServer(
new FizzApiDispatchHttpHandler(
new DefaultWebSessionManager(),
adapter.getCodecConfigurer(),
adapter.getLocaleContextResolver(),
adapter.getForwardedHeaderTransformer()
)
);
server.start();
log.info("fizz api dispatch web server listen on {}", port);
}
@PreDestroy
public void stop() {
server.stop();
}
}

13
fizz-core/src/main/java/we/util/WebUtils.java
View File

@@ -43,6 +43,7 @@ import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.Stream;
@@ -587,6 +588,18 @@ public abstract class WebUtils {
return id;
}
public static void traceDebug(Logger log, Function<Boolean, String> messageFactory) {
if (log.isDebugEnabled()) {
boolean traceEnabled = log.isTraceEnabled();
String logMessage = messageFactory.apply(traceEnabled);
if (traceEnabled) {
log.trace(logMessage);
} else {
log.debug(logMessage);
}
}
}
private static final String s0 = "{\"";
private static final String s1 = "\":";
private static final String s2 = ",\"";