CPUController.java Source code

Java tutorial

Introduction

Here is the source code for CPUController.java

Source

/* 
************************************************************************
    
# RESTful API for Zabbix based on Java.
#
#  Copyright 2014 Yahya Al-Hazmi, TU Berlin.
# Authors: Mingyuan Wu and Yahya Al-Hazmi
# Licensed 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
#
# This API allows you to request monitoring information from Zabbix-Server 
# through Zabbix-API
    
 ************************************************************************
    
 This class is aimed to achieve the goal of getting cpu metrics from 
 zabbix.
     
 ************************************************************************
*/

import java.util.concurrent.atomic.AtomicLong;

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;

import org.json.simple.JSONObject;
import org.json.simple.JSONArray;
import org.json.simple.parser.ParseException;
import org.json.simple.parser.JSONParser;

import java.util.Properties;
import java.io.FileInputStream;

import java.io.ByteArrayInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;

import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpException;
import org.apache.commons.httpclient.methods.PutMethod;

@Controller
public class CPUController {

    private final AtomicLong counter = new AtomicLong();

    //---------------This Part starts the Zabbix connection---------------
    private static String ZABBIX_API_URL = "http://193.175.132.250/zabbix/api_jsonrpc.php"; // zabbix_server

    @RequestMapping("/cpu")
    public @ResponseBody CPU cpu(
            @RequestParam(value = "authentication", required = false, defaultValue = "Error") String authentication,
            @RequestParam(value = "hostid", required = false, defaultValue = "") String hostid,
            @RequestParam(value = "metricType", required = false, defaultValue = "") String name)
            throws FileNotFoundException, UnsupportedEncodingException, IOException {
        Properties props = new Properties();
        FileInputStream fis = new FileInputStream("properties.xml");
        //loading properites from properties file
        props.loadFromXML(fis);

        String server_ip = props.getProperty("server_ip");
        String ZABBIX_API_URL = "http://" + server_ip + "/api_jsonrpc.php"; // 1.2.3.4 is your zabbix_server_ip

        JSONParser parser = new JSONParser();
        HttpClient client = new HttpClient();

        PutMethod putMethod = new PutMethod(ZABBIX_API_URL);
        putMethod.setRequestHeader("Content-Type", "application/json-rpc"); // content-type is controlled in api_jsonrpc.php, so set it like this

        // create json object for apiinfo.version 
        JSONObject jsonObj = new JSONObject();
        jsonObj.put("jsonrpc", "2.0");
        jsonObj.put("method", "item.get");
        JSONObject params = new JSONObject();
        params.put("output", "extend");
        params.put("hostid", hostid);
        JSONObject search = new JSONObject();
        search.put("key_", "cpu");
        params.put("search", search);
        params.put("sortfield", "name");
        jsonObj.put("params", params);
        jsonObj.put("auth", authentication);// todo
        jsonObj.put("id", new Integer(1));

        putMethod.setRequestBody(jsonObj.toString()); // put the json object as input stream into request body 

        String loginResponse = "";
        String cpu = "";
        String clock = "";
        String metricType = "";

        try {
            client.executeMethod(putMethod); // send to request to the zabbix api

            loginResponse = putMethod.getResponseBodyAsString(); // read the result of the response
            Object obj = parser.parse(loginResponse);
            JSONObject obj2 = (JSONObject) obj;
            String jsonrpc = (String) obj2.get("jsonrpc");
            JSONArray array = (JSONArray) obj2.get("result");

            //         System.out.println(array);

            for (int i = 0; i < array.size(); i++) {
                JSONObject tobj = (JSONObject) array.get(i);
                String key = (String) tobj.get("key_");
                if (!tobj.get("hostid").equals(hostid))
                    continue;
                if (name.equals("idle") && key.contains("idle")) {
                    cpu = (String) tobj.get("lastvalue");
                    clock = (String) tobj.get("lastclock");
                    metricType = "cpu idle time";
                } else if (name.equals("iowait") && key.contains("iowait")) {
                    cpu = (String) tobj.get("lastvalue");
                    clock = (String) tobj.get("lastclock");
                    metricType = "cpu iowait time";
                } else if (name.equals("nice") && key.contains("nice")) {
                    cpu = (String) tobj.get("lastvalue");
                    clock = (String) tobj.get("lastclock");
                    metricType = "cpu nice time";
                } else if (name.equals("system") && key.contains("system")) {
                    cpu = (String) tobj.get("lastvalue");
                    clock = (String) tobj.get("lastclock");
                    metricType = "cpu system time";
                } else if (name.equals("user") && key.contains("user")) {
                    cpu = (String) tobj.get("lastvalue");
                    clock = (String) tobj.get("lastclock");
                    metricType = "cpu user time";
                } else if (name.equals("load") && key.contains("load")) {
                    cpu = (String) tobj.get("lastvalue");
                    clock = (String) tobj.get("lastclock");
                    metricType = "processor load";
                } else if (name.equals("usage") && key.contains("usage")) {
                    cpu = (String) tobj.get("lastvalue");
                    clock = (String) tobj.get("lastclock");
                    metricType = "system cpu usage average";
                }
            }
        } catch (HttpException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ParseException pe) {
            System.out.println("Error");
        }

        if (cpu.equals("")) {
            return new CPU(
                    "Error: Please provide the appropriate input parameters of the metric you are looking for its corresponding monitoring data:");
        }

        return new CPU(hostid, metricType, cpu, clock);
    }
    //--------------This part ends the Zabbix connection------------------
}