/*
    * ====================================================================
    * 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.
   * ====================================================================
   *
   * This software consists of voluntary contributions made by many
   * individuals on behalf of the Apache Software Foundation.  For more
   * information on the Apache Software Foundation, please see
   * <http://www.apache.org/>.
   *
   */
  package org.apache.http.nio.pool;
  
  import java.util.concurrent.atomic.AtomicBoolean;
  import java.util.concurrent.atomic.AtomicReference;
  
  import org.apache.http.concurrent.BasicFuture;
  import org.apache.http.concurrent.Cancellable;
  import org.apache.http.nio.reactor.SessionRequest;
  import org.apache.http.pool.PoolEntry;
  import org.apache.http.util.Asserts;
  
  class LeaseRequest<T, C, E extends PoolEntry<T, C>> implements Cancellable {
  
      private final T route;
      private final Object state;
      private final long connectTimeout;
      private final long deadline;
      private final BasicFuture<E> future;
      private final AtomicReference<SessionRequest> sessionRequestRef;
      private final AtomicBoolean completed;
      private volatile E result;
      private volatile Exception ex;
  
      /**
       * Contructor
       * @param route route
       * @param state state
       * @param connectTimeout http connection timeout
       * @param leaseTimeout timeout to wait in a request queue until kicked off
       * @param future future callback
       */
      public LeaseRequest(
              final T route,
              final Object state,
              final long connectTimeout,
              final long leaseTimeout,
              final BasicFuture<E> future) {
          super();
          this.route = route;
          this.state = state;
          this.connectTimeout = connectTimeout;
          this.deadline = leaseTimeout > 0 ? System.currentTimeMillis() + leaseTimeout : Long.MAX_VALUE;
          this.future = future;
          this.sessionRequestRef = new AtomicReference<SessionRequest>(null);
          this.completed = new AtomicBoolean(false);
      }
  
      public T getRoute() {
          return this.route;
      }
  
      public Object getState() {
          return this.state;
      }
  
      public long getConnectTimeout() {
          return this.connectTimeout;
      }
  
      public long getDeadline() {
          return this.deadline;
      }
  
      public boolean isDone() {
          return this.completed.get();
      }
  
      public void attachSessionRequest(final SessionRequest sessionRequest) {
          Asserts.check(this.sessionRequestRef.compareAndSet(null, sessionRequest), "Session request has already been set");
      }
  
      @Override
      public boolean cancel() {
         final boolean cancelled = this.completed.compareAndSet(false, true);
         final SessionRequest sessionRequest = this.sessionRequestRef.getAndSet(null);
         if (sessionRequest != null) {
             sessionRequest.cancel();
         }
         return cancelled;
     }
 
     public void failed(final Exception ex) {
         if (this.completed.compareAndSet(false, true)) {
             this.ex = ex;
         }
     }
 
     public void completed(final E result) {
         if (this.completed.compareAndSet(false, true)) {
             this.result = result;
         }
     }
 
     public BasicFuture<E> getFuture() {
         return this.future;
     }
 
     public E getResult() {
         return this.result;
     }
 
     public Exception getException() {
         return this.ex;
     }
 
     @Override
     public String toString() {
         final StringBuilder buffer = new StringBuilder();
         buffer.append("[");
         buffer.append(this.route);
         buffer.append("][");
         buffer.append(this.state);
         buffer.append("]");
         return buffer.toString();
     }
 
 }