fizz-gateway-node/src/main/java/com/wehotel/fizz/StepContext.java

/*
 *  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);
	}

}