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.arextest.schedule.service.PlanProduceService Maven / Gradle / Ivy
package com.arextest.schedule.service;
import static com.arextest.schedule.common.CommonConstant.CREATE_PLAN_REDIS_EXPIRE;
import static com.arextest.schedule.common.CommonConstant.ONE_DAY_MILLIS;
import static com.arextest.schedule.common.CommonConstant.OPERATION_MAX_CASE_COUNT;
import static com.arextest.schedule.common.CommonConstant.STOP_PLAN_REDIS_EXPIRE;
import com.arextest.common.cache.CacheProvider;
import com.arextest.common.config.DefaultApplicationConfig;
import com.arextest.schedule.bizlog.BizLogger;
import com.arextest.schedule.common.CommonConstant;
import com.arextest.schedule.dao.mongodb.ReplayActionCaseItemRepository;
import com.arextest.schedule.dao.mongodb.ReplayPlanActionRepository;
import com.arextest.schedule.dao.mongodb.ReplayPlanRepository;
import com.arextest.schedule.exceptions.PlanRunningException;
import com.arextest.schedule.mdc.MDCTracer;
import com.arextest.schedule.model.CaseSourceEnvType;
import com.arextest.schedule.model.CommonResponse;
import com.arextest.schedule.model.ReplayActionCaseItem;
import com.arextest.schedule.model.ReplayActionItem;
import com.arextest.schedule.model.ReplayPlan;
import com.arextest.schedule.model.deploy.DeploymentVersion;
import com.arextest.schedule.model.deploy.ServiceInstance;
import com.arextest.schedule.model.plan.BuildReplayFailReasonEnum;
import com.arextest.schedule.model.plan.BuildReplayPlanRequest;
import com.arextest.schedule.model.plan.BuildReplayPlanResponse;
import com.arextest.schedule.model.plan.PlanStageEnum;
import com.arextest.schedule.model.plan.ReRunReplayPlanRequest;
import com.arextest.schedule.model.plan.StageStatusEnum;
import com.arextest.schedule.plan.PlanContext;
import com.arextest.schedule.plan.PlanContextCreator;
import com.arextest.schedule.plan.builder.BuildPlanValidateResult;
import com.arextest.schedule.plan.builder.ReplayPlanBuilder;
import com.arextest.schedule.planexecution.PlanExecutionMonitor;
import com.arextest.schedule.progress.ProgressEvent;
import com.arextest.schedule.utils.RedisKeyBuildUtils;
import com.arextest.schedule.utils.ReplayParentBinder;
import com.arextest.schedule.utils.StageUtils;
import jakarta.annotation.Resource;
import java.nio.charset.StandardCharsets;
import java.util.Date;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;
/**
* Created by wang_yc on 2021/9/15
*/
@Service
@Slf4j
public class PlanProduceService {
private static final String CASE_SOURCE_TO_OFFSET_MILLIS = "case.source.to.offset.millis";
@Resource
private List replayPlanBuilderList;
@Resource
private PlanContextCreator planContextCreator;
@Resource
private ReplayPlanRepository replayPlanRepository;
@Resource
private ReplayPlanActionRepository replayPlanActionRepository;
@Resource
private PlanConsumeService planConsumeService;
@Resource
private PlanConsumePrepareService planConsumePrepareService;
@Resource
private ProgressEvent progressEvent;
@Resource
private ConfigurationService configurationService;
@Resource
private CacheProvider redisCacheProvider;
@Resource
private PlanExecutionMonitor planExecutionMonitorImpl;
@Resource
private ReplayActionCaseItemRepository replayActionCaseItemRepository;
@Resource
private DefaultApplicationConfig defaultConfig;
public CommonResponse createPlan(BuildReplayPlanRequest request) throws PlanRunningException {
fillOptionalValueIfRequestMissed(request);
progressEvent.onBeforePlanCreate(request);
long planCreateMillis = System.currentTimeMillis();
String appId = request.getAppId();
if (isCreating(appId, request.getTargetEnv())) {
progressEvent.onReplayPlanCreateException(request);
return CommonResponse.badResponse("This appid is creating plan",
new BuildReplayPlanResponse(BuildReplayFailReasonEnum.CREATING));
}
ReplayPlanBuilder planBuilder = select(request);
if (planBuilder == null) {
progressEvent.onReplayPlanCreateException(request);
return CommonResponse.badResponse(
"appId:" + appId + " unsupported replay planType : " + request.getReplayPlanType(),
new BuildReplayPlanResponse(BuildReplayFailReasonEnum.INVALID_REPLAY_TYPE));
}
PlanContext planContext = planContextCreator.createByAppId(appId);
BuildPlanValidateResult result = planBuilder.validate(request, planContext);
if (result.failure()) {
progressEvent.onReplayPlanCreateException(request);
return CommonResponse.badResponse("appId:" + appId + " error: " + result.getRemark(),
new BuildReplayPlanResponse(this.validateToResultReason(result)));
}
List replayActionItemList = planBuilder.buildReplayActionList(request,
planContext);
if (CollectionUtils.isEmpty(replayActionItemList)) {
progressEvent.onReplayPlanCreateException(request);
return CommonResponse.badResponse("appId:" + appId + " error: empty replay actions",
new BuildReplayPlanResponse(BuildReplayFailReasonEnum.NO_INTERFACE_FOUND));
}
ReplayPlan replayPlan = build(request, planContext);
replayPlan.setPlanCreateMillis(planCreateMillis);
replayPlan.setReplayActionItemList(replayActionItemList);
ReplayParentBinder.setupReplayActionParent(replayActionItemList, replayPlan);
// todo: add trans
progressEvent.onReplayPlanStageUpdate(replayPlan, PlanStageEnum.SAVE_PLAN,
StageStatusEnum.ONGOING, System.currentTimeMillis(), null);
if (!replayPlanRepository.save(replayPlan)) {
progressEvent.onReplayPlanCreateException(request);
return CommonResponse.badResponse("save replan plan error, " + replayPlan.toString(),
new BuildReplayPlanResponse(BuildReplayFailReasonEnum.DB_ERROR));
}
isRunning(replayPlan.getId());
MDCTracer.addPlanId(replayPlan.getId());
planExecutionMonitorImpl.register(replayPlan);
if (!replayPlanActionRepository.save(replayActionItemList)) {
progressEvent.onReplayPlanCreateException(request);
return CommonResponse.badResponse("save replay action error, " + replayPlan.toString(),
new BuildReplayPlanResponse(BuildReplayFailReasonEnum.DB_ERROR));
}
progressEvent.onReplayPlanStageUpdate(replayPlan, PlanStageEnum.SAVE_PLAN,
StageStatusEnum.SUCCEEDED,
System.currentTimeMillis(), null);
BizLogger.recordPlanStart(replayPlan);
progressEvent.onReplayPlanCreated(replayPlan);
planConsumeService.runAsyncConsume(replayPlan);
return CommonResponse.successResponse("create plan success！" + result.getRemark(),
new BuildReplayPlanResponse(replayPlan.getId()));
}
public void fillOptionalValueIfRequestMissed(BuildReplayPlanRequest request) {
long currentTimeMillis = System.currentTimeMillis();
Date fromDate = new Date(currentTimeMillis - CommonConstant.ONE_DAY_MILLIS);
Date toDate = new Date(currentTimeMillis -
defaultConfig.getConfigAsLong(CASE_SOURCE_TO_OFFSET_MILLIS, 0L));
if (request.getCaseSourceFrom() == null) {
request.setCaseSourceFrom(fromDate);
}
if (request.getCaseSourceTo() == null || request.getCaseSourceTo().after(toDate)) {
request.setCaseSourceTo(toDate);
}
if (request.getCaseSourceType() == null) {
request.setCaseSourceType(CaseSourceEnvType.TEST.getValue());
}
}
public ReplayPlan build(BuildReplayPlanRequest request, PlanContext planContext) {
String appId = request.getAppId();
ReplayPlan replayPlan = new ReplayPlan();
// init
replayPlan.setReplayPlanStageList(StageUtils.initPlanStageList(StageUtils.INITIAL_STAGES));
progressEvent.onReplayPlanStageUpdate(replayPlan, PlanStageEnum.BUILD_PLAN,
StageStatusEnum.ONGOING, System.currentTimeMillis(), null);
replayPlan.setAppId(appId);
DeploymentVersion deploymentVersion = planContext.getTargetVersion();
if (deploymentVersion != null) {
replayPlan.setTargetImageId(deploymentVersion.getImageId());
replayPlan.setTargetImageName(deploymentVersion.getImage().getName());
}
List serviceInstances = planContext.targetActiveInstance();
replayPlan.setTargetHost(getIpAddress(serviceInstances));
replayPlan.setTargetEnv(request.getTargetEnv());
serviceInstances = planContext.sourceActiveInstance();
if (CollectionUtils.isEmpty(serviceInstances)) {
replayPlan.setSourceEnv(StringUtils.EMPTY);
replayPlan.setSourceHost(StringUtils.EMPTY);
} else {
replayPlan.setSourceEnv(request.getSourceEnv());
replayPlan.setSourceHost(getIpAddress(serviceInstances));
}
replayPlan.setPlanCreateTime(new Date());
replayPlan.setOperator(request.getOperator());
replayPlan.setCaseSourceFrom(request.getCaseSourceFrom());
replayPlan.setCaseSourceTo(request.getCaseSourceTo());
replayPlan.setCaseSourceType(request.getCaseSourceType());
replayPlan.setReplayPlanType(request.getReplayPlanType());
ConfigurationService.Application replayApp = configurationService.application(appId);
if (replayApp != null) {
replayPlan.setArexCordVersion(replayApp.getAgentVersion());
replayPlan.setArexExtVersion(replayApp.getAgentExtVersion());
replayPlan.setCaseRecordVersion(replayApp.getAgentExtVersion());
replayPlan.setAppName(replayApp.getAppName());
}
replayPlan.setPlanName(getReplayName(request, replayApp));
ConfigurationService.ScheduleConfiguration schedule = configurationService.schedule(appId);
replayPlan.setReplaySendMaxQps(
Optional.ofNullable(schedule).map(ConfigurationService.ScheduleConfiguration::getSendMaxQps)
.orElse(null)
);
if (request.getCaseCountLimit() == null || request.getCaseCountLimit() <= 0) {
replayPlan.setCaseCountLimit(OPERATION_MAX_CASE_COUNT);
} else {
replayPlan.setCaseCountLimit(request.getCaseCountLimit());
}
replayPlan.setMinInstanceCount(planContext.determineMinInstanceCount());
// add the condition of "caseTag"
replayPlan.setCaseTags(request.getCaseTags());
progressEvent.onReplayPlanStageUpdate(replayPlan, PlanStageEnum.BUILD_PLAN,
StageStatusEnum.SUCCEEDED, null, System.currentTimeMillis());
return replayPlan;
}
private String getIpAddress(List serviceInstances) {
return serviceInstances.stream().map(ServiceInstance::getIp).collect(Collectors.joining(","));
}
private String getReplayName(BuildReplayPlanRequest request,
ConfigurationService.Application appInfo) {
if (StringUtils.isNotBlank(request.getPlanName())) {
return request.getPlanName();
}
StringBuilder stringBuilder = new StringBuilder();
if (appInfo != null) {
stringBuilder.append(appInfo.getAppName()).append("_");
} else {
stringBuilder.append("unknown-appName").append("_");
}
String currentTimeMillis = String.valueOf(System.currentTimeMillis());
String lastSixDigits = currentTimeMillis.substring(currentTimeMillis.length() - 6);
stringBuilder.append(lastSixDigits);
return stringBuilder.toString();
}
public ReplayPlanBuilder select(BuildReplayPlanRequest request) {
for (ReplayPlanBuilder replayPlanBuilder : replayPlanBuilderList) {
if (replayPlanBuilder.isSupported(request)) {
return replayPlanBuilder;
}
}
return null;
}
public Boolean isCreating(String appId, String targetEnv) {
try {
byte[] key = String.format("schedule_creating_%s_%s", appId, targetEnv)
.getBytes(StandardCharsets.UTF_8);
byte[] value = appId.getBytes(StandardCharsets.UTF_8);
boolean result = redisCacheProvider.putIfAbsent(key, CREATE_PLAN_REDIS_EXPIRE, value);
return !result;
} catch (Exception e) {
LOGGER.error("isCreating error : {}", e.getMessage(), e);
return true;
}
}
public void removeCreating(String appId, String targetEnv) {
try {
byte[] key = String.format("schedule_creating_%s_%s", appId, targetEnv)
.getBytes(StandardCharsets.UTF_8);
redisCacheProvider.remove(key);
} catch (Exception e) {
LOGGER.error("removeCreating error : {}", e.getMessage(), e);
}
}
public Boolean isRunning(String planId) {
try {
byte[] key = RedisKeyBuildUtils.buildPlanRunningRedisKey(planId);
byte[] value = planId.getBytes(StandardCharsets.UTF_8);
if (redisCacheProvider.get(key) != null) {
return true;
}
redisCacheProvider.put(key, ONE_DAY_MILLIS, value);
return false;
} catch (Exception e) {
LOGGER.error("isRunning error : {}", e.getMessage(), e);
return true;
}
}
public void stopPlan(String planId, String operator) {
try {
// set key for other instance to stop internal execution
redisCacheProvider.putIfAbsent(RedisKeyBuildUtils.buildStopPlanRedisKey(planId),
STOP_PLAN_REDIS_EXPIRE, planId.getBytes(StandardCharsets.UTF_8));
// set the canceled status immediately to give quick response to user
progressEvent.onReplayPlanTerminate(planId, "Plan Cancelled by " + operator);
} catch (Exception e) {
LOGGER.error("stopPlan error, planId: {}, message: {}", planId, e.getMessage());
}
}
/**
* map internal validation code to VO reason code
*/
public BuildReplayFailReasonEnum validateToResultReason(BuildPlanValidateResult validateResult) {
switch (validateResult.getCodeValue()) {
case BuildPlanValidateResult.REQUESTED_EMPTY_OPERATION:
return BuildReplayFailReasonEnum.NO_INTERFACE_FOUND;
case BuildPlanValidateResult.UNSUPPORTED_CASE_SOURCE_TYPE:
return BuildReplayFailReasonEnum.INVALID_SOURCE_TYPE;
case BuildPlanValidateResult.REQUESTED_CASE_TIME_RANGE_UNSUPPORTED:
return BuildReplayFailReasonEnum.INVALID_CASE_RANGE;
case BuildPlanValidateResult.REQUESTED_TARGET_ENV_UNAVAILABLE:
case BuildPlanValidateResult.REQUESTED_SOURCE_ENV_UNAVAILABLE:
return BuildReplayFailReasonEnum.NO_ACTIVE_SERVICE_INSTANCE;
default:
return BuildReplayFailReasonEnum.UNKNOWN;
}
}
public CommonResponse reRunPlan(ReRunReplayPlanRequest request) throws PlanRunningException {
final String planId = request.getPlanId();
final String planItemId = request.getPlanItemId();
ReplayPlan replayPlan = replayPlanRepository.query(planId);
if (replayPlan == null) {
progressEvent.onReplayPlanReRunException(replayPlan);
return CommonResponse.badResponse("target plan not found");
}
replayPlan.setPlanCreateMillis(System.currentTimeMillis());
progressEvent.onBeforePlanReRun(replayPlan);
if (replayPlan.getReplayPlanStageList() == null) {
progressEvent.onReplayPlanReRunException(replayPlan);
return CommonResponse.badResponse("The plan's version is too old");
}
List failedCaseList = replayActionCaseItemRepository.failedCaseList(
planId, planItemId);
if (CollectionUtils.isEmpty(failedCaseList)) {
progressEvent.onReplayPlanReRunException(replayPlan);
return CommonResponse.badResponse("No failed case or cases are expired!");
}
if (isRunning(planId)) {
progressEvent.onReplayPlanReRunException(replayPlan);
return CommonResponse.badResponse("This plan is Running");
}
replayPlan.setReRun(Boolean.TRUE);
replayPlan.setOperator(request.getOperator());
try {
ConfigurationService.ScheduleConfiguration schedule = configurationService.schedule(
replayPlan.getAppId());
if (schedule != null) {
replayPlan.setReplaySendMaxQps(schedule.getSendMaxQps());
}
planExecutionMonitorImpl.register(replayPlan);
progressEvent.onReplayPlanReRun(replayPlan);
progressEvent.onReplayPlanStageUpdate(replayPlan, PlanStageEnum.LOADING_CASE,
StageStatusEnum.ONGOING, System.currentTimeMillis(), null);
progressEvent.onUpdateFailedCases(replayPlan, failedCaseList);
planConsumePrepareService.updateFailedActionAndCase(replayPlan, failedCaseList);
if (CollectionUtils.isEmpty(replayPlan.getReplayActionItemList())) {
throw new RuntimeException("no replayActionItem!");
}
progressEvent.onReplayPlanStageUpdate(replayPlan, PlanStageEnum.LOADING_CASE,
StageStatusEnum.SUCCEEDED, null, System.currentTimeMillis());
} catch (Exception e) {
progressEvent.onReplayPlanStageUpdate(replayPlan, PlanStageEnum.LOADING_CASE,
StageStatusEnum.FAILED, null, System.currentTimeMillis());
progressEvent.onReplayPlanStageUpdate(replayPlan, PlanStageEnum.RE_RUN,
StageStatusEnum.FAILED, System.currentTimeMillis(), null);
planExecutionMonitorImpl.deregister(replayPlan);
progressEvent.onReplayPlanReRunException(replayPlan, e);
return CommonResponse.badResponse("ReRun plan failed！");
}
planConsumeService.runAsyncConsume(replayPlan);
return CommonResponse.successResponse("ReRun plan success！",
new BuildReplayPlanResponse(replayPlan.getId()));
}
}
