StatelessBeanManagedTransactionBean.java

/*
 * ModeShape (http://www.modeshape.org)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.modeshape.test.integration;

import java.util.LinkedList;
import javax.annotation.Resource;
import javax.ejb.Stateless;
import javax.ejb.TransactionManagement;
import javax.ejb.TransactionManagementType;
import javax.jcr.Node;
import javax.jcr.NodeIterator;
import javax.jcr.Repository;
import javax.jcr.Session;
import javax.jcr.SimpleCredentials;
import javax.jcr.nodetype.NodeType;
import javax.transaction.UserTransaction;
import org.jboss.logging.Logger;
import org.modeshape.jcr.api.JcrTools;

/**
 * A stateless EJB that accesses a Repository using multiple bean-managed transactions (BMT) and creates a JCR Session a variety
 * of ways.
 */
@Stateless
@TransactionManagement( TransactionManagementType.BEAN )
public class StatelessBeanManagedTransactionBean {

    private static Logger log = Logger.getLogger(StatelessBeanManagedTransactionBean.class);

    @Resource( mappedName = "java:/jcr/artifacts" )
    private Repository repository;

    @Resource
    private UserTransaction utxn;

    /** Stateless utility */
    private final JcrTools tools = new JcrTools();

    /**
     * Create a 3-level tree of nodes under the node at the specified path.
     * 
     * @param path the path of the node under which the tree should be created; this node will be created if missing
     * @param numNodesAtEachLevel
     * @param print flag indicating whether the status should be logged as INFO-level messages
     * @throws Exception
     */
    public void createNodes( String path,
                             int numNodesAtEachLevel,
                             boolean print ) throws Exception {
        if (path == null || path.trim().length() == 0) path = "/";
        utxn.begin();
        try {
            Session session = repository.login(new SimpleCredentials("admin", "admin".toCharArray()));
            try {

                // Find or create the node at the specified path ...
                Node topNode = tools.findOrCreateNode(session, path, NodeType.NT_UNSTRUCTURED);

                for (int i = 1; i <= numNodesAtEachLevel; i++) {
                    Node node = topNode.addNode("node" + i, NodeType.NT_UNSTRUCTURED);
                    node.setProperty("prop", "Property value for " + node.getIdentifier());
                    if (print) log.info("Populating " + node.getPath());
                    for (int j = 1; j <= numNodesAtEachLevel; j++) {
                        Node child = node.addNode("child" + j, NodeType.NT_UNSTRUCTURED);
                        child.setProperty("prop", "Property value for " + child.getIdentifier());
                        if (print) log.info("Populating " + child.getPath());
                        for (int k = 1; k <= numNodesAtEachLevel; k++) {
                            Node grandChild = child.addNode("grandchild" + k, NodeType.NT_UNSTRUCTURED);
                            grandChild.setProperty("prop", "Property value for " + grandChild.getIdentifier());
                        }
                    }
                    if (i != numNodesAtEachLevel) {
                        if (print) log.info("Saving session");
                        session.save();
                        if (print) log.info("Committing transaction");
                        utxn.commit();
                        utxn.begin();
                    }
                }

                if (print) log.info("Saving session");
                session.save();

            } finally {
                session.logout();
            }
        } finally {
            if (print) log.info("Committing transaction");
            utxn.commit();
        }

        if (print) log.info("CreateNodesTest completed");
    }

    /**
     * Verify a 3-level tree of nodes under the node at the specified path.
     * 
     * @param path the path of the node under which the tree should be created; this node will be created if missing
     * @param numNodesAtEachLevel
     * @param print flag indicating whether the status should be logged as INFO-level messages
     * @throws Exception
     */
    public void verifyNodesInTransaction( String path,
                                          int numNodesAtEachLevel,
                                          boolean print ) throws Exception {
        if (path == null || path.trim().length() == 0) path = "/";
        utxn.begin();
        try {
            Session session = repository.login(new SimpleCredentials("admin", "admin".toCharArray()));
            try {

                // Find or create the node at the specified path ...
                Node topNode = session.getNode(path);

                for (int i = 1; i <= numNodesAtEachLevel; i++) {
                    Node node = topNode.getNode("node" + i);
                    if (print) log.info("Verifying " + node.getPath());
                    node.getProperty("prop").getString().equals("Property value for " + node.getIdentifier());
                    for (int j = 1; j <= numNodesAtEachLevel; j++) {
                        Node child = node.getNode("child" + j);
                        if (print) log.info("Verifying " + child.getPath());
                        child.getProperty("prop").getString().equals("Property value for " + node.getIdentifier());
                        for (int k = 1; k <= numNodesAtEachLevel; k++) {
                            Node grandChild = child.getNode("grandchild" + k);
                            if (print) log.info("Verifying " + grandChild.getPath());
                            grandChild.getProperty("prop").getString().equals("Property value for " + node.getIdentifier());
                        }
                    }
                }
            } finally {
                if (session != null) session.logout();
            }
        } finally {
            utxn.commit();
        }
    }

    public void cleanup( String path,
                         boolean removeChildrenOnly ) throws Exception {
        utxn.begin();
        try {
            Session session = repository.login(new SimpleCredentials("admin", "admin".toCharArray()));
            try {
                Node topNode = session.getNode(path);

                if (removeChildrenOnly) {
                    // Record (in reverse order) all the children from the top-level node ...
                    LinkedList<Node> children = new LinkedList<Node>();
                    NodeIterator iter = topNode.getNodes("node*");
                    while (iter.hasNext()) {
                        Node child = iter.nextNode();
                        children.addFirst(child);
                    }

                    // Now remove all of the children (in reverse order) ...
                    for (Node child : children) {
                        child.remove();
                    }
                } else {
                    topNode.remove();
                }

                session.save();
            } finally {
                session.logout();
            }
        } finally {
            utxn.commit();
        }
    }
}