import java.lang.annotation.*;

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface EncryptResponse {
}

@EncryptResponse
@RequestMapping("/getInfo")
public Result info(String uid) {
    // 模擬查詢數(shù)據(jù) 
    String json = "{"uid":"" + uid + "","name":"testUser","pwd":"123456"}"; 
    // 使用AES加密并返回Base64編碼的結(jié)果 
    String encryptedData = SecureUtil.aes(aesKey).encryptBase64(json); return Result.ok(encryptedData);

}

import cn.hutool.core.codec.Base64;
import cn.hutool.crypto.SecureUtil;
import com.qiming.common.utils.Result;
import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

@RestControllerAdvice
public class AESResponseEncryptAdvice implements ResponseBodyAdvice<Object> {

    // 與加密工具類中一致的 AES 密鑰（Base64 解碼）
    private static final byte[] AES_KEY = Base64.decode("ew7UkDs58QfulRGXD24R8A==");

    @Override
    public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
        // 若想全局加密，直接返回 true
        return true;
    }

    @Override
    public Object beforeBodyWrite(Object body, MethodParameter returnType,
                                  MediaType selectedContentType,
                                  Class<? extends HttpMessageConverter<?>> selectedConverterType,
                                  ServerHttpRequest request, ServerHttpResponse response) {

        // 僅處理 Result 類型（或你自定義的統(tǒng)一返回類型），或 JSON 響應(yīng)
        if (body instanceof Result) {
            Result result = (Result) body;

            // 只有成功響應(yīng)，且需要加密時才加密 data 字段（可根據(jù)需求調(diào)整）
            if (result.isSuccess()) {
                try {
                    // 將整個返回的 data 部分（通常是 JSON 字符串或?qū)ο螅┻M(jìn)行加密
                    // 這里簡單起見，直接將整個 Result 轉(zhuǎn)為 JSON 字符串然后加密
                    String originalJson = JsonUtil.toJson(result); 
                    String encrypted = SecureUtil.aes(AES_KEY).encryptBase64(originalJson);
                    // 返回一個特殊結(jié)構(gòu)，或者直接返回加密后的密文
                    return Result.ok(encrypted);
                } catch (Exception e) {
                    e.printStackTrace();
                    return Result.fail("響應(yīng)加密失敗");
                }
            }
        }

        // 非加密接口，直接返回原數(shù)據(jù)
        return body;
    }
}