it.avalz.opendaylight.controller.Controller.java Source code

Introduction

Here is the source code for it.avalz.opendaylight.controller.Controller.java

Source

/*
 * Copyright (c) 2015 Andrea Valenza <avalenza89@gmail.com>.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 */
package it.avalz.opendaylight.controller;

import it.avalz.graph.Vertex;
import it.avalz.graph.exceptions.NoLinkException;
import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLConnection;
import java.util.Iterator;
import java.util.List;
import org.apache.commons.codec.binary.Base64;

/**
 *
 * @author Andrea Valenza <avalenza89@gmail.com>
 */
public class Controller {

    private String IP;
    private final String username;
    private final String password;
    private String baseUrl;

    public Controller(String IP, String username, String password) {
        this.IP = IP;
        this.username = username;
        this.password = password;
        this.baseUrl = "http://" + IP + ":8080/controller/nb/v2";
    }

    public Controller(String IP) {
        this(IP, "admin", "admin");
    }

    public Controller() {
        this("127.0.0.1", "admin", "admin");
    }

    public String getRawEdges() {
        return callControllerApi("/topology/default");
    }

    public String getRawNodes() {
        return callControllerApi("/switchmanager/default/nodes");
    }

    private String callControllerApi(String partialUrl) {
        String authString = this.username + ":" + this.password;

        StringBuilder responseString = new StringBuilder();
        try {
            URL url = new URL(this.baseUrl + partialUrl);
            byte[] authEncoded = Base64.encodeBase64(authString.getBytes());
            String authEncodedString = new String(authEncoded);

            URLConnection connection = url.openConnection();
            connection.setRequestProperty("Authorization", "Basic " + authEncodedString);
            connection.setRequestProperty("Content-Type", "application/json");
            connection.setRequestProperty("Accept", "application/json");

            InputStream is = connection.getInputStream();

            BufferedReader br = new BufferedReader(new InputStreamReader(is));
            String line;
            while ((line = br.readLine()) != null) {
                responseString.append(line);
            }
            br.close();

        } catch (Exception e) {
            e.printStackTrace();
        }

        return responseString.toString();
    }

    /**
     * Factory method for creating a graph from the controller object.
     *
     * @return
     */
    public Network newNetwork() {
        Network n = new Network();
        n.addNodesFromJSON(this.getRawNodes());
        n.addEdgesFromJSON(this.getRawEdges());

        return n;
    }

    public void addFlow(Flow f) {
        Vertex v = f.getNode();

        String authString = this.username + ":" + this.password;

        try {
            StringBuilder urlString = new StringBuilder();

            urlString.append(this.baseUrl).append("/flowprogrammer/default/node/OF/").append(v.getId())
                    .append("/staticFlow/").append(f.getName());
            URL url = new URL(urlString.toString());
            byte[] authEncoded = Base64.encodeBase64(authString.getBytes());
            String authEncodedString = new String(authEncoded);

            HttpURLConnection connection = (HttpURLConnection) url.openConnection();

            connection.setDoOutput(true);
            connection.setRequestMethod("PUT");

            connection.setRequestProperty("Authorization", "Basic " + authEncodedString);
            connection.setRequestProperty("Content-Type", "application/json");
            connection.setRequestProperty("Accept", "application/json");

            OutputStreamWriter out = new OutputStreamWriter(connection.getOutputStream());

            out.write(f.toString());
            out.close();

            connection.getInputStream();
        } catch (Exception e) {
            e.printStackTrace();
        }

        System.out.println("Flow installed on vertex " + v);
        System.out.println(f);
    }

    public void installPath(List<Vertex> sp, int ingressPort, int outputPort, String destination)
            throws NoLinkException {

        Iterator<Vertex> i = sp.listIterator();

        // Setup
        Vertex prev = null;
        Vertex v = i.next();
        Vertex next = i.next();
        Flow f = null;

        int prevPort;
        int nextPort;

        // First step
        f = new Flow(v, destination);
        // prevPort = v.getIncomingPortTo(prev); Not the first step.
        prevPort = ingressPort;
        nextPort = v.getPortTo(next);

        f.setIngressPort(prevPort);
        f.addOutputAction(nextPort);

        this.addFlow(f);

        // Intermediate steps
        while (i.hasNext()) {
            // Rotating values one step
            prev = v;
            v = next;
            next = i.next();

            // Creating a flow for the "v" vertex, with ingressPort and outputPort
            f = new Flow(v, destination);

            prevPort = v.getIncomingPortTo(prev);
            nextPort = v.getPortTo(next);

            f.setIngressPort(prevPort);
            f.addOutputAction(nextPort);

            this.addFlow(f);
        }

        // Last step
        prev = v;
        v = next;
        // next = null; Not needed

        f = new Flow(v, destination);
        prevPort = v.getIncomingPortTo(prev);
        nextPort = outputPort;

        f.setIngressPort(prevPort);
        f.addOutputAction(nextPort);
        this.addFlow(f);
    }

    public void installPath(List<Vertex> sp, int ingressPort, int outputPort) throws NoLinkException {
        this.installPath(sp, ingressPort, outputPort, null);
    }

    /**
     * @return the IP
     */
    public String getIP() {
        return IP;
    }

    /**
     * @param IP the IP to set
     */
    public void setIP(String IP) {
        this.IP = IP;
    }
}