Juergen Hoeller
5 years ago
10 changed files with 386 additions and 78 deletions
@ -0,0 +1,151 @@
@@ -0,0 +1,151 @@
|
||||
/* |
||||
* Copyright 2002-2020 the original author or authors. |
||||
* |
||||
* 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 |
||||
* |
||||
* https://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 org.springframework.jdbc.core; |
||||
|
||||
import java.lang.reflect.Constructor; |
||||
import java.sql.ResultSet; |
||||
import java.sql.SQLException; |
||||
|
||||
import org.springframework.beans.BeanUtils; |
||||
import org.springframework.beans.TypeConverter; |
||||
import org.springframework.core.DefaultParameterNameDiscoverer; |
||||
import org.springframework.core.ParameterNameDiscoverer; |
||||
import org.springframework.core.convert.ConversionService; |
||||
import org.springframework.lang.Nullable; |
||||
import org.springframework.util.Assert; |
||||
|
||||
/** |
||||
* {@link RowMapper} implementation that converts a row into a new instance |
||||
* of the specified mapped target class. The mapped target class must be a |
||||
* top-level class and may either expose a data class constructor with named |
||||
* parameters corresponding to column names or classic bean property setters |
||||
* (or even a combination of both). |
||||
* |
||||
* <p>Note that this class extends {@link BeanPropertyRowMapper} and can |
||||
* therefore serve as a common choice for any mapped target class, flexibly |
||||
* adapting to constructor style versus setter methods in the mapped class. |
||||
* |
||||
* @author Juergen Hoeller |
||||
* @since 5.3 |
||||
* @param <T> the result type |
||||
*/ |
||||
public class DataClassRowMapper<T> extends BeanPropertyRowMapper<T> { |
||||
|
||||
private static final ParameterNameDiscoverer parameterNameDiscoverer = new DefaultParameterNameDiscoverer(); |
||||
|
||||
@Nullable |
||||
private Constructor<T> mappedConstructor; |
||||
|
||||
@Nullable |
||||
private String[] constructorParameterNames; |
||||
|
||||
@Nullable |
||||
private Class<?>[] constructorParameterTypes; |
||||
|
||||
|
||||
/** |
||||
* Create a new {@code DataClassRowMapper} for bean-style configuration. |
||||
* @see #setMappedClass |
||||
* @see #setConversionService |
||||
*/ |
||||
public DataClassRowMapper() { |
||||
} |
||||
|
||||
/** |
||||
* Create a new {@code DataClassRowMapper}. |
||||
* @param mappedClass the class that each row should be mapped to |
||||
*/ |
||||
public DataClassRowMapper(Class<T> mappedClass) { |
||||
super(mappedClass); |
||||
} |
||||
|
||||
|
||||
@SuppressWarnings("unchecked") |
||||
@Override |
||||
protected void initialize(Class<T> mappedClass) { |
||||
super.initialize(mappedClass); |
||||
|
||||
this.mappedConstructor = BeanUtils.findPrimaryConstructor(mappedClass); |
||||
|
||||
if (this.mappedConstructor == null) { |
||||
Constructor<?>[] ctors = mappedClass.getConstructors(); |
||||
if (ctors.length == 1) { |
||||
this.mappedConstructor = (Constructor<T>) ctors[0]; |
||||
} |
||||
else { |
||||
try { |
||||
this.mappedConstructor = mappedClass.getDeclaredConstructor(); |
||||
} |
||||
catch (NoSuchMethodException ex) { |
||||
throw new IllegalStateException("No primary or default constructor found for " + mappedClass, ex); |
||||
} |
||||
} |
||||
} |
||||
|
||||
if (this.mappedConstructor.getParameterCount() > 0) { |
||||
this.constructorParameterNames = BeanUtils.getParameterNames(this.mappedConstructor); |
||||
this.constructorParameterTypes = this.mappedConstructor.getParameterTypes(); |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
protected T constructMappedInstance(ResultSet rs, TypeConverter tc) throws SQLException { |
||||
Assert.state(this.mappedConstructor != null, "Mapped constructor was not initialized"); |
||||
|
||||
Object[] args; |
||||
if (this.constructorParameterNames != null && this.constructorParameterTypes != null) { |
||||
args = new Object[this.constructorParameterNames.length]; |
||||
for (int i = 0; i < args.length; i++) { |
||||
String name = underscoreName(this.constructorParameterNames[i]); |
||||
Class<?> type = this.constructorParameterTypes[i]; |
||||
args[i] = tc.convertIfNecessary(getColumnValue(rs, rs.findColumn(name), type), type); |
||||
} |
||||
} |
||||
else { |
||||
args = new Object[0]; |
||||
} |
||||
|
||||
return BeanUtils.instantiateClass(this.mappedConstructor, args); |
||||
} |
||||
|
||||
|
||||
/** |
||||
* Static factory method to create a new {@code DataClassRowMapper}. |
||||
* @param mappedClass the class that each row should be mapped to |
||||
* @see #newInstance(Class, ConversionService) |
||||
*/ |
||||
public static <T> DataClassRowMapper<T> newInstance(Class<T> mappedClass) { |
||||
return new DataClassRowMapper<>(mappedClass); |
||||
} |
||||
|
||||
/** |
||||
* Static factory method to create a new {@code DataClassRowMapper}. |
||||
* @param mappedClass the class that each row should be mapped to |
||||
* @param conversionService the {@link ConversionService} for binding |
||||
* JDBC values to bean properties, or {@code null} for none |
||||
* @see #newInstance(Class) |
||||
* @see #setConversionService |
||||
*/ |
||||
public static <T> DataClassRowMapper<T> newInstance( |
||||
Class<T> mappedClass, @Nullable ConversionService conversionService) { |
||||
|
||||
DataClassRowMapper<T> rowMapper = newInstance(mappedClass); |
||||
rowMapper.setConversionService(conversionService); |
||||
return rowMapper; |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,45 @@
@@ -0,0 +1,45 @@
|
||||
/* |
||||
* Copyright 2002-2020 the original author or authors. |
||||
* |
||||
* 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 |
||||
* |
||||
* https://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 org.springframework.jdbc.core; |
||||
|
||||
import java.util.List; |
||||
|
||||
import org.junit.jupiter.api.Test; |
||||
|
||||
import org.springframework.jdbc.core.test.ConstructorPerson; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
||||
/** |
||||
* @author Juergen Hoeller |
||||
* @since 5.3 |
||||
*/ |
||||
public class DataClassRowMapperTests extends AbstractRowMapperTests { |
||||
|
||||
@Test |
||||
public void testStaticQueryWithDataClass() throws Exception { |
||||
Mock mock = new Mock(); |
||||
List<ConstructorPerson> result = mock.getJdbcTemplate().query( |
||||
"select name, age, birth_date, balance from people", |
||||
new DataClassRowMapper<>(ConstructorPerson.class)); |
||||
assertThat(result.size()).isEqualTo(1); |
||||
verifyPerson(result.get(0)); |
||||
|
||||
mock.verifyClosed(); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,60 @@
@@ -0,0 +1,60 @@
|
||||
/* |
||||
* Copyright 2002-2020 the original author or authors. |
||||
* |
||||
* 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 |
||||
* |
||||
* https://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 org.springframework.jdbc.core.test; |
||||
|
||||
import java.math.BigDecimal; |
||||
import java.util.Date; |
||||
|
||||
/** |
||||
* @author Juergen Hoeller |
||||
*/ |
||||
public class ConstructorPerson { |
||||
|
||||
private String name; |
||||
|
||||
private long age; |
||||
|
||||
private java.util.Date birth_date; |
||||
|
||||
private BigDecimal balance; |
||||
|
||||
|
||||
public ConstructorPerson(String name, long age, Date birth_date, BigDecimal balance) { |
||||
this.name = name; |
||||
this.age = age; |
||||
this.birth_date = birth_date; |
||||
this.balance = balance; |
||||
} |
||||
|
||||
|
||||
public String name() { |
||||
return name; |
||||
} |
||||
|
||||
public long age() { |
||||
return age; |
||||
} |
||||
|
||||
public Date birth_date() { |
||||
return birth_date; |
||||
} |
||||
|
||||
public BigDecimal balance() { |
||||
return balance; |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue