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.ReplayCaseRemoteLoadService Maven / Gradle / Ivy
package com.arextest.schedule.service;
import com.arextest.model.mock.AREXMocker;
import com.arextest.model.mock.MockCategoryType;
import com.arextest.model.mock.Mocker.Target;
import com.arextest.model.replay.PagedRequestType;
import com.arextest.model.replay.PagedResponseType;
import com.arextest.model.replay.QueryCaseCountResponseType;
import com.arextest.model.replay.SortingOption;
import com.arextest.model.replay.SortingTypeEnum;
import com.arextest.model.replay.ViewRecordRequestType;
import com.arextest.model.replay.ViewRecordResponseType;
import com.arextest.schedule.client.HttpWepServiceApiClient;
import com.arextest.schedule.common.CommonConstant;
import com.arextest.schedule.model.CaseProviderEnum;
import com.arextest.schedule.model.CaseSendStatusType;
import com.arextest.schedule.model.CompareProcessStatusType;
import com.arextest.schedule.model.LogType;
import com.arextest.schedule.model.OperationTypeData;
import com.arextest.schedule.model.ReplayActionCaseItem;
import com.arextest.schedule.model.ReplayActionItem;
import com.arextest.schedule.model.ReplayPlan;
import com.arextest.schedule.utils.MapUtils;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import jakarta.annotation.Resource;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.StopWatch;
/**
* Created by wang_yc on 2021/9/15
*/
@Service
@Slf4j
public class ReplayCaseRemoteLoadService {
private static final int EMPTY_SIZE = 0;
private static final int AUTO_PINED_CASE_LIMIT = 20000;
private static final String CREATE_TIME_COLUMN_NAME = "creationTime";
@Resource
private HttpWepServiceApiClient wepApiClientService;
@Value("${arex.storage.viewRecord.url}")
private String viewRecordUrl;
@Value("${arex.storage.countByRange.url}")
private String countByRangeUrl;
@Value("${arex.storage.replayCase.url}")
private String replayCaseUrl;
@Resource
private ObjectMapper objectMapper;
@Resource
private MetricService metricService;
public int queryCaseCount(ReplayActionItem replayActionItem, String providerName) {
int queryTotalCount = EMPTY_SIZE;
try {
int caseCountLimit =
CaseProviderEnum.AUTO_PINNED.getName().equals(providerName) ? AUTO_PINED_CASE_LIMIT :
replayActionItem.getParent().getCaseCountLimit();
List operationTypes = replayActionItem.getOperationTypes();
if (CollectionUtils.isEmpty(operationTypes)) {
return queryTotalCount;
}
for (OperationTypeData operationTypeData : operationTypes) {
PagedRequestType pagedRequestType = buildPagingSearchCaseRequest(replayActionItem,
caseCountLimit, providerName, operationTypeData.getOperationType());
QueryCaseCountResponseType responseType =
wepApiClientService.jsonPost(countByRangeUrl, pagedRequestType,
QueryCaseCountResponseType.class);
if (responseType == null || responseType.getResponseStatusType().hasError()) {
continue;
}
queryTotalCount =
queryTotalCount + Math.min(replayActionItem.getParent().getCaseCountLimit(),
(int) responseType.getCount());
}
} catch (Exception e) {
LOGGER.error("query case count error,request: {} ", replayActionItem.getId(), e);
}
return queryTotalCount;
}
public ReplayActionCaseItem viewReplayLoad(ReplayActionCaseItem caseItem, String sourceProvider,
String operationType) {
try {
ViewRecordRequestType viewReplayCaseRequest = new ViewRecordRequestType();
viewReplayCaseRequest.setRecordId(caseItem.getRecordId());
viewReplayCaseRequest.setCategoryType(operationType);
viewReplayCaseRequest.setSourceProvider(sourceProvider);
ViewRecordResponseType responseType = wepApiClientService.jsonPost(viewRecordUrl,
viewReplayCaseRequest,
ViewRecordResponseType.class);
if (responseType == null || responseType.getResponseStatusType().hasError()) {
LOGGER.warn("view record response invalid recordId:{},response:{}",
viewReplayCaseRequest.getRecordId(), responseType);
return null;
}
List recordResultList = responseType.getRecordResult();
if (CollectionUtils.isEmpty(recordResultList)) {
LOGGER.warn("view record response empty result recordId:{}",
viewReplayCaseRequest.getRecordId());
return null;
}
return toCaseItem(recordResultList.get(0));
} catch (Throwable e) {
LOGGER.error("view record error: {},recordId: {}", e.getMessage(), caseItem.getRecordId(), e);
}
return null;
}
public ReplayActionCaseItem viewReplayLoad(ReplayActionCaseItem caseItem,
Set operationTypes) {
ReplayActionCaseItem viewReplay = null;
if (CollectionUtils.isEmpty(operationTypes)) {
viewReplay = viewReplayLoad(caseItem, CaseProviderEnum.PINNED.getName(),
caseItem.getParent().getActionType());
} else {
for (String operationType : operationTypes) {
viewReplay = viewReplayLoad(caseItem, CaseProviderEnum.PINNED.getName(), operationType);
if (viewReplay != null) {
break;
}
}
}
return viewReplay;
}
private ReplayActionCaseItem toCaseItem(AREXMocker mainEntry) {
Target targetRequest = mainEntry.getTargetRequest();
if (targetRequest == null) {
return null;
}
ReplayActionCaseItem caseItem = new ReplayActionCaseItem();
caseItem.setTargetRequest(targetRequest);
caseItem.setRecordId(mainEntry.getRecordId());
caseItem.setRecordTime(mainEntry.getCreationTime());
caseItem.setCaseType(mainEntry.getCategoryType().getName());
caseItem.setSendStatus(CaseSendStatusType.WAIT_HANDLING.getValue());
caseItem.setCompareStatus(CompareProcessStatusType.WAIT_HANDLING.getValue());
caseItem.setSourceResultId(StringUtils.EMPTY);
caseItem.setTargetResultId(StringUtils.EMPTY);
caseItem.setExpirationTime(mainEntry.getExpirationTime());
return caseItem;
}
public List pagingLoad(long beginTimeMills, long endTimeMills,
ReplayActionItem replayActionItem, int caseCountLimit, String providerName, String operationType) {
PagedRequestType requestType = buildPagingSearchCaseRequest(replayActionItem,
caseCountLimit, providerName, operationType);
buildSortOption(requestType);
requestType.setBeginTime(beginTimeMills);
requestType.setEndTime(endTimeMills);
PagedResponseType responseType;
StopWatch watch = new StopWatch();
watch.start(LogType.LOAD_CASE_TIME.getValue());
responseType = wepApiClientService.jsonPost(replayCaseUrl, requestType,
PagedResponseType.class);
watch.stop();
LOGGER.info("get replay case app id:{},time used:{} ms, operation:{}",
requestType.getAppId(),
watch.getTotalTimeMillis(), requestType.getOperation()
);
metricService.recordTimeEvent(LogType.LOAD_CASE_TIME.getValue(), replayActionItem.getPlanId(),
replayActionItem.getAppId(), null, watch.getTotalTimeMillis());
if (badResponse(responseType)) {
try {
LOGGER.warn("get replay case is empty,request:{} , response:{}",
objectMapper.writeValueAsString(requestType),
objectMapper.writeValueAsString(responseType));
} catch (JsonProcessingException e) {
LOGGER.error(e.getMessage(), e);
}
return Collections.emptyList();
}
return toCaseItemList(responseType.getRecords());
}
/**
* Flashback to pull data
*/
private static void buildSortOption(PagedRequestType requestType) {
SortingOption sortingOption = new SortingOption();
sortingOption.setLabel(CREATE_TIME_COLUMN_NAME);
sortingOption.setSortingType(SortingTypeEnum.DESCENDING.getCode());
requestType.setSortingOptions(Collections.singletonList(sortingOption));
}
private boolean badResponse(PagedResponseType responseType) {
if (responseType == null) {
return true;
}
if (responseType.getResponseStatusType().hasError()) {
return true;
}
return CollectionUtils.isEmpty(responseType.getRecords());
}
private List toCaseItemList(List source) {
List caseItemList = new ArrayList<>(source.size());
for (AREXMocker soa : source) {
ReplayActionCaseItem caseItem = toCaseItem(soa);
caseItemList.add(caseItem);
}
return caseItemList;
}
private PagedRequestType buildPagingSearchCaseRequest(ReplayActionItem replayActionItem,
int caseCountLimit, String providerName, String operationType) {
ReplayPlan parent = replayActionItem.getParent();
PagedRequestType requestType = new PagedRequestType();
requestType.setAppId(parent.getAppId());
requestType.setPageSize(Math.min(CommonConstant.MAX_PAGE_SIZE, caseCountLimit));
requestType.setEnv(parent.getCaseSourceType());
requestType.setBeginTime(parent.getCaseSourceFrom().getTime());
requestType.setEndTime(parent.getCaseSourceTo().getTime());
requestType.setOperation(replayActionItem.getOperationName());
if (StringUtils.isEmpty(operationType)) {
requestType.setCategory(MockCategoryType.createEntryPoint(replayActionItem.getActionType()));
} else {
requestType.setCategory(MockCategoryType.createEntryPoint(operationType));
}
requestType.setSourceProvider(providerName);
// add the condition of "caseTag"
if (MapUtils.isNotEmpty(parent.getCaseTags())){
requestType.setTags(parent.getCaseTags());
}
return requestType;
}
}
