org.sagacity.sqltoy.solon.DbManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sagacity-sqltoy-solon-plugin Show documentation
sagacity-sqltoy-solon-plugin
The newest version!
package org.sagacity.sqltoy.solon;

import java.util.HashMap;
import java.util.Map;

import javax.sql.DataSource;

import org.noear.solon.core.AppContext;
import org.sagacity.sqltoy.SqlToyContext;
import org.sagacity.sqltoy.dao.LightDao;
import org.sagacity.sqltoy.dao.SqlToyLazyDao;
import org.sagacity.sqltoy.service.SqlToyCRUDService;
import org.sagacity.sqltoy.solon.dao.impl.LightDaoImpl;
import org.sagacity.sqltoy.solon.dao.impl.SqlToyLazyDaoImpl;
import org.sagacity.sqltoy.solon.service.impl.SqlToyCRUDServiceForSolon;
/**
 * @author 夜の孤城
 * @since 1.5
 * */
public class DbManager {
    private static Map daoMap = new HashMap<>();
    private static Map lightDaoMap = new HashMap<>();
    private static Map serviceMap = new HashMap<>();
    private static SqlToyContext context;

    public static void setContext(SqlToyContext context) {
        DbManager.context = context;
    }

    public static synchronized SqlToyLazyDao getDao(DataSource dataSource) {
        SqlToyLazyDao dao = daoMap.get(dataSource);

        if (dao == null) {
            SqlToyLazyDaoImpl sqlToyLazyDao = new SqlToyLazyDaoImpl();
            sqlToyLazyDao.setDataSource(dataSource);
            sqlToyLazyDao.setSqlToyContext(context);
            daoMap.put(dataSource, sqlToyLazyDao);
            dao = sqlToyLazyDao;
        }

        return dao;
    }
    public static synchronized LightDao getLightDao(DataSource dataSource) {
        LightDao dao = lightDaoMap.get(dataSource);

        if (dao == null) {
            LightDaoImpl sqlToyLazyDao = new LightDaoImpl();
            sqlToyLazyDao.setDataSource(dataSource);
            sqlToyLazyDao.setSqlToyContext(context);
            lightDaoMap.put(dataSource, sqlToyLazyDao);
            dao = sqlToyLazyDao;
        }
        return dao;
    }
    public static synchronized SqlToyCRUDService getService(AppContext context, DataSource dataSource) {
        SqlToyCRUDService service = serviceMap.get(dataSource);
        if (service == null) {
            SqlToyCRUDServiceForSolon crudService = context.beanMake(SqlToyCRUDServiceForSolon.class).get();
            crudService.setSqlToyLazyDao(getDao(dataSource));
            serviceMap.put(dataSource, crudService);
            service = crudService;
        }
        return service;
    }
    public static Map getServiceMap() {
        return serviceMap;
    }
}