org.uberfire.commons.cluster.BaseLockExecuteNotifyReleaseTemplate Maven / Gradle / Ivy

Go to download
package org.uberfire.commons.cluster;

import java.util.Map;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.RunnableFuture;

import org.uberfire.commons.message.MessageType;

abstract class BaseLockExecuteNotifyReleaseTemplate {
    // default timeout 30 sec
    public static final int TIMEOUT = Integer.parseInt(System.getProperty("org.uberfire.cluster.timeout", "30000"));

    public V execute( final ClusterService clusterService,
                      final RunnableFuture task ) {
        try {
            clusterService.lock();

            task.run();

            final V result = task.get();

            sendMessage( clusterService );

            return result;
        } catch ( final ExecutionException e ) {
            throwException( e.getCause() );
        } catch ( final Exception e ) {
            throwException( e );
        } finally {
            clusterService.unlock();
        }
        return null;
    }

    private void throwException( final Throwable e ) {
        if ( e instanceof RuntimeException ) {
            throw (RuntimeException) e;
        }
        throw new RuntimeException( e );
    }

    abstract void sendMessage( final ClusterService clusterService );

    public abstract MessageType getMessageType();

    public abstract String getServiceId();

    public abstract Map buildContent();
}