001 /** 002 * ======================================== 003 * JFreeReport : a free Java report library 004 * ======================================== 005 * 006 * Project Info: http://reporting.pentaho.org/ 007 * 008 * (C) Copyright 2000-2007, by Object Refinery Limited, Pentaho Corporation and Contributors. 009 * 010 * This library is free software; you can redistribute it and/or modify it under the terms 011 * of the GNU Lesser General Public License as published by the Free Software Foundation; 012 * either version 2.1 of the License, or (at your option) any later version. 013 * 014 * This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; 015 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 016 * See the GNU Lesser General Public License for more details. 017 * 018 * You should have received a copy of the GNU Lesser General Public License along with this 019 * library; if not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, 020 * Boston, MA 02111-1307, USA. 021 * 022 * [Java is a trademark or registered trademark of Sun Microsystems, Inc. 023 * in the United States and other countries.] 024 * 025 * ------------ 026 * $Id: SQLReportDataFactory.java 3525 2007-10-16 11:43:48Z tmorgner $ 027 * ------------ 028 * (C) Copyright 2000-2005, by Object Refinery Limited. 029 * (C) Copyright 2005-2007, by Pentaho Corporation. 030 */ 031 package org.jfree.report.modules.data.sql; 032 033 import java.sql.Connection; 034 import java.util.HashMap; 035 036 import org.jfree.report.DataSet; 037 import org.jfree.report.ReportData; 038 import org.jfree.report.ReportDataFactoryException; 039 040 /** 041 * Creation-Date: 19.02.2006, 17:37:33 042 * 043 * @author Thomas Morgner 044 */ 045 public class SQLReportDataFactory extends SimpleSQLReportDataFactory 046 { 047 private HashMap querymappings; 048 049 public SQLReportDataFactory(final Connection connection) 050 { 051 super(connection); 052 querymappings = new HashMap(); 053 } 054 055 056 public SQLReportDataFactory(final ConnectionProvider connectionProvider) 057 { 058 super(connectionProvider); 059 querymappings = new HashMap(); 060 } 061 062 public void setQuery(final String name, final String queryString) 063 { 064 if (queryString == null) 065 { 066 querymappings.remove(name); 067 } 068 else 069 { 070 querymappings.put(name, queryString); 071 } 072 } 073 074 /** 075 * Queries a datasource. The string 'query' defines the name of the query. The 076 * Parameterset given here may contain more data than actually needed. 077 * <p/> 078 * The dataset may change between two calls, do not assume anything! 079 * 080 * @param query 081 * @param parameters 082 * @return 083 */ 084 public ReportData queryData(final String query, final DataSet parameters) 085 throws ReportDataFactoryException 086 { 087 if (query == null) 088 { 089 throw new NullPointerException("Query is null."); 090 } 091 final String realQuery = getQuery(query); 092 if (realQuery == null) 093 { 094 throw new ReportDataFactoryException("Query '" + query + "' is not recognized."); 095 } 096 return super.queryData(realQuery, parameters); 097 } 098 099 public String getQuery(final String name) 100 { 101 return (String) querymappings.get(name); 102 } 103 104 public String[] getQueryNames() 105 { 106 return (String[]) querymappings.keySet().toArray( 107 new String[querymappings.size()]); 108 } 109 110 }