/*
    * Copyright 2009 Red Hat, Inc.
    *
    * Red Hat 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.jboss.netty.handler.traffic;
  
  import java.util.concurrent.TimeUnit;
  import java.util.concurrent.atomic.AtomicBoolean;
  import java.util.concurrent.atomic.AtomicLong;
  
  import org.jboss.netty.channel.ChannelHandlerContext;
  import org.jboss.netty.util.Timeout;
  import org.jboss.netty.util.Timer;
  import org.jboss.netty.util.TimerTask;
  
  /**
   * TrafficCounter is associated with {@link AbstractTrafficShapingHandler}.<br>
   * <br>
   * A TrafficCounter has for goal to count the traffic in order to enable to limit the traffic or not,
   * globally or per channel. It compute statistics on read and written bytes at the specified
   * interval and call back the {@link AbstractTrafficShapingHandler} doAccounting method at every
   * specified interval. If this interval is set to 0, therefore no accounting will be done and only
   * statistics will be computed at each receive or write operations.
   *
   * @author The Netty Project (netty-dev@lists.jboss.org)
   * @author Frederic Bregier
   * @version $Rev: 1225 $, $Date: 2012-05-20 10:48:53 +0200 (dim., 20 mai 2012) $
   */
  public class TrafficCounter {
      /**
       * Current written bytes
       */
      private final AtomicLong currentWrittenBytes = new AtomicLong(0);
  
      /**
       * Current read bytes
       */
      private final AtomicLong currentReadBytes = new AtomicLong(0);
  
      /**
       * Long life written bytes
       */
      private final AtomicLong cumulativeWrittenBytes = new AtomicLong(0);
  
      /**
       * Long life read bytes
       */
      private final AtomicLong cumulativeReadBytes = new AtomicLong(0);
  
      /**
       * Last Time where cumulative bytes where reset to zero
       */
      private long lastCumulativeTime;
  
      /**
       * Last writing bandwidth
       */
      private long lastWriteThroughput = 0;
  
      /**
       * Last reading bandwidth
       */
      private long lastReadThroughput = 0;
  
      /**
       * Last Time Check taken
       */
      private final AtomicLong lastTime = new AtomicLong(0);
  
      /**
       * Last written bytes number during last check interval
       */
      private long lastWrittenBytes = 0;
  
      /**
       * Last read bytes number during last check interval
       */
      private long lastReadBytes = 0;
  
      /**
       * Delay between two captures
       */
      AtomicLong checkInterval = new AtomicLong(
              AbstractTrafficShapingHandler.DEFAULT_CHECK_INTERVAL);
  
      // default 1 s
  
      /**
      * Name of this Monitor
      */
     private final String name;
 
     /**
      * The associated TrafficShapingHandler
      */
     private final AbstractTrafficShapingHandler trafficShapingHandler;
 
     /**
      * One Timer for all Counter
      */
     private final Timer timer;  // replace executor
     /**
      * Monitor created once in start()
      */
     private TimerTask timerTask;
     /**
      * used in stop() to cancel the timer
      */
     volatile private Timeout timeout = null;
 
     /**
      * Is Monitor active
      */
     AtomicBoolean monitorActive = new AtomicBoolean(false);
 
     /**
      * Class to implement monitoring at fix delay
      *
      */
     private static class TrafficMonitoringTask implements TimerTask {
         /**
          * The associated TrafficShapingHandler
          */
         private final AbstractTrafficShapingHandler trafficShapingHandler1;
 
         /**
          * The associated TrafficCounter
          */
         private final TrafficCounter counter;
 
         /**
          * @param trafficShapingHandler
          * @param counter
          */
         protected TrafficMonitoringTask(
                 AbstractTrafficShapingHandler trafficShapingHandler,
                 TrafficCounter counter) {
             trafficShapingHandler1 = trafficShapingHandler;
             this.counter = counter;
         }
 
         public void run(Timeout timeout) throws Exception {
             if (!counter.monitorActive.get()) {
                 return;
             }
             long endTime = System.currentTimeMillis();
             counter.resetAccounting(endTime);
             if (trafficShapingHandler1 != null) {
                 trafficShapingHandler1.doAccounting(counter);
             }
             timeout = 
                 counter.timer.newTimeout(this, counter.checkInterval.get(), TimeUnit.MILLISECONDS);                        
         }
     }
 
     /**
      * Start the monitoring process
      *
      */
     public void start() {
         synchronized (lastTime) {
             if (monitorActive.get()) {
                 return;
             }
             lastTime.set(System.currentTimeMillis());
             if (checkInterval.get() > 0) {
                 monitorActive.set(true);
                 timerTask = new TrafficMonitoringTask(trafficShapingHandler, this);
                 timeout = 
                     timer.newTimeout(timerTask, checkInterval.get(), TimeUnit.MILLISECONDS);
             }
         }
     }
 
     /**
      * Stop the monitoring process
      *
      */
     public void stop() {
         synchronized (lastTime) {
             if (!monitorActive.get()) {
                 return;
             }
             monitorActive.set(false);
             resetAccounting(System.currentTimeMillis());
             if (trafficShapingHandler != null) {
                 trafficShapingHandler.doAccounting(this);
             }
             if (timeout != null) {
                 timeout.cancel();
             }
         }
     }
 
     /**
      * Reset the accounting on Read and Write
      *
      * @param newLastTime
      */
     void resetAccounting(long newLastTime) {
         synchronized (lastTime) {
             long interval = newLastTime - lastTime.getAndSet(newLastTime);
             if (interval == 0) {
                 // nothing to do
                 return;
             }
             lastReadBytes = currentReadBytes.getAndSet(0);
             lastWrittenBytes = currentWrittenBytes.getAndSet(0);
             lastReadThroughput = lastReadBytes / interval * 1000;
             // nb byte / checkInterval in ms * 1000 (1s)
             lastWriteThroughput = lastWrittenBytes / interval * 1000;
             // nb byte / checkInterval in ms * 1000 (1s)
         }
     }
 
     /**
      * Constructor with the {@link AbstractTrafficShapingHandler} that hosts it, the Timer to use, its
      * name, the checkInterval between two computations in millisecond
      * @param trafficShapingHandler the associated AbstractTrafficShapingHandler
      * @param timer
      *            Could be a HashedWheelTimer
      * @param name
      *            the name given to this monitor
      * @param checkInterval
      *            the checkInterval in millisecond between two computations
      */
     public TrafficCounter(AbstractTrafficShapingHandler trafficShapingHandler,
             Timer timer, String name, long checkInterval) {
         this.trafficShapingHandler = trafficShapingHandler;
         this.timer = timer;
         this.name = name;
         lastCumulativeTime = System.currentTimeMillis();
         configure(checkInterval);
     }
 
     /**
      * Change checkInterval between
      * two computations in millisecond
      *
      * @param newcheckInterval
      */
     public void configure(long newcheckInterval) {
         long newInterval = (newcheckInterval/10)*10;
         if (checkInterval.get() != newInterval) {
             checkInterval.set(newInterval);
             if (newInterval <= 0) {
                 stop();
                 // No more active monitoring
                 lastTime.set(System.currentTimeMillis());
             } else {
                 // Start if necessary
                 start();
             }
         }
     }
 
     /**
      * Computes counters for Read.
      *
      * @param ctx
      *            the associated channelHandlerContext
      * @param recv
      *            the size in bytes to read
      * @throws InterruptedException
      */
     void bytesRecvFlowControl(ChannelHandlerContext ctx, long recv)
             throws InterruptedException {
         currentReadBytes.addAndGet(recv);
         cumulativeReadBytes.addAndGet(recv);
     }
 
     /**
      * Computes counters for Write.
      *
      * @param write
      *            the size in bytes to write
      * @throws InterruptedException
      */
     void bytesWriteFlowControl(long write) throws InterruptedException {
         currentWrittenBytes.addAndGet(write);
         cumulativeWrittenBytes.addAndGet(write);
     }
 
     /**
      *
      * @return the current checkInterval between two computations of traffic counter
      *         in millisecond
      */
     public long getCheckInterval() {
         return checkInterval.get();
     }
 
     /**
      *
      * @return the Read Throughput in bytes/s computes in the last check interval
      */
     public long getLastReadThroughput() {
         return lastReadThroughput;
     }
 
     /**
      *
      * @return the Write Throughput in bytes/s computes in the last check interval
      */
     public long getLastWriteThroughput() {
         return lastWriteThroughput;
     }
 
     /**
      *
      * @return the number of bytes read during the last check Interval
      */
     public long getLastReadBytes() {
         return lastReadBytes;
     }
 
     /**
      *
      * @return the number of bytes written during the last check Interval
      */
     public long getLastWrittenBytes() {
         return lastWrittenBytes;
     }
 
     /**
     *
     * @return the current number of bytes read since the last checkInterval
     */
     public long getCurrentReadBytes() {
         return currentReadBytes.get();
     }
 
     /**
      *
      * @return the current number of bytes written since the last check Interval
      */
     public long getCurrentWrittenBytes() {
         return currentWrittenBytes.get();
     }
 
     /**
      * @return the Time in millisecond of the last check as of System.currentTimeMillis()
      */
     public long getLastTime() {
         return lastTime.get();
     }
 
     /**
      * @return the cumulativeWrittenBytes
      */
     public long getCumulativeWrittenBytes() {
         return cumulativeWrittenBytes.get();
     }
 
     /**
      * @return the cumulativeReadBytes
      */
     public long getCumulativeReadBytes() {
         return cumulativeReadBytes.get();
     }
 
     /**
      * @return the lastCumulativeTime in millisecond as of System.currentTimeMillis()
      * when the cumulative counters were reset to 0.
      */
     public long getLastCumulativeTime() {
         return lastCumulativeTime;
     }
 
     /**
      * Reset both read and written cumulative bytes counters and the associated time.
      */
     public void resetCumulativeTime() {
         lastCumulativeTime = System.currentTimeMillis();
         cumulativeReadBytes.set(0);
         cumulativeWrittenBytes.set(0);
     }
 
     /**
      * @return the name
      */
     public String getName() {
         return name;
     }
 
     /**
      * String information
      */
     @Override
     public String toString() {
         return "Monitor " + name + " Current Speed Read: " +
                 (lastReadThroughput >> 10) + " KB/s, Write: " +
                 (lastWriteThroughput >> 10) + " KB/s Current Read: " +
                 (currentReadBytes.get() >> 10) + " KB Current Write: " +
                 (currentWrittenBytes.get() >> 10) + " KB";
     }
 }