Java tutorial
/** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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 org.apache.tajo; import org.apache.hadoop.yarn.api.records.ApplicationId; import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationIdPBImpl; import org.apache.hadoop.yarn.proto.YarnProtos.ApplicationAttemptIdProto; import org.apache.hadoop.yarn.proto.YarnProtos.ApplicationAttemptIdProtoOrBuilder; import org.apache.hadoop.yarn.proto.YarnProtos.ApplicationIdProto; import java.text.NumberFormat; /** * 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(); } }