Mybatis 使用 SQLite 不支持 LocalDateTime 的解决方案

增加一个 TypeHandler 就好辣

import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;
import org.springframework.boot.autoconfigure.jackson.JacksonProperties;
import org.springframework.lang.NonNull;
import org.springframework.lang.Nullable;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Calendar;
import java.util.Date;

/**
 * @author Mr.X
 * @since 2022-07-30 12:19
 */
@Component
public class LocalDateTimeHandler extends BaseTypeHandler<LocalDateTime> {

    @Resource
    private JacksonProperties jacksonProperties;

    @Override
    public void setNonNullParameter(@NonNull PreparedStatement ps, int i, LocalDateTime parameter, JdbcType jdbcType) throws SQLException {
        ps.setObject(i, parameter);
    }

    @Override
    public LocalDateTime getNullableResult(@NonNull ResultSet rs, String columnName) throws SQLException {
        long aLong = rs.getLong(columnName);
        return aLong < 167375 ? str2LocalDateTime(rs.getString(columnName)) : ts2LocalDateTime(aLong);
    }

    @Override
    public LocalDateTime getNullableResult(@NonNull ResultSet rs, int columnIndex) throws SQLException {
        return ts2LocalDateTime(rs.getLong(columnIndex));
    }

    @Override
    public LocalDateTime getNullableResult(@NonNull CallableStatement cs, int columnIndex) throws SQLException {
        return ts2LocalDateTime(cs.getLong(columnIndex));
    }

    @Nullable
    private LocalDateTime str2LocalDateTime(@Nullable String time) {
        if (time == null) return null;
        return LocalDateTime.parse(time, DateTimeFormatter.ofPattern(jacksonProperties.getDateFormat()));
    }

    @NonNull
    private LocalDateTime ts2LocalDateTime(@NonNull Long timeStamp) {
        Calendar calendar = Calendar.getInstance();
        calendar.setTime(new Date(timeStamp));
        return LocalDateTime.of(calendar.get(Calendar.YEAR), calendar.get(Calendar.MONTH) + 1, calendar.get(Calendar.DAY_OF_MONTH), calendar.get(Calendar.HOUR_OF_DAY), calendar.get(Calendar.MINUTE));
    }

}

posted @ 2022-07-30 12:48  博麗靈夢  阅读(898)  评论(0编辑  收藏  举报