org.fastcatsearch.db.InternalDBModule.java Source code

Introduction

Here is the source code for org.fastcatsearch.db.InternalDBModule.java

Source

/*
 * Copyright (c) 2013 Websquared, Inc.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v2.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * 
 * Contributors:
 *     swsong - initial API and implementation
 */

package org.fastcatsearch.db;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.List;
import java.util.Properties;

import org.apache.ibatis.builder.xml.XMLMapperBuilder;
import org.apache.ibatis.datasource.pooled.PooledDataSource;
import org.apache.ibatis.session.Configuration;
import org.apache.ibatis.session.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.apache.ibatis.transaction.jdbc.JdbcTransactionFactory;
import org.fastcatsearch.env.Environment;
import org.fastcatsearch.module.AbstractModule;
import org.fastcatsearch.module.ModuleException;
import org.fastcatsearch.service.ServiceManager;
import org.fastcatsearch.settings.Settings;

public class InternalDBModule extends AbstractModule {

    private final static String derbyEmbeddedDriver = "org.apache.derby.jdbc.EmbeddedDriver";
    private final static String detbyUrlPrefix = "jdbc:derby:";
    private final static String detbyUrlSuffix = ";create=true"; //create if not exists
    private final String dbPath;

    protected static InternalDBModule instance;

    private List<URL> mapperFileList;

    private SqlSessionFactory sqlSessionFactory;

    public static InternalDBModule getInstance() {
        return instance;
    }

    public void asSingleton() {
        instance = this;
    }

    public InternalDBModule(String dbPath, List<URL> mapperFileList, Environment environment, Settings settings) {
        super(environment, settings);
        if (new File(dbPath).exists()) {
            this.dbPath = detbyUrlPrefix + dbPath;
        } else {
            //create
            this.dbPath = detbyUrlPrefix + dbPath + detbyUrlSuffix;
        }
        this.mapperFileList = mapperFileList;
    }

    @Override
    protected boolean doLoad() throws ModuleException {
        Properties driverProperties = new Properties();
        driverProperties.setProperty("driver.encoding", "UTF-8");
        //******* driverProperties *****
        //poolMaximumActiveConnections
        //poolMaximumIdleConnections
        //poolMaximumCheckoutTime
        //poolTimeToWait
        //poolPingQuery
        //poolPingEnabled
        //poolPingConnectionsNotUsedFor
        //////////////////////////////////

        PooledDataSource dataSource = new PooledDataSource(derbyEmbeddedDriver, dbPath, driverProperties);
        org.apache.ibatis.mapping.Environment environment = new org.apache.ibatis.mapping.Environment("ID",
                new JdbcTransactionFactory(), dataSource);
        Configuration configuration = new Configuration(environment);

        if (mapperFileList != null) {
            for (URL mapperFile : mapperFileList) {
                addSqlMappings(configuration, mapperFile);
            }
        }

        sqlSessionFactory = new SqlSessionFactoryBuilder().build(configuration);
        logger.info("DBModule[{}] Loaded! with {}", dbPath, mapperFileList);
        return true;
    }

    public SqlSession openSession() {
        return sqlSessionFactory.openSession();
    }

    public SqlSession openBatchSession() {
        return sqlSessionFactory.openSession(ExecutorType.BATCH);
    }

    @Override
    protected boolean doUnload() throws ModuleException {
        //      try {
        logger.info(getClass().getSimpleName() + "[{}] Unloaded!, sqlSessionFactory = {}", dbPath,
                sqlSessionFactory);
        //         DriverManager.getConnection(dbPath+";shutdown=true", "","");
        //      } catch (SQLException e) {
        //shutdown? SQLException? ? OK.
        //         logger.info("{}", e.getMessage());
        //         return true;
        //      }
        return true;
    }

    private void addSqlMappings(Configuration conf, URL mapperFilePath) {
        InputStream is = null;
        try {
            is = mapperFilePath.openStream();
            XMLMapperBuilder xmlParser = new XMLMapperBuilder(is, conf, mapperFilePath.toString(),
                    conf.getSqlFragments());
            xmlParser.parse();
        } catch (IOException e) {
            logger.error("error loading mybatis mapping config file.", e);
        } finally {
            if (is != null) {
                try {
                    is.close();
                } catch (IOException e) {
                }
            }
        }
    }

    public static class MapperSession<T> {
        private SqlSession session;
        private T mapper;

        public SqlSession getSession() {
            return session;
        }

        public MapperSession(SqlSession session, T mapper) {
            this.session = session;
            this.mapper = mapper;
        }

        public T getMapper() {
            return mapper;
        }

        public void commit() {
            if (session != null) {
                session.commit();
            }
        }

        public void rollback() {
            if (session != null) {
                session.rollback();
            }
        }

        public void closeSession() {
            if (session != null) {
                session.commit();
                session.close();
            }
        }
    }

}