HADOOP-18567. LogThrottlingHelper: properly trigger dependent recorders in cases of infrequent logging (#5215)
Signed-off-by: Erik Krogen <xkrogen@apache.org>
Co-authored-by: Chengbing Liu <liuchengbing@qiyi.com>
(cherry picked from commit ca3526da92
)
This commit is contained in:
parent
d43fa95043
commit
bfc916e7b0
@ -262,9 +262,15 @@ public LogAction record(String recorderName, long currentTimeMs,
|
||||
if (primaryRecorderName.equals(recorderName) &&
|
||||
currentTimeMs - minLogPeriodMs >= lastLogTimestampMs) {
|
||||
lastLogTimestampMs = currentTimeMs;
|
||||
for (LoggingAction log : currentLogs.values()) {
|
||||
log.setShouldLog();
|
||||
currentLogs.replaceAll((key, log) -> {
|
||||
LoggingAction newLog = log;
|
||||
if (log.hasLogged()) {
|
||||
// create a fresh log since the old one has already been logged
|
||||
newLog = new LoggingAction(log.getValueCount());
|
||||
}
|
||||
newLog.setShouldLog();
|
||||
return newLog;
|
||||
});
|
||||
}
|
||||
if (currentLog.shouldLog()) {
|
||||
currentLog.setHasLogged();
|
||||
@ -357,6 +363,10 @@ private void setHasLogged() {
|
||||
hasLogged = true;
|
||||
}
|
||||
|
||||
private int getValueCount() {
|
||||
return stats.length;
|
||||
}
|
||||
|
||||
private void recordValues(double... values) {
|
||||
if (values.length != stats.length) {
|
||||
throw new IllegalArgumentException("received " + values.length +
|
||||
|
@ -142,6 +142,18 @@ public void testPrimaryAndDependentLoggers() {
|
||||
assertTrue(helper.record("bar", 0).shouldLog());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInfrequentPrimaryAndDependentLoggers() {
|
||||
helper = new LogThrottlingHelper(LOG_PERIOD, "foo", timer);
|
||||
|
||||
assertTrue(helper.record("foo", 0).shouldLog());
|
||||
assertTrue(helper.record("bar", 0).shouldLog());
|
||||
|
||||
// Both should log once the period has elapsed
|
||||
assertTrue(helper.record("foo", LOG_PERIOD).shouldLog());
|
||||
assertTrue(helper.record("bar", LOG_PERIOD).shouldLog());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testMultipleLoggersWithValues() {
|
||||
helper = new LogThrottlingHelper(LOG_PERIOD, "foo", timer);
|
||||
|
Loading…
Reference in New Issue
Block a user