test/MeterSphere
1
0
Fork 0
mirror of https://gitee.com/fit2cloud-feizhiyun/MeterSphere.git synced 2024-12-02 20:19:16 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

refactor: 服务调用时HttpHeader提取

Browse Source
This commit is contained in:
CaptainB 2022-10-10 16:42:07 +08:00 committed by 刘瑞斌
parent f45e15f025
commit 2344385ea1
1 changed files with 22 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
framework/sdk-parent/sdk/src/main/java/io/metersphere/commons/utils/HttpHeaderUtils.java
View File

@ -29,23 +29,36 @@ public class HttpHeaderUtils {
headers.add(HttpHeaders.ACCEPT, MediaType.APPLICATION_JSON.toString());
headers.add(HttpHeaders.COOKIE, SessionUtils.getHttpHeader(HttpHeaders.COOKIE));
if (StringUtils.isNotBlank(SessionUtils.getHttpHeader(SessionConstants.HEADER_TOKEN))) {
headers.add(SessionConstants.HEADER_TOKEN, SessionUtils.getHttpHeader(SessionConstants.HEADER_TOKEN));
String headerToken = SessionUtils.getHttpHeader(SessionConstants.HEADER_TOKEN);
if (StringUtils.isNotBlank(headerToken)) {
headers.add(SessionConstants.HEADER_TOKEN, headerToken);
}
if (StringUtils.isNotBlank(SessionUtils.getHttpHeader(SessionConstants.CSRF_TOKEN))) {
headers.add(SessionConstants.CSRF_TOKEN, SessionUtils.getHttpHeader(SessionConstants.CSRF_TOKEN));
String csrfToken = SessionUtils.getHttpHeader(SessionConstants.CSRF_TOKEN);
if (StringUtils.isNotBlank(csrfToken)) {
headers.add(SessionConstants.CSRF_TOKEN, csrfToken);
}
if (StringUtils.isNotBlank(SessionUtils.getHttpHeader(SessionConstants.CURRENT_PROJECT))) {
headers.add(SessionConstants.CURRENT_PROJECT, SessionUtils.getHttpHeader(SessionConstants.CURRENT_PROJECT));
String currentProject = SessionUtils.getHttpHeader(SessionConstants.CURRENT_PROJECT);
if (StringUtils.isNotBlank(currentProject)) {
headers.add(SessionConstants.CURRENT_PROJECT, currentProject);
}
if (StringUtils.isNotBlank(SessionUtils.getHttpHeader(SessionConstants.CURRENT_WORKSPACE))) {
headers.add(SessionConstants.CURRENT_WORKSPACE, SessionUtils.getHttpHeader(SessionConstants.CURRENT_WORKSPACE));
String currentWorkspace = SessionUtils.getHttpHeader(SessionConstants.CURRENT_WORKSPACE);
if (StringUtils.isNotBlank(currentWorkspace)) {
headers.add(SessionConstants.CURRENT_WORKSPACE, currentWorkspace);
}
String accessKey = SessionUtils.getHttpHeader(SessionConstants.ACCESS_KEY);
if (StringUtils.isNotBlank(accessKey)) {
headers.add(SessionConstants.ACCESS_KEY, accessKey);
}
String signature = SessionUtils.getHttpHeader(SessionConstants.SIGNATURE);
if (StringUtils.isNotBlank(signature)) {
headers.add(SessionConstants.SIGNATURE, signature);
}
UserDTO user = sessionUserThreadLocal.get();
if (user != null) {
UserKey userKey = getUserKey(user);
String accessKey = userKey.getAccessKey();
accessKey = userKey.getAccessKey();
String secretKey = userKey.getSecretKey();
headers.add(SessionConstants.ACCESS_KEY, accessKey);
headers.add(SessionConstants.SIGNATURE, CodingUtil.aesDecrypt(accessKey + "|" + System.currentTimeMillis(), secretKey, accessKey));