package org.jacorb.notification.engine;

/*
 *        JacORB - a free Java ORB
 *
 *   Copyright (C) 1999-2014 Gerald Brose / The JacORB Team.
 *
 *   This library is free software; you can redistribute it and/or
 *   modify it under the terms of the GNU Library General Public
 *   License as published by the Free Software Foundation; either
 *   version 2 of the License, or (at your option) any later version.
 *
 *   This library is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *   Library General Public License for more details.
 *
 *   You should have received a copy of the GNU Library General Public
 *   License along with this library; if not, write to the Free
 *   Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */

import org.jacorb.notification.interfaces.Disposable;
import org.jacorb.notification.util.DisposableManager;

import java.util.concurrent.Executor;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author Alphonse Bendt
 */

public class DefaultTaskExecutor implements TaskExecutor
{
    private final class DirectExecutor implements Executor
    {
        public void execute(Runnable command)
        {
            // execute in current thread
            command.run();
        }
    }

    private static final class DefaultThreadFactory implements ThreadFactory
    {
        private final AtomicInteger counter_ = new AtomicInteger(0);

        private final String name_;

        private DefaultThreadFactory(String name)
        {
            super();
            name_ = name;
        }

        public Thread newThread(Runnable task)
        {
            Thread _thread = new Thread(task);

            _thread.setDaemon(true);
            _thread.setName(name_ + "#" + (counter_.getAndIncrement()));

            return _thread;
        }
    }

    private final Executor executor_;

    private final DisposableManager disposeHooks_ = new DisposableManager();
    
    ////////////////////////////////////////

    public DefaultTaskExecutor(final String name, int numberOfThreads, boolean mayDie)
    {
        if (numberOfThreads < 0)
        {
            throw new IllegalArgumentException();
        }
        else if (numberOfThreads == 0)
        {
            executor_ = new DirectExecutor();
        }
        else
        {
            ThreadFactory _threadFactory = new DefaultThreadFactory(name);

            final ExecutorService _executor;
            
            if (mayDie)
            {
                _executor = Executors.newCachedThreadPool(_threadFactory);
            }
            else
            {
                _executor = Executors.newFixedThreadPool(numberOfThreads, _threadFactory);
            }
            
            executor_ = _executor;
        }
    }

    public DefaultTaskExecutor(String string, int numberOfThreads)
    {
        this(string, numberOfThreads, false);
    }

    public void dispose()
    {
        if (executor_ instanceof ExecutorService)
        {
            ((ExecutorService) executor_).shutdown();
            ((ExecutorService) executor_).shutdownNow();
        }
        
        disposeHooks_.dispose();
    }

    public void registerDisposable(Disposable d)
    {
        disposeHooks_.addDisposable(d);
    }
    
    public void execute(Runnable runnable)
    {
        executor_.execute(runnable);
    }
}