PluginRegistry.java

/*
 * Copyright 2014 LinkedIn Corp.
 * 
 * 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 azkaban.webapp.plugin;

import java.util.List;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.TreeSet;

public class PluginRegistry {

  private static PluginRegistry registry;

  public TreeSet<ViewerPlugin> viewerPlugins;

	public Map<String, TreeSet<ViewerPlugin>> jobTypeViewerPlugins;

  private PluginRegistry() {
		viewerPlugins = new TreeSet<ViewerPlugin>(ViewerPlugin.COMPARATOR);
		jobTypeViewerPlugins = new HashMap<String, TreeSet<ViewerPlugin>>();
  }

  public void register(ViewerPlugin plugin) {
		viewerPlugins.add(plugin);
		String jobType = plugin.getJobType();
		if (jobType == null) {
			return;
		}
		TreeSet<ViewerPlugin> plugins = null;
		if (!jobTypeViewerPlugins.containsKey(jobType)) {
			plugins = new TreeSet<ViewerPlugin>(ViewerPlugin.COMPARATOR);
			plugins.add(plugin);
			jobTypeViewerPlugins.put(jobType, plugins);
		}
		else {
			plugins = jobTypeViewerPlugins.get(jobType);
			plugins.add(plugin);
		}
  }

	public List<ViewerPlugin> getViewerPlugins() {
		return new ArrayList<ViewerPlugin>(viewerPlugins);
	}

	public List<ViewerPlugin> getViewerPluginsForJobType(String jobType) {
		TreeSet<ViewerPlugin> plugins = jobTypeViewerPlugins.get(jobType);
		if (plugins == null) {
			return null;
		}
		return new ArrayList<ViewerPlugin>(plugins);
	}

  public static PluginRegistry getRegistry() {
    if (registry == null) {
      registry = new PluginRegistry();
    }
    return registry;
  }
}