From 2b602b04852b41e963830fffd39d2d2aa6889fc8 Mon Sep 17 00:00:00 2001
From: Marcos Lopez Gonzalez <asturcon1234@gmail.com>
Date: Sun, 28 Jul 2019 16:12:14 +0200
Subject: [PATCH] actuator httpTrace endpoint

---
 spring-boot-management/pom.xml                | 47 +++++++++++++++++++
 .../spring/boot/management/trace/App.java     | 13 +++++
 .../trace/CustomTraceRepository.java          | 28 +++++++++++
 .../boot/management/trace/EchoController.java | 15 ++++++
 .../management/trace/TraceRequestFilter.java  | 27 +++++++++++
 .../src/main/resources/application.properties |  3 ++
 6 files changed, 133 insertions(+)
 create mode 100644 spring-boot-management/pom.xml
 create mode 100644 spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/App.java
 create mode 100644 spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/CustomTraceRepository.java
 create mode 100644 spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/EchoController.java
 create mode 100644 spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/TraceRequestFilter.java
 create mode 100644 spring-boot-management/src/main/resources/application.properties

diff --git a/spring-boot-management/pom.xml b/spring-boot-management/pom.xml
new file mode 100644
index 0000000000..2b3e8f5d06
--- /dev/null
+++ b/spring-boot-management/pom.xml
@@ -0,0 +1,47 @@
+<?xml version="1.0"?>
+<project
+	xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"
+	xmlns="http://maven.apache.org/POM/4.0.0"
+	xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
+	<modelVersion>4.0.0</modelVersion>
+	<parent>
+		<groupId>com.baeldung</groupId>
+		<artifactId>parent-boot-2</artifactId>
+		<version>0.0.1-SNAPSHOT</version>
+		<relativePath>../parent-boot-2</relativePath>
+	</parent>
+	<groupId>com.baeldung</groupId>
+	<artifactId>spring-boot-management</artifactId>
+	<version>0.0.1-SNAPSHOT</version>
+	<name>spring-boot-management</name>
+
+	<properties>
+		<java.version>1.8</java.version>
+	</properties>
+
+	<dependencies>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-actuator</artifactId>
+		</dependency>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-web</artifactId>
+		</dependency>
+
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-test</artifactId>
+			<scope>test</scope>
+		</dependency>
+	</dependencies>
+
+	<build>
+		<plugins>
+			<plugin>
+				<groupId>org.springframework.boot</groupId>
+				<artifactId>spring-boot-maven-plugin</artifactId>
+			</plugin>
+		</plugins>
+	</build>
+</project>
diff --git a/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/App.java b/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/App.java
new file mode 100644
index 0000000000..150e451c57
--- /dev/null
+++ b/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/App.java
@@ -0,0 +1,13 @@
+package com.baeldung.spring.boot.management.trace;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+
+@SpringBootApplication
+public class App {
+
+    public static void main(String[] args) {
+        SpringApplication.run(App.class);
+    }
+    
+}
diff --git a/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/CustomTraceRepository.java b/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/CustomTraceRepository.java
new file mode 100644
index 0000000000..13a0bdda60
--- /dev/null
+++ b/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/CustomTraceRepository.java
@@ -0,0 +1,28 @@
+package com.baeldung.spring.boot.management.trace;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.springframework.boot.actuate.trace.http.HttpTrace;
+import org.springframework.boot.actuate.trace.http.HttpTraceRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public class CustomTraceRepository implements HttpTraceRepository {
+
+    List<HttpTrace> traces = new ArrayList<>(1);
+
+    @Override
+    public List<HttpTrace> findAll() {
+        return traces;
+    }
+
+    @Override
+    public void add(HttpTrace trace) {
+        if ("GET".equals(trace.getRequest().getMethod())) {
+            traces.clear();
+            traces.add(trace);
+        }
+    }
+
+}
diff --git a/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/EchoController.java b/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/EchoController.java
new file mode 100644
index 0000000000..70bc27a3bb
--- /dev/null
+++ b/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/EchoController.java
@@ -0,0 +1,15 @@
+package com.baeldung.spring.boot.management.trace;
+
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController("echo")
+public class EchoController {
+
+    @GetMapping
+    public String echo(@RequestParam("msg") String msg) {
+        return "echoing " + msg;
+    }
+    
+}
diff --git a/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/TraceRequestFilter.java b/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/TraceRequestFilter.java
new file mode 100644
index 0000000000..fd6312df47
--- /dev/null
+++ b/spring-boot-management/src/main/java/com/baeldung/spring/boot/management/trace/TraceRequestFilter.java
@@ -0,0 +1,27 @@
+package com.baeldung.spring.boot.management.trace;
+
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServletRequest;
+
+import org.springframework.boot.actuate.trace.http.HttpExchangeTracer;
+import org.springframework.boot.actuate.trace.http.HttpTraceRepository;
+import org.springframework.boot.actuate.web.trace.servlet.HttpTraceFilter;
+import org.springframework.stereotype.Component;
+
+@Component
+public class TraceRequestFilter extends HttpTraceFilter {
+  /**
+   * Create a new {@link HttpTraceFilter} instance.
+   *
+   * @param repository the trace repository
+   * @param tracer     used to trace exchanges
+   */
+  public TraceRequestFilter(HttpTraceRepository repository, HttpExchangeTracer tracer) {
+      super(repository, tracer);
+  }
+
+  @Override
+  protected boolean shouldNotFilter(HttpServletRequest request) throws ServletException {
+      return request.getServletPath().contains("actuator");
+  }
+}
diff --git a/spring-boot-management/src/main/resources/application.properties b/spring-boot-management/src/main/resources/application.properties
new file mode 100644
index 0000000000..b6c303afdb
--- /dev/null
+++ b/spring-boot-management/src/main/resources/application.properties
@@ -0,0 +1,3 @@
+management.endpoints.web.exposure.include=httptrace
+management.trace.http.include=REQUEST_HEADERS
+