lepdou
2 years ago
committed by
Haotian Zhang
parent
a39d47c1b2
commit
7e847b81a1
@ -0,0 +1,80 @@
|
|||||||
|
/*
|
||||||
|
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the BSD 3-Clause License (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://opensource.org/licenses/BSD-3-Clause
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software distributed
|
||||||
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||||
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations under the License.
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.tencent.cloud.metadata.core;
|
||||||
|
|
||||||
|
import java.util.Enumeration;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
|
||||||
|
import org.apache.commons.lang.StringUtils;
|
||||||
|
|
||||||
|
import org.springframework.http.HttpHeaders;
|
||||||
|
import org.springframework.util.CollectionUtils;
|
||||||
|
import org.springframework.web.server.ServerWebExchange;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* resolve custom transitive metadata from request.
|
||||||
|
*@author lepdou 2022-05-20
|
||||||
|
*/
|
||||||
|
public class CustomTransitiveMetadataResolver {
|
||||||
|
|
||||||
|
private static final String TRANSITIVE_HEADER_PREFIX = "X-SCT-Metadata-Transitive-";
|
||||||
|
private static final int TRANSITIVE_HEADER_PREFIX_LENGTH = TRANSITIVE_HEADER_PREFIX.length();
|
||||||
|
|
||||||
|
public static Map<String, String> resolve(ServerWebExchange exchange) {
|
||||||
|
Map<String, String> result = new HashMap<>();
|
||||||
|
|
||||||
|
HttpHeaders headers = exchange.getRequest().getHeaders();
|
||||||
|
for (Map.Entry<String, List<String>> entry : headers.entrySet()) {
|
||||||
|
String key = entry.getKey();
|
||||||
|
|
||||||
|
if (StringUtils.isNotBlank(key) &&
|
||||||
|
StringUtils.startsWithIgnoreCase(key, TRANSITIVE_HEADER_PREFIX)
|
||||||
|
&& !CollectionUtils.isEmpty(entry.getValue())) {
|
||||||
|
|
||||||
|
String sourceKey = StringUtils.substring(key, TRANSITIVE_HEADER_PREFIX_LENGTH);
|
||||||
|
result.put(sourceKey, entry.getValue().get(0));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static Map<String, String> resolve(HttpServletRequest request) {
|
||||||
|
Map<String, String> result = new HashMap<>();
|
||||||
|
|
||||||
|
Enumeration<String> headers = request.getHeaderNames();
|
||||||
|
while (headers.hasMoreElements()) {
|
||||||
|
String key = headers.nextElement();
|
||||||
|
|
||||||
|
if (StringUtils.isNotBlank(key) &&
|
||||||
|
StringUtils.startsWithIgnoreCase(key, TRANSITIVE_HEADER_PREFIX)
|
||||||
|
&& StringUtils.isNotBlank(request.getHeader(key))) {
|
||||||
|
|
||||||
|
String sourceKey = StringUtils.substring(key, TRANSITIVE_HEADER_PREFIX_LENGTH);
|
||||||
|
result.put(sourceKey, request.getHeader(key));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,145 @@
|
|||||||
|
/*
|
||||||
|
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||||
|
*
|
||||||
|
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the BSD 3-Clause License (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://opensource.org/licenses/BSD-3-Clause
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software distributed
|
||||||
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||||
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations under the License.
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.tencent.cloud.common.metadata;
|
||||||
|
|
||||||
|
import java.util.Collections;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
import com.tencent.cloud.common.metadata.config.MetadataLocalProperties;
|
||||||
|
import org.apache.commons.lang.StringUtils;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* manage metadata from env/config file.
|
||||||
|
*
|
||||||
|
*@author lepdou 2022-05-20
|
||||||
|
*/
|
||||||
|
public class StaticMetadataManager {
|
||||||
|
private static final Logger LOGGER = LoggerFactory.getLogger(StaticMetadataManager.class);
|
||||||
|
|
||||||
|
private static final String ENV_METADATA_PREFIX = "SCT_METADATA_CONTENT_";
|
||||||
|
private static final int ENV_METADATA_PREFIX_LENGTH = ENV_METADATA_PREFIX.length();
|
||||||
|
private static final String ENV_METADATA_CONTENT_TRANSITIVE = "SCT_METADATA_CONTENT_TRANSITIVE";
|
||||||
|
|
||||||
|
private Map<String, String> envMetadata;
|
||||||
|
private Map<String, String> envTransitiveMetadata;
|
||||||
|
private Map<String, String> configMetadata;
|
||||||
|
private Map<String, String> configTransitiveMetadata;
|
||||||
|
private Map<String, String> mergedStaticMetadata;
|
||||||
|
private Map<String, String> mergedStaticTransitiveMetadata;
|
||||||
|
|
||||||
|
public StaticMetadataManager(MetadataLocalProperties metadataLocalProperties) {
|
||||||
|
parseConfigMetadata(metadataLocalProperties);
|
||||||
|
parseEnvMetadata();
|
||||||
|
merge();
|
||||||
|
}
|
||||||
|
|
||||||
|
private void parseEnvMetadata() {
|
||||||
|
Map<String, String> allEnvs = System.getenv();
|
||||||
|
|
||||||
|
envMetadata = new HashMap<>();
|
||||||
|
// parse all metadata
|
||||||
|
for (Map.Entry<String, String> entry : allEnvs.entrySet()) {
|
||||||
|
String key = entry.getKey();
|
||||||
|
String value = entry.getValue();
|
||||||
|
if (StringUtils.isNotBlank(key) && key.startsWith(ENV_METADATA_PREFIX)
|
||||||
|
&& !key.equals(ENV_METADATA_CONTENT_TRANSITIVE)) {
|
||||||
|
String sourceKey = StringUtils.substring(key, ENV_METADATA_PREFIX_LENGTH);
|
||||||
|
envMetadata.put(sourceKey, value);
|
||||||
|
|
||||||
|
LOGGER.info("[SCT] resolve metadata from env. key = {}, value = {}", sourceKey, value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
envMetadata = Collections.unmodifiableMap(envMetadata);
|
||||||
|
|
||||||
|
envTransitiveMetadata = new HashMap<>();
|
||||||
|
// parse transitive metadata
|
||||||
|
String transitiveKeys = allEnvs.get(ENV_METADATA_CONTENT_TRANSITIVE);
|
||||||
|
if (StringUtils.isNotBlank(transitiveKeys)) {
|
||||||
|
String[] keyArr = StringUtils.split(transitiveKeys, ",");
|
||||||
|
if (keyArr != null && keyArr.length > 0) {
|
||||||
|
for (String key : keyArr) {
|
||||||
|
String value = envMetadata.get(key);
|
||||||
|
if (StringUtils.isNotBlank(value)) {
|
||||||
|
envTransitiveMetadata.put(key, value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
envTransitiveMetadata = Collections.unmodifiableMap(envTransitiveMetadata);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void parseConfigMetadata(MetadataLocalProperties metadataLocalProperties) {
|
||||||
|
Map<String, String> allMetadata = metadataLocalProperties.getContent();
|
||||||
|
List<String> transitiveKeys = metadataLocalProperties.getTransitive();
|
||||||
|
|
||||||
|
Map<String, String> result = new HashMap<>();
|
||||||
|
for (String key : transitiveKeys) {
|
||||||
|
if (allMetadata.containsKey(key)) {
|
||||||
|
result.put(key, allMetadata.get(key));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
configTransitiveMetadata = Collections.unmodifiableMap(result);
|
||||||
|
configMetadata = Collections.unmodifiableMap(allMetadata);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void merge() {
|
||||||
|
// env priority is bigger than config
|
||||||
|
Map<String, String> mergedMetadataResult = new HashMap<>();
|
||||||
|
|
||||||
|
mergedMetadataResult.putAll(configMetadata);
|
||||||
|
mergedMetadataResult.putAll(envMetadata);
|
||||||
|
|
||||||
|
this.mergedStaticMetadata = Collections.unmodifiableMap(mergedMetadataResult);
|
||||||
|
|
||||||
|
Map<String, String> mergedTransitiveMetadataResult = new HashMap<>();
|
||||||
|
mergedTransitiveMetadataResult.putAll(configTransitiveMetadata);
|
||||||
|
mergedTransitiveMetadataResult.putAll(envTransitiveMetadata);
|
||||||
|
|
||||||
|
this.mergedStaticTransitiveMetadata = Collections.unmodifiableMap(mergedTransitiveMetadataResult);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Map<String, String> getAllEnvMetadata() {
|
||||||
|
return envMetadata;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Map<String, String> getEnvTransitiveMetadata() {
|
||||||
|
return envTransitiveMetadata;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Map<String, String> getAllConfigMetadata() {
|
||||||
|
return configMetadata;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Map<String, String> getConfigTransitiveMetadata() {
|
||||||
|
return configTransitiveMetadata;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Map<String, String> getMergedStaticMetadata() {
|
||||||
|
return mergedStaticMetadata;
|
||||||
|
}
|
||||||
|
|
||||||
|
Map<String, String> getMergedStaticTransitiveMetadata() {
|
||||||
|
return mergedStaticTransitiveMetadata;
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue