appserver/tests/appserv-tests/devtests/ejb/ejb30/hello/mdb2/ejb/MessageBean.java

/*
 * Copyright (c) 2017, 2020 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.mdb2;

import jakarta.ejb.MessageDriven;
import jakarta.ejb.Timeout;
import jakarta.ejb.Timer;
import jakarta.ejb.TimerService;
import jakarta.ejb.EJBException;
import jakarta.ejb.NoSuchEJBException;
import jakarta.ejb.EJB;
import jakarta.ejb.TransactionAttribute;
import jakarta.ejb.TransactionAttributeType;
import jakarta.interceptor.AroundInvoke;
import jakarta.interceptor.InvocationContext;
import jakarta.ejb.MessageDrivenContext;

import javax.naming.InitialContext;

import jakarta.jms.MessageListener;
import jakarta.jms.Message;
import jakarta.jms.Queue;
import jakarta.jms.QueueConnectionFactory;
import jakarta.jms.QueueConnection;
import jakarta.jms.QueueSession;
import jakarta.jms.QueueSender;
import jakarta.jms.TextMessage;
import jakarta.jms.Session;
import jakarta.transaction.TransactionManager;
import jakarta.transaction.Status;
import jakarta.annotation.Resource;

import java.util.Collection;

@TransactionAttribute(TransactionAttributeType.NOT_SUPPORTED)
@MessageDriven(messageListenerInterface=MessageListener.class) 
public class MessageBean implements jakarta.ejb.MessageDrivenBean {

    @EJB(name="hello1") private Hello1 hello1;
    @EJB private Hello2 hello2;
    
    boolean onMessageInterceptorCalled = false;
    boolean timeoutInterceptorCalled = false;

    @Resource jakarta.ejb.MessageDrivenContext ctx;
    @Resource jakarta.ejb.TimerService injectedTimerService;

    @Resource(name="jms/MyQueueConnectionFactory") 
        QueueConnectionFactory qcFactory;

    // Values for these will be specified in the standard deployment descriptor
    // and the values will be different than the defaults.
    @Resource String stringValue1 = "undefined";
    @Resource(name="intValue1") int intValue1 = 1;
    @Resource(name="integerValue1") Integer integerValue1 = new Integer(1);
    short sv1 = 1;

    // Value specified in deployment descriptor so this should be called.
    @Resource void setShortValue1(short s) {
        sv1 = s;
    }


    // Values for these will not be specified in the standard deployment
    // descriptor so their defaults should apply.
    @Resource String stringValue2 = "undefined";
    @Resource int intValue2 = 1;
    @Resource Integer integerValue2 = new Integer(1);

    // corresponding env-entry is specified in ejb-jar.xml, 
    // but there is no value so no injection should happen and
    // the default value should be used.
    int intValue3 = 3;

    // No value specified in deployment descriptor so this should never be
    // called.
    @Resource(name="shortValue2") void setShortValue2(short s) {
        throw new IllegalStateException("setShortValue1 param = " + s +
                                        " shouldn't be called");
    }

    @Resource
    private void setMDC(jakarta.ejb.MessageDrivenContext context) {

        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");
        }

    }

    public void setMessageDrivenContext(jakarta.ejb.MessageDrivenContext mdc) {
        System.out.println("In MDB:setMessageDrivenContext");
    }

    public void ejbCreate() {
        System.out.println("In MDB:ejbCreate");
    }

    public void ejbRemove() {
        System.out.println("In MDB:ejbRemove");
    }

    @TransactionAttribute(TransactionAttributeType.NOT_SUPPORTED)
    public void onMessage(Message message) {
        System.out.println("Got message!!!");


        try {

            if( !onMessageInterceptorCalled ) {
                throw new Exception("Error : interceptor wasn't invoked");
            }

            if( intValue3 == 3 ) {
                // Also make sure env-entry is not visible in component env
                try {
                    ctx.lookup("someIntValue3");
                    throw new Exception("shouldn't have found someIntValue3");
                } catch(IllegalArgumentException ise) {
                    System.out.println("Successfully verified that " +
                                       "env-entry someIntValue3 is not " +
                                       "visible in java:comp/env");
                }
            } else {
                throw new Exception("wrong value for intValue3 = " +
                                    intValue3);
            } 

            if( stringValue1.equals("undefined") ||
                (intValue1 == 1) || (integerValue1.intValue() == 1) ||
                (sv1 == 1) ) {
                throw new Exception("env-entry @Resource override error");
            } else {
                System.out.println("stringValue1 = " + stringValue1 +
                                   " intValue1 = " + intValue1 +
                                   " integerValue1 = " + integerValue1 +
                                   " shortValue1 = " + sv1);
            }

            if( !stringValue2.equals("undefined") ||
                (intValue2 != 1) || (integerValue2.intValue() != 1) ) {
                throw new Exception("env-entry @Resource override error");
            }
            
            // Proprietary way to look up tx manager.  
            TransactionManager tm = (TransactionManager)
                new InitialContext().lookup("java:appserver/TransactionManager");
            // Use an implementation-specific check to ensure that there
            // is no tx.  A portable application couldn't make this check
            // since the exact tx behavior for TX_NOT_SUPPORTED is not
            // defined.
            int txStatus = tm.getStatus();
            if( txStatus == Status.STATUS_NO_TRANSACTION ) {
                System.out.println("Successfully verified tx attr = " +
                                   "TX_NOT_SUPPORTED in onMessage()");
            } else {
                throw new Exception("Invalid tx status for TX_NOT_SUPPORTED" +
                                    " method " + txStatus);
            }
            
            System.out.println("Calling hello1 stateless bean");
            hello1.hello("local ejb3.0 stateless");

            System.out.println("Calling hello11 stateless bean");

            // Get context through direct "java:comp/EJBContext" lookup
            MessageDrivenContext ctx2 = (MessageDrivenContext)
                new InitialContext().lookup("java:comp/EJBContext");
            Hello1 hello11 = (Hello1) ctx2.lookup("hello1");
            hello11.hello("local ejb3.0 stateless 2");

            System.out.println("Calling hello2 stateful bean");
            hello2.hello("local ejb3.0 stateful");
            hello2.removeMethod();

            try {
                hello2.hello("this call should not go through");
                throw new Exception("bean should have been removed " +
                                    "after removeMethod()");
            } catch(NoSuchEJBException e) {
                System.out.println("Successfully caught EJBException after " +
                                   " accessing removed SFSB");
            }


            System.out.println("creating timer");
            TimerService timerService = (TimerService) 
                new InitialContext().lookup("java:comp/TimerService");
                
            timerService.createTimer(7000, "created timer");
            
            Collection timers = injectedTimerService.getTimers();
            if( timers.size() != 1 ) {
                throw new IllegalStateException("invalid timer count = " +
                                                timers.size());
            }

        } catch(Exception e) {
            e.printStackTrace();
        }
    }


    @Timeout private void timeout(jakarta.ejb.Timer t) {

        QueueConnection connection = null;        

        try {
            System.out.println("In MessageBean.  Got timeout callback");

            if( HelloStatelessSuper.timeoutHappened ) {
                System.out.println("Verified that stateless session bean " +
                                   " timeout happened");
            } else {
                throw new Exception("Stateless session bean timeout " +
                                    "never happened");
            }

            if( timeoutInterceptorCalled ) {
                throw new Exception("Error : interceptor was invoked." +
                                    "Interceptors should not apply to timeout"
                                    + "methods");
            } else {
                System.out.println("Interceptor was successfully ignored for "
                                   + "timeout method");
            }

            InitialContext ic = new InitialContext();

            // Proprietary way to look up tx manager.  
            TransactionManager tm = (TransactionManager)
                ic.lookup("java:appserver/TransactionManager");
            // Use an implementation-specific check to ensure that there
            // is no tx.  A portable application couldn't make this check
            // since the exact tx behavior for TX_NOT_SUPPORTED is not
            // defined.
            int txStatus = tm.getStatus();
            if( txStatus == Status.STATUS_NO_TRANSACTION ) {
                System.out.println("Successfully verified tx attr = " +
                                   "TX_NOT_SUPPORTED in timeout method");
            } else {
                throw new Exception("Invalid tx status for TX_NOT_SUPPORTED" +
                                    " method " + txStatus);
            }

            Queue queue = (Queue) ic.lookup("java:comp/env/jms/ClientQueue");
            connection = qcFactory.createQueueConnection();
            QueueSession session = connection.createQueueSession(false,
                                   Session.AUTO_ACKNOWLEDGE);
            QueueSender sender = session.createSender(queue);
            TextMessage tmessage = session.createTextMessage();
            tmessage.setText("mdb() invoked");
            System.out.println("Sending message");
            sender.send(tmessage);
            System.out.println("message sent");

        } catch(Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if(connection != null) {
                    connection.close();
                }
            } catch(Exception e) {
                e.printStackTrace();
            }
        }

    }

    @AroundInvoke
    public Object intercept(InvocationContext inv)
	throws Exception
    {
	System.out.println("[mdb] Interceptor invoked...");
        System.out.println("method = " + inv.getMethod());
        System.out.println("params = " + inv.getParameters());
        int i = 0;
        for(Object o : inv.getParameters()) {
            System.out.println("param" + i + " = " + o);
            i++;
        }

        if( inv.getMethod().getName().equals("onMessage") ) {
            onMessageInterceptorCalled = true;
        } else if( inv.getMethod().getName().equals("timeout") ) {
            // This shouldn't happen.  Interceptors don't apply to timeout
            // methods.
            timeoutInterceptorCalled = true;
        }


	Object o = inv.proceed();
	System.out.println("[mdb] Interceptor after proceed()...");

        return o;
    }

}