org.cyclopsgroup.jmxterm.cmd.SubscribeCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jmxterm Show documentation
Command line based interactive JMX client
The newest version!
package org.cyclopsgroup.jmxterm.cmd;

import java.io.IOException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import javax.management.*;
import org.cyclopsgroup.jcli.annotation.Cli;
import org.cyclopsgroup.jcli.annotation.Option;
import org.cyclopsgroup.jmxterm.Command;
import org.cyclopsgroup.jmxterm.Session;

/**
 * Command to subscribe to an MBean notification
 *
 * All notifications will be printed to the output in the form of notification received:
 * timestamp=xxx,class=xxx,source=xxx,type=xxx,message=xxx To remove the subscription call the
 * unsubscribe in the terminal.
 */
@Cli(
    name = "subscribe",
    description = "Subscribe to the notifications of a bean",
    note = "Syntax is \n subscribe ")
public class SubscribeCommand extends Command {
  private static Map listeners =
      new ConcurrentHashMap();

  public static Map getListeners() {
    return listeners;
  }

  public class BeanNotificationListener implements NotificationListener {

    @Override
    public void handleNotification(Notification notification, Object handback) {

      Session session = getSession();

      StringBuilder sb = new StringBuilder("notification received: ");
      sb.append("timestamp=").append(notification.getTimeStamp());
      sb.append(",class=").append(notification.getClass().getName());
      sb.append(",source=").append(notification.getSource());
      sb.append(",type=").append(notification.getType());
      sb.append(",message=").append(notification.getMessage());

      session.output.println(sb.toString());
    }
  }

  private String bean;

  private String domain;

  @Override
  public void execute() throws MalformedObjectNameException, IOException, JMException {
    Session session = getSession();
    String beanName = BeanCommand.getBeanName(bean, domain, session);
    if (beanName == null) {
      throw new IllegalArgumentException(
          "Please specify MBean to invoke either using -b option or bean command");
    }

    ObjectName name = new ObjectName(beanName);
    if (!listeners.containsKey(name)) {
      MBeanServerConnection con = session.getConnection().getServerConnection();

      NotificationListener listener = new BeanNotificationListener();
      con.addNotificationListener(name, listener, null, null);
      listeners.put(name, listener);

      session.output.printMessage("Subscribed to " + name);
    }
  }

  /** @param bean Bean under which the operation is */
  @Option(name = "b", longName = "bean", description = "MBean to invoke")
  public final void setBean(String bean) {
    this.bean = bean;
  }

  /** @param domain Domain under which is bean is */
  @Option(name = "d", longName = "domain", description = "Domain of MBean to invoke")
  public final void setDomain(String domain) {
    this.domain = domain;
  }
}