release: 0.9.6 (#105)

Author: devxb

src/main/kotlin/org/gitanimals/core/OrchestratorExtension.kt

@@ -4,6 +4,7 @@ import org.gitanimals.core.filter.MDCFilter.Companion.TRACE_ID
 import org.rooftop.netx.api.Context
 import org.rooftop.netx.api.ContextOrchestrate
 import org.rooftop.netx.api.ContextRollback
+import org.slf4j.LoggerFactory
 import org.slf4j.MDC
 
 open class TraceIdContextOrchestrator<T : Any, V : Any>(
@@ -12,10 +13,19 @@ open class TraceIdContextOrchestrator<T : Any, V : Any>(
 
     override fun orchestrate(context: Context, request: T): V {
         MDC.put(TRACE_ID, context.decodeContext(TRACE_ID, String::class))
-        return orchestrate.orchestrate(context, request).also {
+        return runCatching {
+            orchestrate.orchestrate(context, request)
+        }.getOrElse {
+            logger.warn("Orchestrating fail, rollback start. request: \"$request\"", it)
+            throw it
+        }.also {
             MDC.remove(TRACE_ID)
         }
     }
+
+    private companion object {
+        private val logger = LoggerFactory.getLogger(this::class.simpleName)
+    }
 }
 
 open class TraceIdContextRollback<T : Any, V : Any?>(
@@ -24,8 +34,17 @@ open class TraceIdContextRollback<T : Any, V : Any?>(
 
     override fun rollback(context: Context, request: T): V {
         MDC.put(TRACE_ID, context.decodeContext(TRACE_ID, String::class))
-        return rollback.rollback(context, request).also {
+        return runCatching {
+            rollback.rollback(context, request)
+        }.getOrElse {
+            logger.error("Rollback fail need to fix. request: \"$request\"", it)
+            throw it
+        }.also {
             MDC.remove(TRACE_ID)
         }
     }
+
+    private companion object {
+        private val logger = LoggerFactory.getLogger(this::class.simpleName)
+    }
 }