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.field;
030:
031: import com.caucho.amber.table.Column;
032: import com.caucho.amber.type.EntityType;
033: import com.caucho.bytecode.JMethod;
034: import com.caucho.config.ConfigException;
035: import com.caucho.java.JavaWriter;
036: import com.caucho.log.Log;
037: import com.caucho.util.L10N;
038:
039: import java.io.IOException;
040: import java.util.ArrayList;
041: import java.util.logging.Level;
042: import java.util.logging.Logger;
043:
044: /**
045: * Represents a map to entities.
046: */
047: public class EntityMapField extends AbstractField {
048: private static final L10N L = new L10N(EntityMapField.class);
049: protected static final Logger log = Log.open(EntityMapField.class);
050:
051: private ArrayList<Column> _indexColumns;
052: private JMethod _mapMethod;
053:
054: private EntityType _targetType;
055:
056: private IdField _id;
057: private IdField _index;
058:
059: public EntityMapField(EntityType entityType) {
060: super (entityType);
061: }
062:
063: /**
064: * Sets the field name.
065: */
066: public void setName(String name) {
067: // hack for EJB maps
068: try {
069: super .setName(name);
070: } catch (ConfigException e) {
071: log.log(Level.FINEST, e.toString(), e);
072: }
073:
074: setJavaType(java.util.Map.class);
075: }
076:
077: /**
078: * Sets the target type.
079: */
080: public void setTargetType(EntityType type) {
081: _targetType = type;
082: }
083:
084: /**
085: * Returns true if the methods are abstract.
086: */
087: public boolean isUpdateable() {
088: return false;
089: }
090:
091: /**
092: * Sets the map method.
093: */
094: public void setMapMethod(JMethod method) {
095: _mapMethod = method;
096: }
097:
098: /**
099: * Sets the id field.
100: */
101: public void setId(IdField id) {
102: _id = id;
103: }
104:
105: /**
106: * Sets the index field.
107: */
108: public void setIndex(IdField index) {
109: _index = index;
110: }
111:
112: /**
113: * Sets the index columns.
114: */
115: public void setIndexColumns(ArrayList<Column> columns) {
116: _indexColumns = columns;
117: }
118:
119: /**
120: * Sets the index columns.
121: */
122: public ArrayList<Column> getIndexColumns() {
123: return _indexColumns;
124: }
125:
126: /**
127: * Generates loading cache
128: */
129: public void generateUpdate(JavaWriter out, String mask,
130: String pstmt, String index) throws IOException {
131: }
132:
133: /**
134: * Generates loading cache
135: */
136: public void generateLoadFromObject(JavaWriter out, String obj)
137: throws IOException {
138: }
139:
140: /**
141: * Generates loading cache
142: */
143: public void generateUpdateFromObject(JavaWriter out, String obj)
144: throws IOException {
145: }
146:
147: /**
148: * Generates the select clause.
149: */
150: public String generateLoadSelect(String id) {
151: return null;
152: }
153:
154: /**
155: * Updates the cached copy.
156: */
157: public void generateCopyUpdateObject(JavaWriter out, String dst,
158: String src, int updateIndex) throws IOException {
159: }
160:
161: /**
162: * Updates the cached copy.
163: */
164: public void generateCopyLoadObject(JavaWriter out, String dst,
165: String src, int loadIndex) throws IOException {
166: }
167:
168: /**
169: * Generates the get property.
170: */
171: public void generateSuperGetter(JavaWriter out) throws IOException {
172: }
173:
174: /**
175: * Generates the get property.
176: */
177: public void generateSuperSetter(JavaWriter out) throws IOException {
178: }
179:
180: /**
181: * Generates the set property.
182: */
183: public void generateGetProperty(JavaWriter out) throws IOException {
184: if (getGetterMethod() != null) {
185: out.println();
186: out.println("public " + getJavaTypeName() + " "
187: + getGetterName() + "()");
188: out.println("{");
189: out.pushDepth();
190:
191: out.println("return null;");
192:
193: out.popDepth();
194: out.println("}");
195: }
196:
197: if (_mapMethod != null) {
198: out.println();
199: out.print("public ");
200: out.print(_mapMethod.getReturnType().getPrintName());
201: out.print(" " + _mapMethod.getName() + "(");
202: out.print(_mapMethod.getParameterTypes()[0].getPrintName());
203: out.println(" a0)");
204: out.println("{");
205: out.pushDepth();
206:
207: out.println("if (__caucho_session == null)");
208: out.println(" return null;");
209: out.println();
210:
211: out.println("try {");
212: out.pushDepth();
213:
214: out.println("com.caucho.amber.AmberQuery query;");
215:
216: EntityType targetType = _targetType;
217:
218: String table = targetType.getName();
219:
220: out.print("String sql = \"SELECT o");
221: out.print(" FROM " + table + " o");
222: out.print(" WHERE ");
223:
224: EntityType sourceType = (EntityType) getSourceType();
225: ArrayList<IdField> keys = sourceType.getId().getKeys();
226:
227: out.print("o." + _index.getName() + "=?1");
228:
229: for (int i = 0; i < keys.size(); i++) {
230: IdField key = keys.get(i);
231:
232: out.print(" and ");
233:
234: out.print("o." + _id.getName() + "." + key.getName()
235: + "=?" + (i + 2));
236: }
237:
238: out.println("\";");
239:
240: out.println("query = __caucho_session.prepareQuery(sql);");
241:
242: out.println("int index = 1;");
243: _index.getType().generateSet(out, "query", "index", "a0");
244:
245: for (int i = 0; i < keys.size(); i++) {
246: IdField key = keys.get(i);
247:
248: key.generateSet(out, "query", "index", "this");
249: }
250:
251: out.print("return (");
252: out.print(_mapMethod.getReturnType().getPrintName());
253: out.println(") query.getSingleResult();");
254:
255: out.popDepth();
256: out.println("} catch (Exception e) {");
257: out
258: .println(" throw com.caucho.amber.AmberRuntimeException.create(e);");
259: out.println("}");
260:
261: out.popDepth();
262: out.println("}");
263: }
264: }
265: }
|