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.criteria;
017
018import org.apache.commons.lang.StringUtils;
019import org.kuali.rice.core.api.criteria.GenericQueryResults;
020import org.kuali.rice.core.api.criteria.LookupCustomizer;
021import org.kuali.rice.core.api.criteria.QueryByCriteria;
022import org.kuali.rice.krad.bo.ModuleConfiguration;
023import org.kuali.rice.krad.service.KRADServiceLocatorWeb;
024import org.kuali.rice.krad.service.KualiModuleService;
025import org.kuali.rice.krad.service.ModuleService;
026import org.kuali.rice.krad.util.LegacyUtils;
027
028import java.util.Collections;
029import java.util.HashMap;
030import java.util.Map;
031
032public class CriteriaLookupDaoProxy implements CriteriaLookupDao {
033    CriteriaLookupDao criteriaLookupDaoOjb;
034
035    private static KualiModuleService kualiModuleService;
036    private static Map<String, CriteriaLookupDao> lookupDaoValues = Collections.synchronizedMap(new HashMap<String, CriteriaLookupDao>());
037
038        public void setCriteriaLookupDaoOjb(CriteriaLookupDao lookupDaoOjb) {
039                this.criteriaLookupDaoOjb = lookupDaoOjb;
040        }
041
042    private CriteriaLookupDao getDao(Class clazz) {
043        ModuleService moduleService = getKualiModuleService().getResponsibleModuleService(clazz);
044        if (moduleService != null) {
045            ModuleConfiguration moduleConfig = moduleService.getModuleConfiguration();
046            String dataSourceName = "";
047            if (moduleConfig != null) {
048                dataSourceName = moduleConfig.getDataSourceName();
049            }
050
051            if (StringUtils.isNotEmpty(dataSourceName)) {
052                if (lookupDaoValues.get(dataSourceName) != null) {
053                    return lookupDaoValues.get(dataSourceName);
054                } else {
055                                CriteriaLookupDaoOjb classSpecificLookupDaoOjb = new CriteriaLookupDaoOjb();
056                    classSpecificLookupDaoOjb.setJcdAlias(dataSourceName);
057                    lookupDaoValues.put(dataSourceName, classSpecificLookupDaoOjb);
058                    return classSpecificLookupDaoOjb;
059                }
060
061            }
062        }
063        return criteriaLookupDaoOjb;
064    }
065
066    @Override
067    public <T> GenericQueryResults<T> lookup(Class<T> queryClass, QueryByCriteria criteria) {
068        return getDao(queryClass).lookup(queryClass, criteria);
069    }
070
071    @Override
072    public <T> GenericQueryResults<T> lookup(Class<T> queryClass, QueryByCriteria criteria,
073            LookupCustomizer<T> customizer) {
074        return getDao(queryClass).lookup(queryClass, criteria, customizer);
075    }
076
077    private static KualiModuleService getKualiModuleService() {
078        if (kualiModuleService == null) {
079            kualiModuleService = KRADServiceLocatorWeb.getKualiModuleService();
080        }
081        return kualiModuleService;
082    }
083}