01: /*
02: * JBoss, Home of Professional Open Source.
03: * Copyright 2006, Red Hat Middleware LLC, and individual contributors
04: * as indicated by the @author tags. See the copyright.txt file in the
05: * distribution for a full listing of individual contributors.
06: *
07: * This is free software; you can redistribute it and/or modify it
08: * under the terms of the GNU Lesser General Public License as
09: * published by the Free Software Foundation; either version 2.1 of
10: * the License, or (at your option) any later version.
11: *
12: * This software is distributed in the hope that it will be useful,
13: * but WITHOUT ANY WARRANTY; without even the implied warranty of
14: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15: * Lesser General Public License for more details.
16: *
17: * You should have received a copy of the GNU Lesser General Public
18: * License along with this software; if not, write to the Free
19: * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20: * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21: */
22: package org.jboss.ejb.plugins.cmp.ejbql;
23:
24: import org.jboss.ejb.plugins.cmp.jdbc.JDBCUtil;
25: import org.jboss.ejb.plugins.cmp.jdbc.JDBCResultSetReader;
26: import org.jboss.logging.Logger;
27:
28: import java.sql.ResultSet;
29: import java.sql.SQLException;
30: import java.util.Collection;
31:
32: /**
33: * @author <a href="mailto:alex@jboss.org">Alexey Loubyansky</a>
34: * @version <tt>$Revision: 57209 $</tt>
35: */
36: public abstract class AggregateFunction extends SimpleNode implements
37: SelectFunction {
38: private final Logger log;
39: private JDBCResultSetReader resultReader;
40: private Class resultType;
41:
42: public String distinct = "";
43:
44: public AggregateFunction(int i) {
45: super (i);
46: log = Logger.getLogger(getClass());
47: }
48:
49: public void setResultType(Class type) {
50: if (Collection.class.isAssignableFrom(type)) {
51: resultType = getDefaultResultType();
52: } else {
53: this .resultType = type;
54: }
55: this .resultReader = JDBCUtil.getResultReaderByType(resultType);
56: }
57:
58: protected Class getDefaultResultType() {
59: return Double.class;
60: }
61:
62: // SelectFunction implementation
63:
64: public Object readResult(ResultSet rs) throws SQLException {
65: return resultReader.get(rs, 1, resultType, log);
66: }
67: }
|