com.streamsets.pipeline.api.impl.ErrorMessage Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2017 StreamSets Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.streamsets.pipeline.api.impl;

import com.streamsets.pipeline.api.AntennaDoctorMessage;
import com.streamsets.pipeline.api.ErrorCode;
import com.streamsets.pipeline.api.StageException;

import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.Collections;
import java.util.List;

public class ErrorMessage implements LocalizableString {
  private static final Object[] NULL_ONE_ARG = {null};

  private final String errorCode;
  private final LocalizableString localizableMessage;
  private final long timestamp;
  private final boolean preppendErrorCode;
  private final String stackTrace;
  private final List antennaDoctorMessages;

  public ErrorMessage(final StageException ex) {
    errorCode = ex.getErrorCode().getCode();
    timestamp = System.currentTimeMillis();
    stackTrace = toStackTrace(ex);

    localizableMessage = new LocalizableString() {
      @Override
      public String getNonLocalized() {
        return ex.getMessage();
      }

      @Override
      public String getLocalized() {
        return ex.getLocalizedMessage();
      }
    };
    preppendErrorCode = false;
    antennaDoctorMessages = Collections.emptyList();
  }

  public ErrorMessage(String errorCode, final String nonLocalizedMsg, long timestamp) {
    this.errorCode = errorCode;
    this.localizableMessage = new LocalizableString() {
      @Override
      public String getNonLocalized() {
        return nonLocalizedMsg;
      }

      @Override
      public String getLocalized() {
        return nonLocalizedMsg;
      }
    };
    this.timestamp = timestamp;
    stackTrace = null;
    preppendErrorCode = true;
    antennaDoctorMessages = Collections.emptyList();
  }

  public ErrorMessage(List antennaDoctorMessages, ErrorCode errorCode, Object... params) {
    this(antennaDoctorMessages, Utils.checkNotNull(errorCode, "errorCode").getClass().getName() + "-bundle", errorCode, params);
  }

  public ErrorMessage(ErrorCode errorCode, Object... params) {
    this(Collections.emptyList(), Utils.checkNotNull(errorCode, "errorCode").getClass().getName() + "-bundle", errorCode, params);
  }

  public ErrorMessage(String resourceBundle, ErrorCode errorCode, Object... params) {
    this(Collections.emptyList(), resourceBundle, errorCode, params);
  }

  public List getAntennaDoctorMessages() {
    return antennaDoctorMessages;
  }

  public ErrorMessage(List antennaDoctorMessages, String resourceBundle, ErrorCode errorCode, Object... params) {
    this.errorCode = Utils.checkNotNull(errorCode, "errorCode").getCode();
    if ( params != null && params.length > 0 && params[params.length-1] instanceof Throwable){
      stackTrace = toStackTrace((Throwable)params[params.length-1]);
    } else {
      stackTrace = null;
    }

    localizableMessage = new LocalizableMessage(
        errorCode.getClass().getClassLoader(),
        resourceBundle,
        errorCode.getCode(),
        errorCode.getMessage(),
        (params != null) ? params : NULL_ONE_ARG
    );
    timestamp = System.currentTimeMillis();
    preppendErrorCode = true;
    this.antennaDoctorMessages = Collections.unmodifiableList(antennaDoctorMessages);
  }

  /*
    Utility function, accessible from tests
  */
  public static String toStackTrace(Throwable e){
    StringWriter writer = new StringWriter();
    e.printStackTrace(new PrintWriter(writer));
    return writer.toString();
  }

  public String getErrorStackTrace(){
    return stackTrace;
  }

  public String getErrorCode() {
    return errorCode;
  }

  public long getTimestamp() {
    return timestamp;
  }

  @Override
  public String getNonLocalized() {
    return preppendErrorCode ? Utils.format("{} - {}", getErrorCode(), localizableMessage.getNonLocalized()) :
        localizableMessage.getNonLocalized();
  }

  @Override
  public String getLocalized() {
    return preppendErrorCode ? Utils.format("{} - {}", getErrorCode(), localizableMessage.getLocalized()) :
        localizableMessage.getLocalized();
  }

  @Override
  public String toString() {
    return getNonLocalized();
  }
}