org.mybatis.guice.datasource.dbcp.PerUserPoolDataSourceModule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mybatis-guice Show documentation
The MyBatis Guice module is easy-to-use Google Guice bridge for MyBatis sql mapping framework.
There is a newer version: 4.0.0
/**
 *    Copyright 2009-2015 the original author or authors.
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */
package org.mybatis.guice.datasource.dbcp;

import com.google.inject.AbstractModule;
import com.google.inject.Provider;
import com.google.inject.Scopes;
import com.google.inject.TypeLiteral;

import java.lang.annotation.Annotation;
import java.util.Map;

/**
 * @version $Id$
 */
public final class PerUserPoolDataSourceModule extends AbstractModule {

    private static final TypeLiteral> STRING_BOOLEAN_MAP = new TypeLiteral>(){};

    private static final TypeLiteral> STRING_INTEGER_MAP = new TypeLiteral>(){};

    private final Class>> perUserDefaultAutoCommitProviderClass;

    private final Class>> perUserDefaultReadOnlyProviderClass;

    private final Class>> perUserDefaultTransactionIsolationProviderClass;

    private final Class>> perUserMaxActiveProviderClass;

    private final Class>> perUserMaxIdleProviderClass;

    private final Class>> perUserMaxWaitProviderClass;

    private PerUserPoolDataSourceModule(
            Class>> perUserDefaultAutoCommitProviderClass,
            Class>> perUserDefaultReadOnlyProviderClass,
            Class>> perUserDefaultTransactionIsolationProviderClass,
            Class>> perUserMaxActiveProviderClass,
            Class>> perUserMaxIdleProviderClass,
            Class>> perUserMaxWaitProviderClass) {
        this.perUserDefaultAutoCommitProviderClass = perUserDefaultAutoCommitProviderClass;
        this.perUserDefaultReadOnlyProviderClass = perUserDefaultReadOnlyProviderClass;
        this.perUserDefaultTransactionIsolationProviderClass = perUserDefaultTransactionIsolationProviderClass;
        this.perUserMaxActiveProviderClass = perUserMaxActiveProviderClass;
        this.perUserMaxIdleProviderClass = perUserMaxIdleProviderClass;
        this.perUserMaxWaitProviderClass = perUserMaxWaitProviderClass;
    }

    @Override
    protected void configure() {
        doBind(perUserDefaultAutoCommitProviderClass, STRING_BOOLEAN_MAP, PerUserDefaultAutoCommit.class);
        doBind(perUserDefaultReadOnlyProviderClass, STRING_BOOLEAN_MAP, PerUserDefaultReadOnly.class);
        doBind(perUserDefaultTransactionIsolationProviderClass, STRING_INTEGER_MAP, PerUserDefaultTransactionIsolation.class);
        doBind(perUserMaxActiveProviderClass, STRING_INTEGER_MAP, PerUserMaxActive.class);
        doBind(perUserMaxIdleProviderClass, STRING_INTEGER_MAP, PerUserMaxIdle.class);
        doBind(perUserMaxWaitProviderClass, STRING_INTEGER_MAP, PerUserMaxWait.class);
    }

    private  void doBind(Class> providerClass, TypeLiteral typeLiteral, Class annotation) {
        if (providerClass != null) {
            bind(typeLiteral)
                  .annotatedWith(annotation)
                  .toProvider(providerClass)
                  .in(Scopes.SINGLETON);
        }
    }

    public static final class Builder {

        private Class>> perUserDefaultAutoCommitProviderClass;

        private Class>> perUserDefaultReadOnlyProviderClass;

        private Class>> perUserDefaultTransactionIsolationProviderClass;

        private Class>> perUserMaxActiveProviderClass;

        private Class>> perUserMaxIdleProviderClass;

        private Class>> perUserMaxWaitProviderClass;

        /**
         * @param perUserDefaultAutoCommitProviderClass the perUserDefaultAutoCommitProviderClass to set
         */
        public Builder setPerUserDefaultAutoCommitProviderClass(
                Class>> perUserDefaultAutoCommitProviderClass) {
            this.perUserDefaultAutoCommitProviderClass = perUserDefaultAutoCommitProviderClass;
            return this;
        }

        /**
         * @param perUserDefaultReadOnlyProviderClass the perUserDefaultReadOnlyProviderClass to set
         */
        public Builder setPerUserDefaultReadOnlyProviderClass(
                Class>> perUserDefaultReadOnlyProviderClass) {
            this.perUserDefaultReadOnlyProviderClass = perUserDefaultReadOnlyProviderClass;
            return this;
        }

        /**
         * @param perUserDefaultTransactionIsolationProviderClass the perUserDefaultTransactionIsolationProviderClass to set
         */
        public Builder setPerUserDefaultTransactionIsolationProviderClass(
                Class>> perUserDefaultTransactionIsolationProviderClass) {
            this.perUserDefaultTransactionIsolationProviderClass = perUserDefaultTransactionIsolationProviderClass;
            return this;
        }

        /**
         * @param perUserMaxActiveProviderClass the perUserMaxActiveProviderClass to set
         */
        public Builder setPerUserMaxActiveProviderClass(
                Class>> perUserMaxActiveProviderClass) {
            this.perUserMaxActiveProviderClass = perUserMaxActiveProviderClass;
            return this;
        }

        /**
         * @param perUserMaxIdleProviderClass the perUserMaxIdleProviderClass to set
         */
        public Builder setPerUserMaxIdleProviderClass(
                Class>> perUserMaxIdleProviderClass) {
            this.perUserMaxIdleProviderClass = perUserMaxIdleProviderClass;
            return this;
        }

        /**
         * @param perUserMaxWaitProviderClass the perUserMaxWaitProviderClass to set
         */
        public Builder setPerUserMaxWaitProviderClass(
                Class>> perUserMaxWaitProviderClass) {
            this.perUserMaxWaitProviderClass = perUserMaxWaitProviderClass;
            return this;
        }

        public PerUserPoolDataSourceModule create() {
            return new PerUserPoolDataSourceModule(perUserDefaultAutoCommitProviderClass,
                    perUserDefaultReadOnlyProviderClass,
                    perUserDefaultTransactionIsolationProviderClass,
                    perUserMaxActiveProviderClass,
                    perUserMaxIdleProviderClass,
                    perUserMaxWaitProviderClass);
        }

    }

}