一个基于 Java开发的微服务聚合网关，是拥有自主知识产权的应用网关国产化替代方案

/* * 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 com.fizzgate.util; import org.apache.commons.lang3.StringUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.core.io.buffer.DataBuffer; import org.springframework.http.HttpHeaders; import org.springframework.http.HttpMethod; import org.springframework.http.HttpStatus; import org.springframework.http.MediaType; import org.springframework.http.server.reactive.ServerHttpRequest; import org.springframework.http.server.reactive.ServerHttpResponse; import org.springframework.lang.Nullable; import org.springframework.util.CollectionUtils; import org.springframework.util.MultiValueMap; import org.springframework.web.reactive.function.client.ClientResponse; import org.springframework.web.server.ServerWebExchange; import com.fizzgate.config.SystemConfig; import com.fizzgate.filter.FilterResult; import com.fizzgate.plugin.auth.ApiConfig; import com.fizzgate.plugin.auth.AuthPluginFilter; import com.fizzgate.proxy.Route; import com.fizzgate.util.Consts; import com.fizzgate.util.JacksonUtils; import com.fizzgate.util.NettyDataBufferUtils; import com.fizzgate.util.Result; import com.fizzgate.util.ThreadContext; import com.fizzgate.util.Utils; import reactor.core.publisher.Mono; import java.io.UnsupportedEncodingException; import java.net.URI; import java.net.URLDecoder; import java.nio.ByteBuffer; import java.nio.charset.StandardCharsets; import java.util.*; import java.util.function.Function; import java.util.stream.Collectors; import java.util.stream.Stream; /** * @author hongqiaowei */ public abstract class WebUtils { // TODO: don't log in this class private static final Logger log = LoggerFactory.getLogger(WebUtils.class); private static final String clientService = "cs@"; private static final String xForwardedFor = "X-Forwarded-For"; private static final String unknown = "unknown"; private static final String loopBack = "127.0.0.1"; private static final String binaryAddress = "0:0:0:0:0:0:0:1"; private static final String directResponse = "dr@"; private static final String response = " response "; private static final String originIp = "oi@"; private static final String clientRequestPath = "crp@"; private static final String clientRequestPathPrefix = "crpp@"; private static final String clientRequestQuery = "crq@"; private static String gatewayPrefix = SystemConfig.DEFAULT_GATEWAY_PREFIX; private static List<String> appHeaders = Stream.of(SystemConfig.FIZZ_APP_ID) .collect(Collectors.toList()); private static List<String> signHeaders = Stream.of(SystemConfig.FIZZ_SIGN) .collect(Collectors.toList()); private static List<String> timestampHeaders = Stream.of(SystemConfig.FIZZ_TIMESTAMP).collect(Collectors.toList()); public static final String TRACE_ID = "traid@"; public static final String BACKEND_SERVICE = "bs@"; public static final String FILTER_CONTEXT = "fc@"; public static final String APPEND_HEADERS = "ahs@"; public static final String PREV_FILTER_RESULT = "pfr@"; public static final String BACKEND_PATH = "bp@"; public static final String ROUTE = "rout@"; public static boolean LOG_RESPONSE_BODY = false; public static Set<String> LOG_HEADER_SET = Collections.emptySet(); public static final String ADMIN_REQUEST = "ar@"; public static final String FIZZ_REQUEST = "fr@"; public static final String FAV_REQUEST = "fa@"; public static final String BODY_ENCRYPT = "b-ecyt"; public static final String ORIGINAL_ERROR = "origerr@"; public static final String IGNORE_PLUGIN = "ignPlg@"; private WebUtils() { } public static boolean ignorePlugin(ServerWebExchange exchange) { return exchange.getAttributes().containsKey(IGNORE_PLUGIN); } public static boolean isFavReq(ServerWebExchange exchange) { return exchange.getAttribute(FAV_REQUEST) != null; } public static boolean isAdminReq(ServerWebExchange exchange) { return exchange.getAttribute(ADMIN_REQUEST) != null; } public static boolean isFizzReq(ServerWebExchange exchange) { return exchange.getAttribute(FIZZ_REQUEST) != null; } public static void setGatewayPrefix(String p) { gatewayPrefix = p; } public static void setAppHeaders(List<String> hdrs) { appHeaders = hdrs; } public static void setSignHeaders(List<String> hdrs) { signHeaders = hdrs; } public static void setTimestampHeaders(List<String> hdrs) { timestampHeaders = hdrs; } public static String getHeaderValue(ServerWebExchange exchange, String header) { return exchange.getRequest().getHeaders().getFirst(header); } public static List<String> getHeaderValues(ServerWebExchange exchange, String header) { return exchange.getRequest().getHeaders().get(header); } public static boolean isDedicatedLineRequest(ServerWebExchange exchange) { String v = exchange.getRequest().getHeaders().getFirst(SystemConfig.FIZZ_DL_ID); return v != null; } public static String getAppId(ServerWebExchange exchange) { HttpHeaders headers = exchange.getRequest().getHeaders(); for (int i = 0; i < appHeaders.size(); i++) { String v = headers.getFirst(appHeaders.get(i)); if (v != null) { return v; } } return null; } public static String getTimestamp(ServerWebExchange exchange) { HttpHeaders headers = exchange.getRequest().getHeaders(); for (int i = 0; i < timestampHeaders.size(); i++) { String v = headers.getFirst(timestampHeaders.get(i)); if (v != null) { return v; } } return null; } public static String getSign(ServerWebExchange exchange) { HttpHeaders headers = exchange.getRequest().getHeaders(); for (int i = 0; i < signHeaders.size(); i++) { String v = headers.getFirst(signHeaders.get(i)); if (v != null) { return v; } } return null; } public static String getDedicatedLineId(ServerWebExchange exchange) { return getHeaderValue(exchange, SystemConfig.FIZZ_DL_ID); } public static String getDedicatedLineTimestamp(ServerWebExchange exchange) { return getHeaderValue(exchange, SystemConfig.FIZZ_DL_TS); } p