com.microsoft.azure.servicebus.primitives.Timer Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of azure-servicebus Show documentation

Java library for Azure Service Bus. Please note, a newer package com.azure:azure-messaging-servicebus for Azure Service Bus is available as of December 2020. While this package will continue to receive critical bug fixes, we strongly encourage you to upgrade. Read the migration guide at https://aka.ms/azsdk/java/migrate/sb for more details.

There is a newer version: 3.6.7

Show newest version

/*
 * Copyright (c) Microsoft. All rights reserved.
 * Licensed under the MIT license. See LICENSE file in the project root for full license information.
 */
package com.microsoft.azure.servicebus.primitives;

import java.time.Duration;
import java.util.HashSet;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


/**
 * An abstraction for a Scheduler functionality - which can later be replaced by a light-weight Thread
 */
final public class Timer
{
	private static ScheduledExecutorService executor = null;

	private static final Logger TRACE_LOGGER = LoggerFactory.getLogger(Timer.class);
	private static final HashSet references = new HashSet();
	private static final Object syncReferences = new Object();

	private Timer() 
	{
	}

	
	// runFrequency implemented only for TimeUnit granularity - Seconds	 
	public static ScheduledFuture schedule(Runnable runnable, Duration runFrequency, TimerType timerType)
	{
		switch (timerType)
		{
		case OneTimeRun:
			return executor.schedule(runnable, runFrequency.toMillis(), TimeUnit.MILLISECONDS);

		case RepeatRun:
			return executor.scheduleWithFixedDelay(runnable, runFrequency.toMillis(), runFrequency.toMillis(), TimeUnit.MILLISECONDS);

		default:
			throw new UnsupportedOperationException("Unsupported timer pattern.");
		}
	}

	static void register(final String clientId)
	{
		synchronized (syncReferences)
		{
			if (references.size() == 0 && (executor == null || executor.isShutdown()))
			{
				final int corePoolSize = Math.max(Runtime.getRuntime().availableProcessors(), 4);
				TRACE_LOGGER.debug("Starting ScheduledThreadPoolExecutor with coreThreadPoolSize:{}", corePoolSize);
				
				executor = Executors.newScheduledThreadPool(corePoolSize);
			}

			references.add(clientId);
		}
	}

	static void unregister(final String clientId)
	{
		synchronized (syncReferences)
		{
			if (references.remove(clientId) && references.size() == 0 && executor != null)
			{				
				TRACE_LOGGER.debug("Shuting down ScheduledThreadPoolExecutor");
				executor.shutdownNow();
			}
		}
	}
}