src/main/java/bench/LogWorker.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

package bench;

import org.apache.logging.log4j.Logger;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicLong;

public class LogWorker implements Runnable {
    private final Logger logger;
    private final String message;
    private final CountDownLatch startLatch;
    private final AtomicBoolean running;
    private final AtomicLong eventCounter;
    private final AtomicLong targetEvents;
    private final BenchConfig.Mode mode;

    public LogWorker(Logger logger, String message, CountDownLatch startLatch,
                     AtomicBoolean running, AtomicLong eventCounter,
                     AtomicLong targetEvents, BenchConfig.Mode mode) {
        this.logger = logger;
        this.message = message;
        this.startLatch = startLatch;
        this.running = running;
        this.eventCounter = eventCounter;
        this.targetEvents = targetEvents;
        this.mode = mode;
    }

    @Override
    public void run() {
        try {
            startLatch.await();
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
            return;
        }

        if (mode == BenchConfig.Mode.DURATION) {
            while (running.get()) {
                logger.info(message);
                eventCounter.incrementAndGet();
            }
        } else {
            while (running.get() && eventCounter.get() < targetEvents.get()) {
                logger.info(message);
                eventCounter.incrementAndGet();
            }
        }
    }
}