Add JBoss Logging and log exceptions when calling the thread context

providers
smallrye · Aug 21, 2020 · 3060d43 · 3060d43
1 parent 62aa0eb
commit 3060d43
Show file tree

Hide file tree

Showing 5 changed files with 117 additions and 16 deletions.
diff --git a/core/pom.xml b/core/pom.xml
@@ -33,6 +33,20 @@
             <artifactId>microprofile-config-api</artifactId>
             <version>${version.microprofile.config}</version>
         </dependency>
+        <dependency>
+            <groupId>org.jboss.logging</groupId>
+            <artifactId>jboss-logging</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.jboss.logging</groupId>
+            <artifactId>jboss-logging-annotations</artifactId>
+            <scope>provided</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.jboss.logging</groupId>
+            <artifactId>jboss-logging-processor</artifactId>
+            <scope>provided</scope>
+        </dependency>
         <dependency>
             <groupId>org.jboss.threads</groupId>
             <artifactId>jboss-threads</artifactId>

diff --git a/core/src/main/java/io/smallrye/context/impl/ActiveContextState.java b/core/src/main/java/io/smallrye/context/impl/ActiveContextState.java
@@ -1,5 +1,7 @@
 package io.smallrye.context.impl;
 
+import static io.smallrye.context.logging.SmallRyeContextPropagationLogger.ROOT_LOGGER;
+
 import java.util.ArrayList;
 import java.util.List;
 
@@ -13,16 +15,38 @@ public class ActiveContextState {
     private List<ThreadContextController> activeContext;
 
     public ActiveContextState(SmallRyeContextManager context, List<ThreadContextSnapshot> threadContext) {
-        activeContext = new ArrayList<>(threadContext.size());
-        for (ThreadContextSnapshot threadContextSnapshot : threadContext) {
-            activeContext.add(threadContextSnapshot.begin());
+        try {
+            activeContext = new ArrayList<>(threadContext.size());
+            for (ThreadContextSnapshot threadContextSnapshot : threadContext) {
+                activeContext.add(threadContextSnapshot.begin());
+            }
+        } catch (Throwable t) {
+            ROOT_LOGGER.errorBeginningThreadContextSnapshot(t.getLocalizedMessage());
+            if (ROOT_LOGGER.isDebugEnabled()) {
+                t.printStackTrace();
+            }
+            if (t instanceof RuntimeException) {
+                throw (RuntimeException) t;
+            }
+            throw new RuntimeException(t);
         }
     }
 
     public void endContext() {
-        // restore in reverse order
-        for (int i = activeContext.size() - 1; i >= 0; i--) {
-            activeContext.get(i).endContext();
+        try {
+            // restore in reverse order
+            for (int i = activeContext.size() - 1; i >= 0; i--) {
+                activeContext.get(i).endContext();
+            }
+        } catch (Throwable t) {
+            ROOT_LOGGER.errorEndingContext(t.getLocalizedMessage());
+            if (ROOT_LOGGER.isDebugEnabled()) {
+                t.printStackTrace();
+            }
+            if (t instanceof RuntimeException) {
+                throw (RuntimeException) t;
+            }
+            throw new RuntimeException(t);
         }
     }
 }
diff --git a/core/src/main/java/io/smallrye/context/impl/CapturedContextState.java b/core/src/main/java/io/smallrye/context/impl/CapturedContextState.java
@@ -1,5 +1,7 @@
 package io.smallrye.context.impl;
 
+import static io.smallrye.context.logging.SmallRyeContextPropagationLogger.ROOT_LOGGER;
+
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Map;
@@ -16,18 +18,27 @@ public class CapturedContextState {
 
     public CapturedContextState(SmallRyeContextManager context, ThreadContextProviderPlan plan,
             Map<String, String> props) {
-        this.context = context;
-        for (ThreadContextProvider provider : plan.propagatedProviders) {
-            ThreadContextSnapshot snapshot = provider.currentContext(props);
-            if (snapshot != null) {
-                threadContext.add(snapshot);
+        try {
+            this.context = context;
+            for (ThreadContextProvider provider : plan.propagatedProviders) {
+                ThreadContextSnapshot snapshot = provider.currentContext(props);
+                if (snapshot != null) {
+                    threadContext.add(snapshot);
+                }
             }
-        }
-        for (ThreadContextProvider provider : plan.clearedProviders) {
-            ThreadContextSnapshot snapshot = provider.clearedContext(props);
-            if (snapshot != null) {
-                threadContext.add(snapshot);
+            for (ThreadContextProvider provider : plan.clearedProviders) {
+                ThreadContextSnapshot snapshot = provider.clearedContext(props);
+                if (snapshot != null) {
+                    threadContext.add(snapshot);
+                }
+            }
+        } catch (Throwable t) {
+            ROOT_LOGGER.errorGettingSnapshot(t.getLocalizedMessage());
+            t.printStackTrace();
+            if (t instanceof RuntimeException) {
+                throw (RuntimeException) t;
             }
+            throw new RuntimeException(t);
         }
     }
 

diff --git a/core/src/main/java/io/smallrye/context/logging/SmallRyeContextPropagationLogger.java b/core/src/main/java/io/smallrye/context/logging/SmallRyeContextPropagationLogger.java
@@ -0,0 +1,30 @@
+package io.smallrye.context.logging;
+
+import static org.jboss.logging.Logger.Level.ERROR;
+
+import org.jboss.logging.BasicLogger;
+import org.jboss.logging.Logger;
+import org.jboss.logging.annotations.LogMessage;
+import org.jboss.logging.annotations.Message;
+import org.jboss.logging.annotations.MessageLogger;
+
+/**
+ * @author <a href="mailto:[email protected]">Kabir Khan</a>
+ */
+@MessageLogger(projectCode = "SRCP", length = 5)
+public interface SmallRyeContextPropagationLogger extends BasicLogger {
+    SmallRyeContextPropagationLogger ROOT_LOGGER = Logger.getMessageLogger(SmallRyeContextPropagationLogger.class,
+            "io.smallrye.context");
+
+    @LogMessage(level = ERROR)
+    @Message(id = 1, value = "An error occurred beginning the ThreadContextSnapshot: %s")
+    void errorBeginningThreadContextSnapshot(String errorMsg);
+
+    @LogMessage(level = ERROR)
+    @Message(id = 2, value = "An error occurred ending the ThreadContext: %s")
+    void errorEndingContext(String errorMsg);
+
+    @LogMessage(level = ERROR)
+    @Message(id = 3, value = "An error occurred getting the ThreadContextSnapshot: %s")
+    void errorGettingSnapshot(String errorMsg);
+}
diff --git a/pom.xml b/pom.xml
@@ -40,6 +40,8 @@
 		<version.microprofile.context-propagation>1.0.2</version.microprofile.context-propagation>
 		<version.microprofile.config>1.4</version.microprofile.config>
 		<version.smallrye.config>1.7.0</version.smallrye.config>
+        <version.jboss.logging>3.4.1.Final</version.jboss.logging>
+        <version.jboss.logging.tools>2.2.1.Final</version.jboss.logging.tools>
         <version.jboss.threads>3.1.1.Final</version.jboss.threads>
         <version.jta>1.3</version.jta>
         <version.narayana>5.10.4.Final</version.narayana>
@@ -84,6 +86,26 @@
                 <version>${version.microprofile.context-propagation}</version>
             </dependency>
 
+            <dependency>
+                <groupId>org.jboss.logging</groupId>
+                <artifactId>jboss-logging</artifactId>
+                <version>${version.jboss.logging}</version>
+            </dependency>
+
+            <dependency>
+                <groupId>org.jboss.logging</groupId>
+                <artifactId>jboss-logging-annotations</artifactId>
+                <scope>provided</scope>
+                <version>${version.jboss.logging.tools}</version>
+            </dependency>
+
+            <dependency>
+                <groupId>org.jboss.logging</groupId>
+                <artifactId>jboss-logging-processor</artifactId>
+                <scope>provided</scope>
+                <version>${version.jboss.logging.tools}</version>
+            </dependency>
+
             <dependency>
                 <groupId>io.reactivex.rxjava2</groupId>
                 <artifactId>rxjava</artifactId>