KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > jboss > ejb > plugins > cmp > jdbc > keygen > JDBCKeyGeneratorCreateCommand


1 /*
2 * JBoss, Home of Professional Open Source
3 * Copyright 2005, JBoss Inc., and individual contributors as indicated
4 * by the @authors tag. See the copyright.txt in the distribution for a
5 * full listing of individual contributors.
6 *
7 * This is free software; you can redistribute it and/or modify it
8 * under the terms of the GNU Lesser General Public License as
9 * 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 javax.ejb.CreateException JavaDoc;
25 import javax.naming.InitialContext JavaDoc;
26 import javax.naming.NamingException JavaDoc;
27
28 import org.jboss.deployment.DeploymentException;
29 import org.jboss.ejb.EntityEnterpriseContext;
30 import org.jboss.ejb.plugins.cmp.jdbc.bridge.JDBCCMPFieldBridge;
31 import org.jboss.ejb.plugins.cmp.jdbc.metadata.JDBCEntityCommandMetaData;
32 import org.jboss.ejb.plugins.cmp.jdbc.JDBCInsertPKCreateCommand;
33 import org.jboss.ejb.plugins.cmp.jdbc.JDBCStoreManager;
34 import org.jboss.ejb.plugins.keygenerator.KeyGenerator;
35 import org.jboss.ejb.plugins.keygenerator.KeyGeneratorFactory;
36
37 /**
38  * JDBCKeyGeneratorCreateCommand executes an INSERT INTO query.
39  * This command will ask the corresponding key generator for a
40  * value for the primary key before inserting the row.
41  *
42  * @author <a HREF="mailto:loubyansky@hotmail.com">Alex Loubyansky</a>
43  *
44  * @version $Revision: 37459 $
45  */
46 public class JDBCKeyGeneratorCreateCommand extends JDBCInsertPKCreateCommand
47 {
48    protected KeyGenerator keyGenerator;
49    protected JDBCCMPFieldBridge pkField;
50
51    public void init(JDBCStoreManager manager) throws DeploymentException
52    {
53       super.init(manager);
54       pkField = getGeneratedPKField();
55    }
56
57    protected void initEntityCommand(JDBCEntityCommandMetaData entityCommand) throws DeploymentException
58    {
59       super.initEntityCommand(entityCommand);
60
61       String JavaDoc factoryName = entityCommand.getAttribute("key-generator-factory");
62       if(factoryName == null)
63       {
64          throw new DeploymentException("key-generator-factory attribute must be set for entity " + entity.getEntityName());
65       }
66
67       try
68       {
69          KeyGeneratorFactory keyGeneratorFactory = (KeyGeneratorFactory) new InitialContext JavaDoc().lookup(factoryName);
70          keyGenerator = keyGeneratorFactory.getKeyGenerator();
71       }
72       catch(NamingException JavaDoc e)
73       {
74          throw new DeploymentException("Error: can't find key generator factory: " + factoryName, e);
75       }
76       catch(Exception JavaDoc e)
77       {
78          throw new DeploymentException("Error: can't create key generator instance; key generator factory: " + factoryName, e);
79       }
80    }
81
82    protected void generateFields(EntityEnterpriseContext ctx) throws CreateException JavaDoc
83    {
84       super.generateFields(ctx);
85
86       Object JavaDoc pk = keyGenerator.generateKey();
87       log.debug("Generated new pk: " + pk);
88       pkField.setInstanceValue(ctx, pk);
89    }
90 }
91
Popular Tags