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 package org.apache.geronimo.naming.reference; 018 019 import javax.naming.NameNotFoundException; 020 021 import org.apache.geronimo.gbean.AbstractName; 022 import org.apache.geronimo.gbean.AbstractNameQuery; 023 import org.apache.geronimo.kernel.GBeanNotFoundException; 024 import org.apache.geronimo.kernel.Kernel; 025 import org.apache.geronimo.kernel.repository.Artifact; 026 027 /** 028 * @version $Rev: 534748 $ $Date: 2007-05-03 04:07:09 -0400 (Thu, 03 May 2007) $ 029 */ 030 public class PersistenceUnitReference extends ConfigurationAwareReference { 031 032 033 public PersistenceUnitReference(Artifact[] configId, AbstractNameQuery abstractNameQuery) { 034 super(configId, abstractNameQuery); 035 } 036 037 public String getClassName() { 038 return "javax.persistence.EntityManagerFactory"; 039 } 040 041 public Object getContent() throws NameNotFoundException { 042 Kernel kernel = getKernel(); 043 044 AbstractName target; 045 try { 046 target = resolveTargetName(); 047 } catch (GBeanNotFoundException e) { 048 throw (NameNotFoundException) new NameNotFoundException("Could not resolve name query: " + abstractNameQueries).initCause(e); 049 } 050 051 Object entityManagerFactory; 052 try { 053 entityManagerFactory = kernel.getAttribute(target, "entityManagerFactory"); 054 } catch (Exception e) { 055 throw (IllegalStateException) new IllegalStateException("Could not get EntityManagerFactory").initCause(e); 056 } 057 if (entityManagerFactory == null) { 058 throw new IllegalStateException("entity manager not returned. Target " + target + " not started"); 059 } 060 return entityManagerFactory; 061 } 062 }