package io.dropwizard.jdbi.args;

import org.skife.jdbi.v2.StatementContext;
import org.skife.jdbi.v2.tweak.Argument;

import javax.annotation.Nullable;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.sql.Timestamp;
import java.sql.Types;
import java.time.OffsetDateTime;
import java.util.Calendar;
import java.util.Optional;

An Argument for OffsetDateTime objects.
/** * An {@link Argument} for {@link OffsetDateTime} objects. */
public class OffsetDateTimeArgument implements Argument { @Nullable private final OffsetDateTime value; private final Optional<Calendar> calendar; OffsetDateTimeArgument(@Nullable final OffsetDateTime value, final Optional<Calendar> calendar) { this.value = value; this.calendar = calendar; } @Override public void apply(final int position, final PreparedStatement statement, final StatementContext ctx) throws SQLException { if (value != null) { if (calendar.isPresent()) { // We need to make a clone, because Calendar is not thread-safe // and some JDBC drivers mutate it during time calculations final Calendar calendarClone = (Calendar) calendar.get().clone(); statement.setTimestamp(position, Timestamp.from(value.toInstant()), calendarClone); } else { statement.setTimestamp(position, Timestamp.from(value.toInstant())); } } else { statement.setNull(position, Types.TIMESTAMP); } } }