java - 自定义 WebArgumentResolver，例如 @PathVariable

Question

我想为 id -> 实体使用自定义 WebArgumentResolver。如果我使用请求参数就很容易了：使用参数键来确定实体类型并相应地查找。

但我希望它像 @PathVariable 注释。

例如。

http://mysite.xzy/something/enquiryId/itemId会触发这个方法

@RequestMapping(value = "/something/{enquiry}/{item}")
public String method(@Coerce Enquiry enquiry, @Coerce Item item) 

@Coerce 注释会告诉 WebArgumentResolver 根据其类型使用特定服务。

问题是确定哪个 uri 部分属于实体。

有人可以解释 PathVariable 注释是如何做到的。是否可以用我的自定义注释来模拟它。

谢谢。

Answer 1

您可以使用@InitBinder让 spring 知道如何将给定的 String 强制转换为您的自定义类型。

你会想要这样的东西：

@RequestMapping(value = "/something/{enquiry}")
public String method(@PathVariable Enquiry enquiry) {...}


@InitBinder
public void initBinder(WebDataBinder binder) {
    binder.registerCustomEditor(Enquiry.class, new PropertyEditorSupport() {
        @Override
        public String getAsText() {
            return ((Enquiry) this.getValue()).toString();
        }

        @Override
        public void setAsText(String text) throws IllegalArgumentException {
            setValue(new Enquiry(text));
        }
    });
}

Answer 2

对于不需要为每个实体指定一个 PropertyEditor 的通用方法，您可以使用 ConditionalGenericConverter：

public final class GenericIdToEntityConverter implements ConditionalGenericConverter {
    private static final Logger log = LoggerFactory.getLogger(GenericIdToEntityConverter.class);

    private final ConversionService conversionService;

    @PersistenceContext
    private EntityManager entityManager;

    @Autowired
    public GenericIdToEntityConverter(ConversionService conversionService) {
        this.conversionService = conversionService;
    }

    public Set<ConvertiblePair> getConvertibleTypes() {
        return ImmutableSet.of(new ConvertiblePair(Number.class, AbstractEntity.class),
                new ConvertiblePair(CharSequence.class, AbstractEntity.class));

    }

    public boolean matches(TypeDescriptor sourceType, TypeDescriptor targetType) {
        return AbstractEntity.class.isAssignableFrom(targetType.getType())
        && this.conversionService.canConvert(sourceType, TypeDescriptor.valueOf(Long.class));
    }

    public Object convert(Object source, TypeDescriptor sourceType, TypeDescriptor targetType) {
        if (source == null) {
            return null;
        }

        Long id = (Long) this.conversionService.convert(source, sourceType, TypeDescriptor.valueOf(Long.class));

        Object entity = entityManager.find(targetType.getType(), id);
        if (entity == null) {
            log.info("Did not find an entity with id {} of type {}", id,  targetType.getType());
            return null;
        }

        return entity;
    }

}