Java tutorial
/* * Copyright 2015 JBoss 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 io.apiman.gateway.engine.impl; import io.apiman.gateway.engine.async.IAsyncResult; import io.apiman.gateway.engine.async.IAsyncResultHandler; import io.apiman.gateway.engine.components.jdbc.IJdbcClient; import io.apiman.gateway.engine.components.jdbc.IJdbcConnection; import io.apiman.gateway.engine.components.jdbc.IJdbcResultSet; import io.apiman.gateway.engine.components.jdbc.JdbcOptionsBean; import java.sql.Driver; import java.sql.SQLException; import java.util.ArrayList; import java.util.List; import org.apache.commons.dbcp.BasicDataSource; import org.junit.Assert; import org.junit.Before; import org.junit.Test; /** * Full test for the default JDBC component. * * @author eric.wittmann@redhat.com */ @SuppressWarnings("nls") public class DefaultJdbcComponentTest { List<Throwable> errors = new ArrayList<>(); @Before public void before() { System.out.println("Before!"); errors.clear(); } @Test public void testStandalone() throws Throwable { DefaultJdbcComponent component = new DefaultJdbcComponent(); JdbcOptionsBean config = new JdbcOptionsBean(); config.setAutoCommit(true); config.setJdbcUrl("jdbc:h2:mem:testStandalone;DB_CLOSE_DELAY=-1"); config.setUsername("sa"); config.setPassword(""); IJdbcClient client = component.createStandalone(config); doAllTests(client); } @Test public void testShared() throws Throwable { DefaultJdbcComponent component = new DefaultJdbcComponent(); JdbcOptionsBean config = new JdbcOptionsBean(); config.setAutoCommit(true); config.setJdbcUrl("jdbc:h2:mem:testShared;DB_CLOSE_DELAY=-1"); config.setUsername("sa"); config.setPassword(""); IJdbcClient client = component.createShared("sharedDS", config); doAllTests(client); } @Test public void testDataSource() throws Throwable { DefaultJdbcComponent component = new DefaultJdbcComponent(); BasicDataSource ds = new BasicDataSource(); ds.setDriverClassName(Driver.class.getName()); ds.setUsername("sa"); ds.setPassword(""); ds.setUrl("jdbc:h2:mem:testDataSource;DB_CLOSE_DELAY=-1"); try { IJdbcClient client = component.create(ds); doAllTests(client); } finally { try { ds.close(); } catch (SQLException e) { e.printStackTrace(); } } } /** * @param client */ private void doAllTests(IJdbcClient client) throws Throwable { client.connect(new IAsyncResultHandler<IJdbcConnection>() { @Override public void handle(IAsyncResult<IJdbcConnection> result) { if (result.isError()) { errors.add(result.getError()); } else { doCreate(result.getResult()); } } }); try { Throwable firstError = null; for (Throwable throwable : errors) { System.out.println("------------ ERROR DETECTED ----------------"); throwable.printStackTrace(); System.out.println("------------ -------------- ----------------"); if (firstError == null) { firstError = throwable; } } if (firstError != null) { throw firstError; } } catch (Throwable t) { throw t; } } /** * @param connection */ protected void doCreate(final IJdbcConnection connection) { String createSql = "CREATE TABLE users (" + "userId VARCHAR(255) NOT NULL, " + "name VARCHAR(255) NOT NULL, " + "age INTEGER, " + "isAdmin BIT, " + "PRIMARY KEY (userId))"; IAsyncResultHandler<Void> afterCreateTable = new IAsyncResultHandler<Void>() { @Override public void handle(IAsyncResult<Void> result) { if (result.isError()) { errors.add(result.getError()); } else { System.out.println("Create worked, inserting."); doInserts(connection); } } }; connection.execute(afterCreateTable, createSql); } /** * @param connection */ protected void doInserts(IJdbcConnection connection) { String insertSql = "INSERT INTO users (userId, name, age, isAdmin) VALUES (?, ?, ?, ?)"; IAsyncResultHandler<Void> dfltHandler = new IAsyncResultHandler<Void>() { @Override public void handle(IAsyncResult<Void> result) { if (result.isError()) { errors.add(result.getError()); } else { System.out.println("Insert successful."); } } }; connection.execute(dfltHandler, insertSql, "ewittman", "Eric Wittmann", 17, Boolean.TRUE); connection.execute(dfltHandler, insertSql, "msavy", "Marc Savy", 50, Boolean.FALSE); connection.execute(dfltHandler, insertSql, "ryordan", "Rachel Yordan", 92, Boolean.FALSE); doQueries(connection); } /** * @param connection */ private void doQueries(IJdbcConnection connection) { IAsyncResultHandler<IJdbcResultSet> queryHandler = new IAsyncResultHandler<IJdbcResultSet>() { @Override public void handle(IAsyncResult<IJdbcResultSet> result) { if (result.isError()) { errors.add(result.getError()); } else { try { doQueryAssertions(result.getResult()); } catch (Throwable t) { errors.add(t); } } } }; String querySql = "SELECT u.userId, u.name, u.age, u.isAdmin FROM users u WHERE userId = ?"; connection.query(queryHandler, querySql, "ewittman"); } /** * @param result */ protected void doQueryAssertions(IJdbcResultSet result) { Assert.assertTrue("Expected the result set to have at least one row.", result.hasNext()); result.next(); Assert.assertEquals("ewittman", result.getString(1)); Assert.assertEquals("Eric Wittmann", result.getString(2)); Assert.assertEquals(new Integer(17), result.getInteger(3)); Assert.assertEquals(Boolean.TRUE, result.getBoolean(4)); Assert.assertFalse(result.hasNext()); } }