自定义@RequestBody注解如何获取JSON数据

自定义@RequestBody注解可以方便我们在处理请求数据时进行更细致的控制和处理。其实，要自定义@RequestBody注解获取JSON数据很简单，只需要通过反射机制获取请求体中的数据并进行处理即可。

以下是具体的步骤：

1. 定义自定义注解

定义一个自定义注解并使用@Target(ElementType.PARAMETER)来标识该注解只能用在参数上，例如：

@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface MyRequestBody {
}

1. 在请求处理方法中使用自定义注解

在请求处理方法中使用自定义注解，例如：

@PostMapping("/user")
public User addUser(@MyRequestBody User user) {
    // 处理接收到的user对象
}

1. 反射获取请求体中的JSON数据

使用反射机制获取请求体中的数据并进行处理，例如：

@PostMapping("/user")
public User addUser(@MyRequestBody User user) {
    ObjectMapper objectMapper = new ObjectMapper();
    try {
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        String json = request.getReader().lines().collect(Collectors.joining(System.lineSeparator()));
        user = objectMapper.readValue(json, User.class);
        // 处理接收到的user对象
    } catch (IOException e) {
        e.printStackTrace();
    }
}

示例一：在Spring Boot项目中使用自定义注解获取JSON数据

@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface MyRequestBody {
}

@RestController
public class UserController {
    @PostMapping("/user")
    public User addUser(@MyRequestBody User user) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
            String json = request.getReader().lines().collect(Collectors.joining(System.lineSeparator()));
            user = objectMapper.readValue(json, User.class);
            System.out.println(user);
            // 处理接收到的user对象
        } catch (IOException e) {
            e.printStackTrace();
        }
        return user;
    }
}

// 测试用例
@Test
void testAddUser() throws Exception {
    User user = new User();
    user.setName("张三");
    user.setAge(18);
    user.setSex("男");
    ObjectMapper objectMapper = new ObjectMapper();
    String json = objectMapper.writeValueAsString(user);
    mockMvc.perform(post("/user")
        .content(json)
        .contentType(MediaType.APPLICATION_JSON_VALUE)
    )
        .andDo(print())
        .andExpect(status().isOk())
        .andExpect(jsonPath("$.name", is(user.getName())));
}

示例二：在Spring MVC项目中使用自定义注解获取JSON数据

@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface MyRequestBody {
}

@Controller
public class UserController {
    @ResponseBody
    @PostMapping("/user")
    public User addUser(@MyRequestBody User user) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
            String json = request.getReader().lines().collect(Collectors.joining(System.lineSeparator()));
            user = objectMapper.readValue(json, User.class);
            System.out.println(user);
            // 处理接收到的user对象
        } catch (IOException e) {
            e.printStackTrace();
        }
        return user;
    }
}

// 测试用例
@Test
void testAddUser() throws Exception {
    User user = new User();
    user.setName("张三");
    user.setAge(18);
    user.setSex("男");
    ObjectMapper objectMapper = new ObjectMapper();
    String json = objectMapper.writeValueAsString(user);
    MvcResult mvcResult = mockMvc.perform(post("/user")
        .content(json)
        .contentType(MediaType.APPLICATION_JSON_VALUE)
    )
        .andDo(print())
        .andExpect(status().isOk())
        .andReturn();
    String content = mvcResult.getResponse().getContentAsString();
    user = objectMapper.readValue(content, User.class);
    assertEquals("张三", user.getName());
}

综上所述，我们可以通过自定义@RequestBody注解来获取请求体中的JSON数据，方便我们进行更细致的控制和处理。