com.tacitknowledge.util.migration.RollbackBroadcaster Maven / Gradle / Ivy

Go to download
/* Copyright 2004 Tacit Knowledge
 *  
 * 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.tacitknowledge.util.migration;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

/**
 * Manages the MessageListener that are associated with a
 * Rollback instance.
 *
 * @author Artie Pesh-Imam ([email protected])
 */
class RollbackBroadcaster
{
    /**
     * Used by notifyListeners to indicate that listeners should
     * be informed that a task is about to start.
     */
    public static final int TASK_START = 1;

    /**
     * Used by notifyListeners to indicate that listeners should
     * be informed that a task has successfully completed.
     */
    public static final int TASK_SUCCESS = 2;

    /**
     * Used by notifyListeners to indicate that listeners should
     * be informed that a task failed.
     */
    public static final int TASK_FAILED = 3;

    /**
     * The listeners interested in being notified of migration task events.
     */
    private List listeners = new ArrayList();

    /**
     * Notifies all registered listeners of a migration task event.
     *
     * @param task          the task that is being or that has been executed
     * @param context       the context in which the task was executed
     * @param eventType     TASK_START, TASK_SUCCESS, or TASK_FAIL
     * @param rollbackLevel
     * @param e             the exception thrown by the task if the task failed
     * @throws MigrationException if one of the listeners threw an exception
     */
    public void notifyListeners(RollbackableMigrationTask task,
            MigrationContext context, MigrationException e, int eventType, int rollbackLevel)
            throws MigrationException
    {
        for (Iterator i = listeners.iterator(); i.hasNext();)
        {
            RollbackListener listener = (RollbackListener) i.next();
            switch (eventType)
            {
                case TASK_START:
                    listener.rollbackStarted(task, context);
                    break;

                case TASK_SUCCESS:
                    listener.rollbackSuccessful(task, rollbackLevel, context);
                    break;

                case TASK_FAILED:
                    listener.rollbackFailed(task, context, e);
                    break;

                default:
                    throw new IllegalArgumentException("Unknown event type");
            }
        }
    }

    /**
     * Notifies all registered listeners of a migration task event.
     *
     * @param task      the task that is being or that has been executed
     * @param context   the context in which the task was executed
     * @param eventType TASK_START, TASK_SUCCESS, or TASK_FAIL
     * @throws MigrationException if one of the listeners threw an exception
     */
    public void notifyListeners(RollbackableMigrationTask task,
            MigrationContext context, int eventType, int rollbackLevel) throws MigrationException
    {
        notifyListeners(task, context, null, eventType, rollbackLevel);
    }

    /**
     * Registers the given MigrationListener as being interested
     * in migration task events.
     *
     * @param listener the listener to add; may not be null
     */
    public void addListener(RollbackListener listener)
    {
        if (listener == null)
        {
            throw new IllegalArgumentException("listener cannot be null");
        }
        listeners.add(listener);
    }

    /**
     * Removes the given MigrationListener from the list of
     * listeners associated with the Migration instance.
     *
     * @param listener the listener to add; may not be null
     * @return true if the listener was located and removed,
     *         otherwise false.
     */
    public boolean removeListener(RollbackListener listener)
    {
        if (listener == null)
        {
            throw new IllegalArgumentException("listener cannot be null");
        }
        return listeners.remove(listener);
    }

    /**
     * Get the list of listeners
     *
     * @return List of MigrationListener objects
     */
    public List getListeners()
    {
        return listeners;
    }
}