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