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/BladeServletContext.java | 80 ++++++++++++++++++++++++++++++++++++++++ 1 files changed, 80 insertions(+), 0 deletions(-) diff --git a/Source/BladeX-Tool/blade-core-context/src/main/java/org/springblade/core/context/BladeServletContext.java b/Source/BladeX-Tool/blade-core-context/src/main/java/org/springblade/core/context/BladeServletContext.java new file mode 100644 index 0000000..a7efafa --- /dev/null +++ b/Source/BladeX-Tool/blade-core-context/src/main/java/org/springblade/core/context/BladeServletContext.java @@ -0,0 +1,80 @@ +/* + * 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.ThreadLocalUtil; +import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication; +import org.springframework.http.HttpHeaders; +import org.springframework.lang.Nullable; + +import java.util.function.Function; + +import static org.springblade.core.tool.constant.BladeConstant.CONTEXT_KEY; + +/** + * blade servlet 涓婁笅鏂囷紝璺ㄧ嚎绋嬪け鏁� + * + * @author L.cm + */ +@RequiredArgsConstructor +@ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.SERVLET) +public class BladeServletContext implements BladeContext { + private final BladeContextProperties contextProperties; + private final BladeHttpHeadersGetter httpHeadersGetter; + + @Nullable + @Override + public String getRequestId() { + return get(contextProperties.getHeaders().getRequestId()); + } + + @Nullable + @Override + public String getAccountId() { + return get(contextProperties.getHeaders().getAccountId()); + } + + @Nullable + @Override + public String getTenantId() { + return get(contextProperties.getHeaders().getTenantId()); + } + + @Nullable + @Override + public String get(String ctxKey) { + HttpHeaders headers = ThreadLocalUtil.getIfAbsent(CONTEXT_KEY, httpHeadersGetter::get); + if (headers == null || headers.isEmpty()) { + return null; + } + return headers.getFirst(ctxKey); + } + + @Nullable + @Override + public <T> T get(String ctxKey, Function<String, T> function) { + String ctxValue = get(ctxKey); + if (StringUtil.isBlank(ctxValue)) { + return null; + } + return function.apply(ctxKey); + } + +} -- Gitblit v1.9.3