RpcMetrics.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:4k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.ipc.metrics;
  19. import org.apache.commons.logging.Log;
  20. import org.apache.commons.logging.LogFactory;
  21. import org.apache.hadoop.ipc.Server;
  22. import org.apache.hadoop.metrics.MetricsContext;
  23. import org.apache.hadoop.metrics.MetricsRecord;
  24. import org.apache.hadoop.metrics.MetricsUtil;
  25. import org.apache.hadoop.metrics.Updater;
  26. import org.apache.hadoop.metrics.util.MetricsBase;
  27. import org.apache.hadoop.metrics.util.MetricsIntValue;
  28. import org.apache.hadoop.metrics.util.MetricsRegistry;
  29. import org.apache.hadoop.metrics.util.MetricsTimeVaryingRate;
  30. /**
  31.  * 
  32.  * This class is for maintaining  the various RPC statistics
  33.  * and publishing them through the metrics interfaces.
  34.  * This also registers the JMX MBean for RPC.
  35.  * <p>
  36.  * This class has a number of metrics variables that are publicly accessible;
  37.  * these variables (objects) have methods to update their values;
  38.  * for example:
  39.  *  <p> {@link #rpcQueueTime}.inc(time)
  40.  *
  41.  */
  42. public class RpcMetrics implements Updater {
  43.   public MetricsRegistry registry = new MetricsRegistry();
  44.   private MetricsRecord metricsRecord;
  45.   private Server myServer;
  46.   private static Log LOG = LogFactory.getLog(RpcMetrics.class);
  47.   RpcActivityMBean rpcMBean;
  48.   
  49.   public RpcMetrics(String hostName, String port, Server server) {
  50.     myServer = server;
  51.     MetricsContext context = MetricsUtil.getContext("rpc");
  52.     metricsRecord = MetricsUtil.createRecord(context, "metrics");
  53.     metricsRecord.setTag("port", port);
  54.     LOG.info("Initializing RPC Metrics with hostName=" 
  55.         + hostName + ", port=" + port);
  56.     context.registerUpdater(this);
  57.     
  58.     // Need to clean up the interface to RpcMgt - don't need both metrics and server params
  59.     rpcMBean = new RpcActivityMBean(registry, hostName, port);
  60.   }
  61.   
  62.   
  63.   /**
  64.    * The metrics variables are public:
  65.    *  - they can be set directly by calling their set/inc methods
  66.    *  -they can also be read directly - e.g. JMX does this.
  67.    */
  68.   public MetricsTimeVaryingRate rpcQueueTime =
  69.           new MetricsTimeVaryingRate("RpcQueueTime", registry);
  70.   public MetricsTimeVaryingRate rpcProcessingTime =
  71.           new MetricsTimeVaryingRate("RpcProcessingTime", registry);
  72.   public MetricsIntValue numOpenConnections = 
  73.           new MetricsIntValue("NumOpenConnections", registry);
  74.   public MetricsIntValue callQueueLen = 
  75.           new MetricsIntValue("callQueueLen", registry);
  76.   
  77.   /**
  78.    * Push the metrics to the monitoring subsystem on doUpdate() call.
  79.    */
  80.   public void doUpdates(MetricsContext context) {
  81.     
  82.     synchronized (this) {
  83.       // ToFix - fix server to use the following two metrics directly so
  84.       // the metrics do not have be copied here.
  85.       numOpenConnections.set(myServer.getNumOpenConnections());
  86.       callQueueLen.set(myServer.getCallQueueLen());
  87.       for (MetricsBase m : registry.getMetricsList()) {
  88.         m.pushMetric(metricsRecord);
  89.       }
  90.     }
  91.     metricsRecord.update();
  92.   }
  93.   public void shutdown() {
  94.     if (rpcMBean != null) 
  95.       rpcMBean.shutdown();
  96.   }
  97. }