/*
    * 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));
   }
 
 }