1 files changed, 157 insertions, 0 deletions
diff --git a/java/perftests/visualisation-jfc/src/main/java/org/apache/qpid/disttest/charting/seriesbuilder/JdbcSeriesBuilder.java b/java/perftests/visualisation-jfc/src/main/java/org/apache/qpid/disttest/charting/seriesbuilder/JdbcSeriesBuilder.java
new file mode 100644
index 0000000000..180aa54c6d
--- /dev/null
+++ b/java/perftests/visualisation-jfc/src/main/java/org/apache/qpid/disttest/charting/seriesbuilder/JdbcSeriesBuilder.java
@@ -0,0 +1,157 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ *
+ */
+package org.apache.qpid.disttest.charting.seriesbuilder;
+
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.Iterator;
+import java.util.List;
+
+import org.apache.commons.lang.builder.ToStringBuilder;
+import org.apache.commons.lang.builder.ToStringStyle;
+import org.apache.qpid.disttest.charting.ChartingException;
+import org.apache.qpid.disttest.charting.definition.SeriesDefinition;
+import org.jfree.data.general.Dataset;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * A {@link SeriesBuilder} that uses JDBC to read series data.
+ * The actual JDBC URL used is determined by my {@link JdbcUrlGenerator}.
+ */
+public class JdbcSeriesBuilder implements SeriesBuilder
+{
+    private static final Logger LOGGER = LoggerFactory.getLogger(JdbcSeriesBuilder.class);
+
+    private DatasetHolder _datasetHolder;
+
+    private final JdbcUrlGenerator _jdbcUrlGenerator;
+
+    /**
+     * @param providedJdbcUrl the JDBC URL. Provide null if the value should be
+     * inferred by {@link #_jdbcUrlGenerator}.
+     */
+    public JdbcSeriesBuilder(String jdbcDriverClass, String providedJdbcUrl)
+    {
+        registerDriver(jdbcDriverClass);
+        _jdbcUrlGenerator = new JdbcUrlGenerator(providedJdbcUrl);
+        LOGGER.info("Created: " + this);
+    }
+
+    @Override
+    public void setDatasetHolder(DatasetHolder callback)
+    {
+        _datasetHolder = callback;
+    }
+
+    @Override
+    public Dataset build(List<SeriesDefinition> seriesDefinitions)
+    {
+        for (Iterator<SeriesDefinition> iterator = seriesDefinitions.iterator(); iterator.hasNext();)
+        {
+            SeriesDefinition series = iterator.next();
+            buildDataSetForSingleSeries(series);
+        }
+        return _datasetHolder.getPopulatedDataset();
+    }
+
+    private void buildDataSetForSingleSeries(SeriesDefinition seriesDefinition)
+    {
+        Connection conn = null;
+        Statement stmt = null;
+        try
+        {
+            String jdbcUrl = _jdbcUrlGenerator.getJdbcUrl(seriesDefinition);
+            conn = DriverManager.getConnection(jdbcUrl);
+
+            final String seriesStatement = seriesDefinition.getSeriesStatement();
+
+            stmt = conn.createStatement();
+            ResultSet results = stmt.executeQuery(seriesStatement);
+            int columnCount = results.getMetaData().getColumnCount();
+            _datasetHolder.beginSeries(seriesDefinition);
+            while (results.next())
+            {
+                Object[] row = new Object[columnCount];
+                for (int i = 0; i < row.length; i++)
+                {
+                    row[i] = results.getObject(i+1);
+                }
+
+                SeriesRow seriesRow = SeriesRow.createValidSeriesRow(_datasetHolder.getNumberOfDimensions(), row);
+                _datasetHolder.addDataPointToSeries(seriesDefinition, seriesRow);
+            }
+            _datasetHolder.endSeries(seriesDefinition);
+        }
+        catch (SQLException e)
+        {
+            throw new ChartingException("Failed to create chart dataset", e);
+        }
+        finally
+        {
+            if (stmt != null)
+            {
+                try
+                {
+                    stmt.close();
+                }
+                catch (SQLException e)
+                {
+                    throw new RuntimeException("Failed to close statement", e);
+                }
+            }
+            if (conn != null)
+            {
+                try
+                {
+                    conn.close();
+                }
+                catch (SQLException e)
+                {
+                    throw new RuntimeException("Failed to close connection", e);
+                }
+            }
+        }
+    }
+
+    private void registerDriver(String driverClassName) throws ExceptionInInitializerError
+    {
+        try
+        {
+            Class.forName(driverClassName);
+            LOGGER.info("Loaded JDBC driver class " + driverClassName);
+        }
+        catch (ClassNotFoundException e)
+        {
+            throw new RuntimeException("Could not load JDBC driver " + driverClassName, e);
+        }
+    }
+
+    @Override
+    public String toString()
+    {
+        return new ToStringBuilder(this, ToStringStyle.SHORT_PREFIX_STYLE)
+            .append("jdbcUrlGenerator", _jdbcUrlGenerator)
+            .toString();
+    }
+}