lepdou
3 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