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.keygen;
23:
24: import java.sql.Connection;
25: import java.sql.PreparedStatement;
26: import java.sql.ResultSet;
27: import java.sql.SQLException;
28: import javax.ejb.EJBException;
29: import org.jboss.ejb.EntityEnterpriseContext;
30: import org.jboss.ejb.plugins.cmp.jdbc.JDBCIdentityColumnCreateCommand;
31: import org.jboss.ejb.plugins.cmp.jdbc.JDBCUtil;
32:
33: /**
34: * Create method that uses the identity_val_local() function in DB2 to get
35: * get the ID of the last inserted row, and populate it into the EJB
36: * object being created.
37: *
38: * @author <a href="mailto:dwintschel@esports.com">Daniel Wintschel</a>
39: * @version $Revision: 57209 $
40: */
41: public class JDBCDB2IdentityValLocalCreateCommand extends
42: JDBCIdentityColumnCreateCommand {
43: private static final String SQL = "values (identity_val_local())";
44:
45: protected int executeInsert(int paramIndex, PreparedStatement ps,
46: EntityEnterpriseContext ctx) throws SQLException {
47: int rows = ps.executeUpdate();
48: ResultSet results = null;
49: try {
50: Connection conn = ps.getConnection();
51: results = conn.prepareStatement(SQL).executeQuery();
52: if (!results.next()) {
53: throw new EJBException(
54: "identity_val_local() returned an empty ResultSet");
55: }
56: pkField.loadInstanceResults(results, 1, ctx);
57: } catch (RuntimeException e) {
58: throw e;
59: } catch (Exception e) {
60: // throw EJBException to force a rollback as the row has been inserted
61: throw new EJBException(
62: "Error extracting identity_val_local()", e);
63: } finally {
64: JDBCUtil.safeClose(results);
65: }
66: return rows;
67: }
68: }
|