Spring Data Reactive Mongo:类型转换器

时间:2018-05-23 16:33:34

标签: java spring-boot spring-data-mongodb spring-webflux

我使用Spring boot 2 webflux和反应式mongo存储库。我试图坚持并查询Class<>字段类型的实体。但我无法弄清楚如何配置我的应用程序以使用此类型。 这是我想要存储在mongo中的实体:

import lombok.Data;
import org.springframework.data.annotation.Id;

@Data
public class ServiceEntity {

  @Id
  private String id;
  private String name;
  private Class inputClass;
}

这是我的mongo配置:

import static java.util.Arrays.asList;

import com.mongodb.MongoClient;
import com.mongodb.MongoException;
import com.mongodb.async.client.MongoClientSettings;
import lombok.extern.slf4j.Slf4j;
import org.bson.BsonReader;
import org.bson.BsonWriter;
import org.bson.codecs.Codec;
import org.bson.codecs.DecoderContext;
import org.bson.codecs.EncoderContext;
import org.bson.codecs.configuration.CodecProvider;
import org.bson.codecs.configuration.CodecRegistries;
import org.bson.codecs.configuration.CodecRegistry;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.convert.converter.Converter;
import org.springframework.data.mongodb.core.convert.MongoCustomConversions;

@Configuration
@Slf4j
public class MongoConfiguration {

  @Bean
  public MongoClientSettings settings() {
    log.debug("Configure mongo settings");
    return MongoClientSettings.builder()
        .codecRegistry(CodecRegistries.fromRegistries(
            CodecRegistries.fromProviders(new ClassCodecProvider()),
            MongoClient.getDefaultCodecRegistry()))
        .build();
  }

  //  @Bean
  public MongoCustomConversions customConversions() {
    log.debug("Configure mongo custom conversions");
    return new MongoCustomConversions(asList(new StringToClassConverter()));
  }

  private static class ClassCodec implements Codec<Class> {

    @Override
    public Class decode(BsonReader reader, DecoderContext decoderContext) {
      try {
        return Class.forName(reader.readString());
      } catch (ClassNotFoundException e) {
        throw new MongoException("Couldn't read value as class type", e);
      }
    }

    @Override
    public void encode(BsonWriter writer, Class value, EncoderContext encoderContext) {
      writer.writeString(value.getName());
    }

    @Override
    public Class<Class> getEncoderClass() {
      return Class.class;
    }
  }

  private static class ClassCodecProvider implements CodecProvider {

    @Override
    public <T> Codec<T> get(Class<T> clazz, CodecRegistry registry) {
      if (clazz == Class.class) {
        return (Codec<T>) new ClassCodec();
      }
      return null;
    }
  }

  private static class StringToClassConverter implements Converter<String, Class> {

    @Override
    public Class convert(String source) {
      try {
        return Class.forName(source);
      } catch (ClassNotFoundException e) {
        throw new MongoException("Couldn't read string as class type", e);
      }

    }
  }

}

问题是当我尝试从mongo读取实体时出现错误:

org.springframework.core.convert.ConverterNotFoundException: No converter found capable of converting from type [java.lang.String] to type [java.lang.Class<?>]
    at org.springframework.core.convert.support.GenericConversionService.handleConverterNotFound(GenericConversionService.java:321)
    at org.springframework.core.convert.support.GenericConversionService.convert(GenericConversionService.java:194)
    at org.springframework.core.convert.support.GenericConversionService.convert(GenericConversionService.java:174)
    at org.springframework.data.mongodb.core.convert.MappingMongoConverter.getPotentiallyConvertedSimpleRead(MappingMongoConverter.java:887)
    at org.springframework.data.mongodb.core.convert.MappingMongoConverter.readValue(MappingMongoConverter.java:1392)

我还创建了String to Class转换器(请参阅配置中的StringToClassConverter)并尝试将其用作自定义转换器。但是当我将这样的配置放到上下文中(为了在@Bean方法上取消注释customConversions()注释)时,它已被用于任何字符串转换。因此,在保存实体的同时:

org.springframework.core.convert.ConversionFailedException: Failed to convert from type [java.lang.String] to type [java.lang.Class<?>] for value 'service1'; nested exception is com.mongodb.MongoException: Couldn't read string as class type
    at org.springframework.core.convert.support.ConversionUtils.invokeConverter(ConversionUtils.java:46)
    at org.springframework.core.convert.support.GenericConversionService.convert(GenericConversionService.java:191)
    at org.springframework.core.convert.support.GenericConversionService.convert(GenericConversionService.java:174)
    at org.springframework.data.mongodb.core.convert.MappingMongoConverter.getPotentiallyConvertedSimpleWrite(MappingMongoConverter.java:849)
    at org.springframework.data.mongodb.core.convert.MappingMongoConverter.writeSimpleInternal(MappingMongoConverter.java:829)
    at org.springframework.data.mongodb.core.convert.MappingMongoConverter.writeProperties(MappingMongoConverter.java:488)

github上提供完整代码。请参阅com.example.mongoclasspertist.ServiceRepositoryTest#test以重现。

我通过向Spring回购提交PR来看下一步如何解决这个问题(但可能导致另一个问题): *在ClassToStringConverter中创建StringToClassConverterorg.springframework.data.mongodb.core.convert.MongoConverters *仅当源和目标类型适用于自定义转换器时,才能更改org.springframework.data.mongodb.core.convert.MappingMongoConverter#getPotentiallyConvertedSimpleWrite方法逻辑以进行转换。

请帮我解决这个问题。

1 个答案:

答案 0 :(得分:4)

在转换器上使用注释@ReadingConverter似乎可以完成这项工作。请试一试。

@ReadingConverter
private static class StringToClassConverter implements Converter<String, Class> 
{