hbase-client/src/main/java/org/apache/hadoop/hbase/client/RetryingTimeTracker.java

   1 /*
   2  * Licensed to the Apache Software Foundation (ASF) under one or more
   3  * contributor license agreements.  See the NOTICE file distributed with
   4  * this work for additional information regarding copyright ownership.
   5  * The ASF licenses this file to you under the Apache License, Version 2.0
   6  * (the "License"); you may not use this file except in compliance with
   7  * the License.  You may obtain a copy of the License at
   8  *
   9  * http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17 package org.apache.hadoop.hbase.client;
  18
  19 import org.apache.hadoop.hbase.util.EnvironmentEdgeManager;
  20
  21 /**
  22  * Tracks the amount of time remaining for an operation.
  23  */
  24 class RetryingTimeTracker {
  25   private long globalStartTime = -1;
  26
  27   public RetryingTimeTracker start() {
  28     if (this.globalStartTime < 0) {
  29       this.globalStartTime = EnvironmentEdgeManager.currentTime();
  30     }
  31     return this;
  32   }
  33
  34   public int getRemainingTime(int callTimeout) {
  35     if (callTimeout <= 0) {
  36       return 0;
  37     } else {
  38       if (callTimeout == Integer.MAX_VALUE) {
  39         return Integer.MAX_VALUE;
  40       }
  41       long remaining = EnvironmentEdgeManager.currentTime() - this.globalStartTime;
  42       long remainingTime = callTimeout - remaining;
  43       if (remainingTime < 1) {
  44         // If there is no time left, we're trying anyway. It's too late.
  45         // 0 means no timeout, and it's not the intent here. So we secure both cases by
  46         // resetting to the minimum.
  47         remainingTime = 1;
  48       }
  49       if (remainingTime > Integer.MAX_VALUE) {
  50         throw new RuntimeException("remainingTime=" + remainingTime +
  51             " which is > Integer.MAX_VALUE");
  52       }
  53       return (int)remainingTime;
  54     }
  55   }
  56
  57   public long getStartTime() {
  58     return this.globalStartTime;
  59   }
  60 }