001: /*
002: * Copyright (c) 1998-2008 Caucho Technology -- all rights reserved
003: *
004: * This file is part of Resin(R) Open Source
005: *
006: * Each copy or derived work must preserve the copyright notice and this
007: * notice unmodified.
008: *
009: * Resin Open Source is free software; you can redistribute it and/or modify
010: * it under the terms of the GNU General Public License as published by
011: * the Free Software Foundation; either version 2 of the License, or
012: * (at your option) any later version.
013: *
014: * Resin Open Source is distributed in the hope that it will be useful,
015: * but WITHOUT ANY WARRANTY; without even the implied warranty of
016: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
017: * of NON-INFRINGEMENT. See the GNU General Public License for more
018: * details.
019: *
020: * You should have received a copy of the GNU General Public License
021: * along with Resin Open Source; if not, write to the
022: * Free Software Foundation, Inc.
023: * 59 Temple Place, Suite 330
024: * Boston, MA 02111-1307 USA
025: *
026: * @author Scott Ferguson
027: */
028:
029: package com.caucho.amber.expr.fun;
030:
031: import com.caucho.amber.expr.*;
032: import com.caucho.amber.query.*;
033: import com.caucho.amber.table.*;
034: import com.caucho.util.CharBuffer;
035: import com.caucho.util.L10N;
036:
037: import java.util.ArrayList;
038:
039: /**
040: * SIZE function expression
041: */
042: public class SizeFunExpr extends FunExpr {
043: private static final L10N L = new L10N(SizeFunExpr.class);
044:
045: /**
046: * Creates a new expression
047: */
048: protected SizeFunExpr(QueryParser parser, ArrayList<AmberExpr> args) {
049: super (parser, "size", args, false);
050: }
051:
052: public static FunExpr create(QueryParser parser,
053: ArrayList<AmberExpr> args) {
054: return new SizeFunExpr(parser, args);
055: }
056:
057: /**
058: * Generates the where expression.
059: */
060: public void generateWhere(CharBuffer cb) {
061: generateInternalWhere(cb, true);
062: }
063:
064: /**
065: * Generates the (update) where expression.
066: */
067: public void generateUpdateWhere(CharBuffer cb) {
068: generateInternalWhere(cb, false);
069: }
070:
071: /**
072: * Generates the where clause.
073: */
074: void generateInternalWhere(CharBuffer cb, boolean select) {
075: cb.append("count(");
076:
077: AmberExpr arg = _args.get(0);
078:
079: if (arg instanceof OneToManyExpr) {
080:
081: // jpa/119m
082:
083: OneToManyExpr oneToMany = (OneToManyExpr) arg;
084:
085: FromItem fromItem = oneToMany.getChildFromItem();
086:
087: cb.append(fromItem.getName());
088: cb.append('.');
089:
090: LinkColumns linkColumns = oneToMany.getLinkColumns();
091: ForeignColumn fkColumn = linkColumns.getColumns().get(0);
092:
093: cb.append(fkColumn.getName());
094: } else {
095: if (select)
096: arg.generateWhere(cb);
097: else
098: arg.generateUpdateWhere(cb);
099: }
100:
101: cb.append(')');
102: }
103: }
|