001/**
002 * Copyright 2005-2016 The Kuali Foundation
003 *
004 * Licensed under the Educational Community License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.opensource.org/licenses/ecl2.php
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package org.kuali.rice.krad.dao.proxy;
017
018import org.apache.commons.lang.StringUtils;
019import org.kuali.rice.core.api.config.ConfigurationException;
020import org.kuali.rice.core.framework.persistence.jpa.OrmUtils;
021import org.kuali.rice.krad.bo.ModuleConfiguration;
022import org.kuali.rice.krad.dao.BusinessObjectDao;
023import org.kuali.rice.krad.dao.DocumentDao;
024import org.kuali.rice.krad.dao.impl.DocumentDaoJpa;
025import org.kuali.rice.krad.dao.impl.DocumentDaoOjb;
026import org.kuali.rice.krad.document.Document;
027import org.kuali.rice.krad.service.DocumentAdHocService;
028import org.kuali.rice.krad.service.KRADServiceLocatorWeb;
029import org.kuali.rice.krad.service.KualiModuleService;
030import org.kuali.rice.krad.service.ModuleService;
031import org.springframework.transaction.annotation.Transactional;
032
033import javax.persistence.EntityManager;
034import java.util.List;
035import java.util.Map;
036import java.util.concurrent.ConcurrentHashMap;
037
038@Transactional
039public class DocumentDaoProxy implements DocumentDao {
040
041    private DocumentDao documentDaoJpa;
042    private DocumentDao documentDaoOjb;
043
044    private static KualiModuleService kualiModuleService;
045    private static Map<String, DocumentDao> documentDaoValues = new ConcurrentHashMap<String, DocumentDao>();
046
047    private DocumentDao getDao(Class<? extends Document> clazz) {
048        ModuleService moduleService = getKualiModuleService().getResponsibleModuleService(clazz);
049        if (moduleService != null) {
050            ModuleConfiguration moduleConfig = moduleService.getModuleConfiguration();
051            String dataSourceName = "";
052            EntityManager entityManager = null;
053            if (moduleConfig != null) {
054                dataSourceName = moduleConfig.getDataSourceName();
055                entityManager = moduleConfig.getEntityManager();
056            }
057
058            if (StringUtils.isNotEmpty(dataSourceName)) {
059                if (documentDaoValues.get(dataSourceName) != null) {
060                    return documentDaoValues.get(dataSourceName);
061                }
062                if (OrmUtils.isJpaAnnotated(clazz) && (OrmUtils.isJpaEnabled() || OrmUtils.isJpaEnabled("rice.krad"))) {
063                        //using JPA
064                        if (entityManager != null) {
065                                // we set the entity manager directly in the constructor
066                                DocumentDaoJpa documentDaoJpaInstance =
067                                        new DocumentDaoJpa(entityManager, this.documentDaoJpa.getBusinessObjectDao(),
068                                                        this.documentDaoJpa.getDocumentAdHocService());
069
070                                documentDaoValues.put(dataSourceName, documentDaoJpaInstance);
071                                return documentDaoJpaInstance;
072                        }
073                        throw new ConfigurationException("EntityManager is null. EntityManager must be set in the Module Configuration bean in the appropriate spring beans xml. (see nested exception for details).");
074                }
075                //using OJB
076                DocumentDaoOjb documentDaoOjbInstance =
077                        new DocumentDaoOjb(
078                                        this.documentDaoOjb.getBusinessObjectDao(),
079                                        this.documentDaoOjb.getDocumentAdHocService());
080
081                // set the data source alias
082                documentDaoOjbInstance.setJcdAlias(dataSourceName);
083
084                documentDaoValues.put(dataSourceName, documentDaoOjbInstance);
085                return documentDaoOjbInstance;
086
087                }
088
089        }
090        return (OrmUtils.isJpaAnnotated(clazz) && OrmUtils.isJpaEnabled()) ? documentDaoJpa : documentDaoOjb;
091    }
092    
093    /**
094         * @see org.kuali.rice.krad.dao.DocumentDao#save(org.kuali.rice.krad.document.Document)
095         */
096    @Override
097        public <T extends Document> T save(T document) {
098                return getDao(document.getClass()).save(document);
099        }
100    
101        /**
102         * @see org.kuali.rice.krad.dao.DocumentDao#findByDocumentHeaderId(java.lang.Class, java.lang.String)
103         */
104    @Override
105        public <T extends Document> T findByDocumentHeaderId(Class<T> clazz, String id) {
106                return getDao(clazz).findByDocumentHeaderId(clazz, id);
107        }
108
109        /**
110         * @see org.kuali.rice.krad.dao.DocumentDao#findByDocumentHeaderIds(java.lang.Class, java.util.List)
111         */
112    @Override
113        public <T extends Document> List<T> findByDocumentHeaderIds(Class<T> clazz, List<String> idList) {
114                return getDao(clazz).findByDocumentHeaderIds(clazz, idList);
115        }
116
117        /**
118         * @see org.kuali.rice.krad.dao.DocumentDao#getBusinessObjectDao()
119         */
120    @Override
121        public BusinessObjectDao getBusinessObjectDao() {
122                if (OrmUtils.isJpaEnabled()) {
123                        return documentDaoJpa.getBusinessObjectDao();
124                }
125                return documentDaoOjb.getBusinessObjectDao();
126        }
127
128        /**
129         * @see org.kuali.rice.krad.dao.DocumentDao#getDocumentAdHocService()
130         */
131    @Override
132        public DocumentAdHocService getDocumentAdHocService() {
133                if (OrmUtils.isJpaEnabled()) {
134                        return documentDaoJpa.getDocumentAdHocService();
135                }
136                return documentDaoOjb.getDocumentAdHocService();
137    }
138        
139        public void setDocumentDaoJpa(DocumentDao documentDaoJpa) {
140                this.documentDaoJpa = documentDaoJpa;
141        }
142
143        public void setDocumentDaoOjb(DocumentDao documentDaoOjb) {
144                this.documentDaoOjb = documentDaoOjb;
145        }
146
147    private synchronized static KualiModuleService getKualiModuleService() {
148        if (kualiModuleService == null) {
149            kualiModuleService = KRADServiceLocatorWeb.getKualiModuleService();
150        }
151        return kualiModuleService;
152    }
153
154}