com.nesscomputing.jdbi.SqlObjectBinder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ness-jdbi Show documentation
Ness JDBI support
There is a newer version: 1.6.0
/**
 * Copyright (C) 2012 Ness Computing, Inc.
 *
 * 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 com.nesscomputing.jdbi;

import static java.lang.String.format;

import java.lang.annotation.Annotation;
import java.security.ProviderException;

import org.skife.jdbi.v2.IDBI;

import com.google.inject.Binder;
import com.google.inject.Inject;
import com.google.inject.Injector;
import com.google.inject.Key;
import com.google.inject.Provider;
import com.google.inject.Scopes;
import com.google.inject.name.Names;

public final class SqlObjectBinder
{
    private static enum Style
    {
        ON_DEMAND, OPEN
    };

    private SqlObjectBinder()
    {
    }

    public static  void bindSqlObjectOnDemand(final Binder binder, final Class sqlObjectClass, final String dbName)
    {
        bindSqlObjectOnDemand(binder, sqlObjectClass, Names.named(dbName));
    }

    public static  void bindSqlObjectOnDemand(final Binder binder, final Class sqlObjectClass, final Annotation annotation)
    {
        binder.bind(sqlObjectClass).toProvider(new SqlObjectProvider(sqlObjectClass, annotation, Style.ON_DEMAND)).in(Scopes.SINGLETON);
    }

    public static  void bindSqlObjectOpen(final Binder binder, final Class sqlObjectClass, final String dbName)
    {
        bindSqlObjectOnDemand(binder, sqlObjectClass, Names.named(dbName));
    }

    public static  void bindSqlObjectOpen(final Binder binder, final Class sqlObjectClass, final Annotation annotation)
    {
        binder.bind(sqlObjectClass).toProvider(new SqlObjectProvider(sqlObjectClass, annotation, Style.OPEN)).in(Scopes.SINGLETON);
    }


    static class SqlObjectProvider implements Provider
    {
        private final Class sqlObjectClass;
        private final Annotation annotation;
        private final Style style;

        private IDBI idbi = null;

        private SqlObjectProvider(final Class sqlObjectClass, final Annotation annotation, final Style style)
        {
            this.sqlObjectClass = sqlObjectClass;
            this.annotation = annotation;
            this.style = style;
        }

        @Inject
        void setInjector(final Injector injector)
        {
            this.idbi = injector.getInstance(Key.get(IDBI.class, annotation));
        }

        @Override
        public T get()
        {
            switch (style) {
                case ON_DEMAND:
                    return idbi.onDemand(sqlObjectClass);
                case OPEN:
                    return idbi.open(sqlObjectClass);
            }
            throw new ProviderException(format("Unknown provider style: %s", style));
        }
    }
}