001    /**
002     *  Licensed to the Apache Software Foundation (ASF) under one or more
003     *  contributor license agreements.  See the NOTICE file distributed with
004     *  this work for additional information regarding copyright ownership.
005     *  The ASF licenses this file to You under the Apache License, Version 2.0
006     *  (the "License"); you may not use this file except in compliance with
007     *  the License.  You may obtain a copy of the License at
008     *
009     *     http://www.apache.org/licenses/LICENSE-2.0
010     *
011     *  Unless required by applicable law or agreed to in writing, software
012     *  distributed under the License is distributed on an "AS IS" BASIS,
013     *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     *  See the License for the specific language governing permissions and
015     *  limitations under the License.
016     */
017    
018    package org.apache.geronimo.naming.reference;
019    
020    import java.util.Map;
021    
022    import javax.naming.NameNotFoundException;
023    
024    import org.apache.geronimo.kernel.repository.Artifact;
025    import org.apache.geronimo.kernel.Kernel;
026    import org.apache.geronimo.kernel.GBeanNotFoundException;
027    import org.apache.geronimo.gbean.AbstractNameQuery;
028    import org.apache.geronimo.gbean.AbstractName;
029    
030    /**
031     * @version $Rev: 534748 $ $Date: 2007-05-03 04:07:09 -0400 (Thu, 03 May 2007) $
032     */
033    public class PersistenceContextReference extends ConfigurationAwareReference {
034    
035        private boolean transactionScoped;
036        private Map properties;
037    
038        public PersistenceContextReference(Artifact[] configId, AbstractNameQuery abstractNameQuery, boolean transactionScoped, Map properties) {
039            super(configId, abstractNameQuery);
040            this.transactionScoped = transactionScoped;
041            this.properties = properties;
042        }
043    
044        public String getClassName() {
045            return "javax.persistence.EntityManager";
046        }
047        
048        public Object getContent() throws NameNotFoundException {
049        Kernel kernel = getKernel();
050    
051        AbstractName target;
052        try {
053            target = resolveTargetName();
054        } catch (GBeanNotFoundException e) {
055            throw (NameNotFoundException) new NameNotFoundException("Could not resolve name query: " + abstractNameQueries).initCause(e);
056        }
057    
058        Object entityManager;
059        try {
060            entityManager = kernel.invoke(target, "getEntityManager", new Object[] {Boolean.valueOf(transactionScoped), properties}, new String[] {boolean.class.getName(), Map.class.getName()});
061        } catch (Exception e) {
062            throw (IllegalStateException) new IllegalStateException("Could not get entityManager").initCause(e);
063        }
064        if (entityManager == null) {
065            throw new IllegalStateException("entity manager not returned. Target " + target + " not started");
066        }
067        return entityManager;
068    }
069    }