开张大吉。
This commit is contained in:
linwaiwai
549
src/main/java/com/wehotel/fizz/StepContext.java
Normal file
549
src/main/java/com/wehotel/fizz/StepContext.java
Normal file
@@ -0,0 +1,549 @@
|
||||
/*
|
||||
* Copyright (C) 2020 the original author or authors.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
package com.wehotel.fizz;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
|
||||
import com.wehotel.constants.CommonConstants;
|
||||
|
||||
/**
|
||||
*
|
||||
* @author linwaiwai
|
||||
* @author francis
|
||||
*
|
||||
* @param <K>
|
||||
* @param <V>
|
||||
*/
|
||||
@SuppressWarnings("unchecked")
|
||||
public class StepContext<K, V> extends ConcurrentHashMap<K, V> {
|
||||
|
||||
public static final String ELAPSED_TIMES = "elapsedTimes";
|
||||
public static final String DEBUG = "debug";
|
||||
public static final String RETURN_CONTEXT = "returnContext";
|
||||
|
||||
public void setDebug(Boolean debug) {
|
||||
this.put((K)DEBUG, (V)debug);
|
||||
}
|
||||
|
||||
public String getTraceId() {
|
||||
return (String) this.get(CommonConstants.TRACE_ID);
|
||||
}
|
||||
|
||||
public void setTraceId(String traceId) {
|
||||
this.put((K)CommonConstants.TRACE_ID, (V)traceId);
|
||||
}
|
||||
|
||||
/**
|
||||
* 是否调试模式
|
||||
* @return
|
||||
*/
|
||||
public Boolean isDebug() {
|
||||
return this.get(DEBUG) == null ? false : (Boolean) this.get(DEBUG);
|
||||
}
|
||||
|
||||
/**
|
||||
* 是否在响应体里返回上下文
|
||||
* @return
|
||||
*/
|
||||
public boolean returnContext() {
|
||||
return Boolean.valueOf((String)getInputReqHeader(RETURN_CONTEXT));
|
||||
}
|
||||
|
||||
public synchronized void addElapsedTime(String actionName, Long milliSeconds) {
|
||||
List<Map<String, Long>> elapsedTimes = (List<Map<String, Long>>) this.get(ELAPSED_TIMES);
|
||||
if (elapsedTimes == null) {
|
||||
elapsedTimes = new ArrayList<Map<String, Long>>();
|
||||
this.put((K) ELAPSED_TIMES, (V) elapsedTimes);
|
||||
}
|
||||
Map<String, Long> record = new HashMap<>();
|
||||
record.put(actionName, milliSeconds);
|
||||
elapsedTimes.add(record);
|
||||
}
|
||||
|
||||
public V getElapsedTimes() {
|
||||
return this.get(ELAPSED_TIMES);
|
||||
}
|
||||
|
||||
private Map<String, Object> getStepRequest(String stepName, String requestName) {
|
||||
StepResponse stepResponse = (StepResponse) this.get(stepName);
|
||||
if (stepResponse == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Map<String, Object>> requests = (Map<String, Map<String, Object>>) stepResponse.getRequests();
|
||||
if (requests == null) {
|
||||
requests = new HashMap<>();
|
||||
stepResponse.setRequests(requests);
|
||||
requests.put(requestName, new HashMap<String, Object>());
|
||||
}
|
||||
return (Map<String, Object>) requests.get(requestName);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置Step里调用接口的请求头
|
||||
*
|
||||
* @param stepName
|
||||
* @param requestName
|
||||
* @param headerName
|
||||
* @param headerValue
|
||||
*/
|
||||
public void setStepReqHeader(String stepName, String requestName, String headerName, Object headerValue) {
|
||||
Map<String, Object> request = getStepRequest(stepName, requestName);
|
||||
if (request == null) {
|
||||
return;
|
||||
}
|
||||
Map<String, Object> req = (Map<String, Object>) request.get("request");
|
||||
if (req == null) {
|
||||
req = new HashMap<>();
|
||||
request.put("request", req);
|
||||
}
|
||||
Map<String, Object> headers = (Map<String, Object>) req.get("headers");
|
||||
if (headers == null) {
|
||||
headers = new HashMap<>();
|
||||
req.put("headers", headers);
|
||||
}
|
||||
headers.put(headerName, headerValue);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取Step里调用接口的请求头
|
||||
*
|
||||
* @param stepName
|
||||
* @param requestName
|
||||
* @param headerName
|
||||
*/
|
||||
public Object getStepReqHeader(String stepName, String requestName, String headerName) {
|
||||
Map<String, Object> request = getStepRequest(stepName, requestName);
|
||||
if (request == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> req = (Map<String, Object>) request.get("request");
|
||||
if (req == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> headers = (Map<String, Object>) req.get("headers");
|
||||
if (headers == null) {
|
||||
return null;
|
||||
}
|
||||
return headers.get(headerName);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置Step里调用接口的请求body
|
||||
*
|
||||
* @param stepName
|
||||
* @param requestName
|
||||
* @param key
|
||||
* @param value
|
||||
*/
|
||||
public void setStepReqBody(String stepName, String requestName, String key, Object value) {
|
||||
Map<String, Object> request = getStepRequest(stepName, requestName);
|
||||
if (request == null) {
|
||||
return;
|
||||
}
|
||||
Map<String, Object> req = (Map<String, Object>) request.get("request");
|
||||
if (req == null) {
|
||||
req = new HashMap<>();
|
||||
request.put("request", req);
|
||||
}
|
||||
Map<String, Object> body = (Map<String, Object>) req.get("body");
|
||||
if (body == null) {
|
||||
body = new HashMap<>();
|
||||
req.put("body", body);
|
||||
}
|
||||
body.put(key, value);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置Step里调用接口的请求body
|
||||
*
|
||||
* @param stepName
|
||||
* @param requestName
|
||||
* @param key
|
||||
*/
|
||||
public Object getStepReqBody(String stepName, String requestName, String key) {
|
||||
Map<String, Object> request = getStepRequest(stepName, requestName);
|
||||
if (request == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> req = (Map<String, Object>) request.get("request");
|
||||
if (req == null) {
|
||||
req = new HashMap<>();
|
||||
request.put("request", req);
|
||||
}
|
||||
Map<String, Object> body = (Map<String, Object>) req.get("body");
|
||||
if (body == null) {
|
||||
return null;
|
||||
}
|
||||
return body.get(key);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置Step里调用接口的请求body
|
||||
*
|
||||
* @param stepName
|
||||
* @param requestName
|
||||
*/
|
||||
public Object getStepReqBody(String stepName, String requestName) {
|
||||
Map<String, Object> request = getStepRequest(stepName, requestName);
|
||||
if (request == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> req = (Map<String, Object>) request.get("request");
|
||||
if (req == null) {
|
||||
req = new HashMap<>();
|
||||
request.put("request", req);
|
||||
}
|
||||
return req.get("body");
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置Step里调用接口响应头
|
||||
*
|
||||
* @param stepName
|
||||
* @param requestName
|
||||
* @param headerName
|
||||
* @param headerValue
|
||||
*/
|
||||
public void setStepRespHeader(String stepName, String requestName, String headerName, Object headerValue) {
|
||||
Map<String, Object> request = getStepRequest(stepName, requestName);
|
||||
if (request == null) {
|
||||
return;
|
||||
}
|
||||
Map<String, Object> response = (Map<String, Object>) request.get("response");
|
||||
if (response == null) {
|
||||
response = new HashMap<>();
|
||||
request.put("response", response);
|
||||
}
|
||||
Map<String, Object> headers = (Map<String, Object>) response.get("headers");
|
||||
if (headers == null) {
|
||||
headers = new HashMap<>();
|
||||
response.put("headers", headers);
|
||||
}
|
||||
headers.put(headerName, headerValue);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取Step里调用接口响应头
|
||||
*
|
||||
* @param stepName
|
||||
* @param requestName
|
||||
* @param headerName
|
||||
*/
|
||||
public Object getStepRespHeader(String stepName, String requestName, String headerName) {
|
||||
Map<String, Object> request = getStepRequest(stepName, requestName);
|
||||
if (request == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> response = (Map<String, Object>) request.get("response");
|
||||
if (response == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> headers = (Map<String, Object>) response.get("headers");
|
||||
if (headers == null) {
|
||||
return null;
|
||||
}
|
||||
return headers.get(headerName);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置Step里调用接口的响应body
|
||||
*
|
||||
* @param stepName
|
||||
* @param requestName
|
||||
* @param key
|
||||
* @param value
|
||||
*/
|
||||
public void setStepRespBody(String stepName, String requestName, String key, Object value) {
|
||||
Map<String, Object> request = getStepRequest(stepName, requestName);
|
||||
if (request == null) {
|
||||
return;
|
||||
}
|
||||
Map<String, Object> response = (Map<String, Object>) request.get("response");
|
||||
if (response == null) {
|
||||
response = new HashMap<>();
|
||||
request.put("response", response);
|
||||
}
|
||||
Map<String, Object> body = (Map<String, Object>) response.get("body");
|
||||
if (body == null) {
|
||||
body = new HashMap<>();
|
||||
response.put("body", body);
|
||||
}
|
||||
body.put(key, value);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取Step里调用接口的响应body
|
||||
*
|
||||
* @param stepName
|
||||
* @param requestName
|
||||
* @param key
|
||||
*/
|
||||
public Object getStepRespBody(String stepName, String requestName, String key) {
|
||||
Map<String, Object> request = getStepRequest(stepName, requestName);
|
||||
if (request == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> response = (Map<String, Object>) request.get("response");
|
||||
if (response == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> body = (Map<String, Object>) response.get("body");
|
||||
if (body == null) {
|
||||
return null;
|
||||
}
|
||||
return body.get(key);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取Step里调用接口的响应body
|
||||
*
|
||||
* @param stepName
|
||||
* @param requestName
|
||||
*/
|
||||
public Object getStepRespBody(String stepName, String requestName) {
|
||||
Map<String, Object> request = getStepRequest(stepName, requestName);
|
||||
if (request == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> response = (Map<String, Object>) request.get("response");
|
||||
if (response == null) {
|
||||
return null;
|
||||
}
|
||||
return response.get("body");
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置Step的结果
|
||||
*
|
||||
* @param stepName
|
||||
* @param key
|
||||
* @param value
|
||||
*/
|
||||
public void setStepResult(String stepName, String key, Object value) {
|
||||
StepResponse stepResponse = (StepResponse) this.get(stepName);
|
||||
if (stepResponse == null) {
|
||||
return;
|
||||
}
|
||||
Map<String, Object> result = (Map<String, Object>) stepResponse.getResult();
|
||||
if (result == null) {
|
||||
result = new HashMap<>();
|
||||
stepResponse.setResult(result);
|
||||
}
|
||||
result.put(key, value);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取Step的结果
|
||||
*
|
||||
* @param stepName
|
||||
* @param key
|
||||
*/
|
||||
public Object getStepResult(String stepName, String key) {
|
||||
StepResponse stepResponse = (StepResponse) this.get(stepName);
|
||||
if (stepResponse == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> result = (Map<String, Object>) stepResponse.getResult();
|
||||
if (result == null) {
|
||||
return null;
|
||||
}
|
||||
return result.get(key);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取Step的结果
|
||||
*
|
||||
* @param stepName
|
||||
*/
|
||||
public Object getStepResult(String stepName) {
|
||||
StepResponse stepResponse = (StepResponse) this.get(stepName);
|
||||
if (stepResponse == null) {
|
||||
return null;
|
||||
}
|
||||
return stepResponse.getResult();
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置聚合接口的响应头
|
||||
*
|
||||
* @param headerName
|
||||
* @param headerValue
|
||||
*/
|
||||
public void setInputRespHeader(String headerName, Object headerValue) {
|
||||
Map<String, Object> input = (Map<String, Object>) this.get("input");
|
||||
if (input == null) {
|
||||
return;
|
||||
}
|
||||
Map<String, Object> response = (Map<String, Object>) input.get("response");
|
||||
if (response == null) {
|
||||
return;
|
||||
}
|
||||
Map<String, Object> headers = (Map<String, Object>) response.get("headers");
|
||||
if (headers == null) {
|
||||
headers = new HashMap<>();
|
||||
response.put("headers", headers);
|
||||
}
|
||||
headers.put(headerName, headerValue);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取聚合接口的响应头
|
||||
*
|
||||
* @param headerName
|
||||
*/
|
||||
public Object getInputRespHeader(String headerName) {
|
||||
Map<String, Object> input = (Map<String, Object>) this.get("input");
|
||||
if (input == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> response = (Map<String, Object>) input.get("response");
|
||||
if (response == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> headers = (Map<String, Object>) response.get("headers");
|
||||
if (headers == null) {
|
||||
return null;
|
||||
}
|
||||
return headers.get(headerName);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取聚合接口的请求头
|
||||
*
|
||||
* @param headerName
|
||||
*/
|
||||
public Object getInputReqHeader(String headerName) {
|
||||
Map<String, Object> input = (Map<String, Object>) this.get("input");
|
||||
if (input == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> request = (Map<String, Object>) input.get("request");
|
||||
if (request == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> headers = (Map<String, Object>) request.get("headers");
|
||||
if (headers == null) {
|
||||
return null;
|
||||
}
|
||||
return headers.get(headerName);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置聚合接口的响应body
|
||||
*
|
||||
* @param key
|
||||
* @param value
|
||||
*/
|
||||
public void setInputRespBody(String key, Object value) {
|
||||
Map<String, Object> input = (Map<String, Object>) this.get("input");
|
||||
if (input == null) {
|
||||
return;
|
||||
}
|
||||
Map<String, Object> response = (Map<String, Object>) input.get("response");
|
||||
if (response == null) {
|
||||
response = new HashMap<>();
|
||||
input.put("response", response);
|
||||
}
|
||||
Map<String, Object> body = (Map<String, Object>) response.get("body");
|
||||
if (body == null) {
|
||||
body = new HashMap<>();
|
||||
response.put("body", body);
|
||||
}
|
||||
body.put(key, value);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取聚合接口的响应body
|
||||
*
|
||||
* @param key
|
||||
*/
|
||||
public Object getInputRespBody(String key) {
|
||||
Map<String, Object> input = (Map<String, Object>) this.get("input");
|
||||
if (input == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> response = (Map<String, Object>) input.get("response");
|
||||
if (response == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> body = (Map<String, Object>) response.get("body");
|
||||
if (body == null) {
|
||||
return null;
|
||||
}
|
||||
return body.get(key);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取聚合接口的响应body
|
||||
*
|
||||
*/
|
||||
public Object getInputRespBody() {
|
||||
Map<String, Object> input = (Map<String, Object>) this.get("input");
|
||||
if (input == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> response = (Map<String, Object>) input.get("response");
|
||||
if (response == null) {
|
||||
return null;
|
||||
}
|
||||
return response.get("body");
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取聚合接口的请求body
|
||||
*
|
||||
* @param key
|
||||
*/
|
||||
public Object getInputReqBody(String key) {
|
||||
Map<String, Object> body = (Map<String, Object>) getInputReqAttr("body");
|
||||
if (body == null) {
|
||||
return null;
|
||||
}
|
||||
return body.get(key);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取聚合接口的请求body
|
||||
*
|
||||
*/
|
||||
public Object getInputReqBody() {
|
||||
return getInputReqAttr("body");
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取聚合接口请求属性<br/>
|
||||
* 可选属性：path,method,headers,params,body
|
||||
*
|
||||
*/
|
||||
public Object getInputReqAttr(String key) {
|
||||
Map<String, Object> input = (Map<String, Object>) this.get("input");
|
||||
if (input == null) {
|
||||
return null;
|
||||
}
|
||||
Map<String, Object> request = (Map<String, Object>) input.get("request");
|
||||
if (request == null) {
|
||||
return null;
|
||||
}
|
||||
return request.get(key);
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user