NotificationDefinitionParser.java example

Explorer
MULE-master
- mule-mule-4.x
/*
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package org.mule.runtime.config.spring.parsers.specific;

import org.mule.runtime.config.spring.parsers.PreProcessor;
import org.mule.runtime.config.spring.parsers.assembly.configuration.PropertyConfiguration;
import org.mule.runtime.config.spring.parsers.collection.ChildMapEntryDefinitionParser;
import org.mule.runtime.config.spring.parsers.processors.CheckExclusiveAttributes;
import org.mule.runtime.config.spring.parsers.processors.CheckRequiredAttributes;
import org.mule.runtime.core.api.context.notification.ClusterNodeNotificationListener;
import org.mule.runtime.core.api.context.notification.ComponentMessageNotificationListener;
import org.mule.runtime.core.api.context.notification.ConnectionNotificationListener;
import org.mule.runtime.core.api.context.notification.ConnectorMessageNotificationListener;
import org.mule.runtime.core.api.context.notification.CustomNotificationListener;
import org.mule.runtime.core.api.context.notification.ExceptionNotificationListener;
import org.mule.runtime.core.api.context.notification.ErrorHandlerNotificationListener;
import org.mule.runtime.core.api.context.notification.ManagementNotificationListener;
import org.mule.runtime.core.api.context.notification.MessageProcessorNotificationListener;
import org.mule.runtime.core.api.context.notification.MuleContextNotificationListener;
import org.mule.runtime.core.api.context.notification.RegistryNotificationListener;
import org.mule.runtime.core.api.context.notification.RoutingNotificationListener;
import org.mule.runtime.core.api.context.notification.SecurityNotificationListener;
import org.mule.runtime.core.api.context.notification.TransactionNotificationListener;
import org.mule.runtime.core.context.notification.AsyncMessageNotification;
import org.mule.runtime.core.context.notification.ComponentMessageNotification;
import org.mule.runtime.core.context.notification.ConnectionNotification;
import org.mule.runtime.core.context.notification.ConnectorMessageNotification;
import org.mule.runtime.core.context.notification.CustomNotification;
import org.mule.runtime.core.context.notification.ExceptionNotification;
import org.mule.runtime.core.context.notification.ErrorHandlerNotification;
import org.mule.runtime.core.context.notification.ManagementNotification;
import org.mule.runtime.core.context.notification.MessageProcessorNotification;
import org.mule.runtime.core.context.notification.MuleContextNotification;
import org.mule.runtime.core.context.notification.PipelineMessageNotification;
import org.mule.runtime.core.context.notification.RegistryNotification;
import org.mule.runtime.core.context.notification.RoutingNotification;
import org.mule.runtime.core.context.notification.SecurityNotification;
import org.mule.runtime.core.context.notification.TransactionNotification;

import java.util.HashMap;
import java.util.Map;

import org.w3c.dom.Element;

public class NotificationDefinitionParser extends ChildMapEntryDefinitionParser {

  private static final String ENDPOINT_MESSAGE_NOTIFICATION_CLASS =
      "com.mulesoft.mule.compatibility.core.context.notification.EndpointMessageNotification";
  private static final String ENDPOINT_MESSAGE_NOTIFICATION_LISTENER_CLASS =
      "com.mulesoft.mule.compatibility.core.api.context.notification.EndpointMessageNotificationListener";
  public static final Map EVENT_MAP;
  public static final Map INTERFACE_MAP;
  public static final String INTERFACE = "interface";
  public static final String INTERFACE_CLASS = "interface-class";
  public static final String EVENT = "event";
  public static final String EVENT_CLASS = "event-class";
  public static final String[][] INTERFACE_ATTRIBUTES = new String[][] {new String[] {INTERFACE}, new String[] {INTERFACE_CLASS}};
  public static final String[][] EVENT_ATTRIBUTES = new String[][] {new String[] {EVENT}, new String[] {EVENT_CLASS}};
  public static final String[][] ALL_ATTRIBUTES =
      new String[][] {new String[] {EVENT}, new String[] {EVENT_CLASS}, new String[] {INTERFACE}, new String[] {INTERFACE_CLASS}};

  static {
    EVENT_MAP = new HashMap();
    EVENT_MAP.put("CONTEXT", MuleContextNotification.class.getName());
    EVENT_MAP.put("SECURITY", SecurityNotification.class.getName());
    EVENT_MAP.put("ENDPOINT-MESSAGE", ENDPOINT_MESSAGE_NOTIFICATION_CLASS);
    EVENT_MAP.put("CONNECTOR-MESSAGE", ConnectorMessageNotification.class.getName());
    EVENT_MAP.put("COMPONENT-MESSAGE", ComponentMessageNotification.class.getName());
    EVENT_MAP.put("MANAGEMENT", ManagementNotification.class.getName());
    EVENT_MAP.put("MESSAGE-PROCESSOR", MessageProcessorNotification.class.getName());
    EVENT_MAP.put("EXCEPTION-STRATEGY", ErrorHandlerNotification.class.getName());
    EVENT_MAP.put("CONNECTION", ConnectionNotification.class.getName());
    EVENT_MAP.put("REGISTRY", RegistryNotification.class.getName());
    EVENT_MAP.put("CUSTOM", CustomNotification.class.getName());
    EVENT_MAP.put("EXCEPTION", ExceptionNotification.class.getName());
    EVENT_MAP.put("TRANSACTION", TransactionNotification.class.getName());
    EVENT_MAP.put("ROUTING", RoutingNotification.class.getName());
    EVENT_MAP.put("PIPELINE-MESSAGE", PipelineMessageNotification.class.getName());
    EVENT_MAP.put("ASYNC-MESSAGE", AsyncMessageNotification.class.getName());
    // EVENT_MAP.put("MESSAGE", ConnectorMessageNotification.class.getName());

    INTERFACE_MAP = new HashMap();
    INTERFACE_MAP.put("CONTEXT", MuleContextNotificationListener.class.getName());
    INTERFACE_MAP.put("SECURITY", SecurityNotificationListener.class.getName());
    INTERFACE_MAP.put("MANAGEMENT", ManagementNotificationListener.class.getName());
    INTERFACE_MAP.put("MESSAGE-PROCESSOR", MessageProcessorNotificationListener.class.getName());
    INTERFACE_MAP.put("EXCEPTION-STRATEGY", ErrorHandlerNotificationListener.class.getName());
    INTERFACE_MAP.put("CONNECTION", ConnectionNotificationListener.class.getName());
    INTERFACE_MAP.put("REGISTRY", RegistryNotificationListener.class.getName());
    INTERFACE_MAP.put("CUSTOM", CustomNotificationListener.class.getName());
    INTERFACE_MAP.put("ENDPOINT-MESSAGE", ENDPOINT_MESSAGE_NOTIFICATION_LISTENER_CLASS);
    INTERFACE_MAP.put("CONNECTOR-MESSAGE", ConnectorMessageNotificationListener.class.getName());
    INTERFACE_MAP.put("COMPONENT-MESSAGE", ComponentMessageNotificationListener.class.getName());
    INTERFACE_MAP.put("EXCEPTION", ExceptionNotificationListener.class.getName());
    INTERFACE_MAP.put("TRANSACTION", TransactionNotificationListener.class.getName());
    INTERFACE_MAP.put("ROUTING", RoutingNotificationListener.class.getName());
    INTERFACE_MAP.put("CLUSTER-NODE", ClusterNodeNotificationListener.class.getName());
    INTERFACE_MAP.put("PIPELINE-MESSAGE", PipelineMessageNotification.class.getName());
    INTERFACE_MAP.put("ASYNC-MESSAGE", AsyncMessageNotification.class.getName());

    // Deprecated
    EVENT_MAP.put("MESSAGE", ENDPOINT_MESSAGE_NOTIFICATION_CLASS);
  }

  public NotificationDefinitionParser() {
    super("interfaceToType", INTERFACE_CLASS, EVENT_CLASS);
    addMapping(EVENT, EVENT_MAP);
    addAlias(EVENT, VALUE);
    addMapping(INTERFACE, INTERFACE_MAP);
    addAlias(INTERFACE, KEY);
    registerPreProcessor(new CheckExclusiveAttributes(INTERFACE_ATTRIBUTES));
    registerPreProcessor(new CheckExclusiveAttributes(EVENT_ATTRIBUTES));
    registerPreProcessor(new CheckRequiredAttributes(INTERFACE_ATTRIBUTES));
    registerPreProcessor(new CheckRequiredAttributes(EVENT_ATTRIBUTES));
    registerPreProcessor(new SetDefaults());
  }

  /**
   * If only one of event or interface is set, use it as default for the other
   */
  private class SetDefaults implements PreProcessor {

    @Override
    public void preProcess(PropertyConfiguration config, Element element) {
      copy(element, INTERFACE, EVENT, EVENT_CLASS);
      copy(element, EVENT, INTERFACE, INTERFACE_CLASS);
    }

    private void copy(Element element, String from, String to, String blocker) {
      if (element.hasAttribute(from) && !element.hasAttribute(to) && !element.hasAttribute(blocker)) {
        element.setAttribute(to, element.getAttribute(from));
      }
    }

  }

}