Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.pamirs.pradar.pressurement.ClusterTestUtils Maven / Gradle / Ivy
package com.pamirs.pradar.pressurement;
import com.pamirs.pradar.*;
import com.pamirs.pradar.exception.PressureMeasureError;
import com.pamirs.pradar.internal.config.MatchConfig;
import com.pamirs.pradar.json.ResultSerializer;
import com.pamirs.pradar.pressurement.agent.shared.exit.ArbiterHttpExit;
import com.pamirs.pradar.pressurement.agent.shared.service.ErrorReporter;
import com.shulie.instrument.simulator.api.ThrowableUtils;
import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.Map;
/**
* @author xiaobin.zfb
* @since 2020/7/8 5:26 下午
*/
public final class ClusterTestUtils {
private final static Logger LOGGER = LoggerFactory.getLogger(ClusterTestUtils.class);
/**
* 判断是否是压测流量
*
* @param value
* @return
*/
public final static boolean isClusterTestRequest(String value) {
if (StringUtils.equals(Pradar.PRADAR_CLUSTER_TEST_ON, value)
|| StringUtils.equalsIgnoreCase(value, Boolean.TRUE.toString())) {
return true;
}
return StringUtils.equals(value, Pradar.PRADAR_CLUSTER_TEST_HTTP_USER_AGENT_SUFFIX);
}
/**
* 判断是否是 debug 流量
*
* @param value
* @return
*/
public final static boolean isDebugRequest(String value) {
if (StringUtils.equals(Pradar.PRADAR_DEBUG_ON, value)
|| StringUtils.equalsIgnoreCase(value, Boolean.TRUE.toString())) {
return true;
}
return false;
}
/**
* 判断是否是压测流量
*
* @param context
* @return
*/
public final static boolean isClusterTestRequest(Map context) {
if (context == null) {
return false;
}
return isClusterTestRequest(PradarService.PRADAR_CLUSTER_TEST_KEY);
}
/**
* rpc的压测流量验证, 使用类名+方法名的组合方式,如果方法为空则单独匹配 className
* 会匹配白名单中匹配的 className 和 className#methodName 的名单
*
* @param className 类名
* @param methodName 方法名称
*/
public final static MatchConfig validateRpcClusterTest(String className, String methodName) {
validateClusterTest();
try {
/**
* 如果是压测流量则需要验证 rpc 的白名单,不在白名单中则报错
*/
if (Pradar.isClusterTest()) {
MatchConfig matchConfig = ArbiterHttpExit.shallWePassRpc(className, methodName);
if (!matchConfig.isSuccess()) {
throw new PressureMeasureError("WhiteListError: [" + AppNameUtils.appName() + "] interface [" + className + "#" + methodName + "] is not allowed in WhiteList.");
}
return matchConfig;
}
} catch (PressureMeasureError e) {
String message = "WhiteListError: [" + AppNameUtils.appName() + "] interface [" + className + "#" + methodName + "] is not allowed in WhiteList.";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
ErrorReporter.buildError()
.setErrorType(ErrorTypeEnum.AgentError)
.setErrorCode("whiteList-0001")
.setMessage(message)
.setDetail(ResultSerializer.serializeObject(e, 2))
.closePradar(ConfigNames.URL_WHITE_LIST)
.report();
throw e;
}
} catch (Throwable e) {
String message = "WhiteListError: [" + AppNameUtils.appName() + "] interface [" + className + "#" + methodName + "] is not allowed in WhiteList.";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
ErrorReporter.buildError()
.setErrorType(ErrorTypeEnum.AgentError)
.setErrorCode("whiteList-0001")
.setMessage(message)
.setDetail(ResultSerializer.serializeObject(e, 2))
.closePradar(ConfigNames.URL_WHITE_LIST)
.report();
throw new PressureMeasureError(message, e);
}
}
return ArbiterHttpExit.failure();
}
/**
* rpc的压测流量验证, 使用类名+方法名的组合方式,如果方法为空则单独匹配 className
* 会匹配白名单中匹配的 className 和 className#methodName 的名单
*
* @param className 类名
* @param methodName 方法名称
*/
public final static MatchConfig rpcClusterTest(String className, String methodName) {
validateClusterTest();
try {
/**
* 如果是压测流量则需要验证 rpc 的白名单,不在白名单中则报错
*/
if (Pradar.isClusterTest()) {
return ArbiterHttpExit.shallWePassRpc(className, methodName);
}
} catch (PressureMeasureError e) {
String message = "WhiteListError: [" + AppNameUtils.appName() + "] interface [" + className + "#" + methodName + "] is not allowed in WhiteList.";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
ErrorReporter.buildError()
.setErrorType(ErrorTypeEnum.AgentError)
.setErrorCode("whiteList-0001")
.setMessage(message)
.setDetail(ResultSerializer.serializeObject(e, 2))
.closePradar(ConfigNames.URL_WHITE_LIST)
.report();
throw e;
}
} catch (Throwable e) {
String message = "WhiteListError: [" + AppNameUtils.appName() + "] interface [" + className + "#" + methodName + "] is not allowed in WhiteList.";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
ErrorReporter.buildError()
.setErrorType(ErrorTypeEnum.AgentError)
.setErrorCode("whiteList-0001")
.setMessage(message)
.setDetail(ResultSerializer.serializeObject(e, 2))
.closePradar(ConfigNames.URL_WHITE_LIST)
.report();
throw new PressureMeasureError(message, e);
}
}
return ArbiterHttpExit.failure();
}
private static void reportFastDebugErrorInfo(String code, Throwable e, String message) {
if (Pradar.isDebug()) {
ErrorReporter.buildError()
.setErrorType(ErrorTypeEnum.AgentError)
.setErrorCode(code)
.setMessage(message)
.setDetail(ThrowableUtils.toString(e))
.printFastDebugLog();
}
}
/**
* 验证压测流量的有效性,如果是压测流量但是压测未开启则直接报错
*/
public final static MatchConfig validateHttpClusterTest(String url) {
validateClusterTest();
try {
if (Pradar.isClusterTest()) {
MatchConfig matchConfig = ArbiterHttpExit.shallWePassHttpString(url);
if (!matchConfig.isSuccess()) {
throw new PressureMeasureError("WhiteListError: [" + AppNameUtils.appName() + "] url [" + url + "] is not allowed in WhiteList.");
}
return matchConfig;
}
} catch (PressureMeasureError e) {
String message = "WhiteListError: [" + AppNameUtils.appName() + "] url [" + url + "] is not allowed in WhiteList.";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
ErrorReporter.buildError()
.setErrorType(ErrorTypeEnum.AgentError)
.setErrorCode("whiteList-0001")
.setMessage(message)
.setDetail(ResultSerializer.serializeObject(e, 2))
.closePradar(ConfigNames.URL_WHITE_LIST)
.report();
throw e;
}
} catch (Throwable e) {
String message = "WhiteListError: [" + AppNameUtils.appName() + "] url [" + url + "] is not allowed in WhiteList.";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
ErrorReporter.buildError()
.setErrorType(ErrorTypeEnum.AgentError)
.setErrorCode("whiteList-0001")
.setMessage(message)
.setDetail(ResultSerializer.serializeObject(e, 2))
.closePradar(ConfigNames.URL_WHITE_LIST)
.report();
throw new PressureMeasureError(message, e);
}
}
return ArbiterHttpExit.failure();
}
/**
* 验证压测流量的有效性,如果是压测流量但是压测未开启则直接报错
*/
public final static MatchConfig httpClusterTest(String url) {
validateClusterTest();
try {
if (Pradar.isClusterTest()) {
return ArbiterHttpExit.shallWePassHttpString(url);
}
} catch (PressureMeasureError e) {
String message = "WhiteListError: [" + AppNameUtils.appName() + "] url [" + url + "] is not allowed in WhiteList.";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
ErrorReporter.buildError()
.setErrorType(ErrorTypeEnum.AgentError)
.setErrorCode("whiteList-0001")
.setMessage(message)
.setDetail(ResultSerializer.serializeObject(e, 2))
.closePradar(ConfigNames.URL_WHITE_LIST)
.report();
throw e;
}
} catch (Throwable e) {
String message = "WhiteListError: [" + AppNameUtils.appName() + "] url [" + url + "] is not allowed in WhiteList.";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
ErrorReporter.buildError()
.setErrorType(ErrorTypeEnum.AgentError)
.setErrorCode("whiteList-0001")
.setMessage(message)
.setDetail(ResultSerializer.serializeObject(e, 2))
.closePradar(ConfigNames.URL_WHITE_LIST)
.report();
throw new PressureMeasureError(message, e);
}
}
return ArbiterHttpExit.failure();
}
/**
* 验证压测流量的有效性,如果是压测流量但是压测未开启则直接报错
*/
public final static void validateClusterTest(boolean isClusterTestRequest) {
try {
if (Pradar.isDebug()) {
return;
}
if (!PradarSwitcher.isClusterTestEnabled() && isClusterTestRequest) {
throw new PressureMeasureError("ClusterTestSwitcherError: [" + AppNameUtils.appName() + "] " + PradarSwitcher.getErrorCode() + " " + PradarSwitcher.getErrorMsg(), true);
}
} catch (PressureMeasureError e) {
String message = "ClusterTestSwitcherError: [" + AppNameUtils.appName() + "] validateClusterTest err!";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
reportFastDebugErrorInfo("agent-0008", e, message);
throw e;
}
} catch (Throwable e) {
String message = "ClusterTestSwitcherError: [" + AppNameUtils.appName() + "] validateClusterTest err!";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
reportFastDebugErrorInfo("agent-0008", e, message);
throw new PressureMeasureError(message, e);
}
}
}
/**
* 验证压测流量的有效性,如果是压测流量但是压测未开启则直接报错
*/
public final static void validateClusterTest() {
try {
if (Pradar.isDebug()) {
return;
}
if (!PradarSwitcher.isClusterTestEnabled() && Pradar.isClusterTest()) {
throw new PressureMeasureError("ClusterTestSwitcherError: [" + AppNameUtils.appName() + "] " + PradarSwitcher.getErrorCode() + " " + PradarSwitcher.getErrorMsg());
}
} catch (PressureMeasureError e) {
String message = "ClusterTestSwitcherError: [" + AppNameUtils.appName() + "] validateClusterTest err!";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
reportFastDebugErrorInfo("agent-0008", e, message);
throw e;
}
} catch (Throwable e) {
String message = "ClusterTestSwitcherError: [" + AppNameUtils.appName() + "] validateClusterTest err!";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
reportFastDebugErrorInfo("agent-0008", e, message);
throw new PressureMeasureError(message, e);
}
}
}
/**
* 校验压测流量
*
* @param context
*/
public final static void validateClusterTest(Map context) {
try {
if (Pradar.isDebug()) {
return;
}
boolean isClusterTestRequest = isClusterTestRequest(context);
if (isClusterTestRequest && !PradarSwitcher.isClusterTestEnabled()) {
throw new PressureMeasureError("ClusterTestSwitcherError: [" + AppNameUtils.appName() + "] " + PradarSwitcher.getErrorCode() + " " + PradarSwitcher.getErrorMsg());
}
} catch (PressureMeasureError e) {
String message = "ClusterTestSwitcherError: [" + AppNameUtils.appName() + "] validateClusterTest err!";
LOGGER.error(message, e);
if (Pradar.isClusterTest()) {
reportFastDebugErrorInfo("agent-0008", e, message);
throw e;
}
} catch (Throwable t) {
String message = "ClusterTestSwitcherError: [" + AppNameUtils.appName() + "] validateClusterTest err!";
LOGGER.error(message, t);
if (Pradar.isClusterTest()) {
reportFastDebugErrorInfo("agent-0008", t, message);
throw new PressureMeasureError(message, t);
}
}
}
}
