com.arextest.schedule.web.controller.ReplayLocalController Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arex-schedule-web-api
There is a newer version: 2.0.4
package com.arextest.schedule.web.controller;

import com.arextest.schedule.mdc.MDCTracer;
import com.arextest.schedule.model.CommonResponse;
import com.arextest.schedule.model.bizlog.LogUploadRequest;
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.PostSendRequest;
import com.arextest.schedule.model.plan.PreSendRequest;
import com.arextest.schedule.model.plan.QueryReplaySenderParametersRequest;
import com.arextest.schedule.model.plan.QueryReplaySenderParametersResponse;
import com.arextest.schedule.model.plan.ReRunReplayPlanRequest;
import com.arextest.schedule.service.LocalReplayService;
import com.arextest.schedule.service.PlanBizLogService;
import com.arextest.schedule.service.PlanProduceService;
import javax.validation.Valid;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

/**
 * @author wildeslam.
 * @create 2023/11/15 17:23
 */
@Slf4j
@Controller
@RequestMapping(path = "/api/replay/local/", produces = {MediaType.APPLICATION_JSON_VALUE})
@RequiredArgsConstructor
public class ReplayLocalController {

  private static final String SUCCESS_DESC = "success";

  private final LocalReplayService localReplayService;
  private final PlanProduceService planProduceService;
  private final PlanBizLogService planBizLogService;

  @PostMapping(value = "/queryCaseId")
  @ResponseBody
  public CommonResponse queryCaseId(@Valid @RequestBody BuildReplayPlanRequest request) {
    try {
      planProduceService.fillOptionalValueIfRequestMissed(request);
      return localReplayService.queryReplayCaseId(request);
    } catch (Exception e) {
      LOGGER.error("queryCaseId error: {} , request: {}", e.getMessage(), request, e);
      return CommonResponse.badResponse("queryCaseId error！" + e.getMessage(),
          new BuildReplayPlanResponse(BuildReplayFailReasonEnum.UNKNOWN));
    } finally {
      planProduceService.removeCreating(request.getAppId(), request.getTargetEnv());
    }
  }

  @PostMapping(value = "/queryReplaySenderParameters")
  @ResponseBody
  public CommonResponse queryCases(@Valid @RequestBody QueryReplaySenderParametersRequest request) {
    if (CollectionUtils.isEmpty(request.getCaseIds())) {
      return CommonResponse.badResponse("No caseId!");
    }
    try {
      QueryReplaySenderParametersResponse response = localReplayService.queryReplaySenderParameters(
          request);
      return CommonResponse.successResponse(SUCCESS_DESC, response);
    } finally {
      MDCTracer.clear();
    }
  }

  @PostMapping(value = "/preSend")
  @ResponseBody
  public CommonResponse preSend(@Valid @RequestBody PreSendRequest request) {
    boolean success = localReplayService.preSend(request);
    if (success) {
      return CommonResponse.successResponse(SUCCESS_DESC, success);
    } else {
      return CommonResponse.badResponse("SendLimiter break!", success);
    }
  }

  @PostMapping(value = "/postSend")
  @ResponseBody
  public CommonResponse postSend(@Valid @RequestBody PostSendRequest request) {
    localReplayService.postSend(request);
    return CommonResponse.successResponse(SUCCESS_DESC, true);
  }

  @PostMapping(value = "/queryReRunCaseId")
  @ResponseBody
  public CommonResponse queryReRunCaseId(@Valid @RequestBody ReRunReplayPlanRequest request) {
    try {
      return localReplayService.queryReRunCaseId(request);
    } catch (Exception e) {
      LOGGER.error("queryReRunCaseId error: {} , request: {}", e.getMessage(), request, e);
      return CommonResponse.badResponse("queryReRunCaseId error！" + e.getMessage(),
          new BuildReplayPlanResponse(BuildReplayFailReasonEnum.UNKNOWN));
    }
  }

  @PostMapping(value = "/log")
  @ResponseBody
  public CommonResponse log(@RequestBody LogUploadRequest request) {
    planBizLogService.upload(request.getLogs());
    return CommonResponse.successResponse(SUCCESS_DESC, true);
  }
}