io.micronaut.data.jdbc.mapper.ColumnNameResultSetReader Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of micronaut-data-jdbc Show documentation
Show all versions of micronaut-data-jdbc Show documentation
Data Repository Support for Micronaut
/*
* Copyright 2017-2020 original 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 io.micronaut.data.jdbc.mapper;
import io.micronaut.core.annotation.NonNull;
import io.micronaut.core.annotation.Nullable;
import io.micronaut.core.convert.ConversionService;
import io.micronaut.core.convert.exceptions.ConversionErrorException;
import io.micronaut.core.reflect.ReflectionUtils;
import io.micronaut.core.util.StringUtils;
import io.micronaut.data.exceptions.DataAccessException;
import io.micronaut.data.model.DataType;
import io.micronaut.data.runtime.convert.DataConversionService;
import io.micronaut.data.runtime.mapper.ResultReader;
import java.math.BigDecimal;
import java.sql.Blob;
import java.sql.Clob;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Time;
import java.util.Date;
/**
* A {@link ResultReader} for JDBC that uses the column name.
*
* @author graemerocher
* @since 1.0.0
*/
public final class ColumnNameResultSetReader implements ResultReader {
private final ConversionService conversionService;
public ColumnNameResultSetReader() {
this(null);
}
/**
* Constructs a new instance.
*
* @param conversionService The data conversion service
* @since 3.1
*/
public ColumnNameResultSetReader(DataConversionService conversionService) {
// Backwards compatibility should be removed in the next version
this.conversionService = conversionService == null ? ConversionService.SHARED : conversionService;
}
@Override
public ConversionService getConversionService() {
return conversionService;
}
@Nullable
@Override
public Object readDynamic(@NonNull ResultSet resultSet, @NonNull String index, @NonNull DataType dataType) {
Object val = ResultReader.super.readDynamic(resultSet, index, dataType);
try {
return resultSet.wasNull() ? null : val;
} catch (SQLException e) {
throw exceptionForColumn(index, e);
}
}
@Override
public boolean next(ResultSet resultSet) {
try {
return resultSet.next();
} catch (SQLException e) {
throw new DataAccessException("Error calling next on SQL result set: " + e.getMessage(), e);
}
}
@Override
public T convertRequired(@NonNull Object value, Class type) {
//noinspection ConstantConditions
if (value == null) {
throw new DataAccessException("Cannot convert type null value to target type: " + type + ". Consider defining a TypeConverter bean to handle this case.");
}
Class wrapperType = ReflectionUtils.getWrapperType(type);
if (wrapperType.isInstance(value)) {
return (T) value;
}
return conversionService.convert(
value,
type
).orElseThrow(() ->
new DataAccessException("Cannot convert type [" + value.getClass() + "] with value [" + value + "] to target type: " + type + ". Consider defining a TypeConverter bean to handle this case.")
);
}
@Override
public Date readTimestamp(ResultSet resultSet, String index) {
try {
return resultSet.getTimestamp(index);
} catch (SQLException e) {
throw exceptionForColumn(index, e);
}
}
@Override
public Time readTime(ResultSet resultSet, String index) {
try {
return resultSet.getTime(index);
} catch (SQLException e) {
throw exceptionForColumn(index, e);
}
}
@Override
public long readLong(ResultSet resultSet, String name) {
try {
return resultSet.getLong(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public char readChar(ResultSet resultSet, String name) {
try {
String strValue = resultSet.getString(name);
if (StringUtils.isNotEmpty(strValue)) {
return strValue.charAt(0);
}
return 0;
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public Date readDate(ResultSet resultSet, String name) {
try {
return resultSet.getDate(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Nullable
@Override
public String readString(ResultSet resultSet, String name) {
try {
return resultSet.getString(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public int readInt(ResultSet resultSet, String name) {
try {
return resultSet.getInt(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public boolean readBoolean(ResultSet resultSet, String name) {
try {
return resultSet.getBoolean(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public float readFloat(ResultSet resultSet, String name) {
try {
return resultSet.getFloat(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public byte readByte(ResultSet resultSet, String name) {
try {
return resultSet.getByte(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public short readShort(ResultSet resultSet, String name) {
try {
return resultSet.getShort(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public double readDouble(ResultSet resultSet, String name) {
try {
return resultSet.getDouble(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public BigDecimal readBigDecimal(ResultSet resultSet, String name) {
try {
return resultSet.getBigDecimal(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public byte[] readBytes(ResultSet resultSet, String name) {
try {
return resultSet.getBytes(name);
} catch (SQLException e) {
throw exceptionForColumn(name, e);
}
}
@Override
public T getRequiredValue(ResultSet resultSet, String name, Class type) throws DataAccessException {
try {
Object o;
if (Blob.class.isAssignableFrom(type)) {
o = resultSet.getBlob(name);
} else if (Clob.class.isAssignableFrom(type)) {
o = resultSet.getClob(name);
} else {
o = resultSet.getObject(name);
}
if (o == null) {
return null;
}
if (type.isInstance(o)) {
//noinspection unchecked
return (T) o;
} else {
return convertRequired(o, type);
}
} catch (SQLException | ConversionErrorException e) {
throw exceptionForColumn(name, e);
}
}
private DataAccessException exceptionForColumn(String name, Exception e) {
return new DataAccessException("Error reading object for name [" + name + "] from result set: " + e.getMessage(), e);
}
}