Java tutorial
/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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.apache.synapse.startup.quartz; import org.apache.axiom.om.OMElement; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.apache.synapse.ManagedLifecycle; import org.apache.synapse.SynapseException; import org.apache.synapse.commons.util.PropertyHelper; import org.apache.synapse.core.SynapseEnvironment; import org.apache.synapse.startup.AbstractStartup; import org.apache.synapse.task.*; import javax.xml.namespace.QName; import java.net.InetAddress; import java.net.UnknownHostException; import java.util.*; public class StartUpController extends AbstractStartup { private static final Log logger = LogFactory.getLog(StartUpController.class.getName()); private TaskDescription taskDescription; private SynapseTaskManager synapseTaskManager; private Object task = null; public QName getTagQName() { return SimpleQuartzFactory.TASK; } public void destroy() { if (!destroyTask()) { return; } if (synapseTaskManager.isInitialized()) { TaskScheduler taskScheduler = synapseTaskManager.getTaskScheduler(); if (taskScheduler != null && taskScheduler.isInitialized()) { taskScheduler.deleteTask(taskDescription.getName(), taskDescription.getTaskGroup()); } TaskDescriptionRepository repository = synapseTaskManager.getTaskDescriptionRepository(); if (repository != null) { repository.removeTaskDescription(taskDescription.getName()); } } } public void init(SynapseEnvironment synapseEnvironment) { if (taskDescription == null) { handleException("Error while initializing the startup. TaskDescription is null."); } initSynapseTaskManager(synapseEnvironment); TaskDescriptionRepository repository = synapseTaskManager.getTaskDescriptionRepository(); if (repository == null) { handleException("Task Description Repository cannot be found"); return; } repository.addTaskDescription(taskDescription); if (!processPinnedServers(taskDescription, synapseEnvironment)) { return; } resolveTaskImpl(taskDescription, synapseEnvironment); loadTaskProperties(); initializeTask(synapseEnvironment); if (taskDescription.getResource(TaskDescription.INSTANCE) == null || taskDescription.getResource(TaskDescription.CLASSNAME) == null) { taskDescription.addResource(TaskDescription.INSTANCE, task); taskDescription.addResource(TaskDescription.CLASSNAME, task.getClass().getName()); } try { Map<String, Object> map = new HashMap<String, Object>(); map.put(TaskConstants.SYNAPSE_ENV, synapseEnvironment); TaskScheduler taskScheduler = synapseTaskManager.getTaskScheduler(); TaskManager taskManager = synapseTaskManager.getTaskManagerImpl(); if (taskManager == null) { logger.error("Could not initialize Start up controller. TaskManager not found."); return; } taskManager.setProperties(map); taskScheduler.init(synapseEnvironment.getSynapseConfiguration().getProperties(), taskManager); submitTask(taskScheduler, taskDescription); logger.debug("Submitted task [" + taskDescription.getName() + "] to Synapse task scheduler."); } catch (Exception e) { String msg = "Error starting up Scheduler : " + e.getLocalizedMessage(); logger.fatal(msg, e); throw new SynapseException(msg, e); } } private void initializeTask(SynapseEnvironment synapseEnvironment) { if (task instanceof ManagedLifecycle) { ((ManagedLifecycle) task).init(synapseEnvironment); } } private boolean destroyTask() { if (taskDescription == null) { logger.debug("No task found to delete."); return false; } if (task instanceof ManagedLifecycle) { ((ManagedLifecycle) task).destroy(); } return true; } private void loadTaskProperties() { Set properties = taskDescription.getXmlProperties(); for (Object property : properties) { OMElement prop = (OMElement) property; logger.debug("loaded task property : " + prop.toString()); PropertyHelper.setStaticProperty(prop, task); } } private boolean initSynapseTaskManager(SynapseEnvironment synapseEnvironment) { synapseTaskManager = synapseEnvironment.getTaskManager(); if (!synapseTaskManager.isInitialized()) { logger.warn("SynapseTaskManager is not properly initialized. Initializing now with " + "default parameters."); synapseTaskManager.init(null, null, null); } return true; } private boolean submitTask(TaskScheduler taskScheduler, TaskDescription taskDescription) { if (taskDescription == null) { return false; } if (taskScheduler != null) { return taskScheduler.scheduleTask(taskDescription); } else { if (logger.isDebugEnabled()) { logger.debug("TaskScheduler cannot be found for :" + TaskConstants.TASK_SCHEDULER + " , " + "therefore ignore scheduling of Task " + taskDescription); } return false; } } private boolean processPinnedServers(TaskDescription taskDescription, SynapseEnvironment synapseEnvironment) { String thisServerName = synapseEnvironment.getServerContextInformation().getServerConfigurationInformation() .getServerName(); if (thisServerName == null || thisServerName.equals("")) { try { InetAddress address = InetAddress.getLocalHost(); thisServerName = address.getHostName(); } catch (UnknownHostException e) { logger.warn("Could not get the host name", e); } if (thisServerName == null || thisServerName.equals("")) { thisServerName = "localhost"; } } logger.debug("Synapse server name : " + thisServerName); List pinnedServers = taskDescription.getPinnedServers(); if (pinnedServers != null && !pinnedServers.isEmpty()) { if (!pinnedServers.contains(thisServerName)) { logger.info("Server name not in pinned servers list. Not starting Task : " + getName()); return false; // do not continue the caller of this method. } } return true; } private boolean resolveTaskImpl(TaskDescription taskDescription, SynapseEnvironment synapseEnvironment) { if (synapseEnvironment == null) { return false; } String taskImplClassName = taskDescription.getTaskImplClassName(); if (taskImplClassName == null || taskImplClassName.isEmpty()) { taskImplClassName = "org.apache.synapse.startup.tasks.MessageInjector"; } taskDescription.setTaskImplClassName(taskImplClassName); try { task = getClass().getClassLoader().loadClass(taskDescription.getTaskImplClassName()).newInstance(); if (!(task instanceof Task)) { logger.warn("Task implementation is not a Synapse Task."); } } catch (Exception e) { handleException("Cannot instantiate task : " + taskDescription.getTaskImplClassName(), e); } return true; } private static void handleException(String message) { logger.error(message); throw new SynapseException(message); } private static void handleException(String message, Exception e) { logger.error(message, e); throw new SynapseException(message, e); } public TaskDescription getTaskDescription() { return taskDescription; } public void setTaskDescription(TaskDescription taskDescription) { this.taskDescription = taskDescription; } }