package cn.finopen.boot.autoconfigure.aop;@Configuration
@EnableAspectJAutoProxy
@Order
public class EndpointLogAopConfiguration {/*** 請求方法白名單*/private static final String[] METHOD_WHITE_LIST = {"get", "unreadCount", "find", "findAll"};/*** 防止表單重復提交*/@Aspect@Service@Slf4jpublic static class EndpointLogAspect {@Pointcut("execution(* *.*.boot.autoconfigure..*.endpoint.*.*(..))")public void endpointPointcut() {}@Resourceprivate SaleGroupUserLogApiService logApiService;@Resourceprivate WechatService wechatService;@Resourceprivate StorageDataProperties properties;@Around("endpointPointcut()")public Object doAround(ProceedingJoinPoint pjp) throws Throwable {long start = System.currentTimeMillis();try {Object[] args = pjp.getArgs();Class<?> targetCls = pjp.getTarget().getClass();MethodSignature methodSignature = (MethodSignature) pjp.getSignature();String simpleName = targetCls.getSimpleName();String method = methodSignature.getName();if (method.startsWith("find")) {return proceed(pjp);}for (String i : METHOD_WHITE_LIST) {if (method.equals(i)) {return proceed(pjp);}}String targetObjectMethodName = simpleName + "." + method;String targetMethodParams = Arrays.toString(args);if (targetMethodParams.startsWith("[{") && targetMethodParams.endsWith("}]")) {JSONObject jsonObject;try {jsonObject = JSON.parseObject(targetMethodParams.substring(targetMethodParams.indexOf("{"), targetMethodParams.lastIndexOf("}") + 1));String loginUserId = jsonObject.getString("loginUserId");String loginGroupId = jsonObject.getString("loginGroupId");String loginGroupName = jsonObject.getString("loginGroupName");String requestIp = jsonObject.getString("requestIp");String body = JSONObject.toJSONString(jsonObject, SerializerFeature.WriteNullListAsEmpty);if (body.length() > 1024) {//內容過長截斷body = body.substring(0, 1024);}if (loginGroupId != null && loginUserId != null) {logApiService.create(SaleGroupUserLogCreateReq.newBuilder().setLoginGroupId(Long.parseLong(loginGroupId)).setLoginUserId(loginUserId).setLoginGroupName(loginGroupName).setReqUrl(targetObjectMethodName).setReqIp(requestIp).setReqBody(body).build());}} catch (Exception ignore) {return proceed(pjp);}}return proceed(pjp);} catch (Throwable e) {throw print(e, pjp);} finally {long take = System.currentTimeMillis() - start;if (take >= 1000) {if (log.isWarnEnabled()) {log.warn("around endpoint [{}] take {}ms", pjp.getTarget().getClass().getSimpleName(), (System.currentTimeMillis() - start));}} else {if (log.isInfoEnabled()) {log.info("around endpoint [{}] take {}ms", pjp.getTarget().getClass().getSimpleName(), (System.currentTimeMillis() - start));}}}}/*** 為了捕獲異步異常** @param pjp* @return* @throws Throwable*/private Object proceed(ProceedingJoinPoint pjp) throws Throwable {String url = properties.getUrl();
// 測試環境屏蔽boolean isTest = url.startsWith("https://test") || url.startsWith("http://test");Object result = pjp.proceed();if (isTest) {return result;}//這里是異步請求，所以直接用try cache 是無效的if (result instanceof Flux) {Flux<?> r = (Flux<?>) (result);return r.onErrorMap(Throwable.class, throwable -> print(throwable, pjp));} else if (result instanceof Mono) {Mono<?> r = (Mono<?>) (result);return r.onErrorMap(Throwable.class, throwable -> print(throwable, pjp));}return result;}private final static String WEBHOOK_URL = "https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=*";/*** 發送異常信息到企業微信** @param e* @param pjp*/private Throwable print(Throwable e, ProceedingJoinPoint pjp) {
//            if (e instanceof FibException) {
//                return e;
//            }ThreadPoolUtils.execute(() -> {Object[] args = pjp.getArgs();Class<?> targetCls = pjp.getTarget().getClass();MethodSignature methodSignature = (MethodSignature) pjp.getSignature();String simpleName = targetCls.getSimpleName();String method = methodSignature.getName();String targetObjectMethodName = simpleName + "." + method;String targetMethodParams = format(Arrays.toString(args));if (targetMethodParams.length() >= 512) {targetMethodParams = targetMethodParams.substring(0, 512);}String errorMessage = getStackTrace(e);String content = "<font color=\"info\">接口地址：</font>" + targetObjectMethodName + "\n<font color=\"info\">請求報文：</font>" + targetMethodParams + "\n<font color=\"warning\">異常消息：</font>" + errorMessage;//markdown內容，最長不超過4096個字節，必須是utf8編碼if (content.length() >= 2048) {content = truncateUtf8(content);}WechatWebhookReq webhookReq = WechatWebhookReq.newBuilder().setUrl(WEBHOOK_URL).setMsgType("markdown").setContent(content).build();wechatService.send(webhookReq);});return e;}private String format(String targetMethodParams) {if (targetMethodParams.startsWith("[{") && targetMethodParams.endsWith("}]")) {try {JSONObject jsonObject = JSON.parseObject(targetMethodParams.substring(targetMethodParams.indexOf("{"), targetMethodParams.lastIndexOf("}") + 1));return JSONObject.toJSONString(jsonObject, SerializerFeature.WriteNullListAsEmpty);} catch (Exception ignore) {return targetMethodParams;}} else {return targetMethodParams;}}/**打印完整的日志信息**/String getStackTrace(Throwable e) {StringWriter sw = new StringWriter();PrintWriter pw = new PrintWriter(sw, true);// 打印當前異常的堆棧信息e.printStackTrace(pw);// 遍歷并打印所有被抑制的異常（如果有）for (Throwable suppressed : e.getSuppressed()) {suppressed.printStackTrace(pw);}// 遞歸打印異常的原因鏈（如果存在）Throwable cause = e.getCause();while (cause != null) {cause.printStackTrace(pw);cause = cause.getCause();}return sw.toString();}/*** 截取UTF-8編碼的字符串，確保字節長度不超過指定的最大值。** @param str 待截取的字符串* @return 截取后的字符串*/private final static int MARK_DOWN_MAX = 4096;/*** 簡單地根據字節數截斷UTF-8編碼的字符串，可能會導致多字節字符被截斷。** @param str 待截取的字符串* @return 截取后的字符串*/public static String truncateUtf8(String str) {byte[] bytes = str.getBytes(StandardCharsets.UTF_8);if (bytes.length <= MARK_DOWN_MAX) {return str;}// 直接截取到指定字節數對應的字符位置return new String(bytes, 0, MARK_DOWN_MAX, StandardCharsets.UTF_8);}}}