/* * The MIT License (MIT) * * Copyright (c) 2018 by rumatoest at github.com * * Permission is hereby granted, free of charge, to any person obtaining a copy * of this software and associated documentation files (the "Software"), to deal * in the Software without restriction, including without limitation the rights * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell * copies of the Software, and to permit persons to whom the Software is * furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in all * copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ package com.github.jneat.mybatis; import org.apache.ibatis.type.BaseTypeHandler; import org.apache.ibatis.type.JdbcType; import org.apache.ibatis.type.MappedTypes; import java.sql.CallableStatement; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; import java.sql.Timestamp; import java.time.OffsetDateTime; import java.time.ZoneId; import java.time.ZoneOffset; import java.time.temporal.ChronoField; import java.util.Calendar; import java.util.GregorianCalendar; /** * Map Java 8 Instant <-> java.sql.Timestamp with timezone. */ @MappedTypes(OffsetDateTime.class) public class OffsetDateTimeTypeHandler extends BaseTypeHandler<OffsetDateTime> { @Override public void setNonNullParameter(PreparedStatement ps, int i, OffsetDateTime parameter, JdbcType jdbcType) throws SQLException { // Postgres do not work with offsets > than 15 hours, maybe other DBs too int offset = parameter.get(ChronoField.OFFSET_SECONDS); if (Math.abs(offset) > 54000) { parameter = parameter.withOffsetSameInstant(ZoneOffset.ofHours(offset > 0 ? 15 : -15)); } ps.setTimestamp( i, Timestamp.from(parameter.toInstant()), GregorianCalendar.from(parameter.toZonedDateTime()) ); } @Override public OffsetDateTime getNullableResult(ResultSet rs, String columnName) throws SQLException { Timestamp ts = rs.getTimestamp(columnName, Calendar.getInstance()); if (ts != null) { return OffsetDateTime.ofInstant(ts.toInstant(), ZoneId.systemDefault()); } return null; } @Override public OffsetDateTime getNullableResult(ResultSet rs, int columnIndex) throws SQLException { Timestamp ts = rs.getTimestamp(columnIndex, Calendar.getInstance()); if (ts != null) { return OffsetDateTime.ofInstant(ts.toInstant(), ZoneId.systemDefault()); } return null; } @Override public OffsetDateTime getNullableResult(CallableStatement cs, int columnIndex) throws SQLException { Timestamp ts = cs.getTimestamp(columnIndex, Calendar.getInstance()); if (ts != null) { return OffsetDateTime.ofInstant(ts.toInstant(), ZoneId.systemDefault()); } return null; } }