com.databasesandlife.util.jooq.UtcInstantViaStringBinding Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of java-common Show documentation
Utility classes developed at Adrian Smith Software (A.S.S.)
The newest version!
package com.databasesandlife.util.jooq;

import org.jooq.*;
import org.jooq.conf.ParamType;
import org.jooq.impl.DSL;

import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.sql.Types;
import java.time.Instant;
import java.time.LocalDateTime;
import java.time.ZoneOffset;
import java.time.format.DateTimeFormatter;
import java.util.Objects;

/**
 * Convert from a MySQL DATETIME (representing a UTC time) to a Java {@link Instant} by using string manipulation.
 *
 * jOOQ uses resultSet.getTimezone() by default. The MySQL JDBC driver does timezone conversions.
 * This means that the {@link LocalDateTime} read from the database isn't the same as the DATETIME database
 * value in the database. Use this class to avoid that conversion.
 *
 * This is only necessary with MySQL; The PostgreSQL JDBC driver does not suffer from this problem.
 */
@SuppressWarnings("unchecked")
public class UtcInstantViaStringBinding
implements Binding {

    @Override
    public Converter converter() {
        return new Converter() {
            @Override public Class fromType() { return (Class) Object.class; }
            @Override public Class toType() { return Instant.class; }
            
            @Override
            public Instant from(Object x) {
                if (x == null) return null;
                return LocalDateTime.parse("" + x, DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")).atOffset(ZoneOffset.UTC).toInstant();
            }

            @Override
            public T to(Instant m) {
                if (m == null) return null;
                return (T) m.atOffset(ZoneOffset.UTC).toLocalDateTime().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
            }
        };
    }

    @Override
    public void sql(BindingSQLContext ctx) throws SQLException {
        // Depending on how you generate your SQL, you may need to explicitly distinguish
        // between jOOQ generating bind variables or inlined literals.
        if (ctx.render().paramType() == ParamType.INLINED)
            ctx.render().visit(DSL.inline(ctx.convert(converter()).value()));
        else
            ctx.render().sql("?");
    }
    
    @Override
    public void register(BindingRegisterContext ctx) throws SQLException {
        ctx.statement().registerOutParameter(ctx.index(), Types.VARCHAR);
    }

    @Override
    public void set(BindingSetStatementContext ctx) throws SQLException {
        ctx.statement().setString(
            ctx.index(), 
            Objects.toString(ctx.convert(converter()).value()));
    }

    @Override /// XXX
    public void get(BindingGetResultSetContext ctx) throws SQLException {
        ctx.convert(converter()).value((T) ctx.resultSet().getString(ctx.index()));
    }

    @Override
    public void get(BindingGetStatementContext ctx) throws SQLException {
        ctx.convert(converter()).value((T) ctx.statement().getString(ctx.index()));
    }

    @Override
    public void set(BindingSetSQLOutputContext ctx) throws SQLException {
        throw new SQLFeatureNotSupportedException();
    }

    @Override
    public void get(BindingGetSQLInputContext ctx) throws SQLException {
        throw new SQLFeatureNotSupportedException();
    }
}