package org.eclipse.aether.named.support;
   
   /*
    * 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.
   */
  
  import org.eclipse.aether.named.NamedLockFactory;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  import java.util.concurrent.ConcurrentHashMap;
  import java.util.concurrent.ConcurrentMap;
  import java.util.concurrent.atomic.AtomicBoolean;
  import java.util.concurrent.atomic.AtomicInteger;
  
  /**
   * Support class for {@link NamedLockFactory} implementations providing reference counting.
   */
  public abstract class NamedLockFactorySupport implements NamedLockFactory
  {
      protected final Logger logger = LoggerFactory.getLogger( getClass() );
  
      private final ConcurrentMap<String, NamedLockHolder> locks;
  
      public NamedLockFactorySupport()
      {
          this.locks = new ConcurrentHashMap<>();
      }
  
      @Override
      public NamedLockSupport getLock( final String name )
      {
          return locks.compute( name, ( k, v ) ->
          {
              if ( v == null )
              {
                  v = new NamedLockHolder( createLock( k ) );
              }
              v.incRef();
              return v;
          } ).namedLock;
      }
  
      @Override
      public void shutdown()
      {
          // override if needed
      }
  
      public boolean closeLock( final NamedLockSupport lock )
      {
          AtomicBoolean destroyed = new AtomicBoolean( false );
          locks.compute( lock.name(), ( k, v ) ->
          {
              if ( v != null && v.decRef() == 0 )
              {
                  destroyLock( v.namedLock );
                  destroyed.set( true );
                  return null;
              }
              return v;
          } );
          return destroyed.get();
      }
  
  
      @Override
      protected void finalize() throws Throwable
      {
          try
          {
              if ( !locks.isEmpty() )
              {
                  // report leak
                  logger.warn( "Lock leak, referenced locks still exist {}", locks );
              }
          }
          finally
          {
              super.finalize();
          }
      }
  
      protected abstract NamedLockSupport createLock( final String name );
 
     protected void destroyLock( final NamedLockSupport lock )
     {
         // override if needed
     }
 
     private static final class NamedLockHolder
     {
         private final NamedLockSupport namedLock;
 
         private final AtomicInteger referenceCount;
 
         private NamedLockHolder( NamedLockSupport namedLock )
         {
             this.namedLock = namedLock;
             this.referenceCount = new AtomicInteger( 0 );
         }
 
         private int incRef()
         {
             return referenceCount.incrementAndGet();
         }
 
         private int decRef()
         {
             return referenceCount.decrementAndGet();
         }
 
         @Override
         public String toString()
         {
             return "[refCount=" + referenceCount.get() + ", lock=" + namedLock + "]";
         }
     }
 }