/*
    * 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.giraph.comm.netty.handler;
  
  import org.apache.giraph.comm.requests.WritableRequest;
  import org.apache.giraph.time.SystemTime;
  import org.apache.giraph.time.Time;
  import io.netty.channel.ChannelFuture;
  
  import java.net.InetSocketAddress;
  import java.util.Date;
  
  /**
   * Help track requests throughout the system
   */
  public class RequestInfo {
    /** Time class to use */
    private static final Time TIME = SystemTime.get();
    /** Destination of the request */
    private final InetSocketAddress destinationAddress;
    /** When the request was started */
    private final long startedNanos;
    /** Request */
    private final WritableRequest request;
    /** Future of the write of this request*/
    private volatile ChannelFuture writeFuture;
  
    /**
     * Constructor.
     *
     * @param destinationAddress Destination of the request
     * @param request Request that is sent
     */
    public RequestInfo(InetSocketAddress destinationAddress,
                       WritableRequest request) {
      this.destinationAddress = destinationAddress;
      this.request = request;
      this.startedNanos = TIME.getNanoseconds();
    }
  
    public InetSocketAddress getDestinationAddress() {
      return destinationAddress;
    }
  
    /**
     * Get the started msecs.
     *
     * @return Started msecs
     */
    public long getStartedMsecs() {
      return startedNanos / Time.NS_PER_MS;
    }
  
    /**
     * Get the elapsed nanoseconds since the request started.
     *
     * @return Nanoseconds since the request was started
     */
    public long getElapsedNanos() {
      return TIME.getNanoseconds() - startedNanos;
    }
  
    /**
     * Get the elapsed millseconds since the request started.
     *
     * @return Milliseconds since the request was started
     */
    public long getElapsedMsecs() {
      return getElapsedNanos() / Time.NS_PER_MS;
    }
  
  
    public WritableRequest getRequest() {
      return request;
    }
  
    public void setWriteFuture(ChannelFuture writeFuture) {
      this.writeFuture = writeFuture;
    }
  
    public ChannelFuture getWriteFuture() {
      return writeFuture;
    }
 
   @Override
   public String toString() {
     return "(reqId=" + request.getRequestId() +
         ",destAddr=" + destinationAddress.getHostName() + ":" +
         destinationAddress.getPort() +
         ",elapsedNanos=" +
         getElapsedNanos() +
         ",started=" + new Date(getStartedMsecs()) +
         ((writeFuture == null) ? ")" :
             ",writeDone=" + writeFuture.isDone() +
                 ",writeSuccess=" + writeFuture.isSuccess() + ")");
   }
 }