/*
 * Copyright (C) 2014 Brett Wooldridge
 *
 * Licensed 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 com.zaxxer.hikari.pool;

import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.atomic.AtomicInteger;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.zaxxer.hikari.util.ClockSource;
import com.zaxxer.hikari.util.ConcurrentBag.IConcurrentBagEntry;
import com.zaxxer.hikari.util.FastList;

/**
 * Entry used in the ConcurrentBag to track Connection instances.
 *
 * @author Brett Wooldridge
 */
public final class PoolEntry implements IConcurrentBagEntry
{
   private static final Logger LOGGER = LoggerFactory.getLogger(PoolEntry.class);

   Connection connection;
   long lastAccessed;
   long lastBorrowed;
   volatile boolean evict;

   private final FastList<Statement> openStatements;
   private final PoolBase poolBase;
   private final AtomicInteger state;

   private volatile ScheduledFuture<?> endOfLife;

   PoolEntry(final Connection connection, final PoolBase pool)
   {
      this.connection = connection;
      this.poolBase = pool;
      this.state = new AtomicInteger(STATE_NOT_IN_USE);
      this.lastAccessed = ClockSource.INSTANCE.currentTime();
      this.openStatements = new FastList<>(Statement.class, 16);
   }

   /**
    * Release this entry back to the pool.
    *
    * @param lastAccessed last access time-stamp
    */
   void recycle(final long lastAccessed)
   {
      this.lastAccessed = lastAccessed;
      poolBase.releaseConnection(this);
   }

   /**
    * @param endOfLife
    */
   void setFutureEol(final ScheduledFuture<?> endOfLife)
   {
      this.endOfLife = endOfLife;
   }

   Connection createProxyConnection(final ProxyLeakTask leakTask, final long now)
   {
      return ProxyFactory.getProxyConnection(this, connection, openStatements, leakTask, now);
   }

   public void resetConnectionState(final ProxyConnection proxyConnection, final int dirtyBits) throws SQLException
   {
      poolBase.resetConnectionState(connection, proxyConnection, dirtyBits);
   }

   public String getPoolName()
   {
      return poolBase.getPoolName();
   }

   Connection getConnection()
   {
      return connection;
   }

   public boolean isEvicted()
   {
      return evict;
   }

   void evict()
   {
      this.evict = true;
   }

   FastList<Statement> getStatementsList()
   {
      return openStatements;
   }

   /** Returns millis since lastBorrowed */
   public long getElapsedLastBorrowed()
   {
      return ClockSource.INSTANCE.elapsedMillis(lastBorrowed);
   }

   // ***********************************************************************
   //                      IConcurrentBagEntry methods
   // ***********************************************************************

   /** {@inheritDoc} */
   @Override
   public int getState()
   {
      return state.get();
   }

   /** {@inheritDoc} */
   @Override
   public boolean compareAndSet(int expect, int update)
   {
      return state.compareAndSet(expect, update);
   }

   /** {@inheritDoc} */
   @Override
   public void lazySet(int update)
   {
      state.lazySet(update);
   }

   /** {@inheritDoc} */
   @Override
   public String toString()
   {
      return connection
         + ", borrowed " + ClockSource.INSTANCE.elapsedMillis(lastBorrowed) + "ms ago, "
         + ", accessed " + ClockSource.INSTANCE.elapsedMillis(lastAccessed) + "ms ago, "
         + stateToString();
   }

   void close()
   {
      if (endOfLife != null && !endOfLife.isDone() && !endOfLife.cancel(false)) {
         LOGGER.warn("{} - maxLifeTime expiration task cancellation unexpectedly returned false for connection {}", getPoolName(), connection);
      }

      endOfLife = null;
      connection = null;
   }

   private String stateToString()
   {
      switch (state.get()) {
      case STATE_IN_USE:
         return "IN_USE";
      case STATE_NOT_IN_USE:
         return "NOT_IN_USE";
      case STATE_REMOVED:
         return "REMOVED";
      case STATE_RESERVED:
         return "RESERVED";
      default:
         return "Invalid";
      }
   }
}

@@ -3775,32 +3775,91 @@
 toString()%0A   %7B%0A
+      final long now = ClockSource.INSTANCE.currentTime();%0A
       return con
@@ -3940,16 +3940,21 @@
 Borrowed
+, now
 ) + %22ms 
@@ -4034,16 +4034,21 @@
 Accessed
+, now
 ) + %22ms 

correct time