package com.trendmicro.grid.acl.ds.jmx;
   
   import com.trendmicro.grid.acl.ds.RepositoriesContext;
   import com.trendmicro.grid.acl.ds.RepositorySelector;
   import org.springframework.jmx.export.annotation.ManagedResource;
   import org.springframework.jmx.export.annotation.ManagedAttribute;
   import org.springframework.stereotype.Service;
   import org.slf4j.Logger;
   import org.slf4j.LoggerFactory;
  
  import javax.annotation.PostConstruct;
  import javax.annotation.PreDestroy;
  import javax.annotation.Resource;
  import javax.management.*;
  import java.util.List;
  import java.util.ArrayList;
  import java.util.Iterator;
  
  /**
   * Publishes all registered repositories to JMX monitoring.
   *
   * @author juergen_kellerer, 2010-05-05
   * @version 1.0
   */
  @Service("jmxRepositoriesAdapter")
  @ManagedResource(objectName = RepositoriesAdapter.OBJECT_NAME, description = "" +
  		"Shows the active repository implementations and allows " +
  		"to change the selection of the implementation that is currently active.")
  public class RepositoriesAdapter {
  	private static final Logger log = LoggerFactory.getLogger(RepositoriesAdapter.class);
  
  	static final String JMX_DOMAIN = "com.trendmicro.grid.acl:";
  	static final String JMX_NAME = "RepositorySelection";
  	static final String OBJECT_NAME = JMX_DOMAIN + "type=" + JMX_NAME + ",name=Context";
  
  	@Resource
  	MBeanServer server;
  	@Resource
  	RepositoriesContext context;
  
  	private List<ObjectName> registeredNames = new ArrayList<ObjectName>();
  
  	@PostConstruct
  	synchronized void publishSelectorsToMonitoring() {
  		for (RepositorySelector<?> selector : context.getAllSelectors()) {
  			String className = selector.getRepositoryClass();
  			try {
  				className = className.substring(className.lastIndexOf('.') + 1);
  				ObjectName name = new ObjectName(JMX_DOMAIN + "type=" + JMX_NAME + ",name=" + className);
  				server.registerMBean(selector, name);
  				registeredNames.add(name);
  
  				if (log.isDebugEnabled())
  					log.debug("Registered the repository selector MBean '{}'", name);
  
  			} catch (Exception e) {
  				log.error("TMACL-00160:Failed to register selector for '" + className +
  						"' with JMX monitoring.", e);
  			}
  		}
  	}
  
  	@PreDestroy
  	synchronized void removeSelectorsFromMonitoring() {
  		for (Iterator<ObjectName> i = registeredNames.iterator(); i.hasNext();) {
  			ObjectName name = i.next();
  			try {
  				server.unregisterMBean(name);
  				i.remove();
  			} catch (Exception e) {
  				log.error("TMACL-00180:Failed to unregister selector '" + name + "' from JMX.", e);
  			}
  		}
  	}
  
  	@ManagedAttribute
  	public int getRegisteredRepositoryImplementationCount() {
  		int count = 0;
  		for (RepositorySelector<?> selector : context.getAllSelectors())
  			count += selector.getAvailableKeys().size();
  		return count;
  	}
  }