summaryrefslogtreecommitdiffstats
path: root/test/mocks/pnfsimulator/src/main/java/org/onap/pnfsimulator/simulator/Simulator.java
blob: 75ce71e5ef86c6d353af8c17c1ce173f867a2f1b (plain)
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
package org.onap.pnfsimulator.simulator;

import java.time.Duration;
import java.time.Instant;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.json.JSONObject;
import org.onap.pnfsimulator.simulator.client.HttpClientProvider;

public class Simulator {

    private static final Logger logger = LogManager.getLogger(HttpClientProvider.class);
    private HttpClientProvider clientProvider;
    private JSONObject messageBody;
    private Duration duration;
    private Duration interval;

    public Simulator(String vesServerUrl, JSONObject messageBody, Duration duration, Duration interval) {
        this.messageBody = messageBody;
        this.duration = duration;
        this.interval = interval;
        this.clientProvider = new HttpClientProvider(vesServerUrl);
    }

    public void start() {
        logger.info("SIMULATOR STARTED - DURATION: {}s, INTERVAL: {}s", duration.getSeconds(), interval.getSeconds());

        Instant endTime = Instant.now().plus(duration);
        while (runningTimeNotExceeded(endTime)) {
            try {
                logger.info("MESSAGE TO BE SENT:\n{}", messageBody.toString(4));
                clientProvider.sendMsg(messageBody.toString());
                Thread.sleep(interval.toMillis());
            } catch (InterruptedException e) {
                logger.error("SIMULATOR INTERRUPTED");
                break;
            }
        }
        logger.info("SIMULATOR FINISHED");
    }

    private boolean runningTimeNotExceeded(Instant endTime) {
        return Instant.now().isBefore(endTime);
    }
}