io.micronaut.data.jdbc.mapper.JdbcQueryStatement 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.convert.ConversionService;
import io.micronaut.data.exceptions.DataAccessException;
import io.micronaut.data.runtime.convert.DataConversionService;
import io.micronaut.data.runtime.mapper.QueryStatement;
import io.micronaut.data.model.DataType;
import java.math.BigDecimal;
import java.sql.*;
import java.time.Instant;
import java.util.Date;
/**
* A {@link QueryStatement} for a SQL {@link PreparedStatement}.
*
* @author graemerocher
* @since 1.0.0
*/
public class JdbcQueryStatement implements QueryStatement {
private final ConversionService conversionService;
public JdbcQueryStatement() {
this(null);
}
/**
* Constructs a new instance.
*
* @param conversionService The data conversion service
* @since 3.1
*/
public JdbcQueryStatement(DataConversionService conversionService) {
// Backwards compatibility should be removed in the next version
this.conversionService = conversionService == null ? ConversionService.SHARED : conversionService;
}
@Override
public ConversionService getConversionService() {
return conversionService;
}
@Override
public QueryStatement setDynamic(@NonNull PreparedStatement statement, @NonNull Integer index, @NonNull DataType dataType, Object value) {
if (value == null) {
try {
switch (dataType) {
case ENTITY:
throw new IllegalStateException("Cannot set null value as ENTITY data type!");
case LONG:
statement.setNull(index, Types.BIGINT);
return this;
case STRING:
case JSON:
statement.setNull(index, Types.VARCHAR);
return this;
case DATE:
statement.setNull(index, Types.DATE);
return this;
case BOOLEAN:
statement.setNull(index, Types.BOOLEAN);
return this;
case INTEGER:
statement.setNull(index, Types.INTEGER);
return this;
case TIMESTAMP:
statement.setNull(index, Types.TIMESTAMP);
return this;
case TIME:
statement.setNull(index, Types.TIME);
return this;
case OBJECT:
statement.setNull(index, Types.OTHER);
return this;
case CHARACTER:
statement.setNull(index, Types.CHAR);
return this;
case DOUBLE:
statement.setNull(index, Types.DOUBLE);
return this;
case BYTE_ARRAY:
statement.setNull(index, Types.BINARY);
return this;
case FLOAT:
statement.setNull(index, Types.FLOAT);
return this;
case BIGDECIMAL:
statement.setNull(index, Types.DECIMAL);
return this;
case BYTE:
statement.setNull(index, Types.BIT);
return this;
case SHORT:
statement.setNull(index, Types.TINYINT);
return this;
case UUID:
statement.setNull(index, Types.OTHER, "uuid");
return this;
default:
if (dataType.isArray()) {
statement.setNull(index, Types.ARRAY);
} else {
statement.setNull(index, Types.NULL);
}
return this;
}
} catch (SQLException e) {
throw new DataAccessException("Error setting JDBC null value: " + e.getMessage(), e);
}
} else {
return QueryStatement.super.setDynamic(statement, index, dataType, value);
}
}
@Override
public QueryStatement setTimestamp(PreparedStatement statement, Integer name, Instant instant) {
try {
if (instant == null) {
statement.setNull(name, Types.TIMESTAMP);
} else {
statement.setTimestamp(name, Timestamp.from(instant));
}
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@Override
public QueryStatement setTime(PreparedStatement statement, Integer name, Time instant) {
try {
if (instant == null) {
statement.setNull(name, Types.TIME);
} else {
statement.setTime(name, instant);
}
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@Override
public QueryStatement setValue(PreparedStatement statement, Integer index, Object value) throws DataAccessException {
try {
if (value instanceof Clob) {
statement.setClob(index, (Clob) value);
} else if (value instanceof Blob) {
statement.setBlob(index, (Blob) value);
} else if (value instanceof Array) {
statement.setArray(index, (Array) value);
} else if (value != null) {
if (value.getClass().isEnum()) {
statement.setObject(index, value, java.sql.Types.OTHER);
} else {
statement.setObject(index, value);
}
}
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setLong(PreparedStatement statement, Integer name, long value) {
try {
statement.setLong(name, value);
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setChar(PreparedStatement statement, Integer name, char value) {
try {
statement.setString(name, String.valueOf(value));
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setDate(PreparedStatement statement, Integer name, Date date) {
try {
if (date == null) {
statement.setNull(name, Types.DATE);
} else {
statement.setDate(name, new java.sql.Date(date.getTime()));
}
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@Override
public QueryStatement setString(PreparedStatement statement, Integer name, String string) {
try {
if (string == null) {
statement.setNull(name, Types.VARCHAR);
} else {
statement.setString(name, string);
}
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setInt(PreparedStatement statement, Integer name, int integer) {
try {
statement.setInt(name, integer);
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setBoolean(PreparedStatement statement, Integer name, boolean bool) {
try {
statement.setBoolean(name, bool);
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setFloat(PreparedStatement statement, Integer name, float f) {
try {
statement.setFloat(name, f);
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setByte(PreparedStatement statement, Integer name, byte b) {
try {
statement.setByte(name, b);
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setShort(PreparedStatement statement, Integer name, short s) {
try {
statement.setShort(name, s);
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setDouble(PreparedStatement statement, Integer name, double d) {
try {
statement.setDouble(name, d);
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setBigDecimal(PreparedStatement statement, Integer name, BigDecimal bd) {
try {
statement.setBigDecimal(name, bd);
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setBytes(PreparedStatement statement, Integer name, byte[] bytes) {
try {
statement.setBytes(name, bytes);
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
@NonNull
@Override
public QueryStatement setArray(PreparedStatement statement, Integer name, Object array) {
try {
if (array == null) {
statement.setNull(name, Types.ARRAY);
} else if (array instanceof Array) {
statement.setArray(name, (Array) array);
} else {
statement.setObject(name, array);
}
} catch (SQLException e) {
throw newDataAccessException(e);
}
return this;
}
private DataAccessException newDataAccessException(SQLException e) {
return new DataAccessException("Unable to set PreparedStatement value: " + e.getMessage(), e);
}
}