net.decix.jatlasx.ripe.atlas.api.handler.ProbeListHandler.java Source code

Introduction

Here is the source code for net.decix.jatlasx.ripe.atlas.api.handler.ProbeListHandler.java

Source

/*
 * This file is part of jAtlasX
 *
 * Copyright (c) 2015 DE-CIX Management GmbH <http://www.de-cix.net> - All rights
 * reserved.
 * 
 * @author: Sascha Bleidner <sascha.bleidner@de-cix.net>
 *
 * This software is licensed under the Apache License, version 2.0. A copy of 
 * the license agreement is included in this distribution.
 */
package net.decix.jatlasx.ripe.atlas.api.handler;

import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.List;

import org.json.JSONArray;
import org.json.JSONObject;

import net.decix.jatlasx.data.IpAddress;
import net.decix.jatlasx.data.Probe;
import net.decix.jatlasx.ripe.atlas.api.ApiKeys;

public class ProbeListHandler implements ResponseHandler<Probe> {

    private long asn;

    public ProbeListHandler(long asNumber) {
        asn = asNumber;
    }

    public List<Probe> handleResponse(String jsonString) {

        List<Probe> probes = new ArrayList<Probe>();

        JSONObject jsonObj = new JSONObject(jsonString);
        JSONArray probeListArray = (JSONArray) jsonObj.get("objects");

        for (int i = 0; i < probeListArray.length(); i++) {
            JSONObject probeObj = (JSONObject) probeListArray.get(i);
            // a probe has to have an IPv4 address
            if (!probeObj.get(ApiKeys.ipv4).equals(null)) {
                String ipAddress = String.valueOf(probeObj.get(ApiKeys.ipv4));
                Long id = (Long) probeObj.getLong(ApiKeys.id);
                boolean isPublic = (Boolean) probeObj.get(ApiKeys.isPublic);
                Long asNum = 0L;
                if (!probeObj.get(ApiKeys.asn).equals(null)) {
                    asNum = (Long) probeObj.getLong(ApiKeys.asn);
                } else {
                    // Ignore probe if asNum is not specified
                    continue;
                }

                String status = (String) probeObj.get(ApiKeys.status);
                // only select probes which provide an IPv4 address
                if (ipAddress != null && asn == asNum && status.equals("Connected") && isPublic) {
                    Probe newProbe = new Probe(ApiKeys.probes, id, 1);
                    newProbe.setAsn(asNum);
                    try {
                        newProbe.setIpAddress(new IpAddress(ipAddress));
                    } catch (UnknownHostException e) {
                        String errorMsg = "Could not create IP-address:" + ipAddress;
                        System.err.println(
                                e.getClass().getName() + ":" + errorMsg + "(" + this.getClass().getName() + ")");
                    }
                    probes.add(newProbe);
                }
            }
        }
        return probes;
    }
}