@Override public List<T> extractData(ResultSet rs) throws SQLException, DataAccessException { return mapper.forEach(rs, new ListCollector<T>()).getList(); }
@Test public void testTuple2OnString() throws Exception { JdbcMapperBuilder<Tuple2<String, String>> builder = JdbcMapperFactoryHelper.asm().newBuilder(Tuples.typeDef(String.class, String.class)); builder.addMapping("element0"); builder.addMapping("element1"); final JdbcMapper<Tuple2<String, String>> mapper = builder.mapper(); DbHelper.testQuery( new TestRowHandler<PreparedStatement>() { @Override public void handle(PreparedStatement preparedStatement) throws Exception { ResultSet rs = preparedStatement.executeQuery(); try { List<Tuple2<String, String>> list = mapper.forEach(rs, new ListCollector<Tuple2<String, String>>()).getList(); assertEquals(1, list.size()); Tuple2<String, String> tuple2 = list.get(0); assertEquals("1", tuple2.getElement0()); assertEquals("2", tuple2.getElement1()); } finally { try { rs.close(); } catch (Exception e) { } } } }, "select '1', '2' from TEST_DB_OBJECT where id = 1" ); }
@Test public void testTuple3OnString() throws Exception { JdbcMapperBuilder<Tuple3<String, String, Long>> builder = JdbcMapperFactoryHelper.asm().newBuilder(Tuples.typeDef(String.class, String.class, Long.class)); builder.addMapping("element0"); builder.addMapping("element1"); builder.addMapping("element2"); final JdbcMapper<Tuple3<String, String, Long>> mapper = builder.mapper(); DbHelper.testQuery( new TestRowHandler<PreparedStatement>() { @Override public void handle(PreparedStatement preparedStatement) throws Exception { ResultSet rs = preparedStatement.executeQuery(); try { List<Tuple3<String, String, Long>> list = mapper.forEach(rs, new ListCollector<Tuple3<String, String, Long>>()).getList(); assertEquals(1, list.size()); Tuple3<String, String, Long> tuple = list.get(0); assertEquals("1", tuple.getElement0()); assertEquals("2", tuple.getElement1()); assertEquals(3l, tuple.getElement2().longValue()); } finally { try { rs.close(); } catch (Exception e) {} } } }, "select '1', '2', 3 from TEST_DB_OBJECT where id = 1" ); }
private Object[] getDefineProperties(AnnotatedElement annotatedElement) { ListCollector<Object> properties = new ListCollector<Object>(); AnnotationToPropertyService annotationToPropertyService = AnnotationToPropertyUtil.getAnnotationToPropertyService(); for(Annotation annotation : annotatedElement.getAnnotations()) { annotationToPropertyService.generateProperty(annotation, properties); } return properties.getList().toArray(); }
@Test public void testTuple4OnString() throws Exception { JdbcMapperBuilder<Tuple4<String, String, Long, Integer>> builder = JdbcMapperFactoryHelper.asm().newBuilder(Tuples.typeDef(String.class, String.class, Long.class, Integer.class)); builder.addMapping("element0"); builder.addMapping("element1"); builder.addMapping("element2"); builder.addMapping("element3"); final JdbcMapper<Tuple4<String, String, Long, Integer>> mapper = builder.mapper(); DbHelper.testQuery( new TestRowHandler<PreparedStatement>() { @Override public void handle(PreparedStatement preparedStatement) throws Exception { ResultSet rs = preparedStatement.executeQuery(); try { List<Tuple4<String, String, Long, Integer>> list = mapper.forEach(rs, new ListCollector<Tuple4<String, String, Long, Integer>>()).getList(); assertEquals(1, list.size()); Tuple4<String, String, Long, Integer> tuple = list.get(0); assertEquals("1", tuple.getElement0()); assertEquals("2", tuple.getElement1()); assertEquals(3l, tuple.getElement2().longValue()); assertEquals(4, tuple.getElement3().intValue()); } finally { try { rs.close(); } catch (Exception e) {} } } }, "select '1', '2', 3, 4 from TEST_DB_OBJECT where id = 1" ); }
private Object[] getDefineProperties(AnnotatedElement annotatedElement) { ListCollector<Object> properties = new ListCollector<Object>(); AnnotationToPropertyService annotationToPropertyService = AnnotationToPropertyUtil.getAnnotationToPropertyService(); for(Annotation annotation : annotatedElement.getAnnotations()) { annotationToPropertyService.generateProperty(annotation, properties); } return properties.getList().toArray(); }
ResultSet rs = preparedStatement.executeQuery(); try { List<Tuple5<String, String, Long, Integer, Float>> list = mapper.forEach(rs, new ListCollector<Tuple5<String, String, Long, Integer, Float>>()).getList();
@SuppressWarnings("unchecked") public static <K extends FieldKey<K>> MapperKey<K>[] generateKeys(KeyProducer<K> producer) throws Exception { ListCollector<MapperKey<K>> collector = new ListCollector<MapperKey<K>>(); for(int i = 0; i < 1000; i++) { producer.produces(collector, names(random)); } return collector.getList().toArray(new MapperKey[0]); }
@Override public void handle(PreparedStatement ps) throws Exception { List<DbObject> objects = mapper.forEach(ps.executeQuery(), new ListCollector<DbObject>()).getList(); assertEquals(1, objects.size()); DbHelper.assertDbObjectMapping(objects.get(0)); } });
@Override public void handle(PreparedStatement t) throws Exception { List<DbExtendedType> list = mapper.forEach(t.executeQuery(), new ListCollector<DbExtendedType>()).getList(); assertEquals(1, list.size()); DbExtendedType o = list.get(0); assertDbExtended(o); }
private void assertMapPsDbObjectWithAlias(ResultSet rs, JdbcMapper<DbObjectWithAlias> mapper) throws Exception, ParseException { List<DbObjectWithAlias> list = mapper.forEach(rs, new ListCollector<DbObjectWithAlias>()).getList(); assertEquals(1, list.size()); DbHelper.assertDbObjectWithAliasMapping(list.get(0)); }
@Override public void handle(PreparedStatement ps) throws Exception { List<DbFinalObject> objects = mapper.forEach(ps.executeQuery(), new ListCollector<DbFinalObject>()).getList(); assertEquals(1, objects.size()); DbHelper.assertDbObjectMapping(objects.get(0)); } });
@Override public void handle(PreparedStatement ps) throws Exception { List<DbObject> objects = mapper.forEach(ps.executeQuery(), new ListCollector<DbObject>()).getList(); assertEquals(1, objects.size()); DbHelper.assertDbObjectMapping(objects.get(0)); } });
private void assertMapPsDbObjectWithEmptyAlias(ResultSet rs, JdbcMapper<DbObjectWithEmptyAlias> mapper) throws Exception, ParseException { List<DbObjectWithEmptyAlias> list = mapper.forEach(rs, new ListCollector<DbObjectWithEmptyAlias>()).getList(); assertEquals(1, list.size()); DbHelper.assertDbObjectWithEmptyAliasMapping(list.get(0)); }
private void assertMapPsDbObject(ResultSet rs, JdbcMapper<DbObject> mapper) throws Exception, ParseException { List<DbObject> list = mapper.forEach(rs, new ListCollector<DbObject>()).getList(); assertEquals(1, list.size()); DbHelper.assertDbObjectMapping(list.get(0)); }
private void assertMapPsFinalDbObject(ResultSet rs, JdbcMapper<DbFinalObject> mapper) throws Exception, ParseException { List<DbFinalObject> list = mapper.forEach(rs, new ListCollector<DbFinalObject>()).getList(); assertEquals(1, list.size()); DbHelper.assertDbObjectMapping(list.get(0)); } }
@Test public void testMapCat() { try { mapper.forEach( new Object[][]{ {"cat", "sname", 1}, {"professor", "pname", 2} }, new ListCollector<Person>()).getList(); fail(); } catch (MappingException e) { //expected } }
private void test501(JdbcMapperBuilder<C501> builder) throws SQLException { try { builder.addMapping("zone_id", 1, Types.VARCHAR); JdbcMapper<C501> mapper = builder.mapper(); String zoneId = ZoneId.getAvailableZoneIds().iterator().next(); ResultSet rs = mock(ResultSet.class); when(rs.next()).thenReturn(true, false); when(rs.getString(1)).thenReturn(zoneId); List<C501> l = mapper.forEach(rs, new ListCollector<C501>()).getList(); assertEquals(1, l.size()); assertEquals(ZoneId.of(zoneId), l.get(0).zoneId); } catch (Throwable e) { System.out.println("XXXXXX = " + e); e.printStackTrace(System.out); throw e; } }
@Test public void test509ZoneIdvsStr() throws SQLException { JdbcMapperBuilder<C509> builder = JdbcMapperFactoryHelper.noAsm() .newBuilder(C509.class); builder.addMapping("zoneId", 1, Types.VARCHAR); JdbcMapper<C509> mapper = builder.mapper(); ResultSet rs = mock(ResultSet.class); when(rs.next()).thenReturn(true, false); when(rs.getString(1)).thenReturn("UTC"); List<C509> l = mapper.forEach(rs, new ListCollector<C509>()).getList(); assertEquals(1, l.size()); assertEquals("UTC", l.get(0).zoneId.getId()); }
private void checkJoins(JoinMapper<Object[], Object[][], DbListObject, RuntimeException> joinMapper) { Object[][] data = new Object[][] { {1, 1l, "name1"}, {1, 2l, "name2"}, {2, 3l, "name3"} }; checkList(joinMapper.forEach(data, new ListCollector<DbListObject>()).getList()); //IFJAVA8_START checkList(joinMapper.stream(data).collect(Collectors.<DbListObject>toList())); //IFJAVA8_END List<DbListObject> list = new ArrayList<DbListObject>(); Iterator<DbListObject> iterator = joinMapper.iterator(data); while(iterator.hasNext()) { list.add(iterator.next()); } checkList(list); }