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.jdbc;
23:
24: import java.lang.reflect.Method;
25: import java.util.List;
26: import java.util.ArrayList;
27:
28: import org.jboss.ejb.EntityEnterpriseContext;
29: import org.jboss.ejb.plugins.cmp.jdbc.bridge.JDBCCMRFieldBridge;
30: import org.jboss.ejb.plugins.cmp.jdbc.bridge.JDBCEntityBridge;
31: import org.jboss.ejb.plugins.cmp.jdbc.bridge.JDBCFieldBridge;
32:
33: /**
34: * This command establishes relationships for CMR fields that have
35: * foreign keys mapped to primary keys.
36: *
37: * @author <a href="mailto:aloubyansky@hotmail.com">Alex Loubyansky</a>
38: *
39: * @version $Revision: 57209 $
40: */
41: public final class JDBCPostCreateEntityCommand {
42: // Attributes ------------------------------------
43: private final JDBCEntityBridge entity;
44: private final JDBCCMRFieldBridge[] cmrWithFKMappedToCMP;
45:
46: // Constructors ----------------------------------
47: public JDBCPostCreateEntityCommand(JDBCStoreManager manager) {
48: entity = (JDBCEntityBridge) manager.getEntityBridge();
49: JDBCFieldBridge[] cmrFields = entity.getCMRFields();
50: List fkToCMPList = new ArrayList(4);
51: for (int i = 0; i < cmrFields.length; ++i) {
52: JDBCCMRFieldBridge cmrField = (JDBCCMRFieldBridge) cmrFields[i];
53: JDBCCMRFieldBridge relatedCMRField = (JDBCCMRFieldBridge) cmrField
54: .getRelatedCMRField();
55: if (cmrField.hasFKFieldsMappedToCMPFields()
56: || relatedCMRField.hasFKFieldsMappedToCMPFields()) {
57: fkToCMPList.add(cmrField);
58: }
59: }
60: if (fkToCMPList.isEmpty())
61: cmrWithFKMappedToCMP = null;
62: else
63: cmrWithFKMappedToCMP = (JDBCCMRFieldBridge[]) fkToCMPList
64: .toArray(new JDBCCMRFieldBridge[fkToCMPList.size()]);
65: }
66:
67: // Public ----------------------------------------
68: public Object execute(Method m, Object[] args,
69: EntityEnterpriseContext ctx) {
70: if (cmrWithFKMappedToCMP == null)
71: return null;
72:
73: for (int i = 0; i < cmrWithFKMappedToCMP.length; ++i) {
74: JDBCCMRFieldBridge cmrField = cmrWithFKMappedToCMP[i];
75: JDBCCMRFieldBridge relatedCMRField = (JDBCCMRFieldBridge) cmrField
76: .getRelatedCMRField();
77: if (cmrField.hasFKFieldsMappedToCMPFields()) {
78: Object relatedId = cmrField
79: .getRelatedIdFromContext(ctx);
80: if (relatedId != null) {
81: try {
82: if (cmrField.isForeignKeyValid(relatedId)) {
83: cmrField
84: .createRelationLinks(ctx, relatedId);
85: } else {
86: relatedCMRField.addRelatedPKWaitingForMyPK(
87: relatedId, ctx.getId());
88: }
89: } catch (Exception e) {
90: // no such object
91: }
92: }
93: } else if (relatedCMRField.hasFKFieldsMappedToCMPFields()) {
94: cmrField.addRelatedPKsWaitedForMe(ctx);
95: }
96: }
97: return null;
98: }
99: }
|