src/main/java/net/onrc/onos/ofcontroller/devicemanager/internal/DeviceStorageImpl.java

package net.onrc.onos.ofcontroller.devicemanager.internal;

import java.util.List;
import org.openflow.util.HexString;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.collect.Lists;
import com.thinkaurelius.titan.core.TitanException;
import net.floodlightcontroller.devicemanager.IDevice;
import net.floodlightcontroller.devicemanager.SwitchPort;
import net.onrc.onos.ofcontroller.core.INetMapTopologyObjects.IDeviceObject;
import net.onrc.onos.ofcontroller.core.INetMapTopologyObjects.IPortObject;
import net.onrc.onos.ofcontroller.core.internal.SwitchStorageImpl;
import net.onrc.onos.ofcontroller.devicemanager.IDeviceStorage;
import net.onrc.onos.util.GraphDBConnection;
import net.onrc.onos.util.GraphDBConnection.Transaction;

public class DeviceStorageImpl implements IDeviceStorage {
	
	public GraphDBConnection conn;
	protected static Logger log = LoggerFactory.getLogger(SwitchStorageImpl.class);

	@Override
	public void init(String conf) {
		conn = GraphDBConnection.getInstance(conf);
	}	

	public void finalize() {		
		close();
	}
	
	@Override
	public void close() {
		conn.close();
	}

	@Override
	public IDeviceObject addDevice(IDevice device) {
		// TODO Auto-generated method stub
		IDeviceObject obj = null;
 		try {
            if ((obj = conn.utils().searchDevice(conn, device.getMACAddressString())) != null) {
                log.debug("Adding device {}: found existing device",device.getMACAddressString());
            } else {
            	obj = conn.utils().newDevice(conn);
                log.debug("Adding device {}: creating new device",device.getMACAddressString());
            }
            changeDeviceAttachments(device, obj);
            
 			obj.setIPAddress(device.getIPv4Addresses().toString());
 			obj.setMACAddress(device.getMACAddressString());
 			obj.setType("device");
 			obj.setState("ACTIVE");
 			conn.endTx(Transaction.COMMIT);
 			
 			log.debug("Adding device {}",device.getMACAddressString());
		} catch (Exception e) {
            // TODO: handle exceptions
          	conn.endTx(Transaction.ROLLBACK);
			log.error(":addDevice mac:{} failed", device.getMACAddressString());
		}	
		
		return obj;
	}

	@Override
	public IDeviceObject updateDevice(IDevice device) {
		return addDevice(device);
	}

	@Override
	public void removeDevice(IDevice device) {
		// TODO Auto-generated method stub
		IDeviceObject dev;
		try {
			if ((dev = conn.utils().searchDevice(conn, device.getMACAddressString())) != null) {
             	conn.utils().removeDevice(conn, dev);
              	conn.endTx(Transaction.COMMIT);
            	log.error("DeviceStorage:removeDevice mac:{} done", device.getMACAddressString());
            }
		} catch (Exception e) {
             // TODO: handle exceptions
          	conn.endTx(Transaction.ROLLBACK);
			log.error("DeviceStorage:removeDevice mac:{} failed", device.getMACAddressString());
		}
	}

	@Override
	public IDeviceObject getDeviceByMac(String mac) {
		return conn.utils().searchDevice(conn, mac);
	}

	@Override
	public IDeviceObject getDeviceByIP(String ip) {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public void changeDeviceAttachments(IDevice device) {
		// TODO Auto-generated method stub
		IDeviceObject obj = null;
 		try {
            if ((obj = conn.utils().searchDevice(conn, device.getMACAddressString())) != null) {
                log.debug("Changing device ports {}: found existing device",device.getMACAddressString());
                changeDeviceAttachments(device, obj);
     			conn.endTx(Transaction.COMMIT);
           } else {
   				log.debug("failed to search device...now adding {}",device.getMACAddressString());
   				addDevice(device);
           }            			
		} catch (Exception e) {
            // TODO: handle exceptions
          	conn.endTx(Transaction.ROLLBACK);
			log.error(":addDevice mac:{} failed", device.getMACAddressString());
		}	
	}
	
	public void changeDeviceAttachments(IDevice device, IDeviceObject obj) {
		SwitchPort[] attachmentPoints = device.getAttachmentPoints();
		List<IPortObject> attachedPorts = Lists.newArrayList(obj.getAttachedPorts());

        for (SwitchPort ap : attachmentPoints) {
       	 IPortObject port = conn.utils().searchPort(conn,
       			 									HexString.toHexString(ap.getSwitchDPID()),
       												(short) ap.getPort());
       	if (attachedPorts.contains(port)) {
       		attachedPorts.remove(port);
       	} else {
               log.debug("Adding device {}: attaching to port",device.getMACAddressString());
               port.setDevice(obj);
       		//obj.setHostPort(port);
       	}            		
       }
       for (IPortObject port: attachedPorts) {
       		port.removeDevice(obj);
       	//	obj.removeHostPort(port);
       }	
	}

	@Override
	public void changeDeviceIPv4Address(IDevice device) {
		// TODO Auto-generated method stub
		IDeviceObject obj;
  		try {
  			if ((obj = conn.utils().searchDevice(conn, device.getMACAddressString())) != null) {
            	obj.setIPAddress(device.getIPv4Addresses().toString());
              	conn.endTx(Transaction.COMMIT); 
  			} else {
            	log.error(":changeDeviceIPv4Address mac:{} failed", device.getMACAddressString());
             }		
  		} catch (TitanException e) {
            // TODO: handle exceptions
          	conn.endTx(Transaction.ROLLBACK);
			log.error(":changeDeviceIPv4Address mac:{} failed due to exception {}", device.getMACAddressString(),e);
		}
	}

}