From 80b6cbfc9c861469146318d0b3dd5f8b8b525b8a Mon Sep 17 00:00:00 2001 From: xiejun <xiejun@vci-tech.com> Date: 星期五, 01 十一月 2024 15:11:19 +0800 Subject: [PATCH] Revert "集成获取mdm分发通用数据格式接口集成" --- Source/BladeX-Tool/blade-core-context/src/main/java/org/springblade/core/context/ServletHttpHeadersGetter.java | 75 +++++++++++++++++++++++++++++++++++++ 1 files changed, 75 insertions(+), 0 deletions(-) diff --git a/Source/BladeX-Tool/blade-core-context/src/main/java/org/springblade/core/context/ServletHttpHeadersGetter.java b/Source/BladeX-Tool/blade-core-context/src/main/java/org/springblade/core/context/ServletHttpHeadersGetter.java new file mode 100644 index 0000000..06781c6 --- /dev/null +++ b/Source/BladeX-Tool/blade-core-context/src/main/java/org/springblade/core/context/ServletHttpHeadersGetter.java @@ -0,0 +1,75 @@ +/* + * Copyright (c) 2018-2028, DreamLu All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * Neither the name of the dreamlu.net developer nor the names of its + * contributors may be used to endorse or promote products derived from + * this software without specific prior written permission. + * Author: DreamLu 鍗㈡槬姊� (596392912@qq.com) + */ +package org.springblade.core.context; + +import lombok.RequiredArgsConstructor; +import org.springblade.core.context.props.BladeContextProperties; +import org.springblade.core.tool.utils.StringUtil; +import org.springblade.core.tool.utils.WebUtil; +import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication; +import org.springframework.http.HttpHeaders; +import org.springframework.lang.Nullable; + +import javax.servlet.http.HttpServletRequest; +import java.util.Enumeration; +import java.util.List; + +/** + * HttpHeaders 鑾峰彇鍣� + * + * @author L.cm + */ +@RequiredArgsConstructor +@ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.SERVLET) +public class ServletHttpHeadersGetter implements BladeHttpHeadersGetter { + private final BladeContextProperties properties; + + @Nullable + @Override + public HttpHeaders get() { + HttpServletRequest request = WebUtil.getRequest(); + if (request == null) { + return null; + } + return get(request); + } + + @Nullable + @Override + public HttpHeaders get(HttpServletRequest request) { + HttpHeaders headers = new HttpHeaders(); + List<String> crossHeaders = properties.getCrossHeaders(); + // 浼犻�掕姹傚ご + Enumeration<String> headerNames = request.getHeaderNames(); + if (headerNames != null) { + List<String> allowed = properties.getHeaders().getAllowed(); + while (headerNames.hasMoreElements()) { + String key = headerNames.nextElement(); + // 鍙敮鎸侀厤缃殑 header + if (crossHeaders.contains(key) || allowed.contains(key)) { + String values = request.getHeader(key); + // header value 涓嶄负绌虹殑 浼犻�� + if (StringUtil.isNotBlank(values)) { + headers.add(key, values); + } + } + } + } + return headers; + } + +} -- Gitblit v1.9.3