在SpringBoot中使用ResponseBodyAdvice自定义响应的代码实现

import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

@ControllerAdvice
public class CustomResponseBodyAdvice implements ResponseBodyAdvice<Object> {

    @Override
    public boolean supports(MethodParameter returnType, Class converterType) {
        // This method is called to determine if the advice should be applied
        // based on the return type and converter type.
        // Return true if you want to apply the advice, false otherwise.
        return true;
    }

    @Override
    public Object beforeBodyWrite(
            Object body,
            MethodParameter returnType,
            MediaType selectedContentType,
            Class selectedConverterType,
            ServerHttpRequest request,
            ServerHttpResponse response) {
        // This method is called just before the response body is written to the client.
        // You can modify the body or the response before it's sent to the client.

        // For example, you can wrap the original response in a custom wrapper.
        CustomResponseWrapper wrapper = new CustomResponseWrapper(body);
        return wrapper;
    }
}

public class CustomResponseWrapper {

    private Object data;

    public CustomResponseWrapper(Object data) {
        this.data = data;
    }

    public Object getData() {
        return data;
    }

    // You can add more methods or properties as needed
}

@RestController
public class MyController {

    @GetMapping("/api/data")
    public ResponseEntity<String> getData() {
        // Your original response
        String responseData = "Hello, World!";
        return ResponseEntity.ok(responseData);
    }
}