package de.laures.cewolf.dp;

import java.io.Serializable;

import java.sql.Connection;

import java.sql.ResultSet;

import java.util.Map;

import javax.naming.Context;

import javax.naming.InitialContext;

import javax.naming.NamingException;

import javax.sql.DataSource;

import org.jfree.data.xy.XYSeries;

import de.laures.cewolf.DatasetProduceException;

/**

 * @author glaures

 *

 * To change this generated comment edit the template variable "typecomment":

 * Window>Preferences>Java>Templates.

 * To enable and disable the creation of type comments go to

 * Window>Preferences>Java>Code Generation.

 */

public class DataSourceXYSeries implements Serializable {

	private String dataSourceName;

	private String query;

	private String xCol = "x";

	private String yCol = "y";

	private String seriesName = "name";

	/**

	 * Constructor for DataSourceXYSeries.

	 */

	public DataSourceXYSeries(String dataSourceName, String query) {

		this.dataSourceName = dataSourceName;

		this.query = query;

	}

	protected DataSource getDataSource() throws NamingException {

		Context initCtx = new InitialContext();

		Context envCtx = (Context) initCtx.lookup("java:comp/env");

		return (DataSource) envCtx.lookup(dataSourceName);

	}

	/**

	 * @see de.laures.cewolf.DatasetProducer#produceDataset(Map)

	 */

	public XYSeries produceXYSeries() throws DatasetProduceException {

		XYSeries series = new XYSeries(seriesName);

		try {

			DataSource ds = getDataSource();

			Connection con = ds.getConnection();

			ResultSet rs = con.createStatement().executeQuery(query);

			int xColIndex = rs.findColumn(xCol);

			int yColIndex = rs.findColumn(yCol);

			while(rs.next()){

				series.add((Number)rs.getObject(xColIndex), (Number)rs.getObject(yColIndex));

			}

		} catch (Exception namingEx) {

			namingEx.printStackTrace();

			throw new DatasetProduceException(namingEx.getMessage());

		}

		return series;

	}

}