appserver/tests/appserv-tests/devtests/deployment/ejb30/ear/session2/ejb/SfulEJB.java - glassfish - Git at Google

 /*
  * Copyright (c) 2002, 2018 Oracle and/or its affiliates. All rights reserved.
  *
  * This program and the accompanying materials are made available under the
  * terms of the Eclipse Public License v. 2.0, which is available at
  * http://www.eclipse.org/legal/epl-2.0.
  *
  * This Source Code may also be made available under the following Secondary
  * Licenses when the conditions for such availability set forth in the
  * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
  * version 2 with the GNU Classpath Exception, which is available at
  * https://www.gnu.org/software/classpath/license.html.
  *
  * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
  */

 package com.sun.s1asdev.ejb.ejb30.hello.session2;

 import jakarta.ejb.Stateful;
 import jakarta.ejb.Remote;
 import jakarta.ejb.EJB;
 import jakarta.annotation.PostConstruct;
 import jakarta.interceptor.Interceptors;
 import jakarta.ejb.EJBs;
 import jakarta.ejb.Remove;
 import jakarta.annotation.PreDestroy;
 import jakarta.ejb.EJBException;
 import jakarta.ejb.SessionContext;
 import jakarta.ejb.TransactionManagement;
 import jakarta.ejb.TransactionManagementType;
 import javax.naming.InitialContext;

 import jakarta.annotation.Resource;
 import jakarta.transaction.UserTransaction;

 import java.util.Collection;
 import java.util.HashSet;

 @Stateful
 @TransactionManagement(TransactionManagementType.BEAN)

 @EJBs(
  { @EJB(name="ejb/TypeLevelSless1", beanName="SlessEJB",
        beanInterface=Sless.class),
    @EJB(name="ejb/TypeLevelSless2", beanName="SlessEJB2",
        beanInterface=Sless.class) ,
    @EJB(name="ejb/SfulEJB2", beanInterface=Sful2.class)
  })

 @Interceptors(MyCallbackHandler.class)
 @EJB(name="ejb/TypeLevelSless3", beanInterface=SlessSub.class)
 @Remote({Sful.class})
 // ejb is not required to put business interface in implements clause
 public class SfulEJB
 {

     private @Resource SessionContext sc;

     private @EJB(beanName="SlessEJB") Sless sless;
     private @EJB(beanName="SlessEJB2") Sless sless2;
     private @EJB(beanName="SlessEJB3") SlessSub sless3;

     private Sless sless4;
     private @EJB(beanName="SlessEJB") void setSless4(Sless sless) {
         sless4 = sless;
     }

     private Sless sless5;
     @EJB(beanName="SlessEJB2") void setSless5(Sless sless) {
         sless5 = sless;
     }

     private SlessSub sless6;
     public @EJB void setSless6(SlessSub sless) {
         sless6 = sless;
     }

     @Resource
     private void setSessionContext(SessionContext context) {

         try {
             context.getUserTransaction();
             throw new RuntimeException("Should have gotten IllegalStateEx");
         } catch(IllegalStateException ise) {
             System.out.println("Successfully got exception when accessing " +
                                "context.getUserTransaction() in " +
                                "setContext method of BMT SFSB");
         }

         try {
             context.getTimerService();
             throw new RuntimeException("Should have gotten IllegalStateEx");
         } catch(IllegalStateException ise) {
             System.out.println("Successfully got exception when accessing " +
                                "context.getTimerService() in " +
                                "setContext method");
         }

         try {
             context.getBusinessObject(Sful.class);
             throw new RuntimeException("Should have gotten IllegalStateEx");
         } catch(IllegalStateException ise) {
             System.out.println("Successfully got exception when accessing " +
                                "getBusinessObject() in " +
                                "setContext method");
         }

     }

     private String state;

     public void set(String value) {
         state = value;
     }

     public String get() {
         return state;
     }

     public String hello() {
         System.out.println("In SfulEJB:hello()");

         try {

             testSlessRefs(sless, sless2, sless3);

             testSlessRefs(sless4, sless5, sless6);


             InitialContext ic = new InitialContext();

             Sless sless7 = (Sless)
                 ic.lookup("java:comp/env/ejb/TypeLevelSless1");
             Sless sless8 = (Sless)
                 ic.lookup("java:comp/env/ejb/TypeLevelSless2");
             SlessSub sless9 = (SlessSub)
                 ic.lookup("java:comp/env/ejb/TypeLevelSless3");

             testSlessRefs(sless7, sless8, sless9);

             Sless sless10 = (Sless)
                 sc.lookup("ejb/TypeLevelSless1");
             Sless sless11 = (Sless)
                 sc.lookup("ejb/TypeLevelSless2");
             SlessSub sless12 = (SlessSub)
                 sc.lookup("ejb/TypeLevelSless3");
             testSlessRefs(sless10, sless11, sless12);

             // Make sure we can getUserTransaction via SessionContext.
             // This must work if bean has bean-managed transactions.
             UserTransaction ut = sc.getUserTransaction();

             int status = ut.getStatus();
             System.out.println("context ut status = " + status);

             // Do bogus lookup to ensure exception is thrown
             try {
                 sc.lookup("abcdef");
                 throw new Exception("lookup should have thrown exception");
             } catch(IllegalArgumentException iae) {
                 System.out.println("Got expected exception for bogus lookup");
             }

         } catch(Exception e) {
             e.printStackTrace();
             throw new EJBException("hello failure", e);
         }

         return "hello";
     }

     public void foo() {}
     public void foo(int a, String b) {
     }

     private void testSlessRefs(Sless s1, Sless s2, SlessSub s3)
         throws Exception {

         String sless1Id = s1.getId();
         String sless2Id = s2.getId();
         String sless3Id = s3.getId();

         System.out.println("sless1Id = " + sless1Id);
         System.out.println("sless2Id = " + sless2Id);
         System.out.println("sless3Id = " + sless3Id);

         if( sless1Id.equals(sless2Id) ||
             sless1Id.equals(sless3Id) ||
             sless2Id.equals(sless3Id) ) {
             throw new EJBException("sless bean ejb linking error");
         }

         Sless r1 = s1.roundTrip(s1);
         s1.roundTrip(s2);
         s1.roundTrip(s3);

         s2.roundTrip(s1);
         Sless r2 = s2.roundTrip(s2);
         s2.roundTrip(s3);

         s3.roundTrip(s1);
         s3.roundTrip(s2);
         Sless r3 = s3.roundTrip(s3);

         sless1Id = r1.getId();
         sless2Id = r2.getId();
         sless3Id = r3.getId();

         System.out.println("sless1Id = " + sless1Id);
         System.out.println("sless2Id = " + sless2Id);
         System.out.println("sless3Id = " + sless3Id);

         if( sless1Id.equals(sless2Id) ||
             sless1Id.equals(sless3Id) ||
             sless2Id.equals(sless3Id) ) {
             throw new EJBException("remote 3.0 param passing error");
         }

         Collection c = new HashSet();
         c.add(s2);
         s1.roundTrip2(c);
         s2.roundTrip2(c);
         s3.roundTrip2(c);

         s3.hello3();

     }

     @PostConstruct
     public void afterCreate() {
         System.out.println("In SfulEJB::afterCreate() marked as PostConstruct");

         // jndi lookup should work within postconstruct
         Sless sless = (Sless) sc.lookup("ejb/TypeLevelSless1");

         // Access to getBusinessObject is allowed here
         sc.getBusinessObject(Sful.class);

     }

     @Remove
     public void remove() {
         System.out.println("In SfulEJB " + state + " @Remove method");
         // Access to getBusinessObject is allowed here
         sc.getBusinessObject(Sful.class);

     }

     @Remove(retainIfException=true)
     public void removeRetainIfException(boolean throwException)
         throws Exception {

         System.out.println("In SfulEJB " + state + " removeRetainIfException");
         System.out.println("throwException = " + throwException);
         if( throwException ) {
             throw new Exception("throwing app exception from @Remove method");
         }
     }

     @Remove
     public void removeNotRetainIfException(boolean throwException)
         throws Exception {

         System.out.println("In SfulEJB " + state +
                            "removeNotRetainIfException");
         System.out.println("throwException = " + throwException);
         if( throwException ) {
             throw new Exception("throwing app exception from @Remove method");
         }
     }

     @Remove
     public void removeMethodThrowSysException(boolean throwException) {

         System.out.println("In SfulEJB " + state +
                            "removeMethodThrowSysException");
         System.out.println("throwException = " + throwException);
         if( throwException ) {
             throw new EJBException
                 ("throwing system exception from @Remove method");
         }
     }

     public void doRemoveMethodSessionSyncTests() {

         Sful2 sful2_1 = (Sful2) sc.lookup("ejb/SfulEJB2");
         Sful2 sful2_2 = (Sful2) sc.lookup("ejb/SfulEJB2");
         Sful2 sful2_3 = (Sful2) sc.lookup("ejb/SfulEJB2");

         sful2_1.hello();
         try {
             sful2_1.removeRetainIfException(true);
         } catch(Exception e) {
             e.printStackTrace();
         }

         // all SessionSynch callbacks should have happened
         if( SfulEJB2.afterBeginCalled &&
             SfulEJB2.beforeCompletionCalled &&
             SfulEJB2.afterCompletionCalled ) {
             System.out.println("Got expected SessionSynch behavior for " +
                                "removeRetainIfException(true)");
         } else {
             throw new EJBException("SessionSynch failure for " +
                                    "removeRetainIfException(true)");
         }

         // invocation should still work since bean wasn't removed
         sful2_1.hello();

         try {
             sful2_1.removeRetainIfException(false);
         } catch(Exception e) {
             throw new EJBException(e);
         }

         // bean should have been removed and completion-time SessionSynch
         // callbacks should not have been called.
         if( SfulEJB2.afterBeginCalled &&
             !SfulEJB2.beforeCompletionCalled &&
             !SfulEJB2.afterCompletionCalled ) {
             System.out.println("Got expected SessionSynch behavior for " +
                                "removeRetainIfException(false)");
         } else {
             throw new EJBException("SessionSynch failure for " +
                                    "removeRetainIfException(true)");
         }


     }


 }
	/*
	* Copyright (c) 2002, 2018 Oracle and/or its affiliates. All rights reserved.
	*
	* This program and the accompanying materials are made available under the
	* terms of the Eclipse Public License v. 2.0, which is available at
	* http://www.eclipse.org/legal/epl-2.0.
	*
	* This Source Code may also be made available under the following Secondary
	* Licenses when the conditions for such availability set forth in the
	* Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
	* version 2 with the GNU Classpath Exception, which is available at
	* https://www.gnu.org/software/classpath/license.html.
	*
	* SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
	*/

	package com.sun.s1asdev.ejb.ejb30.hello.session2;

	import jakarta.ejb.Stateful;
	import jakarta.ejb.Remote;
	import jakarta.ejb.EJB;
	import jakarta.annotation.PostConstruct;
	import jakarta.interceptor.Interceptors;
	import jakarta.ejb.EJBs;
	import jakarta.ejb.Remove;
	import jakarta.annotation.PreDestroy;
	import jakarta.ejb.EJBException;
	import jakarta.ejb.SessionContext;
	import jakarta.ejb.TransactionManagement;
	import jakarta.ejb.TransactionManagementType;
	import javax.naming.InitialContext;

	import jakarta.annotation.Resource;
	import jakarta.transaction.UserTransaction;

	import java.util.Collection;
	import java.util.HashSet;

	@Stateful
	@TransactionManagement(TransactionManagementType.BEAN)

	@EJBs(
	{ @EJB(name="ejb/TypeLevelSless1", beanName="SlessEJB",
	beanInterface=Sless.class),
	@EJB(name="ejb/TypeLevelSless2", beanName="SlessEJB2",
	beanInterface=Sless.class) ,
	@EJB(name="ejb/SfulEJB2", beanInterface=Sful2.class)
	})

	@Interceptors(MyCallbackHandler.class)
	@EJB(name="ejb/TypeLevelSless3", beanInterface=SlessSub.class)
	@Remote({Sful.class})
	// ejb is not required to put business interface in implements clause
	public class SfulEJB
	{

	private @Resource SessionContext sc;

	private @EJB(beanName="SlessEJB") Sless sless;
	private @EJB(beanName="SlessEJB2") Sless sless2;
	private @EJB(beanName="SlessEJB3") SlessSub sless3;

	private Sless sless4;
	private @EJB(beanName="SlessEJB") void setSless4(Sless sless) {
	sless4 = sless;
	}

	private Sless sless5;
	@EJB(beanName="SlessEJB2") void setSless5(Sless sless) {
	sless5 = sless;
	}

	private SlessSub sless6;
	public @EJB void setSless6(SlessSub sless) {
	sless6 = sless;
	}

	@Resource
	private void setSessionContext(SessionContext context) {

	try {
	context.getUserTransaction();
	throw new RuntimeException("Should have gotten IllegalStateEx");
	} catch(IllegalStateException ise) {
	System.out.println("Successfully got exception when accessing " +
	"context.getUserTransaction() in " +
	"setContext method of BMT SFSB");
	}

	try {
	context.getTimerService();
	throw new RuntimeException("Should have gotten IllegalStateEx");
	} catch(IllegalStateException ise) {
	System.out.println("Successfully got exception when accessing " +
	"context.getTimerService() in " +
	"setContext method");
	}

	try {
	context.getBusinessObject(Sful.class);
	throw new RuntimeException("Should have gotten IllegalStateEx");
	} catch(IllegalStateException ise) {
	System.out.println("Successfully got exception when accessing " +
	"getBusinessObject() in " +
	"setContext method");
	}

	}

	private String state;

	public void set(String value) {
	state = value;
	}

	public String get() {
	return state;
	}

	public String hello() {
	System.out.println("In SfulEJB:hello()");

	try {

	testSlessRefs(sless, sless2, sless3);

	testSlessRefs(sless4, sless5, sless6);


	InitialContext ic = new InitialContext();

	Sless sless7 = (Sless)
	ic.lookup("java:comp/env/ejb/TypeLevelSless1");
	Sless sless8 = (Sless)
	ic.lookup("java:comp/env/ejb/TypeLevelSless2");
	SlessSub sless9 = (SlessSub)
	ic.lookup("java:comp/env/ejb/TypeLevelSless3");

	testSlessRefs(sless7, sless8, sless9);

	Sless sless10 = (Sless)
	sc.lookup("ejb/TypeLevelSless1");
	Sless sless11 = (Sless)
	sc.lookup("ejb/TypeLevelSless2");
	SlessSub sless12 = (SlessSub)
	sc.lookup("ejb/TypeLevelSless3");
	testSlessRefs(sless10, sless11, sless12);

	// Make sure we can getUserTransaction via SessionContext.
	// This must work if bean has bean-managed transactions.
	UserTransaction ut = sc.getUserTransaction();

	int status = ut.getStatus();
	System.out.println("context ut status = " + status);

	// Do bogus lookup to ensure exception is thrown
	try {
	sc.lookup("abcdef");
	throw new Exception("lookup should have thrown exception");
	} catch(IllegalArgumentException iae) {
	System.out.println("Got expected exception for bogus lookup");
	}

	} catch(Exception e) {
	e.printStackTrace();
	throw new EJBException("hello failure", e);
	}

	return "hello";
	}

	public void foo() {}
	public void foo(int a, String b) {
	}

	private void testSlessRefs(Sless s1, Sless s2, SlessSub s3)
	throws Exception {

	String sless1Id = s1.getId();
	String sless2Id = s2.getId();
	String sless3Id = s3.getId();

	System.out.println("sless1Id = " + sless1Id);
	System.out.println("sless2Id = " + sless2Id);
	System.out.println("sless3Id = " + sless3Id);

	if( sless1Id.equals(sless2Id) \|\|
	sless1Id.equals(sless3Id) \|\|
	sless2Id.equals(sless3Id) ) {
	throw new EJBException("sless bean ejb linking error");
	}

	Sless r1 = s1.roundTrip(s1);
	s1.roundTrip(s2);
	s1.roundTrip(s3);

	s2.roundTrip(s1);
	Sless r2 = s2.roundTrip(s2);
	s2.roundTrip(s3);

	s3.roundTrip(s1);
	s3.roundTrip(s2);
	Sless r3 = s3.roundTrip(s3);

	sless1Id = r1.getId();
	sless2Id = r2.getId();
	sless3Id = r3.getId();

	System.out.println("sless1Id = " + sless1Id);
	System.out.println("sless2Id = " + sless2Id);
	System.out.println("sless3Id = " + sless3Id);

	if( sless1Id.equals(sless2Id) \|\|
	sless1Id.equals(sless3Id) \|\|
	sless2Id.equals(sless3Id) ) {
	throw new EJBException("remote 3.0 param passing error");
	}

	Collection c = new HashSet();
	c.add(s2);
	s1.roundTrip2(c);
	s2.roundTrip2(c);
	s3.roundTrip2(c);

	s3.hello3();

	}

	@PostConstruct
	public void afterCreate() {
	System.out.println("In SfulEJB::afterCreate() marked as PostConstruct");

	// jndi lookup should work within postconstruct
	Sless sless = (Sless) sc.lookup("ejb/TypeLevelSless1");

	// Access to getBusinessObject is allowed here
	sc.getBusinessObject(Sful.class);

	}

	@Remove
	public void remove() {
	System.out.println("In SfulEJB " + state + " @Remove method");
	// Access to getBusinessObject is allowed here
	sc.getBusinessObject(Sful.class);

	}

	@Remove(retainIfException=true)
	public void removeRetainIfException(boolean throwException)
	throws Exception {

	System.out.println("In SfulEJB " + state + " removeRetainIfException");
	System.out.println("throwException = " + throwException);
	if( throwException ) {
	throw new Exception("throwing app exception from @Remove method");
	}
	}

	@Remove
	public void removeNotRetainIfException(boolean throwException)
	throws Exception {

	System.out.println("In SfulEJB " + state +
	"removeNotRetainIfException");
	System.out.println("throwException = " + throwException);
	if( throwException ) {
	throw new Exception("throwing app exception from @Remove method");
	}
	}

	@Remove
	public void removeMethodThrowSysException(boolean throwException) {

	System.out.println("In SfulEJB " + state +
	"removeMethodThrowSysException");
	System.out.println("throwException = " + throwException);
	if( throwException ) {
	throw new EJBException
	("throwing system exception from @Remove method");
	}
	}

	public void doRemoveMethodSessionSyncTests() {

	Sful2 sful2_1 = (Sful2) sc.lookup("ejb/SfulEJB2");
	Sful2 sful2_2 = (Sful2) sc.lookup("ejb/SfulEJB2");
	Sful2 sful2_3 = (Sful2) sc.lookup("ejb/SfulEJB2");

	sful2_1.hello();
	try {
	sful2_1.removeRetainIfException(true);
	} catch(Exception e) {
	e.printStackTrace();
	}

	// all SessionSynch callbacks should have happened
	if( SfulEJB2.afterBeginCalled &&
	SfulEJB2.beforeCompletionCalled &&
	SfulEJB2.afterCompletionCalled ) {
	System.out.println("Got expected SessionSynch behavior for " +
	"removeRetainIfException(true)");
	} else {
	throw new EJBException("SessionSynch failure for " +
	"removeRetainIfException(true)");
	}

	// invocation should still work since bean wasn't removed
	sful2_1.hello();

	try {
	sful2_1.removeRetainIfException(false);
	} catch(Exception e) {
	throw new EJBException(e);
	}

	// bean should have been removed and completion-time SessionSynch
	// callbacks should not have been called.
	if( SfulEJB2.afterBeginCalled &&
	!SfulEJB2.beforeCompletionCalled &&
	!SfulEJB2.afterCompletionCalled ) {
	System.out.println("Got expected SessionSynch behavior for " +
	"removeRetainIfException(false)");
	} else {
	throw new EJBException("SessionSynch failure for " +
	"removeRetainIfException(true)");
	}





	}


	}