src/main/java/org/apache/hadoop/chukwa/extraction/hbase/SystemMetrics.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/**
 * Demux parser for system metrics data collected through
 * org.apache.hadoop.chukwa.datacollection.adaptor.sigar.SystemMetrics.
 */
package org.apache.hadoop.chukwa.extraction.hbase;

import java.nio.charset.Charset;
import java.security.NoSuchAlgorithmException;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import org.apache.hadoop.chukwa.extraction.engine.ChukwaRecord;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.json.simple.JSONValue;

public class SystemMetrics extends AbstractProcessor {

  public SystemMetrics() throws NoSuchAlgorithmException {
    super();
  }

  @Override
  protected void parse(byte[] recordEntry) throws Throwable {
    String buffer = new String(recordEntry, Charset.forName("UTF-8"));
    JSONObject json = (JSONObject) JSONValue.parse(buffer);
    time = ((Long) json.get("timestamp")).longValue();
    ChukwaRecord record = new ChukwaRecord();
    JSONArray cpuList = (JSONArray) json.get("cpu");
    double combined = 0.0;
    double user = 0.0;
    double sys = 0.0;
    double idle = 0.0;
    int actualSize = 0;
    for (int i = 0; i < cpuList.size(); i++) {
      JSONObject cpu = (JSONObject) cpuList.get(i);
      // Work around for sigar returning null sometimes for cpu metrics on
      // pLinux
      if (cpu.get("combined") == null) {
        continue;
      }
      actualSize++;
      combined = combined + Double.parseDouble(cpu.get("combined").toString());
      user = user + Double.parseDouble(cpu.get("user").toString());
      sys = sys + Double.parseDouble(cpu.get("sys").toString());
      idle = idle + Double.parseDouble(cpu.get("idle").toString());
      for(Entry<String, Object> entry : (Set<Map.Entry>) cpu.entrySet()) {
        String key = entry.getKey();
        addRecord("cpu." + key + "." + i, String.valueOf(entry.getValue()));
      }
    }
    combined = combined / actualSize;
    user = user / actualSize;
    sys = sys / actualSize;
    idle = idle / actualSize;
    addRecord("cpu.combined", Double.toString(combined));
    addRecord("cpu.user", Double.toString(user));
    addRecord("cpu.idle", Double.toString(idle));
    addRecord("cpu.sys", Double.toString(sys));

    addRecord("Uptime", json.get("uptime").toString());
    JSONArray loadavg = (JSONArray) json.get("loadavg");
    addRecord("LoadAverage.1", loadavg.get(0).toString());
    addRecord("LoadAverage.5", loadavg.get(1).toString());
    addRecord("LoadAverage.15", loadavg.get(2).toString());

    record = new ChukwaRecord();
    JSONObject memory = (JSONObject) json.get("memory");
    for(Entry<String, Object> entry : (Set<Map.Entry>) memory.entrySet()) {
      String key = entry.getKey();
      addRecord("memory." + key, String.valueOf(entry.getValue()));
    }

    record = new ChukwaRecord();
    JSONObject swap = (JSONObject) json.get("swap");
    for(Map.Entry<String, Object> entry : (Set<Map.Entry>) swap.entrySet()) {
      addRecord("swap." + entry.getKey(), String.valueOf(entry.getValue()));
    }

    double rxBytes = 0;
    double rxDropped = 0;
    double rxErrors = 0;
    double rxPackets = 0;
    double txBytes = 0;
    double txCollisions = 0;
    double txErrors = 0;
    double txPackets = 0;
    record = new ChukwaRecord();
    JSONArray netList = (JSONArray) json.get("network");
    for (int i = 0; i < netList.size(); i++) {
      JSONObject netIf = (JSONObject) netList.get(i);
      for(Map.Entry<String, Object> entry : (Set<Map.Entry>) netIf.entrySet()) {
        String key = entry.getKey();
        long value = 0;
        if(entry.getValue() instanceof Long) {
          value = (Long) entry.getValue();
        }
        record.add(key + "." + i, String.valueOf(entry.getValue()));
        if (i != 0) {
          if (key.equals("RxBytes")) {
            rxBytes = rxBytes + value;
          } else if (key.equals("RxDropped")) {
            rxDropped = rxDropped + value;
          } else if (key.equals("RxErrors")) {
            rxErrors = rxErrors + value;
          } else if (key.equals("RxPackets")) {
            rxPackets = rxPackets + value;
          } else if (key.equals("TxBytes")) {
            txBytes = txBytes + value;
          } else if (key.equals("TxCollisions")) {
            txCollisions = txCollisions + value;
          } else if (key.equals("TxErrors")) {
            txErrors = txErrors + value;
          } else if (key.equals("TxPackets")) {
            txPackets = txPackets + value;
          }
        }
      }
    }

    addRecord("network.RxBytes", Double.toString(rxBytes));
    addRecord("network.RxDropped", Double.toString(rxDropped));
    addRecord("network.RxErrors", Double.toString(rxErrors));
    addRecord("network.RxPackets", Double.toString(rxPackets));
    addRecord("network.TxBytes", Double.toString(txBytes));
    addRecord("network.TxCollisions", Double.toString(txCollisions));
    addRecord("network.TxErrors", Double.toString(txErrors));
    addRecord("network.TxPackets", Double.toString(txPackets));

    double readBytes = 0;
    double reads = 0;
    double writeBytes = 0;
    double writes = 0;
    double total = 0;
    double used = 0;
    record = new ChukwaRecord();
    JSONArray diskList = (JSONArray) json.get("disk");
    for (int i = 0; i < diskList.size(); i++) {
      JSONObject disk = (JSONObject) diskList.get(i);
      for(Entry<String, Object> entry : (Set<Map.Entry>) disk.entrySet()) {
        String key = entry.getKey();
        long value = 0;
        if(entry.getValue() instanceof Long) {
          value = (Long) entry.getValue();
        }
        record.add(key + "." + i, String.valueOf(entry.getValue()));
        if (key.equals("ReadBytes")) {
          readBytes = readBytes + value;
        } else if (key.equals("Reads")) {
          reads = reads + Long.valueOf(value);;
        } else if (key.equals("WriteBytes")) {
          writeBytes = writeBytes + value;
        } else if (key.equals("Writes")) {
          writes = writes + value;
        } else if (key.equals("Total")) {
          total = total + value;
        } else if (key.equals("Used")) {
          used = used + value;
        }
      }
    }
    double percentUsed = used / total;
    addRecord("disk.ReadBytes", Double.toString(readBytes));
    addRecord("disk.Reads", Double.toString(reads));
    addRecord("disk.WriteBytes", Double.toString(writeBytes));
    addRecord("disk.Writes", Double.toString(writes));
    addRecord("disk.Total", Double.toString(total));
    addRecord("disk.Used", Double.toString(used));
    addRecord("disk.PercentUsed", Double.toString(percentUsed));
  }

}