View Javadoc

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.chukwa.datacollection.agent.metrics;
19  
20  import org.apache.hadoop.metrics.MetricsContext;
21  import org.apache.hadoop.metrics.MetricsRecord;
22  import org.apache.hadoop.metrics.MetricsUtil;
23  import org.apache.hadoop.metrics.Updater;
24  import org.apache.hadoop.metrics.util.MetricsBase;
25  import org.apache.hadoop.metrics.util.MetricsIntValue;
26  import org.apache.hadoop.metrics.util.MetricsRegistry;
27  import org.apache.hadoop.metrics.util.MetricsTimeVaryingInt;
28  
29  public class AgentMetrics implements Updater {
30    public static final AgentMetrics agentMetrics = new AgentMetrics("chukwaAgent", "metrics");
31    
32    public MetricsRegistry registry = new MetricsRegistry();
33    private MetricsRecord metricsRecord;
34    private AgentActivityMBean agentActivityMBean;
35  
36    public MetricsIntValue adaptorCount =
37      new MetricsIntValue("adaptorCount", registry,"number of new adaptor");
38  
39    public MetricsTimeVaryingInt addedAdaptor =
40      new MetricsTimeVaryingInt("addedAdaptor", registry,"number of added adaptor");
41    
42    public MetricsTimeVaryingInt removedAdaptor =
43      new MetricsTimeVaryingInt("removedAdaptor", registry,"number of removed adaptor");
44    
45    /** Creates a new instance of AgentMetrics 
46     * @param processName 
47     * @param recordName */
48    public AgentMetrics(String processName, String recordName) {
49        MetricsContext context = MetricsUtil.getContext(processName);
50        metricsRecord = MetricsUtil.createRecord(context, recordName);
51        metricsRecord.setTag("process", processName);
52        agentActivityMBean = new AgentActivityMBean(registry, recordName);
53        context.registerUpdater(this);
54        
55    }
56  
57  
58    /**
59     * Since this object is a registered updater, this method will be called
60     * periodically, e.g. every 5 seconds.
61     */
62    public void doUpdates(MetricsContext unused) {
63      synchronized (this) {
64        for (MetricsBase m : registry.getMetricsList()) {
65          m.pushMetric(metricsRecord);
66        }
67      }
68      metricsRecord.update();
69    }
70  
71    public void shutdown() {
72      if (agentActivityMBean != null)
73        agentActivityMBean.shutdown();
74    }
75  
76  }