content: // 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 com.cloud.alert;
  
  import java.util.Map;
  
  import javax.inject.Inject;
  import javax.naming.ConfigurationException;
  
  import org.springframework.stereotype.Component;
  
  import com.cloud.cluster.ClusterManager;
  import com.cloud.cluster.ClusterNodeJoinEventArgs;
  import com.cloud.cluster.ClusterNodeLeftEventArgs;
  import com.cloud.cluster.ManagementServerHostVO;
  import com.cloud.cluster.dao.ManagementServerHostDao;
  import com.cloud.utils.component.AdapterBase;
  import com.cloud.utils.events.EventArgs;
  import com.cloud.utils.events.SubscriptionMgr;
  
  @Component
  public class ClusterAlertAdapter extends AdapterBase implements AlertAdapter {
  
  
      @Inject
      private AlertManager _alertMgr;
      @Inject
      private ManagementServerHostDao _mshostDao;
  
      public void onClusterAlert(Object sender, EventArgs args) {
          if (logger.isDebugEnabled()) {
              logger.debug("Receive cluster alert, EventArgs: " + args.getClass().getName());
          }
  
          if (args instanceof ClusterNodeJoinEventArgs) {
              onClusterNodeJoined(sender, (ClusterNodeJoinEventArgs)args);
          } else if (args instanceof ClusterNodeLeftEventArgs) {
              onClusterNodeLeft(sender, (ClusterNodeLeftEventArgs)args);
          } else {
              logger.error("Unrecognized cluster alert event");
          }
      }
  
      private void onClusterNodeJoined(Object sender, ClusterNodeJoinEventArgs args) {
          if (logger.isDebugEnabled()) {
              for (ManagementServerHostVO mshost : args.getJoinedNodes()) {
                  logger.debug("Handle cluster node join alert, joined node: {} ({})", mshost.getServiceIP(), mshost);
              }
          }
  
          for (ManagementServerHostVO mshost : args.getJoinedNodes()) {
              if (mshost.getId() == args.getSelf().longValue()) {
                  if (logger.isDebugEnabled()) {
                      logger.debug("Management server node {} ({}) is up, send alert", mshost.getServiceIP(), mshost);
                  }
  
                  _alertMgr.sendAlert(AlertManager.AlertType.ALERT_TYPE_MANAGEMENT_NODE, 0, new Long(0), "Management server node " + mshost.getServiceIP() + " is up", "");
                  break;
              }
          }
      }
  
      private void onClusterNodeLeft(Object sender, ClusterNodeLeftEventArgs args) {
  
          if (logger.isDebugEnabled()) {
              for (ManagementServerHostVO mshost : args.getLeftNodes()) {
                  logger.debug("Handle cluster node left alert, leaving node: {} ({})", mshost.getServiceIP(), mshost);
              }
          }
  
          for (ManagementServerHostVO mshost : args.getLeftNodes()) {
              if (mshost.getId() != args.getSelf().longValue()) {
                  if (_mshostDao.increaseAlertCount(mshost.getId()) > 0) {
                      if (logger.isDebugEnabled()) {
                          logger.debug("Detected management server node {} ({}) is down, send alert", mshost.getServiceIP(), mshost);
                      }
                      _alertMgr.sendAlert(AlertManager.AlertType.ALERT_TYPE_MANAGEMENT_NODE, 0, new Long(0), "Management server node " + mshost.getServiceIP() + " is down",
                          "");
                  } else {
                      if (logger.isDebugEnabled()) {
                          logger.debug("Detected management server node {} ({}) is down, but alert has already been set", mshost.getServiceIP(), mshost);
                      }
                  }
              }
          }
      }
  
      @Override
      public boolean configure(String name, Map<String, Object> params) throws ConfigurationException {
  
          if (logger.isInfoEnabled()) {
              logger.info("Start configuring cluster alert manager : " + name);
          }
  
          try {
              SubscriptionMgr.getInstance().subscribe(ClusterManager.ALERT_SUBJECT, this, "onClusterAlert");
          } catch (SecurityException e) {
              throw new ConfigurationException("Unable to register cluster event subscription");
          } catch (NoSuchMethodException e) {
              throw new ConfigurationException("Unable to register cluster event subscription");
          }
  
          return true;
      }
  }
  