com.arextest.storage.trace.MDCTracer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arex-storage-web-api Show documentation

There is a newer version: 2.0.7

package com.arextest.storage.trace;

import com.arextest.model.mock.MockCategoryType;
import com.arextest.model.mock.Mocker;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.MDC;

/**
 * @author jmo
 * @since 2021/11/5
 */
public final class MDCTracer {

  private static final String CATEGORY = "category";
  private static final String REPLAY_ID = "replayId";
  private static final String RECORD_ID = "recordId";
  private static final String APP_TYPE = "app-type";
  private static final String AREX_STORAGE = "arex-storage";
  private static final String CONVERT_CACHE_KEY = "convertCacheKey";

  private MDCTracer() {

  }

  public static void addTrace(Mocker requestType) {
    if (requestType == null) {
      return;
    }
    addCategory(requestType.getCategoryType());
    addRecordId(requestType.getRecordId());
    addReplayId(requestType.getReplayId());
    addAppId(requestType.getAppId());
  }

  private static void put(String name, String value) {
    if (StringUtils.isNotEmpty(value)) {
      MDC.put(name, value);
    }
  }

  public static void addAppType() {
    MDC.put(APP_TYPE, AREX_STORAGE);
  }

  public static void addReplayId(String replayId) {
    if (StringUtils.isEmpty(replayId)) {
      return;
    }
    addAppType();
    put(REPLAY_ID, replayId);
  }

  public static void addRecordId(String recordId) {
    addAppType();
    put(RECORD_ID, recordId);
  }

  public static void addCategory(MockCategoryType category) {
    addAppType();
    if (category != null) {
      put(CATEGORY, category.getName());
    }
  }

  public static void addConvertCacheKey(String cacheKey) {
    addAppType();
    put(CONVERT_CACHE_KEY, cacheKey);
  }

  public static void addAppId(String appId) {
    addAppType();
    put("appId", appId);
  }

  public static void removeCategory() {
    MDC.remove(CATEGORY);
  }

  public static void removeConvertCacheKey(){
    MDC.remove(CONVERT_CACHE_KEY);
  }

  public static void clear() {
    MDC.clear();
  }
}