edu.jhu.cvrg.timeseriesstore.opentsdb.AnnotationManager.java Source code

Introduction

Here is the source code for edu.jhu.cvrg.timeseriesstore.opentsdb.AnnotationManager.java

Source

package edu.jhu.cvrg.timeseriesstore.opentsdb;

/*
Copyright 2015 Johns Hopkins University Institute for Computational Medicine
    
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.
*/
/**
* @author Chris Jurado, helper:Mackenzie Jurado
* 
*/

import java.io.IOException;
import java.io.OutputStreamWriter;
import java.net.HttpURLConnection;

import org.json.JSONException;
import org.json.JSONObject;

import edu.jhu.cvrg.timeseriesstore.exceptions.OpenTSDBException;

public class AnnotationManager {

    public static final String API_METHOD = "/api/annotation";

    public static String createSinglePointAnnotation(String urlString, long startEpoch, String tsuid,
            String description, String notes) {
        return createIntervalAnnotation(urlString, startEpoch, 0L, tsuid, description, notes);
    }

    public static String createIntervalAnnotation(String urlString, long startEpoch, long endEpoch, String tsuid,
            String description, String notes) {
        urlString = urlString + API_METHOD;
        String result = "";
        try {
            HttpURLConnection httpConnection = TimeSeriesUtility.openHTTPConnectionPOST(urlString);
            OutputStreamWriter wr = new OutputStreamWriter(httpConnection.getOutputStream());
            JSONObject requestObject = new JSONObject();
            requestObject.put("startTime", startEpoch);
            requestObject.put("endTime", endEpoch);
            requestObject.put("tsuid", tsuid);
            requestObject.put("description", description);
            requestObject.put("notes", notes);
            wr.write(requestObject.toString());
            wr.close();
            result = TimeSeriesUtility.readHttpResponse(httpConnection);
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        } catch (OpenTSDBException e) {
            e.printStackTrace();
            result = String.valueOf(e.responseCode);
        } catch (JSONException e) {
            e.printStackTrace();
        }
        return result;
    }

    public static JSONObject queryAnnotation(String urlString, long startEpoch, String tsuid) {

        urlString = urlString + API_METHOD;
        String result = "";
        JSONObject resultObject = null;

        StringBuilder builder = new StringBuilder();

        builder.append("?start_time=");
        builder.append(startEpoch);
        builder.append("&tsuid=");
        builder.append(tsuid);

        try {
            HttpURLConnection httpConnection = TimeSeriesUtility
                    .openHTTPConnectionGET(urlString + builder.toString());
            result = TimeSeriesUtility.readHttpResponse(httpConnection);
        } catch (OpenTSDBException e) {
            result = String.valueOf(e.responseCode);
        } catch (IOException e) {
            e.printStackTrace();
        }

        try {
            if (result.equals("404")) {
                resultObject = new JSONObject();
                resultObject.put("code", result);
            } else {
                resultObject = new JSONObject(result);
            }
        } catch (JSONException e) {
            e.printStackTrace();
        }

        return resultObject;
    }

    public static String editAnnotation(String urlString, long startEpoch, long endEpoch, String tsuid,
            String description, String notes) {
        JSONObject annotation = queryAnnotation(urlString, startEpoch, tsuid);

        String descriptionOld = "";
        String notesOld = "";
        try {
            descriptionOld = annotation.getString("description");
            notesOld = annotation.getString("notes");
        } catch (JSONException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        description = (description.equals("")) ? descriptionOld : description;
        notes = (notes.equals("")) ? notesOld : notes;

        return createIntervalAnnotation(urlString, startEpoch, endEpoch, tsuid, description, notes);
    }

    public static String deleteAnnotation(String urlString, long startEpoch, String tsuid) {
        urlString = urlString + API_METHOD;
        String result = "";
        StringBuilder builder = new StringBuilder();
        builder.append("?start_time=");
        builder.append(startEpoch);
        builder.append("&tsuid=");
        builder.append(tsuid);
        try {
            HttpURLConnection httpConnection = TimeSeriesUtility
                    .openHTTPConnectionDELETE(urlString + builder.toString());
            result = TimeSeriesUtility.readHttpResponse(httpConnection);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (OpenTSDBException e) {
            result = String.valueOf(e.responseCode);
        }
        return result;
    }
}