/*
    * 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.io.IOException;
  import java.util.Collections;
  import java.util.List;
  import java.util.Map;
  import java.util.Random;
  
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.apache.hadoop.conf.Configuration;
  import org.apache.hadoop.hbase.ServerName;
  import org.apache.hadoop.hbase.client.HConnection;
  import org.apache.hadoop.hbase.protobuf.generated.AdminProtos.AdminService;
  import org.apache.hadoop.hbase.replication.HBaseReplicationEndpoint;
  import com.google.common.collect.Lists;
  import com.google.common.collect.Maps;
  
  /**
   * Maintains a collection of peers to replicate to, and randomly selects a
   * single peer to replicate to per set of data to replicate. Also handles
   * keeping track of peer availability.
   */
  public class ReplicationSinkManager {
  
    private static final Log LOG = LogFactory.getLog(ReplicationSinkManager.class);
  
    /**
     * Default maximum number of times a replication sink can be reported as bad before
     * it will no longer be provided as a sink for replication without the pool of
     * replication sinks being refreshed.
     */
    static final int DEFAULT_BAD_SINK_THRESHOLD = 3;
  
    /**
     * Default ratio of the total number of peer cluster region servers to consider
     * replicating to.
     */
    static final float DEFAULT_REPLICATION_SOURCE_RATIO = 0.1f;
  
  
    private final HConnection conn;
  
    private final String peerClusterId;
  
    private final HBaseReplicationEndpoint endpoint;
  
    // Count of "bad replication sink" reports per peer sink
    private final Map<ServerName, Integer> badReportCounts;
  
    // Ratio of total number of potential peer region servers to be used
    private final float ratio;
  
    // Maximum number of times a sink can be reported as bad before the pool of
    // replication sinks is refreshed
    private final int badSinkThreshold;
  
    private final Random random;
  
    // A timestamp of the last time the list of replication peers changed
    private long lastUpdateToPeers;
  
    // The current pool of sinks to which replication can be performed
    private List<ServerName> sinks = Lists.newArrayList();
  
    /**
     * Instantiate for a single replication peer cluster.
     * @param conn connection to the peer cluster
     * @param peerClusterId identifier of the peer cluster
     * @param endpoint replication endpoint for inter cluster replication
     * @param conf HBase configuration, used for determining replication source ratio and bad peer
     *          threshold
     */
    public ReplicationSinkManager(HConnection conn, String peerClusterId,
        HBaseReplicationEndpoint endpoint, Configuration conf) {
      this.conn = conn;
      this.peerClusterId = peerClusterId;
      this.endpoint = endpoint;
      this.badReportCounts = Maps.newHashMap();
      this.ratio = conf.getFloat("replication.source.ratio", DEFAULT_REPLICATION_SOURCE_RATIO);
      this.badSinkThreshold = conf.getInt("replication.bad.sink.threshold",
                                          DEFAULT_BAD_SINK_THRESHOLD);
     this.random = new Random();
   }
 
   /**
    * Get a randomly-chosen replication sink to replicate to.
    *
    * @return a replication sink to replicate to
    */
   public SinkPeer getReplicationSink() throws IOException {
     if (endpoint.getLastRegionServerUpdate() > this.lastUpdateToPeers || sinks.isEmpty()) {
       LOG.info("Current list of sinks is out of date or empty, updating");
       chooseSinks();
     }
 
     if (sinks.isEmpty()) {
       throw new IOException("No replication sinks are available");
     }
     ServerName serverName = sinks.get(random.nextInt(sinks.size()));
     return new SinkPeer(serverName, conn.getAdmin(serverName));
   }
 
   /**
    * Report a {@code SinkPeer} as being bad (i.e. an attempt to replicate to it
    * failed). If a single SinkPeer is reported as bad more than
    * replication.bad.sink.threshold times, it will be removed
    * from the pool of potential replication targets.
    *
    * @param sinkPeer
    *          The SinkPeer that had a failed replication attempt on it
    */
   public void reportBadSink(SinkPeer sinkPeer) {
     ServerName serverName = sinkPeer.getServerName();
     int badReportCount = (badReportCounts.containsKey(serverName)
                     ? badReportCounts.get(serverName) : 0) + 1;
     badReportCounts.put(serverName, badReportCount);
     if (badReportCount > badSinkThreshold) {
       this.sinks.remove(serverName);
       if (sinks.isEmpty()) {
         chooseSinks();
       }
     }
   }
 
   /**
    * Report that a {@code SinkPeer} successfully replicated a chunk of data.
    *
    * @param sinkPeer
    *          The SinkPeer that had a failed replication attempt on it
    */
   public void reportSinkSuccess(SinkPeer sinkPeer) {
     badReportCounts.remove(sinkPeer.getServerName());
   }
 
   void chooseSinks() {
     List<ServerName> slaveAddresses = endpoint.getRegionServers();
     Collections.shuffle(slaveAddresses, random);
     int numSinks = (int) Math.ceil(slaveAddresses.size() * ratio);
     sinks = slaveAddresses.subList(0, numSinks);
     lastUpdateToPeers = System.currentTimeMillis();
     badReportCounts.clear();
   }
 
   List<ServerName> getSinks() {
     return sinks;
   }
 
   /**
    * Wraps a replication region server sink to provide the ability to identify
    * it.
    */
   public static class SinkPeer {
     private ServerName serverName;
     private AdminService.BlockingInterface regionServer;
 
     public SinkPeer(ServerName serverName, AdminService.BlockingInterface regionServer) {
       this.serverName = serverName;
       this.regionServer = regionServer;
     }
 
     ServerName getServerName() {
       return serverName;
     }
 
     public AdminService.BlockingInterface getRegionServer() {
       return regionServer;
     }
 
   }
 
 }