计算机系统应用教程网站

网站首页 > 技术文章 正文

太强 ! SpringBoot中出入参增强的5种方法 : 加解密、脱敏、格式转换

btikc 2024-10-13 01:50:24 技术文章 14 ℃ 0 评论
  • 1. 使用@JsonSerialize和@JsonDeserialize注解
  • 2. 全局配置Jackson的ObjectMapper
  • 3. 使用@ControllerAdvice配合@InitBinder
  • 4. 自定义HttpMessageConverter
  • 5. 使用AOP进行切面编程
  • 结语

在Spring Boot中,对接口的请求入参和出参进行自定义的增强或者修改,通常有以下几种方法:

1. 使用@JsonSerialize和@JsonDeserialize注解

可以在实体类的字段上使用这两个注解来指定自定义的序列化器和反序列化器。

「使用场景」

  • 当需要对某个特定字段进行自定义的序列化和反序列化时。
  • 当实体类中的某些字段类型不是标准的JSON类型,需要转换成JSON能识别的格式时。
  • 需要在序列化和反序列化过程中添加自定义逻辑,如加密、解密、格式转换等。

首先,定义自定义的序列化器和反序列化器:

public class CustomLocalDateTimeSerializer extends JsonSerializer<LocalDateTime> {
    @Override
    public void serialize(LocalDateTime value, JsonGenerator gen, SerializerProvider serializers) throws IOException {
        gen.writeString(value.format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
    }
}


public class CustomLocalDateTimeDeserializer extends JsonDeserializer<LocalDateTime> {
    @Override
    public LocalDateTime deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
        return LocalDateTime.parse(p.getValueAsString(), DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
    }
}

然后,在实体类中使用这些注解:

public class MyEntity {
    @JsonSerialize(using = CustomLocalDateTimeSerializer.class)
    @JsonDeserialize(using = CustomLocalDateTimeDeserializer.class)
    private LocalDateTime dateTime;


    // getters and setters
}

2. 全局配置Jackson的ObjectMapper

通过配置ObjectMapper来全局地改变序列化和反序列化的行为,可以添加自定义的模块或配置属性。

「使用场景」

  • 当项目中多个实体类需要应用相同的序列化和反序列化规则时。
  • 需要在全局范围内统一处理日期、时间、枚举等类型的序列化和反序列化。
  • 需要对ObjectMapper进行全局的配置,如设置默认的时区、日期格式等。

创建一个自定义模块,并注册序列化器和反序列化器:

public class CustomJacksonModule extends SimpleModule {
    public CustomJacksonModule() {
        addSerializer(LocalDateTime.class, new CustomLocalDateTimeSerializer());
        addDeserializer(LocalDateTime.class, new CustomLocalDateTimeDeserializer());
    }
}

然后,配置ObjectMapper:

@Bean
public Jackson2ObjectMapperBuilderCustomizer jsonCustomizer() {
    return builder -> {
        builder.modules(new CustomJacksonModule());
    };
}

3. 使用@ControllerAdvice配合@InitBinder

要对Spring MVC的控制器进行全局的配置,可以使用@ControllerAdvice注解。然后,在这个类中使用@InitBinder注解的方法来注册自定义的属性编辑器。

「注意」:@InitBinder主要用于处理表单数据的绑定,对于JSON数据的序列化和反序列化,它并不是最直接的方法。但如果是处理非JSON格式的请求体(如表单数据),则可以使用此方法。

「使用场景」(对于JSON数据,更偏向于使用其他方法;对于表单数据):

  • 当需要对表单数据的绑定进行自定义处理时。
  • 当需要在多个控制器中复用相同的表单数据绑定逻辑时。

创建一个自定义的属性编辑器:

public class CustomLocalDateTimeEditor extends PropertyEditorSupport {
    @Override
    public void setAsText(String text) throws IllegalArgumentException {
        setValue(LocalDateTime.parse(text, DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
    }


    @Override
    public String getAsText() {
        LocalDateTime value = (LocalDateTime) getValue();
        return value.format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
    }
}

然后,在@ControllerAdvice类中注册这个属性编辑器:

@ControllerAdvice
public class CustomControllerAdvice {
    @InitBinder
    public void initBinder(WebDataBinder binder) {
        binder.registerCustomEditor(LocalDateTime.class, new CustomLocalDateTimeEditor());
    }
}

4. 自定义HttpMessageConverter

可以编写自定义的HttpMessageConverter来处理特定的媒体类型,并在其中实现自定义的序列化和反序列化逻辑。然后,将其注册到Spring MVC的配置中。

「使用场景」

  • 当Spring Boot默认的HttpMessageConverter无法满足自定义的序列化和反序列化需求时。
  • 当需要处理非标准的媒体类型时,如自定义的二进制格式或文本格式。
  • 当需要在序列化和反序列化过程中应用复杂的业务逻辑时。

创建一个自定义的HttpMessageConverter:

public class CustomLocalDateTimeConverter extends MappingJackson2HttpMessageConverter {
    public CustomLocalDateTimeConverter() {
        super();
        ObjectMapper customObjectMapper = new ObjectMapper();
        customObjectMapper.registerModule(new CustomJacksonModule());
        setObjectMapper(customObjectMapper);
    }
}

然后,在配置类中注册这个转换器:

@Configuration
public class WebConfig implements WebMvcConfigurer {
    @Override
    public void extendMessageConverters(List<HttpMessageConverter<?>> converters) {
        converters.add(0, new CustomLocalDateTimeConverter());
    }
}

5. 使用AOP进行切面编程

可以使用Spring AOP来对控制器的方法进行切面,从而在方法执行前后进行自定义的序列化和反序列化操作。

「使用场景」

  • 当需要在不修改原有业务代码的情况下,对方法入参和出参进行额外的处理时。
  • 当需要对多个控制器或方法中的入参和出参应用统一的处理逻辑时。
  • 当处理逻辑与业务逻辑相对独立,且需要保持代码结构清晰时。

首先,定义一个切面:

@Aspect
@Component
public class CustomSerializationAspect {
    @Before("execution(* com.example.controller..*.*(..)) && args(..,@RequestParam(..),@RequestBody(..))")
    public void beforeControllerMethod(JoinPoint joinPoint) {
        // 在这里可以修改入参,但通常不建议这么做,因为这会改变方法的签名
        // 更常见的是在处理响应后进行修改
    }


    @AfterReturning(pointcut = "execution(* com.example.controller..*.*(..)) && @annotation(org.springframework.web.bind.annotation.RequestMapping)", returning = "result")
    public void afterControllerMethod(JoinPoint joinPoint, Object result) {
        // 在这里可以修改出参,例如将LocalDateTime转换为特定格式的字符串
        if (result instanceof MyEntity) {
            MyEntity entity = (MyEntity) result;
            // 假设你想在这里修改entity的dateTime字段
        }
    }
}

注意:AOP通常用于横切关注点的处理,如日志、事务管理等,而不是用于修改方法的入参和出参。如果你需要修改入参和出参,通常建议使用其他方法,如自定义的HttpMessageConverter或@ControllerAdvice。上面的AOP中捕获方法的执行,但实际上修改入参是不推荐的,而出参的修改也通常不是AOP的最佳用途。如确实要在AOP中修改出参,考虑使用@AfterReturning注解。更常见和推荐的做法使用Jackson的序列化特性或@ControllerAdvice来全局处理响应体的格式。

结语

选择哪种方法取决于具体需求。对于简单的自定义需求,使用@JsonSerialize和@JsonDeserialize注解是最直接和简单的方式。而对于更复杂的全局配置或跨多个控制器的需求,则可能需要使用ObjectMapper的配置或@ControllerAdvice。

本文暂时没有评论,来添加一个吧(●'◡'●)

欢迎 发表评论:

最近发表
标签列表