/**
    * 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.hadoop.hbase.replication.regionserver;
  
  import java.util.HashMap;
  import java.util.Map;
  
  import org.apache.hadoop.hbase.classification.InterfaceAudience;
  import org.apache.hadoop.hbase.CompatibilitySingletonFactory;
  import org.apache.hadoop.hbase.HBaseInterfaceAudience;
  import org.apache.hadoop.hbase.util.EnvironmentEdgeManager;
  
  /**
   * This class is for maintaining the various replication statistics for a source and publishing them
   * through the metrics interfaces.
   */
  @InterfaceAudience.LimitedPrivate(HBaseInterfaceAudience.REPLICATION)
  public class MetricsSource {
  
    // tracks last shipped timestamp for each wal group
    private Map<String, Long> lastTimeStamps = new HashMap<String, Long>();
    private long lastHFileRefsQueueSize = 0;
    private String id;
  
    private final MetricsReplicationSourceSource singleSourceSource;
    private final MetricsReplicationSourceSource globalSourceSource;
    private Map<String, MetricsReplicationSourceSource> singleSourceSourceByTable;
  
    /**
     * Constructor used to register the metrics
     *
     * @param id Name of the source this class is monitoring
     */
    public MetricsSource(String id) {
      this.id = id;
      singleSourceSource =
          CompatibilitySingletonFactory.getInstance(MetricsReplicationSourceFactory.class)
              .getSource(id);
      globalSourceSource = CompatibilitySingletonFactory
          .getInstance(MetricsReplicationSourceFactory.class).getGlobalSource();
      singleSourceSourceByTable = new HashMap<>();
    }
  
    /**
     * Constructor for injecting custom (or test) MetricsReplicationSourceSources
     * @param id Name of the source this class is monitoring
     * @param singleSourceSource Class to monitor id-scoped metrics
     * @param globalSourceSource Class to monitor global-scoped metrics
     */
    public MetricsSource(String id, MetricsReplicationSourceSource singleSourceSource,
                         MetricsReplicationSourceSource globalSourceSource,
                         Map<String, MetricsReplicationSourceSource> singleSourceSourceByTable) {
      this.id = id;
      this.singleSourceSource = singleSourceSource;
      this.globalSourceSource = globalSourceSource;
      this.singleSourceSourceByTable = singleSourceSourceByTable;
    }
  
    /**
     * Set the age of the last edit that was shipped
     * @param timestamp write time of the edit
     * @param walGroup which group we are setting
     */
    public void setAgeOfLastShippedOp(long timestamp, String walGroup) {
      long age = EnvironmentEdgeManager.currentTime() - timestamp;
      singleSourceSource.setLastShippedAge(age);
      globalSourceSource.setLastShippedAge(Math.max(age, globalSourceSource.getLastShippedAge()));
      this.lastTimeStamps.put(walGroup, timestamp);
    }
  
    /**
     * Set the age of the last edit that was shipped group by table
     * @param timestamp write time of the edit
     * @param tableName String as group and tableName
     */
    public void setAgeOfLastShippedOpByTable(long timestamp, String tableName) {
      long age = EnvironmentEdgeManager.currentTime() - timestamp;
      if (!this.getSingleSourceSourceByTable().containsKey(tableName)) {
        this.getSingleSourceSourceByTable().put(tableName,
            CompatibilitySingletonFactory.getInstance(MetricsReplicationSourceFactory.class)
            .getSource(tableName));
      }
      this.singleSourceSourceByTable.get(tableName).setLastShippedAge(age);
   }
   /**
    * Convenience method to use the last given timestamp to refresh the age of the last edit. Used
    * when replication fails and need to keep that metric accurate.
    * @param walGroupId id of the group to update
    */
   public void refreshAgeOfLastShippedOp(String walGroupId) {
     Long lastTimestamp = this.lastTimeStamps.get(walGroupId);
     if (lastTimestamp == null) {
       this.lastTimeStamps.put(walGroupId, 0L);
       lastTimestamp = 0L;
     }
     if (lastTimestamp > 0) {
       setAgeOfLastShippedOp(lastTimestamp, walGroupId);
     }
   }
 
   /**
    * Increment size of the log queue.
    */
   public void incrSizeOfLogQueue() {
     singleSourceSource.incrSizeOfLogQueue(1);
     globalSourceSource.incrSizeOfLogQueue(1);
   }
 
   public void decrSizeOfLogQueue() {
     singleSourceSource.decrSizeOfLogQueue(1);
     globalSourceSource.decrSizeOfLogQueue(1);
   }
 
   /**
    * Add on the the number of log edits read
    *
    * @param delta the number of log edits read.
    */
   private void incrLogEditsRead(long delta) {
     singleSourceSource.incrLogReadInEdits(delta);
     globalSourceSource.incrLogReadInEdits(delta);
   }
 
   /** Increment the number of log edits read by one. */
   public void incrLogEditsRead() {
     incrLogEditsRead(1);
   }
 
   /**
    * Add on the number of log edits filtered
    *
    * @param delta the number filtered.
    */
   public void incrLogEditsFiltered(long delta) {
     singleSourceSource.incrLogEditsFiltered(delta);
     globalSourceSource.incrLogEditsFiltered(delta);
   }
 
   /** The number of log edits filtered out. */
   public void incrLogEditsFiltered() {
     incrLogEditsFiltered(1);
   }
 
   /**
    * Convience method to apply changes to metrics do to shipping a batch of logs.
    *
    * @param batchSize the size of the batch that was shipped to sinks.
    */
   public void shipBatch(long batchSize, int sizeInBytes) {
     singleSourceSource.incrBatchesShipped(1);
     globalSourceSource.incrBatchesShipped(1);
 
     singleSourceSource.incrOpsShipped(batchSize);
     globalSourceSource.incrOpsShipped(batchSize);
 
     singleSourceSource.incrShippedBytes(sizeInBytes);
     globalSourceSource.incrShippedBytes(sizeInBytes);
   }
 
   /**
    * Convience method to apply changes to metrics do to shipping a batch of logs.
    *
    * @param batchSize the size of the batch that was shipped to sinks.
    * @param hfiles total number of hfiles shipped to sinks.
    */
   public void shipBatch(long batchSize, int sizeInBytes, long hfiles) {
     shipBatch(batchSize, sizeInBytes);
     singleSourceSource.incrHFilesShipped(hfiles);
     globalSourceSource.incrHFilesShipped(hfiles);
   }
 
   /** increase the byte number read by source from log file */
   public void incrLogReadInBytes(long readInBytes) {
     singleSourceSource.incrLogReadInBytes(readInBytes);
     globalSourceSource.incrLogReadInBytes(readInBytes);
   }
 
   /** Removes all metrics about this Source. */
   public void clear() {
     int lastQueueSize = singleSourceSource.getSizeOfLogQueue();
     globalSourceSource.decrSizeOfLogQueue(lastQueueSize);
     singleSourceSource.decrSizeOfLogQueue(lastQueueSize);
     singleSourceSource.clear();
     globalSourceSource.decrSizeOfHFileRefsQueue(lastHFileRefsQueueSize);
     lastTimeStamps.clear();
     lastHFileRefsQueueSize = 0;
   }
 
   /**
    * Get AgeOfLastShippedOp
    * @return AgeOfLastShippedOp
    */
   public Long getAgeOfLastShippedOp() {
     return singleSourceSource.getLastShippedAge();
   }
 
   /**
    * Get the sizeOfLogQueue
    * @return sizeOfLogQueue
    */
   public int getSizeOfLogQueue() {
     return singleSourceSource.getSizeOfLogQueue();
   }
 
   /**
    * Get the timeStampsOfLastShippedOp, if there are multiple groups, return the latest one
    * @return lastTimestampForAge
    */
   public long getTimeStampOfLastShippedOp() {
     long lastTimestamp = 0L;
     for (long ts : lastTimeStamps.values()) {
       if (ts > lastTimestamp) {
         lastTimestamp = ts;
       }
     }
     return lastTimestamp;
   }
 
   /**
    * Get the slave peer ID
    * @return peerID
    */
   public String getPeerID() {
     return id;
   }
 
   public void incrSizeOfHFileRefsQueue(long size) {
     singleSourceSource.incrSizeOfHFileRefsQueue(size);
     globalSourceSource.incrSizeOfHFileRefsQueue(size);
     lastHFileRefsQueueSize = size;
   }
 
   public void decrSizeOfHFileRefsQueue(int size) {
     singleSourceSource.decrSizeOfHFileRefsQueue(size);
     globalSourceSource.decrSizeOfHFileRefsQueue(size);
     lastHFileRefsQueueSize -= size;
     if (lastHFileRefsQueueSize < 0) {
       lastHFileRefsQueueSize = 0;
     }
   }
 
   public void incrUnknownFileLengthForClosedWAL() {
     singleSourceSource.incrUnknownFileLengthForClosedWAL();
     globalSourceSource.incrUnknownFileLengthForClosedWAL();
   }
 
   public void incrUncleanlyClosedWALs() {
     singleSourceSource.incrUncleanlyClosedWALs();
     globalSourceSource.incrUncleanlyClosedWALs();
   }
 
   public void incrBytesSkippedInUncleanlyClosedWALs(final long bytes) {
     singleSourceSource.incrBytesSkippedInUncleanlyClosedWALs(bytes);
     globalSourceSource.incrBytesSkippedInUncleanlyClosedWALs(bytes);
   }
 
   public void incrRestartedWALReading() {
     singleSourceSource.incrRestartedWALReading();
     globalSourceSource.incrRestartedWALReading();
   }
 
   public void incrRepeatedFileBytes(final long bytes) {
     singleSourceSource.incrRepeatedFileBytes(bytes);
     globalSourceSource.incrRepeatedFileBytes(bytes);
   }
 
   public void incrCompletedWAL() {
     singleSourceSource.incrCompletedWAL();
     globalSourceSource.incrCompletedWAL();
   }
 
   public void incrCompletedRecoveryQueue() {
     singleSourceSource.incrCompletedRecoveryQueue();
     globalSourceSource.incrCompletedRecoveryQueue();
   }
 
   public void incrFailedRecoveryQueue() {
     globalSourceSource.incrFailedRecoveryQueue();
   }
 
   public Map<String, MetricsReplicationSourceSource> getSingleSourceSourceByTable() {
     return singleSourceSourceByTable;
   }
 }