java - 枚举作为映射键转换在 Spring Boot 2.4.1 和 Elastic Search 7.10 中不起作用

Question

我正在使用 SpringBoot 2.2.6、Elastic Search 5.6.13，我决定升级到 SpringBoot 2.4.1 , Elasticsearch 7.10 .
我在 ES 中存储复杂的文档，使用 枚举作为值和映射键 .
对于枚举和字符串(加上自定义序列化器)之间的所有序列化，我使用自定义实体映射器和自定义配置的 Jackson ObjectMapper。
在 ES 4 中，ElasticsearchEntityMapper 接口(interface)已被移除。并且所有的转换都应该作为转换器来实现。
所以我为每个枚举实现了转换器，它正确地写入了 ES。
但他们没有被正确阅读。我发现了几个问题，但我想从一个特定的问题开始。
映射键未转换回枚举 .它们被读取为字符串。
我调试了 Spring Boot 代码，寻找我应该实现的特定接口(interface)，但我找不到任何接口(interface)。
有任何想法吗？
https://docs.spring.io/spring-data/elasticsearch/docs/current-SNAPSHOT/reference/html/#elasticsearch-migration-guide-3.2-4.0.jackson-removal
更新 2022-01:Spring Boot 2.6.x 似乎工作正常，允许使用枚举作为映射键...

Answer 1

我已经设法通过编写自定义 MappingElasticsearchConverter 来解决这个问题

public class CustomMappingElasticsearchConverter extends MappingElasticsearchConverter {

    public CustomMappingElasticsearchConverter(MappingContext<? extends ElasticsearchPersistentEntity<?>, ElasticsearchPersistentProperty> mappingContext) {
        super(mappingContext);
    }

    @Override
    @Nullable
    protected <R> R readValue(@Nullable Object source, ElasticsearchPersistentProperty property, TypeInformation<R> targetType) {
        R r = super.readValue(source, property, targetType);
        return convertMapKeysToEnum(r, targetType);
    }

    @SuppressWarnings("unchecked")
    private <R> R convertMapKeysToEnum(R value, TypeInformation<R> targetType) {
        if (!(value instanceof Map)) return value;
        Class<?> keyClass = targetType.getTypeArguments().get(0).getType();
        if (!Enum.class.isAssignableFrom(keyClass)) return value;
        return (R) ((Map<?, ?>) value).entrySet().stream()
            .collect(toMap(e -> Enum.valueOf((Class<Enum>) keyClass, e.getKey().toString()), e -> e.getValue()));
    }
}

然后只需用您的自定义实现替换默认bean

public class ElasticsearchConfig extends AbstractElasticsearchConfiguration {

    ...

    @Bean
    @Nonnull
    @Override
    public ElasticsearchConverter elasticsearchEntityMapper(SimpleElasticsearchMappingContext elasticsearchMappingContext) {
        MappingElasticsearchConverter elasticsearchConverter = new CustomMappingElasticsearchConverter(
            elasticsearchMappingContext);
        elasticsearchConverter.setConversions(elasticsearchCustomConversions());
        return elasticsearchConverter;
    }
}