org.tio.utils.timer.DefaultTimerTaskService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mica-net-utils Show documentation

Mica net is a net framework.

There is a newer version: 1.0.8

/*
 * Copyright (c) 2019-2029, Dreamlu 卢春梦 ([email protected] & dreamlu.net).
 *
 * 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 org.tio.utils.timer;

import java.util.function.Function;

/**
 * Default TimerTaskService
 *
 * @author L.cm
 */
public class DefaultTimerTaskService implements TimerTaskService {
	private final SystemTimer systemTimer;
	private final TimingWheelThread timingWheelThread;

	public DefaultTimerTaskService() {
		this(1000L, 60);
	}

	public DefaultTimerTaskService(long tickMs, int wheelSize) {
		this(new SystemTimer(tickMs, wheelSize, "DefaultTimerTaskService"));
	}

	public DefaultTimerTaskService(SystemTimer systemTimer) {
		this(systemTimer, new TimingWheelThread(systemTimer));
	}

	public DefaultTimerTaskService(SystemTimer systemTimer, TimingWheelThread timingWheelThread) {
		this.systemTimer = systemTimer;
		this.timingWheelThread = timingWheelThread;
	}

	@Override
	public  T addTask(Function consumer) {
		T timerTask = consumer.apply(this.systemTimer);
		this.systemTimer.add(timerTask);
		return timerTask;
	}

	@Override
	public void start() {
		if (!timingWheelThread.isStarted()) {
			timingWheelThread.start();
		}
	}

	@Override
	public void stop() {
		timingWheelThread.shutdown();
		systemTimer.shutdown();
	}

}