/*
    * 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.eclipse.aether.util.concurrency;
  
  import java.util.concurrent.Callable;
  import java.util.concurrent.CompletableFuture;
  import java.util.concurrent.ExecutorService;
  import java.util.concurrent.Future;
  import java.util.concurrent.RejectedExecutionException;
  import java.util.concurrent.Semaphore;
  
  /**
   * Utilities for executors and sizing them.
   * <em>Big fat note:</em> Do not use this class outside of resolver. This and related classes are not meant as "drop
   * in replacement" for Jave Executors, is used in very controlled fashion only.
   *
   * @since 2.0.11
   */
  public interface SmartExecutor extends AutoCloseable {
      /**
       * Submits a {@link Runnable} to execution.
       */
      void submit(Runnable runnable);
  
      /**
       * Submits a {@link Callable} to execution, returns a {@link CompletableFuture}.
       */
      <T> Future<T> submit(Callable<T> callable);
  
      /**
       * Shut down this instance (ideally used in try-with-resource construct).
       */
      void close();
  
      /**
       * Direct executor (caller executes).
       */
      class Direct implements SmartExecutor {
          @Override
          public void submit(Runnable runnable) {
              runnable.run();
          }
  
          @Override
          public <T> CompletableFuture<T> submit(Callable<T> callable) {
              CompletableFuture<T> future = new CompletableFuture<>();
              try {
                  future.complete(callable.call());
              } catch (Exception e) {
                  future.completeExceptionally(e);
              }
              return future;
          }
  
          @Override
          public void close() {}
      }
  
      /**
       * Pooled executor backed by {@link ExecutorService}.
       */
      class Pooled implements SmartExecutor {
          private final ExecutorService executor;
  
          Pooled(ExecutorService executor) {
              this.executor = executor;
          }
  
          @Override
          public void submit(Runnable runnable) {
              ClassLoader tccl = Thread.currentThread().getContextClassLoader();
              try {
                  executor.submit(() -> {
                      ClassLoader old = Thread.currentThread().getContextClassLoader();
                      Thread.currentThread().setContextClassLoader(tccl);
                      try {
                          runnable.run();
                      } finally {
                          Thread.currentThread().setContextClassLoader(old);
                      }
                  });
              } catch (RejectedExecutionException e) {
                  try {
                     runnable.run();
                 } catch (RuntimeException | Error t) {
                     // swallow to match async submit() semantics where exceptions
                     // are captured by the Future; callers like RunnableErrorForwarder
                     // already record the error before re-throwing
                 }
             }
         }
 
         @Override
         public <T> Future<T> submit(Callable<T> callable) {
             ClassLoader tccl = Thread.currentThread().getContextClassLoader();
             CompletableFuture<T> future = new CompletableFuture<>();
             try {
                 executor.submit(() -> {
                     ClassLoader old = Thread.currentThread().getContextClassLoader();
                     Thread.currentThread().setContextClassLoader(tccl);
                     try {
                         future.complete(callable.call());
                     } catch (Exception e) {
                         future.completeExceptionally(e);
                     } finally {
                         Thread.currentThread().setContextClassLoader(old);
                     }
                 });
             } catch (RejectedExecutionException e) {
                 try {
                     future.complete(callable.call());
                 } catch (Exception ex) {
                     future.completeExceptionally(ex);
                 }
             }
             return future;
         }
 
         @Override
         public void close() {
             executor.shutdown();
         }
     }
 
     /**
      * Limited executor, where the actual goal is to protect accessed resource, like when virtual threads
      * are being used, so the "pool" itself does not provide any kind of back-pressure.
      */
     class Limited implements SmartExecutor {
         private final SmartExecutor executor;
         private final Semaphore semaphore;
 
         Limited(SmartExecutor executor, int limit) {
             this.executor = executor;
             this.semaphore = new Semaphore(limit);
         }
 
         @Override
         public void submit(Runnable runnable) {
             try {
                 semaphore.acquire();
                 try {
                     executor.submit(() -> {
                         try {
                             runnable.run();
                         } finally {
                             semaphore.release();
                         }
                     });
                 } catch (RejectedExecutionException e) {
                     try {
                         runnable.run();
                     } catch (RuntimeException | Error t) {
                         // swallow to match async submit() semantics where exceptions
                         // are captured by the Future; callers like RunnableErrorForwarder
                         // already record the error before re-throwing
                     } finally {
                         semaphore.release();
                     }
                 }
             } catch (InterruptedException e) {
                 Thread.currentThread().interrupt();
             }
         }
 
         @Override
         public <T> Future<T> submit(Callable<T> callable) {
             try {
                 semaphore.acquire();
                 CompletableFuture<T> future = new CompletableFuture<>();
                 try {
                     executor.submit(() -> {
                         try {
                             future.complete(callable.call());
                         } catch (Exception e) {
                             future.completeExceptionally(e);
                         } finally {
                             semaphore.release();
                         }
                     });
                 } catch (RejectedExecutionException e) {
                     try {
                         future.complete(callable.call());
                     } catch (Exception ex) {
                         future.completeExceptionally(ex);
                     } finally {
                         semaphore.release();
                     }
                 }
                 return future;
             } catch (InterruptedException e) {
                 Thread.currentThread().interrupt();
                 CompletableFuture<T> failed = new CompletableFuture<>();
                 failed.completeExceptionally(e);
                 return failed;
             }
         }
 
         @Override
         public void close() {
             executor.close();
         }
     }
 
     /**
      * Wrapper to prevent closing.
      */
     class NonClosing implements SmartExecutor {
         private final SmartExecutor smartExecutor;
 
         NonClosing(SmartExecutor smartExecutor) {
             this.smartExecutor = smartExecutor;
         }
 
         @Override
         public void submit(Runnable runnable) {
             smartExecutor.submit(runnable);
         }
 
         @Override
         public <T> Future<T> submit(Callable<T> callable) {
             return smartExecutor.submit(callable);
         }
 
         @Override
         public void close() {
             // nope; delegate is managed
         }
     }
 }