/
ActorWithMDC.java
65 lines (53 loc) · 1.39 KB
/
ActorWithMDC.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
/*
* Copyright (C) 2018-2021 Lightbend Inc. <https://www.lightbend.com>
*/
package akka.event;
import akka.actor.AbstractActor;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.Map;
public class ActorWithMDC extends AbstractActor {
private final DiagnosticLoggingAdapter logger = Logging.getLogger(this);
@Override
public Receive createReceive() {
return receiveBuilder().match(Log.class, this::receiveLog).build();
}
private void receiveLog(Log log) {
Map<String, Object> mdc;
if (log.message.startsWith("No MDC")) {
mdc = Collections.emptyMap();
} else if (log.message.equals("Null MDC")) {
mdc = null;
} else {
mdc = new LinkedHashMap<String, Object>();
mdc.put("messageLength", log.message.length());
}
logger.setMDC(mdc);
switch (log.level()) {
case 1:
logger.error(log.message);
break;
case 2:
logger.warning(log.message);
break;
case 3:
logger.info(log.message);
break;
default:
logger.debug(log.message);
break;
}
logger.clearMDC();
}
public static class Log {
private final Object level;
public final String message;
public Log(Object level, String message) {
this.level = level;
this.message = message;
}
public int level() {
return (Integer) this.level;
}
}
}