KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > netbeans > modules > j2ee > ejbcore > action > BusinessMethodGenerator


1 /*
2  * The contents of this file are subject to the terms of the Common Development
3  * and Distribution License (the License). You may not use this file except in
4  * compliance with the License.
5  *
6  * You can obtain a copy of the License at http://www.netbeans.org/cddl.html
7  * or http://www.netbeans.org/cddl.txt.
8  *
9  * When distributing Covered Code, include this CDDL Header Notice in each file
10  * and include the License file at http://www.netbeans.org/cddl.txt.
11  * If applicable, add the following below the CDDL Header, with the fields
12  * enclosed by brackets [] replaced by your own identifying information:
13  * "Portions Copyrighted [year] [name of copyright owner]"
14  *
15  * The Original Software is NetBeans. The Initial Developer of the Original
16  * Software is Sun Microsystems, Inc. Portions Copyright 1997-2006 Sun
17  * Microsystems, Inc. All Rights Reserved.
18  */
19
20 package org.netbeans.modules.j2ee.ejbcore.action;
21
22 import java.io.IOException JavaDoc;
23 import java.util.ArrayList JavaDoc;
24 import java.util.Collections JavaDoc;
25 import java.util.List JavaDoc;
26 import javax.lang.model.element.Modifier;
27 import org.netbeans.modules.j2ee.common.method.MethodModel;
28 import org.netbeans.modules.j2ee.dd.api.ejb.EntityAndSession;
29 import org.openide.filesystems.FileObject;
30
31 /**
32  *
33  * @author Martin Adamek
34  */
35 public final class BusinessMethodGenerator extends AbstractMethodGenerator {
36     
37     private BusinessMethodGenerator(EntityAndSession ejb, FileObject ejbClassFileObject) {
38         super(ejb, ejbClassFileObject, null);
39     }
40     
41     public static BusinessMethodGenerator create(EntityAndSession ejb, FileObject ejbClassFileObject) {
42         return new BusinessMethodGenerator(ejb, ejbClassFileObject);
43     }
44     
45     public void generate(MethodModel methodModel, boolean generateLocal, boolean generateRemote) throws IOException JavaDoc {
46         
47         // local interface
48 if (generateLocal && ejb.getLocal() != null) {
49             MethodModel methodModelCopy = MethodModel.create(
50                     methodModel.getName(),
51                     methodModel.getReturnType(),
52                     null,
53                     methodModel.getParameters(),
54                     methodModel.getExceptions(),
55                     methodModel.getModifiers()
56                     );
57             addMethodToInterface(methodModelCopy, ejb.getLocal());
58         }
59         
60         // remote interface, add RemoteException if it's not there
61 if (generateRemote && ejb.getRemote() != null) {
62             List JavaDoc<String JavaDoc> exceptions = new ArrayList JavaDoc<String JavaDoc>(methodModel.getExceptions());
63             if (!methodModel.getExceptions().contains("java.rmi.RemoteException")) {
64                 exceptions.add("java.rmi.RemoteException");
65             }
66             MethodModel methodModelCopy = MethodModel.create(
67                     methodModel.getName(),
68                     methodModel.getReturnType(),
69                     null,
70                     methodModel.getParameters(),
71                     exceptions,
72                     methodModel.getModifiers()
73                     );
74             addMethodToInterface(methodModelCopy, ejb.getRemote());
75         }
76         
77         // ejb class, add 'public' modifier
78 MethodModel methodModelCopy = MethodModel.create(
79                 methodModel.getName(),
80                 methodModel.getReturnType(),
81                 methodModel.getBody(),
82                 methodModel.getParameters(),
83                 methodModel.getExceptions(),
84                 Collections.singleton(Modifier.PUBLIC)
85                 );
86         addMethod(methodModelCopy, ejbClassFileObject, ejb.getEjbClass());
87     }
88     
89 }
90
Popular Tags