Skip to content

Commit 61f2966

Browse files
uaihebertuaihebert
committed
add a first row data converter
1 parent 2cefdc8 commit 61f2966

File tree

2 files changed

+285
-0
lines changed

2 files changed

+285
-0
lines changed

‎src/main/java/org/springframework/data/r2dbc/repository/query/RowDataConverter.java

+160
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,160 @@
1+
/*
2+
* Copyright 2019 the original author or authors.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
package org.springframework.data.r2dbc.repository.query;
18+
19+
import io.r2dbc.spi.Row;
20+
import org.springframework.core.convert.converter.Converter;
21+
import org.springframework.core.convert.converter.ConverterFactory;
22+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToNumberConverterFactory.RowToOffsetDateTimeConverter;
23+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToNumberConverterFactory.RowToStringConverter;
24+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToNumberConverterFactory.RowToUuidConverter;
25+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToNumberConverterFactory.RowToZonedDateTimeConverter;
26+
import org.springframework.util.Assert;
27+
import org.springframework.util.NumberUtils;
28+
29+
import java.time.LocalDate;
30+
import java.time.LocalDateTime;
31+
import java.time.LocalTime;
32+
import java.time.OffsetDateTime;
33+
import java.time.ZonedDateTime;
34+
import java.util.ArrayList;
35+
import java.util.Collection;
36+
import java.util.List;
37+
import java.util.UUID;
38+
39+
/**
40+
* Base class for row data converter.
41+
*
42+
* @author Hebert Coelho
43+
*/
44+
abstract class RowDataConverter {
45+
private RowDataConverter() {
46+
}
47+
48+
static Collection<Object> getConvertersToRegister() {
49+
List<Object> converters = new ArrayList<>();
50+
51+
converters.add(RowToBooleanConverter.INSTANCE);
52+
converters.add(RowToLocalDateConverter.INSTANCE);
53+
converters.add(RowToLocalDateTimeConverter.INSTANCE);
54+
converters.add(RowToLocalTimeConverter.INSTANCE);
55+
converters.add(RowToOffsetDateTimeConverter.INSTANCE);
56+
converters.add(RowToStringConverter.INSTANCE);
57+
converters.add(RowToUuidConverter.INSTANCE);
58+
converters.add(RowToZonedDateTimeConverter.INSTANCE);
59+
60+
return converters;
61+
}
62+
63+
public enum RowToBooleanConverter implements Converter<Row, Boolean> {
64+
INSTANCE;
65+
66+
@Override
67+
public Boolean convert(Row row) {
68+
return row.get(0, Boolean.class);
69+
}
70+
}
71+
72+
public enum RowToLocalDateConverter implements Converter<Row, LocalDate> {
73+
INSTANCE;
74+
75+
@Override
76+
public LocalDate convert(Row row) {
77+
return row.get(0, LocalDate.class);
78+
}
79+
}
80+
81+
public enum RowToLocalDateTimeConverter implements Converter<Row, LocalDateTime> {
82+
INSTANCE;
83+
84+
@Override
85+
public LocalDateTime convert(Row row) {
86+
return row.get(0, LocalDateTime.class);
87+
}
88+
}
89+
90+
public enum RowToLocalTimeConverter implements Converter<Row, LocalTime> {
91+
INSTANCE;
92+
93+
@Override
94+
public LocalTime convert(Row row) {
95+
return row.get(0, LocalTime.class);
96+
}
97+
}
98+
99+
public enum RowToNumberConverterFactory implements ConverterFactory<Row, Number> {
100+
INSTANCE;
101+
102+
@Override
103+
public <T extends Number> Converter<Row, T> getConverter(Class<T> targetType) {
104+
Assert.notNull(targetType, "Target type must not be null");
105+
return new RowToNumber<>(targetType);
106+
}
107+
108+
private static final class RowToNumber<T extends Number> implements Converter<Row, T> {
109+
private final Class<T> targetType;
110+
111+
RowToNumber(Class<T> targetType) {
112+
this.targetType = targetType;
113+
}
114+
115+
@Override
116+
public T convert(Row source) {
117+
118+
Object object = source.get(0, targetType);
119+
120+
return (object != null ? NumberUtils.convertNumberToTargetClass((Number) object, this.targetType) : null);
121+
}
122+
}
123+
124+
public enum RowToOffsetDateTimeConverter implements Converter<Row, OffsetDateTime> {
125+
INSTANCE;
126+
127+
@Override
128+
public OffsetDateTime convert(Row row) {
129+
return row.get(0, OffsetDateTime.class);
130+
}
131+
}
132+
133+
public enum RowToStringConverter implements Converter<Row, String> {
134+
INSTANCE;
135+
136+
@Override
137+
public String convert(Row row) {
138+
return row.get(0, String.class);
139+
}
140+
}
141+
142+
public enum RowToUuidConverter implements Converter<Row, UUID> {
143+
INSTANCE;
144+
145+
@Override
146+
public UUID convert(Row row) {
147+
return row.get(0, UUID.class);
148+
}
149+
}
150+
151+
public enum RowToZonedDateTimeConverter implements Converter<Row, ZonedDateTime> {
152+
INSTANCE;
153+
154+
@Override
155+
public ZonedDateTime convert(Row row) {
156+
return row.get(0, ZonedDateTime.class);
157+
}
158+
}
159+
}
160+
}

‎src/test/java/org/springframework/data/r2dbc/repository/query/RowDataConverterTests.java

+125
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,125 @@
1+
package org.springframework.data.r2dbc.repository.query;
2+
3+
import io.r2dbc.spi.Row;
4+
import org.junit.Test;
5+
import org.springframework.core.convert.converter.Converter;
6+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToBooleanConverter;
7+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToLocalDateConverter;
8+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToLocalDateTimeConverter;
9+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToLocalTimeConverter;
10+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToNumberConverterFactory;
11+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToNumberConverterFactory.RowToOffsetDateTimeConverter;
12+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToNumberConverterFactory.RowToStringConverter;
13+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToNumberConverterFactory.RowToUuidConverter;
14+
import org.springframework.data.r2dbc.repository.query.RowDataConverter.RowToNumberConverterFactory.RowToZonedDateTimeConverter;
15+
16+
import java.time.LocalDate;
17+
import java.time.LocalDateTime;
18+
import java.time.LocalTime;
19+
import java.time.OffsetDateTime;
20+
import java.time.ZonedDateTime;
21+
import java.util.UUID;
22+
23+
import static org.assertj.core.api.Assertions.assertThat;
24+
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
25+
import static org.junit.Assert.assertTrue;
26+
import static org.mockito.Mockito.mock;
27+
import static org.mockito.Mockito.when;
28+
29+
public class RowDataConverterTests {
30+
private static final int TOTAL_REGISTERED_CONVERTERS = 8;
31+
32+
@Test
33+
public void isReturningAllCreatedConverts() {
34+
assertThat(RowDataConverter.getConvertersToRegister().size())
35+
.isEqualTo(TOTAL_REGISTERED_CONVERTERS);
36+
}
37+
38+
@Test
39+
public void isConvertingBoolean() {
40+
Row row = mock(Row.class);
41+
when(row.get(0, Boolean.class)).thenReturn(true);
42+
43+
assertTrue(RowToBooleanConverter.INSTANCE.convert(row));
44+
}
45+
46+
@Test
47+
public void isConvertingLocalDate() {
48+
LocalDate now = LocalDate.now();
49+
Row row = mock(Row.class);
50+
when(row.get(0, LocalDate.class)).thenReturn(now);
51+
52+
assertThat(RowToLocalDateConverter.INSTANCE.convert(row)).isEqualTo(now);
53+
}
54+
55+
@Test
56+
public void isConvertingLocalDateTime() {
57+
LocalDateTime now = LocalDateTime.now();
58+
Row row = mock(Row.class);
59+
when(row.get(0, LocalDateTime.class)).thenReturn(now);
60+
61+
assertThat(RowToLocalDateTimeConverter.INSTANCE.convert(row)).isEqualTo(now);
62+
}
63+
64+
@Test
65+
public void isConvertingLocalTime() {
66+
LocalTime now = LocalTime.now();
67+
Row row = mock(Row.class);
68+
when(row.get(0, LocalTime.class)).thenReturn(now);
69+
70+
assertThat(RowToLocalTimeConverter.INSTANCE.convert(row)).isEqualTo(now);
71+
}
72+
73+
@Test
74+
public void isConvertingOffsetDateTime() {
75+
OffsetDateTime now = OffsetDateTime.now();
76+
Row row = mock(Row.class);
77+
when(row.get(0, OffsetDateTime.class)).thenReturn(now);
78+
79+
assertThat(RowToOffsetDateTimeConverter.INSTANCE.convert(row)).isEqualTo(now);
80+
}
81+
82+
@Test
83+
public void isConvertingString() {
84+
String value = "aValue";
85+
Row row = mock(Row.class);
86+
when(row.get(0, String.class)).thenReturn(value);
87+
88+
assertThat(RowToStringConverter.INSTANCE.convert(row)).isEqualTo(value);
89+
}
90+
91+
@Test
92+
public void isConvertingUUID() {
93+
UUID value = UUID.randomUUID();
94+
Row row = mock(Row.class);
95+
when(row.get(0, UUID.class)).thenReturn(value);
96+
97+
assertThat(RowToUuidConverter.INSTANCE.convert(row)).isEqualTo(value);
98+
}
99+
100+
@Test
101+
public void isConvertingZonedDateTime() {
102+
ZonedDateTime now = ZonedDateTime.now();
103+
Row row = mock(Row.class);
104+
when(row.get(0, ZonedDateTime.class)).thenReturn(now);
105+
106+
assertThat(RowToZonedDateTimeConverter.INSTANCE.convert(row)).isEqualTo(now);
107+
}
108+
109+
@Test
110+
public void isConvertingNumber() {
111+
Row row = mock(Row.class);
112+
when(row.get(0, Integer.class)).thenReturn(33);
113+
114+
final Converter<Row, Integer> converter = RowToNumberConverterFactory.INSTANCE.getConverter(Integer.class);
115+
116+
assertThat(converter.convert(row)).isEqualTo(33);
117+
}
118+
119+
@Test
120+
public void isRaisingExceptionForInvalidNumber() {
121+
assertThatIllegalArgumentException().isThrownBy(
122+
() -> RowToNumberConverterFactory.INSTANCE.getConverter(null)
123+
);
124+
}
125+
}

0 commit comments

Comments
 (0)