コード例 #1
0
ファイル: QueryId.java プロジェクト: niumowm/tajo
 public QueryId() {
   builder = ApplicationAttemptIdProto.newBuilder();
 }
コード例 #2
0
ファイル: QueryId.java プロジェクト: niumowm/tajo
 private synchronized void maybeInitBuilder() {
   if (viaProto || builder == null) {
     builder = ApplicationAttemptIdProto.newBuilder(proto);
   }
   viaProto = false;
 }
コード例 #3
0
ファイル: QueryId.java プロジェクト: niumowm/tajo
/** QueryId represents a unique identifier of a query. */
public class QueryId implements Comparable<QueryId> {
  public static final String PREFIX = "q";
  public static final String SEPARATOR = "_";

  ApplicationAttemptIdProto proto = ApplicationAttemptIdProto.getDefaultInstance();
  ApplicationAttemptIdProto.Builder builder = null;
  boolean viaProto = false;

  private ApplicationId applicationId = null;

  public QueryId() {
    builder = ApplicationAttemptIdProto.newBuilder();
  }

  public QueryId(ApplicationAttemptIdProto proto) {
    this.proto = proto;
    viaProto = true;
  }

  public synchronized ApplicationAttemptIdProto getProto() {
    mergeLocalToProto();
    proto = viaProto ? proto : builder.build();
    viaProto = true;
    return proto;
  }

  private synchronized void mergeLocalToBuilder() {
    if (this.applicationId != null
        && !((ApplicationIdPBImpl) applicationId).getProto().equals(builder.getApplicationId())) {
      builder.setApplicationId(convertToProtoFormat(this.applicationId));
    }
  }

  private synchronized void mergeLocalToProto() {
    if (viaProto) maybeInitBuilder();
    mergeLocalToBuilder();
    proto = builder.build();
    viaProto = true;
  }

  private synchronized void maybeInitBuilder() {
    if (viaProto || builder == null) {
      builder = ApplicationAttemptIdProto.newBuilder(proto);
    }
    viaProto = false;
  }

  public synchronized int getAttemptId() {
    ApplicationAttemptIdProtoOrBuilder p = viaProto ? proto : builder;
    return (p.getAttemptId());
  }

  public synchronized void setAttemptId(int attemptId) {
    maybeInitBuilder();
    builder.setAttemptId((attemptId));
  }

  public synchronized ApplicationId getApplicationId() {
    ApplicationAttemptIdProtoOrBuilder p = viaProto ? proto : builder;
    if (this.applicationId != null) {
      return this.applicationId;
    }
    if (!p.hasApplicationId()) {
      return null;
    }
    this.applicationId = convertFromProtoFormat(p.getApplicationId());
    return this.applicationId;
  }

  public synchronized void setApplicationId(ApplicationId appId) {
    maybeInitBuilder();
    if (appId == null) builder.clearApplicationId();
    this.applicationId = appId;
  }

  private ApplicationIdPBImpl convertFromProtoFormat(ApplicationIdProto p) {
    return new ApplicationIdPBImpl(p);
  }

  private ApplicationIdProto convertToProtoFormat(ApplicationId t) {
    return ((ApplicationIdPBImpl) t).getProto();
  }

  @Override
  public int compareTo(QueryId queryId) {
    int compVal = getApplicationId().compareTo(queryId.getApplicationId());
    if (compVal != 0) {
      return compVal;
    } else {
      return getAttemptId() - queryId.getAttemptId();
    }
  }

  static final ThreadLocal<NumberFormat> appIdFormat =
      new ThreadLocal<NumberFormat>() {
        @Override
        public NumberFormat initialValue() {
          NumberFormat fmt = NumberFormat.getInstance();
          fmt.setGroupingUsed(false);
          fmt.setMinimumIntegerDigits(4);
          return fmt;
        }
      };

  static final ThreadLocal<NumberFormat> attemptIdFormat =
      new ThreadLocal<NumberFormat>() {
        @Override
        public NumberFormat initialValue() {
          NumberFormat fmt = NumberFormat.getInstance();
          fmt.setGroupingUsed(false);
          fmt.setMinimumIntegerDigits(6);
          return fmt;
        }
      };

  @Override
  public int hashCode() {
    // Generated by eclipse.
    final int prime = 31;
    int result = 1;
    ApplicationId appId = getApplicationId();
    result = prime * result + appId.hashCode();
    result = prime * result + getAttemptId();
    return result;
  }

  @Override
  public boolean equals(Object obj) {
    if (this == obj) return true;
    if (obj == null) return false;
    if (getClass() != obj.getClass()) return false;
    QueryId other = (QueryId) obj;
    if (!this.getApplicationId().equals(other.getApplicationId())) return false;
    if (this.getAttemptId() != other.getAttemptId()) return false;
    return true;
  }

  @Override
  public String toString() {
    StringBuilder sb =
        new StringBuilder(PREFIX)
            .append(SEPARATOR)
            .append(this.getApplicationId().getClusterTimestamp())
            .append(SEPARATOR)
            .append(appIdFormat.get().format(this.getApplicationId().getId()))
            .append(SEPARATOR)
            .append(attemptIdFormat.get().format(getAttemptId()));
    return sb.toString();
  }
}