GraphController.java Source code

Introduction

Here is the source code for GraphController.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 graph 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 GraphController {

    private final AtomicLong counter = new AtomicLong();

    //---------------This Part starts the Zabbix connection---------------
    @RequestMapping("/graphs")
    public @ResponseBody Graph graph(
            @RequestParam(value = "authentication", required = false, defaultValue = "Error") String authentication,
            @RequestParam(value = "hostid", required = false, defaultValue = "") String hostid)
            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();
        JSONArray list = new JSONArray();
        jsonObj.put("jsonrpc", "2.0");
        jsonObj.put("method", "graph.get");
        JSONObject params = new JSONObject();
        params.put("output", "extend");
        params.put("hostids", hostid);
        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 = "";

        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);

                JSONObject objret = new JSONObject();

                objret.put("graphId", tobj.get("graphid"));
                objret.put("graphName", tobj.get("name"));

                String type = (String) tobj.get("graphtype");

                if (type.equals("0")) {
                    objret.put("graphType", "normal");
                } else if (type.equals("1")) {
                    objret.put("graphType", "stacked");
                } else if (type.equals("2")) {
                    objret.put("graphType", "pie");
                } else if (type.equals("3")) {
                    objret.put("graphType", "exploded");
                }

                list.add(objret);

            }

            return new Graph(list);

        } catch (HttpException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ParseException pe) {
            pe.printStackTrace();
        }
        return new Graph(
                "Error: please provide the appropriate input parameters of the metric you are looking for its corresponding monitoring data:");
    }
    //--------------This part ends the Zabbix connection------------------
}