mirror of
https://github.com/ccmjga/zhilu-admin
synced 2026-03-14 05:33:42 +08:00
add curl field
This commit is contained in:
@@ -6,18 +6,24 @@ import com.zl.mjga.annotation.SkipAopLog;
|
||||
import com.zl.mjga.repository.UserRepository;
|
||||
import com.zl.mjga.service.AopLogService;
|
||||
import jakarta.servlet.http.HttpServletRequest;
|
||||
import java.io.BufferedReader;
|
||||
import java.io.IOException;
|
||||
import java.lang.reflect.Method;
|
||||
import java.time.Duration;
|
||||
import java.time.Instant;
|
||||
import java.util.Collections;
|
||||
import java.util.Enumeration;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.lang3.ArrayUtils;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.aspectj.lang.ProceedingJoinPoint;
|
||||
import org.aspectj.lang.annotation.Around;
|
||||
import org.aspectj.lang.annotation.Aspect;
|
||||
import org.aspectj.lang.reflect.MethodSignature;
|
||||
import org.jooq.generated.mjga.tables.pojos.AopLog;
|
||||
import org.jooq.generated.mjga.tables.pojos.User;
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
|
||||
import org.springframework.security.core.Authentication;
|
||||
import org.springframework.security.core.context.SecurityContextHolder;
|
||||
import org.springframework.stereotype.Component;
|
||||
@@ -28,6 +34,7 @@ import org.springframework.web.context.request.ServletRequestAttributes;
|
||||
@Component
|
||||
@Slf4j
|
||||
@RequiredArgsConstructor
|
||||
@ConditionalOnProperty(name = "aop.logging.enabled", havingValue = "true", matchIfMissing = true)
|
||||
public class LoggingAspect {
|
||||
|
||||
private final AopLogService aopLogService;
|
||||
@@ -137,6 +144,7 @@ public class LoggingAspect {
|
||||
HttpServletRequest request = attributes.getRequest();
|
||||
aopLog.setIpAddress(getClientIpAddress(request));
|
||||
aopLog.setUserAgent(request.getHeader("User-Agent"));
|
||||
aopLog.setCurl(generateCurlCommand(request));
|
||||
}
|
||||
|
||||
private String getClientIpAddress(HttpServletRequest request) {
|
||||
@@ -179,4 +187,132 @@ public class LoggingAspect {
|
||||
return e.getMessage();
|
||||
}
|
||||
}
|
||||
|
||||
public String generateCurlCommand(HttpServletRequest request) {
|
||||
try {
|
||||
StringBuilder curl = new StringBuilder("curl -X ");
|
||||
|
||||
curl.append(request.getMethod());
|
||||
|
||||
String url = getFullRequestUrl(request);
|
||||
curl.append(" '").append(url).append("'");
|
||||
|
||||
appendHeaders(curl, request);
|
||||
|
||||
if (hasRequestBody(request.getMethod())) {
|
||||
appendRequestBody(curl, request);
|
||||
}
|
||||
return curl.toString();
|
||||
} catch (Exception e) {
|
||||
log.error("Failed to generate curl command", e);
|
||||
return "curl command generation failed: " + e.getMessage();
|
||||
}
|
||||
}
|
||||
|
||||
private String getFullRequestUrl(HttpServletRequest request) {
|
||||
StringBuilder url = new StringBuilder();
|
||||
|
||||
String scheme = request.getScheme();
|
||||
String serverName = request.getServerName();
|
||||
int serverPort = request.getServerPort();
|
||||
|
||||
if (scheme == null) {
|
||||
scheme = "http";
|
||||
}
|
||||
if (serverName == null) {
|
||||
serverName = "localhost";
|
||||
}
|
||||
|
||||
url.append(scheme).append("://").append(serverName);
|
||||
|
||||
if ((scheme.equals("http") && serverPort != 80)
|
||||
|| (scheme.equals("https") && serverPort != 443)) {
|
||||
url.append(":").append(serverPort);
|
||||
}
|
||||
|
||||
url.append(request.getRequestURI());
|
||||
if (request.getQueryString() != null) {
|
||||
url.append("?").append(request.getQueryString());
|
||||
}
|
||||
|
||||
return url.toString();
|
||||
}
|
||||
|
||||
private void appendHeaders(StringBuilder curl, HttpServletRequest request) {
|
||||
Enumeration<String> headerNames = request.getHeaderNames();
|
||||
for (String headerName : Collections.list(headerNames)) {
|
||||
if (shouldSkipHeader(headerName)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
String headerValue = request.getHeader(headerName);
|
||||
curl.append(" -H '").append(headerName).append(": ").append(headerValue).append("'");
|
||||
}
|
||||
}
|
||||
|
||||
private boolean shouldSkipHeader(String headerName) {
|
||||
String lowerName = headerName.toLowerCase();
|
||||
return lowerName.equals("host")
|
||||
|| lowerName.equals("content-length")
|
||||
|| lowerName.equals("connection")
|
||||
|| lowerName.startsWith("sec-")
|
||||
|| lowerName.equals("upgrade-insecure-requests");
|
||||
}
|
||||
|
||||
private boolean hasRequestBody(String method) {
|
||||
return "POST".equalsIgnoreCase(method)
|
||||
|| "PUT".equalsIgnoreCase(method)
|
||||
|| "PATCH".equalsIgnoreCase(method);
|
||||
}
|
||||
|
||||
private void appendRequestBody(StringBuilder curl, HttpServletRequest request) {
|
||||
try {
|
||||
String contentType = request.getContentType();
|
||||
if (StringUtils.contains(contentType, "application/json")) {
|
||||
String body = getRequestBody(request);
|
||||
if (StringUtils.isNotEmpty(body)) {
|
||||
curl.append(" -d '").append(body.replace("'", "\\'")).append("'");
|
||||
}
|
||||
} else if (StringUtils.contains(contentType, "application/x-www-form-urlencoded")) {
|
||||
appendFormData(curl, request);
|
||||
}
|
||||
} catch (Exception e) {
|
||||
log.warn("Failed to append request body to curl command", e);
|
||||
}
|
||||
}
|
||||
|
||||
private String getRequestBody(HttpServletRequest request) {
|
||||
try (BufferedReader reader = request.getReader()) {
|
||||
if (reader == null) {
|
||||
return null;
|
||||
}
|
||||
StringBuilder body = new StringBuilder();
|
||||
String line;
|
||||
while ((line = reader.readLine()) != null) {
|
||||
body.append(line);
|
||||
}
|
||||
return body.toString();
|
||||
} catch (IOException e) {
|
||||
log.warn("Failed to read request body", e);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
private void appendFormData(StringBuilder curl, HttpServletRequest request) {
|
||||
Enumeration<String> paramNames = request.getParameterNames();
|
||||
StringBuilder formData = new StringBuilder();
|
||||
while (paramNames.hasMoreElements()) {
|
||||
String paramName = paramNames.nextElement();
|
||||
String[] paramValues = request.getParameterValues(paramName);
|
||||
for (String paramValue : paramValues) {
|
||||
if (!formData.isEmpty()) {
|
||||
formData.append("&");
|
||||
}
|
||||
formData.append(paramName).append("=").append(paramValue);
|
||||
}
|
||||
}
|
||||
if (!formData.isEmpty()) {
|
||||
curl.append(" -d '").append(formData).append("'");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user