Java DBI.open方法代碼示例

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
public AnetObjectEngine(DBI jdbi) { 
	dbHandle = jdbi.open();
	
	personDao = new PersonDao(dbHandle);
	poamDao = new PoamDao(dbHandle);
	locationDao =  new LocationDao(dbHandle);
	orgDao = new OrganizationDao(dbHandle);
	positionDao = new PositionDao(dbHandle);
	asDao = new ApprovalStepDao(dbHandle);
	approvalActionDao = new ApprovalActionDao(dbHandle);
	reportDao = new ReportDao(dbHandle);
	commentDao = new CommentDao(dbHandle);
	adminDao = new AdminDao(dbHandle);
	savedSearchDao = new SavedSearchDao(dbHandle);
	
	instance = this;
	
	//TODO: maybe do this differently!
	if (DaoUtils.isMsSql(dbHandle)) { 
		searcher = new MssqlSearcher();
	} else { 
		searcher = new SqliteSearcher();
	}
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
@Before
public void setupTests() throws IOException {
    final DataSourceFactory dataSourceFactory = new DataSourceFactory();
    dataSourceFactory.setDriverClass("org.h2.Driver");
    dataSourceFactory.setUrl("jdbc:h2:mem:test-" + System.currentTimeMillis() + "?user=sa");
    dataSourceFactory.setInitialSize(1);
    final DBI dbi = new VavrDBIFactory().build(env, dataSourceFactory, "test");
    try (Handle h = dbi.open()) {
        h.execute("CREATE TABLE tasks (" +
                "id INT PRIMARY KEY, " +
                "assignee VARCHAR(255) NOT NULL, " +
                "start_date TIMESTAMP, " +
                "end_date TIMESTAMP, " +
                "comments VARCHAR(1024) " +
                ")");
    }
    dao = dbi.onDemand(TaskDao.class);

    dao.insert(100, Option.some("Name 1"), LocalDate.parse("2017-08-24"), Option.none(), Option.none());
    dao.insert(200, Option.some("Name 2"), LocalDate.parse("2017-08-25"), Option.none(), Option.some("To be done"));
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
@Test
public void test() throws Exception {
  DBI dbi = new DBI(ds.getDataSource());

  try (Handle h = dbi.open()) {
    h.execute("create table contacts (id int primary key, name varchar(100))");

    h.execute("insert into contacts (id, name) values (?, ?)", 1, "Alice");
    h.execute("insert into contacts (id, name) values (?, ?)", 2, "Bob");

    List<String> names = h.createQuery("select name from contacts order by id")
        .mapTo(String.class)
        .list();
    assertThat(names)
        .containsExactly("Alice", "Bob");

    String name = h.createQuery("select name from contacts where id = :id")
        .bind("id", 1)
        .mapTo(String.class)
        .first();
    assertThat(name)
        .isEqualTo("Alice");

  }
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
@BeforeMethod
public void setupDatabase()
        throws Exception
{
    TypeRegistry typeRegistry = new TypeRegistry();
    dbi = new DBI("jdbc:h2:mem:test" + System.nanoTime());
    dbi.registerMapper(new TableColumn.Mapper(typeRegistry));
    dummyHandle = dbi.open();
    createTablesWithRetry(dbi);

    RaptorConnectorId connectorId = new RaptorConnectorId("raptor");
    InMemoryNodeManager nodeManager = new InMemoryNodeManager();
    nodeManager.addCurrentNodeDatasource(connectorId.toString());
    NodeSupplier nodeSupplier = new RaptorNodeSupplier(nodeManager, connectorId);
    shardManager = new DatabaseShardManager(dbi, nodeSupplier);
    metadata = new RaptorMetadata(connectorId.toString(), dbi, shardManager, SHARD_INFO_CODEC, SHARD_DELTA_CODEC);
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
/**
 * データベースの初期化
 */
public static void initialize() {
	log.info("database initialize.");

	ds.set(createConnection());
	final DBI dbi = new DBI(ds.get());

	final RelationQueries relation = dbi.open(RelationQueries.class);
	relation.createRelationTable();
	relation.close();

	final BeginWord beginWord = dbi.open(BeginWord.class);
	beginWord.createBeginWordTable();
	beginWord.close();

	log.info("database initialized.");
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
@Before
public void before() throws IOException {
    tempLocation = Files.createTempDirectory("jpagen").toFile();
    System.out.println(tempLocation.getAbsolutePath());


    com.mysql.jdbc.jdbc2.optional.MysqlDataSource ds
            = new MysqlDataSource();
    ds.setUrl("jdbc:mysql://localhost/devopsplus");
    ds.setUser("devopsplus");
    ds.setPassword("devopsplus");
    DBI dbi = new DBI(ds);
    Handle handle = dbi.open();

    config = Configuration.fromStream(getClass().getResourceAsStream("/test-2.json"));
    reflector = new ReflectorMySql(config, handle);
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
@Before
public void setupTests() throws IOException {
    final DataSourceFactory dataSourceFactory = new DataSourceFactory();
    dataSourceFactory.setDriverClass("org.h2.Driver");
    dataSourceFactory.setUrl("jdbc:h2:mem:date-time-optional-" + System.currentTimeMillis() + "?user=sa");
    dataSourceFactory.setInitialSize(1);
    final DBI dbi = new DBIFactory().build(env, dataSourceFactory, "test");
    try (Handle h = dbi.open()) {
        h.execute("CREATE TABLE tasks (" +
                "id INT PRIMARY KEY, " +
                "assignee VARCHAR(255) NOT NULL, " +
                "start_date TIMESTAMP, " +
                "end_date TIMESTAMP, " +
                "comments VARCHAR(1024) " +
                ")");
    }
    dao = dbi.onDemand(TaskDao.class);
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
public Optional<UserBean> getUserViaFacebook(String userId) {
	UserBean userBean = null;

	DBI dbConn;
	try {
		dbConn = DataSourceHandler.getInstance().getDataSource(
				JDBISetting.IMG_CONNECTION_SERVICE);
		UserJDBI userJDBI = dbConn.open(UserJDBI.class);
		userBean = userJDBI.selectViaFacebookId(userId);
		userJDBI.close();
		
	} catch (ExecutionException e) {
		log.error("Unable to retrieve user info via gmail", e);
	}

	return Optional.fromNullable(userBean);
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
@Test(groups = "database")
public void shouldFindSubAlarmMetricDefinitions() {
  // This test won't work without the real mysql database so use mini-mon.
  // Warning, this will truncate your mini-mon database
  db = new DBI("jdbc:mysql://192.168.10.4/mon", "monapi", "password");
  handle = db.open();
  repo = new AlarmDefinitionMySqlRepoImpl(db, new PersistUtils());
  beforeMethod();

  assertEquals(
      repo.findSubAlarmMetricDefinitions("123").get("111"),
      new MetricDefinition("hpcs.compute", ImmutableMap.<String, String>builder()
          .put("flavor_id", "777").put("image_id", "888").put("metric_name", "cpu")
          .put("device", "1").build()));

  assertEquals(
      repo.findSubAlarmMetricDefinitions("234").get("222"),
      new MetricDefinition("hpcs.compute", ImmutableMap.<String, String>builder()
          .put("flavor_id", "777").put("image_id", "888").put("metric_name", "mem").build()));

  assertTrue(repo.findSubAlarmMetricDefinitions("asdfasdf").isEmpty());
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
/**
 * execute the schemaCreationScript using JDBI library.
 */
public void createTables() {
	final DBI dbi = LocalRegistry.getDbi();
	final Handle handle = dbi.open();
	try {
		handle.execute(this.schemaCreationScript);
	} finally {
		handle.close();
	}
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
/**
 * execute the schemaDroppScript using JDBI library.
 */
public void dropTables() {
	final DBI dbi = LocalRegistry.getDbi();
	final Handle handle = dbi.open();
	try {
		handle.execute(this.schemaDroppScript);
	} finally {
		handle.close();
	}
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
@Override
public void migrate(Connection connection) throws Exception {
	TempDataSource ds = new TempDataSource(connection);
	DBI dbi = new DBI(ds);
	dbi.registerContainerFactory(new OptionalContainerFactory());
	Handle handle = dbi.open();
	
	PostgresDataAccessLayer dataAccessLayer = new PostgresDataAccessLayer(
			handle.attach(EntryQueryDAO.class),
			handle.attach(IndexDAO.class),
			handle.attach(IndexQueryDAO.class),
			handle.attach(EntryDAO.class),
			handle.attach(EntryItemDAO.class),
			handle.attach(ItemQueryDAO.class),
			handle.attach(ItemDAO.class),
			connection.getSchema(),
			new IndexDriver(),
			ImmutableMap.of(EntryType.user, Arrays.asList(new LatestByKeyIndexFunction(IndexNames.RECORD))));

	IndexDriver indexDriver = new IndexDriver();

	Map<String, Entry> entries = new HashMap<>();
	int currentIndexEntryNumber = 0;

	dataAccessLayer.getEntryIterator().forEachRemaining(entry -> {
		indexDriver.indexEntry(dataAccessLayer, entry, new LatestByKeyIndexFunction(IndexNames.RECORD), entries, currentIndexEntryNumber);
	});
	
	dataAccessLayer.checkpoint();
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
/**
 * provides a DBI Handle for the given register
 * the handle will automatically be closed by the RegisterRule
 */
public Handle handleFor(TestRegister register) {
    DBI dbi = new DBI(register.getDatabaseConnectionString("RegisterRule"));
    
    dbi.registerContainerFactory(new OptionalContainerFactory());
    Handle handle = dbi.open();
    handles.add(handle);
    return handle;
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
@Before
public void setup() {
    MDC.put("register", schema);
    dbi = new DBI(address.getDatabaseConnectionString("PGRegisterTxnFT"));
    dbi.registerContainerFactory(new OptionalContainerFactory());
    handle = dbi.open();
    dao = handle.attach(IndexQueryDAO.class);
}
 

import org.skife.jdbi.v2.DBI; //導入方法依賴的package包/類
@Test
public void test() throws Exception {
  DBI dbi = new DBI(ds.getDataSource());
  dbi.registerMapper(new BeanMapperFactory());
  dbi.registerColumnMapper(new MoneyMapper());
  dbi.registerArgumentFactory(new MoneyArgumentFactory());

  try (Handle h = dbi.open()) {
    Money tenDollars = Money.of(USD, 10);
    Money fiveDollars = Money.of(USD, 5);

    h.execute("create table accounts (id int primary key, name varchar(100), balance decimal)");

    h.execute("insert into accounts (id, name, balance) values (?, ?, ?)", 1, "Alice", tenDollars);
    h.execute("insert into accounts (id, name, balance) values (?, ?, ?)", 2, "Bob", fiveDollars);

    List<Account> list = h.createQuery("select * from accounts order by id")
        .mapTo(Account.class)
        .list();
    assertThat(list)
        .extracting(Account::getId, Account::getName, Account::getBalance)
        .containsExactly(tuple(1, "Alice", tenDollars),
                         tuple(2, "Bob", fiveDollars));

    Account bob = h.createQuery("select * from accounts where id = :id")
        .bind("id", 2)
        .mapTo(Account.class)
        .first();
    assertThat(bob)
        .extracting(Account::getId, Account::getName, Account::getBalance)
        .containsExactly(2, "Bob", fiveDollars);
  }
}