com.arextest.schedule.sender.impl.AbstractReplaySender 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.sender.impl;

import com.arextest.schedule.common.CommonConstant;
import com.arextest.schedule.model.ReplayActionCaseItem;
import com.arextest.schedule.model.ReplayActionItem;
import com.arextest.schedule.model.deploy.ServiceInstance;
import com.arextest.schedule.sender.ReplaySendResult;
import com.arextest.schedule.sender.ReplaySender;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import java.util.Base64;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import javax.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.collections4.MapUtils;
import org.apache.commons.lang3.StringUtils;


@Slf4j
public abstract class AbstractReplaySender implements ReplaySender {

  @Resource
  private MockCachePreLoader mockCachePreLoader;
  @Resource
  private ObjectMapper objectMapper;

  protected void bindSendResult(ReplayActionCaseItem caseItem, ReplaySendResult sendResult) {
    caseItem.setTargetResultId(sendResult.getTraceId());
    caseItem.setSendStatus(sendResult.getStatusType().getValue());
  }

  protected void before(ReplayActionCaseItem caseItem) {
    mockCachePreLoader.prepareCache(caseItem);
  }

  protected Map createHeaders(ReplayActionCaseItem caseItem) {
    ReplayActionItem replayActionItem = caseItem.getParent();
    Map headers = newHeadersIfEmpty(caseItem.requestHeaders());
    headers.remove(CommonConstant.AREX_REPLAY_WARM_UP);
    headers.put(CommonConstant.AREX_RECORD_ID, caseItem.getRecordId());

    /**
     * this header will be transparently passed to the storage service
     * and storage may use it to perform extra operations
     */
    headers.put(CommonConstant.AREX_SCHEDULE_REPLAY,
        Optional.ofNullable(caseItem.getCaseSendScene())
            .map(Enum::name)
            .orElse(Boolean.TRUE.toString()));

    String exclusionOperationConfig = replayActionItem.getExclusionOperationConfig();
    if (StringUtils.isNotEmpty(exclusionOperationConfig)) {
      headers.put(CommonConstant.X_AREX_EXCLUSION_OPERATIONS, exclusionOperationConfig);
    }
    return headers;
  }

  /**
   * Please DO NOT use Removing cache might cause unexpected behavior on rerunning the case
   */
  @Deprecated
  protected void after(ReplayActionCaseItem caseItem) {
    mockCachePreLoader.removeCache(caseItem.getRecordId());
  }

  protected Map newHeadersIfEmpty(Map source) {
    if (MapUtils.isEmpty(source)) {
      return new HashMap<>();
    }
    return source;
  }

  protected boolean isReplayRequest(Map requestHeaders) {
    return MapUtils.isNotEmpty(requestHeaders) && requestHeaders.containsKey(
        CommonConstant.AREX_RECORD_ID);
  }

  protected String encodeResponseAsString(Object responseBody) {
    if (responseBody == null) {
      return null;
    }
    if (responseBody instanceof String) {
      return (String) responseBody;
    }
    if (responseBody instanceof byte[]) {
      return Base64.getEncoder().encodeToString((byte[]) responseBody);
    }
    try {
      return objectMapper.writeValueAsString(responseBody);
    } catch (JsonProcessingException e) {
      LOGGER.warn("encodeAsString error:{}", e.getMessage(), e);
    }
    return null;
  }

  protected ServiceInstance selectLoadBalanceInstance(String caseItemId,
      List serviceInstances) {
    if (CollectionUtils.isEmpty(serviceInstances)) {
      return null;
    }
    int index = Math.abs(caseItemId.hashCode() % serviceInstances.size());
    return serviceInstances.get(index);
  }
}