Spring Data MongoDB 自定义转换器：写入和读取 LocalDateTime

引言

在使用 Spring Data MongoDB 时，有时需要将特定类型（如 LocalDateTime）与 MongoDB 文档中的特定格式（如时间戳）之间进行自定义转换。本篇博客将介绍如何创建自定义转换器来实现这一功能。

先决条件

读者需具备以下知识：

Java、Spring Framework 和 MongoDB 的基础知识。

使用的技术

本文涉及以下技术：

Java 8+
Spring Boot
Spring Data MongoDB

转换器简介

转换器在 Spring Data MongoDB 中的作用：

自定义转换器用于控制 Java 对象与 MongoDB 文档之间的数据转换过程。

WritingConverter：将 LocalDateTime 转换为时间戳

详细解释编写写入转换器的步骤：

步骤 1：创建写入转换器
- 定义一个 @WritingConverter 类，实现 Converter<LocalDateTime, Long> 接口。
- 使用 toEpochMilli() 方法将 LocalDateTime 转换为 Epoch 毫秒值。

java 复制代码

java
复制代码
import org.springframework.core.convert.converter.Converter;
import org.springframework.data.convert.WritingConverter;
import java.time.LocalDateTime;
import java.time.ZoneOffset;

@WritingConverter
public class LocalDateTimeToTimestampConverter implements Converter<LocalDateTime, Long> {

    @Override
    public Long convert(LocalDateTime localDateTime) {
        return localDateTime.atZone(ZoneOffset.UTC).toInstant().toEpochMilli();
    }
}

ReadingConverter：将时间戳转换为 LocalDateTime

详细解释编写读取转换器的步骤：

步骤 2：创建读取转换器
- 定义一个 @ReadingConverter 类，实现 Converter<Long, LocalDateTime> 接口。
- 使用 ofEpochMilli() 方法将时间戳转换为 Instant，然后使用 atZone() 和 toLocalDateTime() 方法将其转换为 LocalDateTime 对象。

java 复制代码

java
复制代码
import org.springframework.core.convert.converter.Converter;
import org.springframework.data.convert.ReadingConverter;
import java.time.Instant;
import java.time.LocalDateTime;
import java.time.ZoneId;

@ReadingConverter
public class TimestampToLocalDateTimeConverter implements Converter<Long, LocalDateTime> {

    @Override
    public LocalDateTime convert(Long timestamp) {
        return LocalDateTime.ofInstant(Instant.ofEpochMilli(timestamp), ZoneId.systemDefault());
    }
}

注册自定义转换器

在 Spring 配置类中注册自定义转换器，以便 Spring Data MongoDB 能够识别并使用它们：

java 复制代码

java
复制代码
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.core.convert.MongoCustomConversions;
import java.util.Arrays;

@Configuration
public class MongoConfig {

    @Bean
    public MongoCustomConversions mongoCustomConversions() {
        return new MongoCustomConversions(Arrays.asList(
            new LocalDateTimeToTimestampConverter(),
            new TimestampToLocalDateTimeConverter()
        ));
    }
}

java 复制代码

package com.cdn.test.mode;

import lombok.Data;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.index.HashIndexed;
import org.springframework.data.mongodb.core.index.Indexed;
import org.springframework.data.mongodb.core.mapping.Document;
import org.springframework.data.mongodb.core.mapping.Field;
import org.springframework.data.mongodb.core.mapping.FieldType;

import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.util.HashMap;
import java.util.List;

/**
 * 蔡定努
 * 2024/07/08 18:42
 */
@Data
@Document
public class Student {

    @Id
    private String id;

    @Indexed
    private String username;

    @HashIndexed
    private String email;

    /**
     *  @Field(targetType = FieldType.DOUBLE)  也能实现简单转换
     */
    @Field(targetType = FieldType.DOUBLE)
    private BigDecimal amount;


    private List<BigDecimal> amount3;

    private Long balance;


    private LocalDateTime date;
}

单独使用@Field(targetType = FieldType.DOUBLE) 也能实现简单转换

结论

总结本文内容：

自定义转换器在 Spring Data MongoDB 中的重要性和使用方法。
如何编写和注册写入和读取转换器，实现 LocalDateTime 与时间戳之间的双向转换。

通过以上步骤，你可以轻松地在 Spring Data MongoDB 中实现对特定数据类型的自定义转换，满足复杂数据映射需求。