/
TurboMarkerTurboFilter.java
76 lines (67 loc) · 2.27 KB
/
TurboMarkerTurboFilter.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
66
67
68
69
70
71
72
73
74
75
76
/*
* SPDX-License-Identifier: CC0-1.0
*
* Copyright 2018-2020 Will Sargent.
*
* Licensed under the CC0 Public Domain Dedication;
* You may obtain a copy of the License at
*
* http://creativecommons.org/publicdomain/zero/1.0/
*/
package com.tersesystems.logback.turbomarker;
import static java.util.Objects.requireNonNull;
import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.Logger;
import ch.qos.logback.classic.turbo.TurboFilter;
import ch.qos.logback.core.spi.FilterReply;
import com.tersesystems.logback.classic.TurboFilterDecider;
import java.util.Spliterator;
import java.util.Spliterators;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
import org.slf4j.Marker;
/**
* This class is a turbo filter that hands off the evaluation of whether a logging event should be
* created to the marker, if it is a predicate marker.
*/
public class TurboMarkerTurboFilter extends TurboFilter implements TurboFilterDecider {
@Override
public FilterReply decide(
Marker rootMarker, Logger logger, Level level, String format, Object[] params, Throwable t) {
if (!isStarted()) {
return FilterReply.NEUTRAL;
}
if (rootMarker == null) {
return FilterReply.NEUTRAL;
}
if (evaluateMarker(rootMarker, rootMarker, logger, level, format, params, t)
== FilterReply.ACCEPT) {
return FilterReply.ACCEPT;
}
return stream(rootMarker)
.map(m -> evaluateMarker(m, rootMarker, logger, level, format, params, t))
.filter(reply -> reply != FilterReply.NEUTRAL)
.findFirst()
.orElse(FilterReply.NEUTRAL);
}
private FilterReply evaluateMarker(
Marker marker,
Marker rootMarker,
Logger logger,
Level level,
String format,
Object[] params,
Throwable t) {
if (marker instanceof TurboFilterDecider) {
TurboFilterDecider decider = (TurboFilterDecider) marker;
return decider.decide(rootMarker, logger, level, format, params, t);
}
return FilterReply.NEUTRAL;
}
@SuppressWarnings("unchecked")
private Stream<Marker> stream(Marker marker) {
requireNonNull(marker);
Spliterator spliterator = Spliterators.spliteratorUnknownSize(marker.iterator(), 0);
return (Stream<Marker>) StreamSupport.stream(spliterator, false);
}
}