org.protege.osgi.jdbc.prefs.DriverTableModel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ontop-protege Show documentation
Ontop OBDA plugin for protégé
There is a newer version: 5.2.1.1
package org.protege.osgi.jdbc.prefs;

import org.osgi.util.tracker.ServiceTracker;
import org.protege.osgi.jdbc.JdbcRegistry;

import javax.swing.table.AbstractTableModel;
import java.sql.Driver;
import java.util.*;

public class DriverTableModel extends AbstractTableModel {
    private static final long serialVersionUID = -7588371899390500462L;
    
    private List drivers = new ArrayList<>();
    private Set foundDrivers = new HashSet<>();
    private ServiceTracker jdbcRegistryTracker;
    
	public enum Column {
		DESCRIPTION("Description"), STATUS("Status");
		
		private String name;
		
		Column(String name) {
			this.name = name;
		}
		
		public String getName() {
			return name;
		}
	}
	

	
	public DriverTableModel(List drivers, ServiceTracker jdbcRegistryTracker) {
		this.drivers  = drivers;
		this.jdbcRegistryTracker = jdbcRegistryTracker;
		refresh();
	}

	public int getColumnCount() {
		return Column.values().length;
	}
	
	@Override
	public String getColumnName(int column) {
	    Column col = Column.values()[column];
	    return col.getName();
	}

	public int getRowCount() {
		return drivers.size();
	}

	public Object getValueAt(int rowIndex, int columnIndex) {
		DriverInfo info = drivers.get(rowIndex);
		switch (Column.values()[columnIndex]) {
		case DESCRIPTION:
			return info != null ? info.getDescription() : null;
		case STATUS:
			return foundDrivers.contains(info.getClassName()) ? "ready" : "down";
	    default:
	    	throw new UnsupportedOperationException("Programmer error");
		}
	}
	
	public void refresh() {
		foundDrivers.clear();
		try {
		    jdbcRegistryTracker.open();
		    for (Object o : jdbcRegistryTracker.getServices()) {
		        JdbcRegistry registry = (JdbcRegistry) o;
		        for (Driver d : registry.getJdbcDrivers()) {
		            foundDrivers.add(d.getClass().getName());
		        }
		    }
		}
		finally {
		    jdbcRegistryTracker.close();
		}
		fireTableStructureChanged();
	}
	
	public List getDrivers() {
	    return drivers;
	}
	
	public void addDriver(DriverInfo driver) {
		DriverInfo toRemove = null;
		for (DriverInfo existingDriver : drivers) {
			if (existingDriver.getClassName().equals(driver.getClassName())) {
				toRemove = existingDriver;
				break;
			}
		}
		if (toRemove != null) {
		    drivers.remove(toRemove);
		}
		drivers.add(driver);
		refresh();
	}
	
	public void removeDrivers(List rows) {
	    Collections.sort(rows);
	    Collections.reverse(rows);
	    for (Integer row : rows) {
	        drivers.remove(row.intValue());
	    }
		refresh();
	}
	
	public void replaceDriver(int row, DriverInfo newDriver) {
	    drivers.remove(row);
	    drivers.add(row, newDriver);
	}


}