Question

我有一个Java Akka应用程序，我想根据消息内部的信息为每个消息处理设置一个单独的MDC上下文，例如我对所有消息都有以下基本接口：

public interface IdMessage {
    String getId();
}

此外，我为所有演员提供以下基础演员：

public class BaseActor extends AbstractActor {

    private final DiagnosticLoggingAdapter log = Logging.apply(this);

    @Override
    public void aroundReceive(PartialFunction<Object, BoxedUnit> receive, Object msg) {
        if (msg instanceof IdMessage) {
            final Map<String, Object> originalMDC = log.getMDC();
            log.setMDC(ImmutableMap.of("id", ((IdMessage) msg).getId()));
            try {
                super.aroundReceive(receive, msg);
            } finally {
                if (originalMDC != null) {
                    log.setMDC(originalMDC);
                } else {
                    log.clearMDC();
                }
            }
        } else {
            super.aroundReceive(receive, msg);
        }
    }
}

实际的演员实施：

public class SomeActor extends BaseActor {
    SomeActor() {
    receive(ReceiveBuilder
                    .match(SomeMessage.class, message -> {
                        ...
                     }
    }
}

我想确保SomeActor#receive()中的所有日志条目都在BaseActor中设置了MDC上下文。要使这项工作SomeActor#receice()需要在与BaseActor#aroundReceive()方法相同的线程中执行。

我没有找到关于aroundReceive行为的任何信息 - 是否总是在与实际receive方法相同的线程中执行？根据我的测试，它总是在同一个线程中执行。

Answer 1

我能够自己弄清楚正确的实现，并希望分享它以防有人面对同样的问题。

aroundReceive将在与receive相同的线程中执行，因此这是设置MDC上下文的正确位置。

我使用org.slf4j.MDC来设置MDC上下文，这里是完整的代码：

import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.MDC;

import com.google.common.collect.ImmutableMap;

import akka.actor.AbstractActor;
import scala.PartialFunction;
import scala.runtime.BoxedUnit;

public class BaseActor extends AbstractActor {

    private final Logger log = LoggerFactory.getLogger(BaseActor.class);

    @Override
    public void aroundReceive(PartialFunction<Object, BoxedUnit> receive, Object msg) {
        if (msg instanceof IdMessage) {
            final Map<String, Object> originalMDC = log.getMDC();
            MDC.setContextMap(ImmutableMap.of("id", ((IdMessage) msg).getId()));
            try {
                super.aroundReceive(receive, msg);
            } finally {
                if (originalMDC != null) {
                    MDC.setContextMap(originalMDC);
                } else {
                    MDC.clear();
                }
            }
        } else {
            super.aroundReceive(receive, msg);
        }
    }
}

使用BaseActor的实现，receive中的所有日志条目都记录有适当的MDC上下文。可以在这个有趣的博客post中找到更多信息（使用Scala实现）。

注意：我无法使用Akka DiagnosticLoggingAdapter获得相同的功能，尽管它有设置MDC上下文的方法。

aroundReceive方法中的Actor MDC上下文

1 个答案: