001 /** 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, software 013 * distributed under the License is distributed on an "AS IS" BASIS, 014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 015 * See the License for the specific language governing permissions and 016 * limitations under the License. 017 */ 018 019 package org.apache.oozie.action.hadoop; 020 021 import org.apache.hadoop.mapred.Counters; 022 import org.apache.hadoop.mapred.Counters.Counter; 023 import org.json.simple.JSONObject; 024 025 /** 026 * Class to collect statistics for Map-Reduce action. 027 */ 028 public class MRStats extends ActionStats { 029 public static final String ACTION_TYPE_LABEL = "ACTION_TYPE"; 030 private Counters counters = null; 031 032 public MRStats(Counters groups) { 033 this.currentActionType = ActionType.MAP_REDUCE; 034 this.counters = groups; 035 } 036 037 @SuppressWarnings("unchecked") 038 @Override 039 public String toJSON() { 040 if (counters == null) { 041 return null; 042 } 043 044 JSONObject groups = new JSONObject(); 045 groups.put(ACTION_TYPE_LABEL, getCurrentActionType().toString()); 046 for (String gName : counters.getGroupNames()) { 047 JSONObject group = new JSONObject(); 048 for (Counter counter : counters.getGroup(gName)) { 049 String cName = counter.getName(); 050 Long cValue = counter.getValue(); 051 group.put(cName, cValue); 052 } 053 groups.put(gName, group); 054 } 055 return groups.toJSONString(); 056 } 057 }