com.enonic.xp.task.TaskProgressReporterContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of core-api Show documentation

core-api

The newest version!

package com.enonic.xp.task;

import com.enonic.xp.annotation.PublicApi;

@PublicApi
public final class TaskProgressReporterContext
{
    private static final ThreadLocal CURRENT = new ThreadLocal<>();

    public static ProgressReporter current()
    {
        return CURRENT.get();
    }

    private static void set( final ProgressReporter req )
    {
        if ( req == null )
        {
            CURRENT.remove();
        }
        else
        {
            CURRENT.set( req );
        }
    }

    public static RunnableTask withContext( final RunnableTask runnableTask )
    {
        return ( id, progressReporter ) ->
        {
            final ProgressReporter old = current();
            set( progressReporter );

            try
            {
                runnableTask.run( id, progressReporter );
            }
            finally
            {
                set( old );
            }
        };
    }
}