com.github.zhengframework.jdbc.Sql2oProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zheng-jdbc-sql2o Show documentation

zheng framework module: jdbc sql2o support

There is a newer version: 1.8.0

package com.github.zhengframework.jdbc;

import com.google.inject.Injector;
import com.google.inject.Key;
import java.lang.annotation.Annotation;
import javax.inject.Provider;
import javax.sql.DataSource;
import org.sql2o.Sql2o;

public class Sql2oProvider implements Provider {

  private final Annotation qualifier;
  private Provider injectorProvider;

  public Sql2oProvider(Annotation qualifier, Provider injectorProvider) {
    this.qualifier = qualifier;
    this.injectorProvider = injectorProvider;
  }

  @Override
  public Sql2o get() {
    DataSource dataSource;

    Injector injector = injectorProvider.get();
    if (qualifier == null) {
      dataSource = injector.getInstance(Key.get(DataSource.class));
    } else {
      dataSource = injector
          .getInstance(Key.get(DataSource.class, qualifier));
    }
    return new Sql2o(dataSource);
  }
}