org.makersoft.shards.cfg.MyBatisConfigurationsWrapper.java Source code

Introduction

Here is the source code for org.makersoft.shards.cfg.MyBatisConfigurationsWrapper.java

Source

/*
 * @(#)MyBatisConfigurationsWrapper.java 2013-3-9 ?23:33:33
 *
 * Copyright (c) 2011-2013 Makersoft.org all rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 *
 */
package org.makersoft.shards.cfg;

import java.util.List;

import org.apache.ibatis.binding.BindingException;
import org.apache.ibatis.mapping.Environment;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.reflection.factory.ObjectFactory;
import org.apache.ibatis.session.Configuration;
import org.apache.ibatis.session.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;

/**
 * MyBatis-ShardsShard??Configuration.
 * (??)
 */
public class MyBatisConfigurationsWrapper extends Configuration {

    private final Configuration configuration;

    private final List<SqlSessionFactory> sqlSessionFactories;

    public MyBatisConfigurationsWrapper(Configuration configuration, List<SqlSessionFactory> sqlSessionFactories) {
        this.configuration = configuration;
        this.sqlSessionFactories = sqlSessionFactories;
    }

    @Override
    public Environment getEnvironment() {
        return configuration.getEnvironment();
    }

    @Override
    public ExecutorType getDefaultExecutorType() {
        return configuration.getDefaultExecutorType();
    }

    @Override
    public ObjectFactory getObjectFactory() {
        return configuration.getObjectFactory();
    }

    @Override
    public <T> T getMapper(Class<T> type, SqlSession sqlSession) {
        for (SqlSessionFactory sqlSessionFactory : getSqlSessionFactories()) {
            T mapper = null;
            try {
                mapper = sqlSessionFactory.getConfiguration().getMapper(type, sqlSession);
            } catch (BindingException e) {
                // ignore exception
            }

            if (mapper != null) {
                return mapper;
            }
        }

        throw new BindingException("Type " + type + " is not known to the MapperRegistry.");

    }

    @Override
    public boolean hasMapper(Class<?> type) {
        for (SqlSessionFactory factory : getSqlSessionFactories()) {
            if (factory.getConfiguration().hasMapper(type)) {
                return true;
            }
        }

        return false;
    }

    @Override
    public MappedStatement getMappedStatement(String id) {
        Exception exception = null;
        MappedStatement mappedStatement = null;
        for (SqlSessionFactory sqlSessionFactory : getSqlSessionFactories()) {
            try {
                mappedStatement = sqlSessionFactory.getConfiguration().getMappedStatement(id);
            } catch (Exception e) {
                // ignore exception
                exception = e;
            }

            if (mappedStatement != null) {
                return mappedStatement;
            }
        }

        throw new BindingException("Invalid bound statement (not found): " + id, exception);
    }

    private List<SqlSessionFactory> getSqlSessionFactories() {
        return sqlSessionFactories;
    }
}