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.jdbc2;
23:
24: import org.jboss.ejb.EntityEnterpriseContext;
25: import org.jboss.ejb.plugins.cmp.jdbc2.bridge.JDBCEntityBridge2;
26: import org.jboss.deployment.DeploymentException;
27:
28: import javax.ejb.CreateException;
29: import javax.ejb.DuplicateKeyException;
30: import java.lang.reflect.Method;
31: import java.sql.SQLException;
32:
33: /**
34: * @author <a href="mailto:alex@jboss.org">Alexey Loubyansky</a>
35: * @version <tt>$Revision: 57209 $</tt>
36: */
37: public class ApplicationPkCreateCommand implements CreateCommand {
38: private JDBCEntityBridge2 entityBridge;
39:
40: public void init(JDBCStoreManager2 manager)
41: throws DeploymentException {
42: this .entityBridge = (JDBCEntityBridge2) manager
43: .getEntityBridge();
44: }
45:
46: public Object execute(Method m, Object[] args,
47: EntityEnterpriseContext ctx) throws CreateException {
48: Object pk;
49: PersistentContext pctx = (PersistentContext) ctx
50: .getPersistenceContext();
51: if (ctx.getId() == null) {
52: pk = entityBridge.extractPrimaryKeyFromInstance(ctx);
53:
54: if (pk == null) {
55: throw new CreateException(
56: "Primary key for created instance is null.");
57: }
58:
59: pctx.setPk(pk);
60: } else {
61: // insert-after-ejb-post-create
62: try {
63: pctx.flush();
64: } catch (SQLException e) {
65: if ("23000".equals(e.getSQLState())) {
66: throw new DuplicateKeyException(
67: "Unique key violation or invalid foreign key value: pk="
68: + ctx.getId());
69: } else {
70: throw new CreateException(
71: "Failed to create instance: pk="
72: + ctx.getId() + ", state="
73: + e.getSQLState() + ", msg="
74: + e.getMessage());
75: }
76: }
77: pk = ctx.getId();
78: }
79: return pk;
80: }
81: }
|