appserver/tests/appserv-tests/devtests/ejb/sfsb/cacheNPE/ejb/SFSBEJB.java

/*
 * Copyright (c) 2017, 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.sfsb.cacheNPE.ejb;

import jakarta.ejb.SessionBean;
import jakarta.ejb.SessionContext;
import jakarta.ejb.*;

import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.*;
import javax.rmi.PortableRemoteObject;

import jakarta.transaction.UserTransaction;

import java.rmi.RemoteException;

public class SFSBEJB
    implements SessionBean
{
    private Context envCtx;
    private Context envSubCtx;
    private Context javaCtx;
    private Context javaCompCtx;

    private transient String message;

    private SessionContext              sessionCtx;
    private Context                     initialCtx;
    private String                      sfsbName;
    private String                      envEntryTagValue;
    private SimpleEntityHome            entityHome;
    private SimpleEntityRemote          entityRemote;
    private SimpleEntityLocalHome       entityLocalHome;
    private SimpleEntityLocal           entityLocal;
    private HomeHandle                  homeHandle;
    private Handle                      handle;
    private UserTransaction             userTransaction1;
    private UserTransaction             userTransaction2;

    public void ejbCreate(String sfsbName) {
        System.out.println ("In SFSB.ejbCreate() for name -> " + sfsbName);
        this.sfsbName = sfsbName;

        try {
            entityRemote = entityHome.create(sfsbName, sfsbName);

            entityLocal = entityLocalHome.findByPrimaryKey(sfsbName);


            homeHandle = entityHome.getHomeHandle();
            handle = entityRemote.getHandle();

            userTransaction1 = sessionCtx.getUserTransaction();
            userTransaction2 = (UserTransaction) new InitialContext().
                lookup("java:comp/UserTransaction");


        } catch (Exception ex) {
            ex.printStackTrace();
            //TODO
        }
    }

    public String getName() {
        return this.sfsbName;
    }

    public boolean checkSessionContext() {
        boolean status = sessionCtx != null;
        status = status && (sessionCtx.getEJBObject() != null);
        return status;
    }

    public boolean checkInitialContext() {
        boolean status = (initialCtx != null);
        //status = status && lookupEntityHome();
        return status;
    }

    public boolean checkEntityHome() {
        boolean status = entityHome != null;
        try {
            status = status && (entityHome.findByPrimaryKey(sfsbName) != null);
        } catch (Exception ex) {
            status = false;
        }

        return status;
    }

    public boolean checkEntityLocalHome() {
        boolean status = entityHome != null;
        try {
            status = status && (entityLocalHome.findByPrimaryKey(sfsbName) != null);
        } catch (Exception ex) {
            status = false;
        }

        return status;
    }

    public boolean checkEntityRemoteRef() {
        boolean status = entityHome != null;
        try {
            status = status && (entityHome.findByPrimaryKey(sfsbName) != null);
        } catch (Exception ex) {
            status = false;
        }

        return status;
    }

    public boolean checkEntityLocalRef() {
        boolean status = entityLocalHome != null;
        try {
            status = status
                && (entityLocalHome.findByPrimaryKey(sfsbName) != null);
        } catch (Exception ex) {
            status = false;
        }

        return status;
    }

    public boolean checkHomeHandle() {
        boolean status = homeHandle != null;
        try {
            if (status) {
                Object homeRef = homeHandle.getEJBHome();
                SimpleEntityHome h = (SimpleEntityHome)
                    PortableRemoteObject.narrow(homeRef, SimpleEntityHome.class);
                EJBMetaData metaData2 = h.getEJBMetaData();
            }
        } catch (Exception ex) {
            ex.printStackTrace();
            status = false;
        }

        return status;
    }

    public boolean checkHandle() {
        boolean status = handle != null;
        try {
            if (status) {
                Object ref = handle.getEJBObject();
                SimpleEntityRemote ejbRef = (SimpleEntityRemote)
                    PortableRemoteObject.narrow(ref, SimpleEntityRemote.class);
                status = ejbRef.getPrimaryKey().equals(sfsbName);
            }
        } catch (Exception ex) {
            ex.printStackTrace();
        }

        return status;
    }

    public boolean checkUserTransaction() {
        boolean status =
            ((userTransaction1 != null) && (userTransaction2 != null));

        try {
            if( status ) {
                userTransaction1.begin();
                userTransaction1.commit();

                userTransaction2.begin();
                userTransaction2.commit();
            }
        } catch (Exception ex) {
            ex.printStackTrace();
            status = false;
        }

        return status;
    }

    public boolean isOK(String name) {
        String fieldName = "Name";
        boolean ok = name.equals(sfsbName);

        try {
            if (ok) {
                fieldName = "SessionContext";
                ok = sessionCtx != null;
            }
            if (ok) {
                fieldName = "InitialContext";
                ok = initialCtx != null;
            }
            if (ok) {
                fieldName = "java:";
                ok = ( (javaCtx != null) &&
                       javaCtx.getNameInNamespace().equals(fieldName) );
            }
            if (ok) {
                fieldName = "java:comp";
                ok = ( (javaCompCtx != null) &&
                       javaCompCtx.getNameInNamespace().equals(fieldName) );
            }
            if (ok) {
                fieldName = "java:comp/env";
                ok = ( (envCtx != null) &&
                       envCtx.getNameInNamespace().equals(fieldName) );

            }
            if (ok) {
                fieldName = "java:comp/env/ejb";
                ok = ( (envSubCtx != null) &&
                       envSubCtx.getNameInNamespace().equals(fieldName) );
            }
            if (ok) {
                fieldName = "env-entry";

                String value1 = (String)
                    initialCtx.lookup("java:comp/env/TagValue");
                String value2 = (String) envCtx.lookup("TagValue");
                ok = (value1 != null)
                    && (value1.equals(value2))
                    && (value1.equals(envEntryTagValue));
            }
        } catch (Exception ex) {
            ex.printStackTrace();
            ok = false;
        }

        this.message = (ok) ? null : (fieldName + " not restored properly");

        return ok;
    }

    public String getMessage() {
        return this.message;
    }

    public void setSessionContext(SessionContext sc) {
        this.sessionCtx = sc;
        try {
            this.initialCtx = new InitialContext();
            this.javaCtx = (Context) initialCtx.lookup("java:");
            this.javaCompCtx = (Context) initialCtx.lookup("java:comp");
            this.envCtx = (Context) initialCtx.lookup("java:comp/env");
            this.envSubCtx = (Context) initialCtx.lookup("java:comp/env/ejb");
            this.envEntryTagValue = (String)
                envCtx.lookup("TagValue");
            lookupEntityHome();
            lookupEntityLocalHome();
        } catch (Throwable th) {
            th.printStackTrace();
        }
    }

    public void ejbRemove() {
        System.out.println ("\n\tIn SFSB.ejbRemove() for name -> " + sfsbName);
        try {
            entityRemote.remove();
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }

    public void ejbActivate() {
        System.out.println ("\n\tIn SFSB.ejbActivate() for name -> " + sfsbName);
    }

    public void ejbPassivate() {
        System.out.println ("\n\t####In SFSB.ejbPassivate() for: " + sfsbName);
    }

    private boolean lookupEntityHome() {
        boolean status = false;
        try {
            Object homeRef = initialCtx.lookup("java:comp/env/ejb/SimpleEntityHome");
            this.entityHome = (SimpleEntityHome)
                PortableRemoteObject.narrow(homeRef, SimpleEntityHome.class);

            status = true;
        } catch (Exception ex) {
            ex.printStackTrace();
        }

        return status;
    }

    private boolean lookupEntityLocalHome() {
        boolean status = false;
        try {
            Object homeRef = envSubCtx.lookup("SimpleEntityLocalHome");
            this.entityLocalHome = (SimpleEntityLocalHome) homeRef;

            status = true;
        } catch (Exception ex) {
            ex.printStackTrace();
        }

        return status;
    }

}