监控OSB中服务的状态并生成日志

import com.bea.wli.config.Ref;
import com.bea.wli.monitoring.*;
import weblogic.management.jmx.MBeanServerInvocationHandler;
import javax.management.MBeanServerConnection;
import javax.management.MalformedObjectNameException;
import javax.management.ObjectName;
import javax.management.remote.JMXConnector;
import javax.management.remote.JMXConnectorFactory;
import javax.management.remote.JMXServiceURL;
import javax.naming.Context;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.net.MalformedURLException;
import java.util.*;
import java.text.SimpleDateFormat;

/**   
 * This class provides sample code showing how to use   
 * ServiceDomainMBean.   
 * It shows how to:   
 * 1. Find business and proxy services enabled for monitoring.   
 * 2. Get statistics for one or more business and proxy services.   
 * 3. Perform reset operation on one or more business and proxy services.   
 * 4. Handle exceptions.   
 * It uses a timer to retrieve statistics, save them in a file, and   
 * perform resets in a recursive manner.
 */

public class ServiceStatisticsRetriever_file {
	private ServiceDomainMBean serviceDomainMbean = null;
	private String serverName = null;
	
	 /**
	  * Retrieve statistics for all business services being monitored in the      
	  *  domain and reset statistics for the same.      
	  *  @throws Exception      
	  */
	void getAndResetStatsForAllMonitoredBizServices() throws Exception {
		 Ref[] serviceRefs = serviceDomainMbean.getMonitoredBusinessServiceRefs();
		// Create a bitwise map for desired resource types.
		 int typeFlag = 0;
		 typeFlag = typeFlag | ResourceType.SERVICE.value();
		 typeFlag = typeFlag | ResourceType.WEBSERVICE_OPERATION.value();
		 typeFlag = typeFlag | ResourceType.URI.value();
		 
		 HashMap<Ref, ServiceResourceStatistic> resourcesMap = null;
		 HashMap<Ref, ServiceResourceStatistic> resourcesMapOnSingleServer = null;
		 // Get cluster-level statistics.
		 try {               
			 // Get statistics.
			 System.out.println("Now trying to get statistics for -" + serviceRefs.length + " business services...");
			 resourcesMap = serviceDomainMbean.getBusinessServiceStatistics(serviceRefs, typeFlag, serverName);
			 // Reset statistics.               
			 long resetRequestTime = serviceDomainMbean.resetStatistics (serviceRefs);
			 // Save retrieved statistics.
			 String fileName = "BizStatistics_" + new SimpleDateFormat("yyyy_MM_dd_HH_mm").format(new Date(System. currentTimeMillis()))+ ".txt";
			 saveStatisticsToFile(resourcesMap, resetRequestTime,fileName);
		 }catch (IllegalArgumentException iae) {
			 System.out.println("===============================\n");
			 System.out.println("Encountered IllegalArgumentException... Details:");
			 System.out.println(iae.getMessage());
			 System.out.println("Check if proxy ref was passed OR flowComp " + "resource was passed OR bitmap is invalid..." +"\nIf so correct it and try again!!!");
			 System.out.println("==================================\n");
			 throw iae;
		 }catch (DomainMonitoringDisabledException dmde) {
			 /** Statistics not available as monitoring is turned off at domain level.*/
			 System.out.println("==================================\n");
			 System.out.println("Statistics not available as monitoring " + "is turned off at domain level.");
			 System.out.println("==============================\n");
			 throw dmde;
		 }catch (MonitoringException me) {
			 // Internal problem... May be aggregation server is crashed...
			 System.out.println("================================\n");
			 System.out.println("ERROR: Statistics is not available..." + "Check if aggregation server is crashed...");
			 System.out.println("=================================\n");
			 throw me;
		 }
	}
	
	/**
	 * Retrieve statistics for all proxy services being monitored in the     
	 * domain and reset statistics for the same.      
	 * @throws Exception      
	 */
	 void getAndResetStatsForAllMonitoredProxyServices() throws Exception {
		 Ref[] serviceRefs = serviceDomainMbean.getMonitoredProxyServiceRefs();
		// Create a bitwise map for desired resource types.
		 int typeFlag = 0;
		 typeFlag = typeFlag | ResourceType.SERVICE.value();
		 typeFlag = typeFlag | ResourceType.FLOW_COMPONENT.value();
		 typeFlag = typeFlag | ResourceType.WEBSERVICE_OPERATION.value();
		 HashMap<Ref, ServiceResourceStatistic> resourcesMap = null;
		// Get cluster-level statistics.
		 try {
			// Get statistics.
			 System.out.println("Now trying to get statistics for -" + serviceRefs.length + " proxy services...");
			 resourcesMap = serviceDomainMbean.getProxyServiceStatistics(serviceRefs,typeFlag, null);
			 // Reset statistics.
			 long resetRequestTime = serviceDomainMbean.resetStatistics(serviceRefs);
			 // Save retrieved statistics.
			 String fileName = "ProxyStatistics_" + new SimpleDateFormat("yyyy_MM_dd_HH_mm").format(new Date(System.currentTimeMillis())) + ".txt";
			 saveStatisticsToFile(resourcesMap, resetRequestTime,fileName);
		 }catch (IllegalArgumentException iae) {
			 System.out.println("===================================\n");
			 System.out.println("Encountered IllegalArgumentException...Details:");
			 System.out.println(iae.getMessage());
			 System.out.println("Check if business ref was passed OR bitmap is " + "invalid...\nIf so correct it and try again!!!");               System.out.println("===================================\n");                    throw iae;
		 }catch (DomainMonitoringDisabledException dmde) {
			 /** Statistics not available as monitoring is turned off at the              
			  *  domain level.              
			  */
			 System.out.println("===================================\n");
			 System.out.println("Statistics not available as monitoring" + "is turned off at domain level.");
			 System.out.println("===================================\n");
			 throw dmde;
		 }catch (MonitoringException me) {
			// Internal problem ... May be aggregation server is crashed ...
			 System.out.println("===================================\n");
			 System.out.println("ERROR: Statistics is not available... " + "Check if aggregation server is crashed...");
			 System.out.println("===================================\n");
			 throw me;
		 }
	 }
	 
	 /**      
	  *  Saves statistics of all services from the specified map.      
	  *   @param statsMap Map containing statistics for one or more services     
	  *   of the same type; i.e., business or proxy.      
	  *    @param resetReqTime Reset request time. This information will be     
	  *    written at the end of the file, provided it is not zero.      
	  *    @param fileName Statistics will be saved in a file with this name.      
	  *    @throws Exception      
	  */
	 private void saveStatisticsToFile(HashMap<Ref, ServiceResourceStatistic> statsMap,long resetReqTime, String fileName) throws Exception {
		 if (statsMap == null) {
			 System.out.println("\nService statistics map is null...Nothing to save.\n");
		 }
		 
		 if (statsMap.size() == 0) {
			 System.out.println("\nService statistics map is empty... Nothing to save.\n");
		 }
		 FileWriter out = new FileWriter(new File(fileName));
		 out.write("*********************************************\n");
		 out.write("This file contains statistics for " + statsMap.size()+ " services.\n");
		 out.write("***********************************************\n");
		 Set<Map.Entry<Ref, ServiceResourceStatistic>> set = statsMap.entrySet();
		 System.out.println(new StringBuffer().append("\nWriting stats to the file - ").append(fileName).append("\n").toString());
		// Print statistical information of each service
		 for (Map.Entry<Ref, ServiceResourceStatistic> mapEntry : set) {
			 out.write(new StringBuffer().append("\n\n======= Pirnting statistics for service ").append(mapEntry.getKey().getFullName()).append("=======\n").toString());
			 ServiceResourceStatistic serviceStats = mapEntry.getValue();
			 out.write(new StringBuffer().append("Statistic collection time is - ").append(new Date(serviceStats.getCollectionTimestamp())).append("\n").toString());
			 ResourceStatistic[] resStatsArray = null;
			 try {                    
				 resStatsArray = serviceStats.getAllResourceStatistics();
			 }catch (MonitoringNotEnabledException mnee) {
				 // Statistics not available
				 out.write("WARNING: Monitoring is not enabled for  " + "this service... Do something...");
				 out.write("=====================================\n");
				 continue;
			 }catch (InvalidServiceRefException isre) {
				 // Invalid service
				 out.write("ERROR: Invlaid Ref. May be this service is" + "deleted. Do something...");
				 out.write("======================================\n");
				 continue;
			 }catch (MonitoringException me) {
				 // Statistics not available  
				 out.write("ERROR: Failed to get statistics for this service... " + "Details: " + me.getMessage());
				 me.printStackTrace();
				 out.write("======================================\n");
				 continue;
			 }
			 
			 for (ResourceStatistic resStats : resStatsArray) {
				// Print resource information
				 out.write("\nResource name: " + resStats.getName());
				 out.write("\tResource type: " + resStats.getResourceType().toString());
				// Now get and print statistics for this resource
				 StatisticValue[] statValues = resStats.getStatistics();
				 for (StatisticValue value : statValues) {
					 out.write("\n\t\tStatistic Name - " + value.getName ());
					 out.write("\n\t\tStatistic Type - " + value.getType().toString());
					// Determine statistics type
					 if ( value.getType() == StatisticType.INTERVAL ) {
						 StatisticValue.IntervalStatistic is = (StatisticValue.IntervalStatistic)value;
						// Print interval statistics values
						 out.write("\n\t\t\t\tCount Value - " + is.getCount());
						 out.write("\n\t\t\t\tMin Value - " + is.getMin()); 
						 out.write("\n\t\t\t\tMax Value - " + is.getMax()); 
						 out.write("\n\t\t\t\tSum Value - " + is.getSum()); 
						 out.write("\n\t\t\t\tAve Value - " + is.getAverage());
					 }else if ( value.getType() == StatisticType.COUNT ) {
						 StatisticValue.CountStatistic cs = (StatisticValue.CountStatistic)value;
						// Print count statistics value
						 out.write("\n\t\t\t\tCount Value - " + cs.getCount());
					 }else if ( value.getType() == StatisticType.STATUS){
						 StatisticValue.StatusStatistic ss = (StatisticValue.StatusStatistic)value;
						// Print count statistics value
						 out.write("\n\t\t\t\t Initial Status - " + ss.getInitialStatus());
						 out.write("\n\t\t\t\t Current Status - " + ss.getCurrentStatus());
					 }
				 }
			 }
			 out.write("\n=========================================\n");
		 }
		 if (resetReqTime > 0) {               
			 // Save reset request time.
			 out.write("\n*****************************************\n");
			 out.write("Statistics for all these services are RESET.\n");
			 out.write("RESET request time is " + new SimpleDateFormat("MM/dd/yyyy HH:mm:ss").format(new Date(resetReqTime)));
			 out.write("\n****************************************\n");
		 }
		 
		// Flush and close file.
		 out.flush(); 
		 out.close();
	 }
			
	 /**      
	  * Init method.      
	  * @param props Properties required for initialization.      
	  *  @throws Exception      
	  */     
	 private void init(HashMap props) throws Exception {
		 Properties properties = new Properties();
		 properties.putAll(props);
		 getServiceDomainMBean(properties.getProperty("HOSTNAME"),Integer.parseInt(properties.getProperty("PORT", "7001")),properties.getProperty("USERNAME"),properties.getProperty("PASSWORD"));
		 serverName = properties.getProperty("SERVER_NAME");
	 }
	 
	 /**       
	  * Gets an instance of ServiceDomainMBean from the weblogic server.       
	  * @param host       
	  * @param port       
	  * @param username       
	  * @param password       
	  * @throws Exception       
	  */
	 private void getServiceDomainMBean(String host, int port, String username, String password) throws Exception {
		 InvocationHandler handler =  new ServiceDomainMBeanInvocationHandler(host, port, username,password);
		 Object proxy = Proxy.newProxyInstance(ServiceDomainMBean.class.getClassLoader(),new Class[]{ServiceDomainMBean.class}, handler);
		 serviceDomainMbean = (ServiceDomainMBean) proxy;
	 }
	 
	 /**      
	  * Invocation handler class for ServiceDomainMBean class.      
	  */
	 public static class ServiceDomainMBeanInvocationHandler implements InvocationHandler {
		 private String jndiURL ="weblogic.management.mbeanservers.domainruntime";
		 private String mbeanName = ServiceDomainMBean.NAME;
		 private String type = ServiceDomainMBean.TYPE;
		 private String protocol = "t3";
		 private String hostname = "localhost";
		 private int port = 7003;
		 private String jndiRoot = "/jndi/";
		 private String username = "weblogic";
		 private String password = "weblogic";
		 private JMXConnector conn = null; 
		 private Object actualMBean = null;
		 
		 public ServiceDomainMBeanInvocationHandler(String hostName, int port, String userName, String password) {
			 this.hostname = hostName;
			 this.port = port;
			 this.username = userName;
			 this.password = password;
		 }
		 
		 /**            
		  * Gets JMX connection            
		  * @return JMX connection            
		  * @throws IOException            
		  * @throws MalformedURLException            
		  */
		 public JMXConnector initConnection() throws IOException, MalformedURLException {
			 JMXServiceURL serviceURL = new JMXServiceURL(protocol,hostname, port, jndiRoot + jndiURL);
			 Hashtable<String, String> h = new Hashtable<String,String>();
			 if (username != null)h.put(Context.SECURITY_PRINCIPAL, username);
			 if (password != null)h.put(Context.SECURITY_CREDENTIALS, password);
			 h.put(JMXConnectorFactory.PROTOCOL_PROVIDER_PACKAGES,"weblogic.management.remote");
			 return JMXConnectorFactory.connect(serviceURL, h);
		 }
		 
		 /**           
		  * Invokes specified method with specified params on specified            
		  * object.           
		  * @param proxy            
		  * @param method           
		  * @param args            
		  * @return           
		  * @throws Throwable            
		  */
		 public Object invoke(Object proxy, Method method, Object[] args)throws Throwable {
			 try {
				 if (conn == null) { 
					 conn = initConnection(); 
				 }
				 if (actualMBean == null) {
					 actualMBean =  findServiceDomain(conn.getMBeanServerConnection(),mbeanName, type, null);
				 }
				 Object returnValue = method.invoke(actualMBean, args);
				 return returnValue; 
			}catch (Exception e) {
				throw e; 
			}
		 }
		 
		 /**           
		  * Finds the specified MBean object           
		  * @param connection - A connection to the MBeanServer.           
		  * @param mbeanName - The name of the MBean instance.          
		  * @param mbeanType - The type of the MBean.          
		  * @param parent - The name of the parent Service. Can be NULL.            
		  * @return Object - The MBean or null if the MBean was not found.           
		  */
		 
		 public Object findServiceDomain(MBeanServerConnection connection,String mbeanName,String mbeanType,String parent) {
			 ServiceDomainMBean serviceDomainbean = null;
			 try {
				 ObjectName on = new ObjectName(ServiceDomainMBean.OBJECT_NAME);
				 serviceDomainbean = (ServiceDomainMBean)MBeanServerInvocationHandler.newProxyInstance(connection, on);
			 }catch (MalformedObjectNameException e) {
				 e.printStackTrace();
				 return null;
		     }
			 return serviceDomainbean;
		 }
		 
	 }
	 
	 /**      Timer task to keep retrieving and resetting service statistics.      
	  */
	 static class GetAndResetStatisticsTask extends TimerTask {
		 private ServiceStatisticsRetriever_file collector;
		 public GetAndResetStatisticsTask(ServiceStatisticsRetriever_file col){collector = col;}
		 public void run() {
			 System.out.println("\n**********************************");
			 System.out.println("Retrieving statistics for all monitored" + "business services.");
			 try {
				 collector.getAndResetStatsForAllMonitoredBizServices();
				 System.out.println("Successfully retrieved and reset statistics for " + "all monitored \n business services at " + new SimpleDateFormat("MM/dd/yyyy HH:mm:ss").format(new Date(System.currentTimeMillis())));
			 } catch (Exception e) {
				 System.out.println("Failed to retrieve and reset statistics for all " + "monitored business service...");
				 e.printStackTrace();
			 }
			 System.out.println("**********************************\n");
			 System.out.println("\n**********************************");
			 System.out.println("Retrieving statistics for all monitored proxy services.");
			 try {
				 collector.getAndResetStatsForAllMonitoredProxyServices();
				 System.out.println("Successfully retrieved and reset statistics " + "for all monitored \nproxy services at " + new SimpleDateFormat("MM/dd/yyyy HH:mm:ss").format(new Date(System.currentTimeMillis())));
			 } catch (Exception e) { 
				 System.out.println("Failed to retrieve and reset statistics " + "for all monitored proxy service...");
				 e.printStackTrace();
			 }
			 System.out.println("*********************************\n");
		 }
	 }


	 /**        
	  * The main method to start the timer task to extract, save, and reset       
	  * statistics for all monitored business and proxy services. It uses      
	  * the following system properties.       
	  * 1. hostname - Hostname of admin server       
	  * 2. port - Listening port of admin server       
	  * 3. username - Login username       
	  * 4. password - Login password       
	  * 5. period - Frequency in hours. This will be used by the timer       
	  * to determine the time gap between two executions.       
	  * @param args Not used.       
	  */
	public static void main(String[] args) {
		try {
			Properties p = System.getProperties();
			HashMap map = new HashMap();
			map.put("HOSTNAME", p.getProperty("hostname","localhost"));
			map.put("PORT", p.getProperty("port", "7003"));
			map.put("USERNAME", p.getProperty("username", "weblogic"));
			map.put("PASSWORD", p.getProperty("password", "weblogic"));
			//set a server name if you want to get the uri status statistics in a cluster
			map.put("SERVER_NAME", p.getProperty("server_name","AdminServer"));
			ServiceStatisticsRetriever_file collector =  new ServiceStatisticsRetriever_file();
			String periodStr = p.getProperty("period", "1");
			int periodInHour = Integer.parseInt(periodStr);
			long periodInMilliSec = periodInHour * 60 * 1000;
			collector.init(map);
			// Start timer.
			Timer timer = new Timer();
			timer.scheduleAtFixedRate(new GetAndResetStatisticsTask(collector),0, periodInMilliSec);
		}catch (Exception e) {
			e.printStackTrace();
		}
		
	}

}