io.datarouter.tasktracker.config.DatarouterTaskTrackerPlugin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of datarouter-task-tracker Show documentation
datarouter-task-tracker
There is a newer version: 0.0.126
Show newest version
/**
 * Copyright © 2009 HotPads ([email protected])
 *
 * 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 io.datarouter.tasktracker.config;

import java.util.List;

import io.datarouter.storage.client.ClientId;
import io.datarouter.storage.dao.Dao;
import io.datarouter.storage.dao.DaosModuleBuilder;
import io.datarouter.tasktracker.storage.DatarouterLongRunningTaskDao;
import io.datarouter.tasktracker.storage.DatarouterLongRunningTaskDao.DatarouterLongRunningTaskDaoParams;
import io.datarouter.tasktracker.web.LongRunningTaskGraphLink;
import io.datarouter.web.config.BaseWebPlugin;
import io.datarouter.web.navigation.DatarouterNavBarCategory;
import io.datarouter.web.navigation.NavBarItem;

public class DatarouterTaskTrackerPlugin extends BaseWebPlugin{

	private final Class longRunningTaskGraphLinkClass;
	private final DatarouterTaskTrackerDaoModule daosModuleBuilder;

	private DatarouterTaskTrackerPlugin(
			Class longRunningTaskGraphLinkClass,
			DatarouterTaskTrackerDaoModule daosModuleBuilder){
		this.longRunningTaskGraphLinkClass = longRunningTaskGraphLinkClass;
		this.daosModuleBuilder = daosModuleBuilder;
		addDatarouterNavBarItem(new NavBarItem(DatarouterNavBarCategory.JOBS,
				new DatarouterTaskTrackerPaths().datarouter.longRunningTasks, "Long running tasks"));
		addSettingRoot(DatarouterTaskTrackerSettingRoot.class);
		addRouteSet(DatarouterTaskTrackerRouteSet.class);
	}

	@Override
	public DatarouterTaskTrackerDaoModule getDaosModuleBuilder(){
		return daosModuleBuilder;
	}

	@Override
	public void configure(){
		if(longRunningTaskGraphLinkClass != null){
			bindActual(LongRunningTaskGraphLink.class, longRunningTaskGraphLinkClass);
		}
	}

	public static class DatarouterTaskTrackerDaoModule extends DaosModuleBuilder{

		private final ClientId datarouterLongRunningTaskClientId;

		public DatarouterTaskTrackerDaoModule(ClientId datarouterLongRunningTaskClientId){
			this.datarouterLongRunningTaskClientId = datarouterLongRunningTaskClientId;
		}

		@Override
		public List> getDaoClasses(){
			return List.of(DatarouterLongRunningTaskDao.class);
		}

		@Override
		public void configure(){
			bind(DatarouterLongRunningTaskDaoParams.class)
					.toInstance(new DatarouterLongRunningTaskDaoParams(datarouterLongRunningTaskClientId));
		}

	}

	public static class DatarouterTaskTrackerPluginBuilder{

		private final ClientId defaultClientId;
		private DatarouterTaskTrackerDaoModule daoModule;

		private Class longRunningTaskGraphLinkClass;

		public DatarouterTaskTrackerPluginBuilder(ClientId defaultClientId){
			this.defaultClientId = defaultClientId;
		}

		public DatarouterTaskTrackerPluginBuilder setLongRunningTaskGraphLinkClass(
				Class longRunningTaskGraphLinkClass){
			this.longRunningTaskGraphLinkClass = longRunningTaskGraphLinkClass;
			return this;
		}

		public DatarouterTaskTrackerPluginBuilder setDaoModule(DatarouterTaskTrackerDaoModule daoModule){
			this.daoModule = daoModule;
			return this;
		}

		public DatarouterTaskTrackerPlugin build(){
			return new DatarouterTaskTrackerPlugin(
					longRunningTaskGraphLinkClass,
					daoModule == null ? new DatarouterTaskTrackerDaoModule(defaultClientId) : daoModule);
		}

	}

}