|
2 | 2 |
|
3 | 3 | import com.javaworm.configme.resources.ConfigSourceResource;
|
4 | 4 | import com.javaworm.configme.sources.HttpSourceConfig;
|
5 |
| -import org.slf4j.Logger; |
6 |
| -import org.slf4j.LoggerFactory; |
7 |
| - |
8 | 5 | import java.io.IOException;
|
9 | 6 | import java.net.URI;
|
10 | 7 | import java.net.http.HttpClient;
|
|
14 | 11 | import java.util.Timer;
|
15 | 12 | import java.util.TimerTask;
|
16 | 13 | import java.util.concurrent.ConcurrentHashMap;
|
| 14 | +import org.slf4j.Logger; |
| 15 | +import org.slf4j.LoggerFactory; |
17 | 16 |
|
18 | 17 | public class HttpResourceScheduler implements ResourceScheduler {
|
19 |
| - private static final Logger log = LoggerFactory.getLogger(HttpResourceScheduler.class); |
20 |
| - private final HttpClient client = HttpClient |
21 |
| - .newBuilder() |
22 |
| - .followRedirects(HttpClient.Redirect.NORMAL) |
23 |
| - .build(); |
| 18 | + private static final Logger log = LoggerFactory.getLogger(HttpResourceScheduler.class); |
| 19 | + private final HttpClient client; |
24 | 20 |
|
25 |
| - private final FetchedDataHandler fetchedDataHandler; |
26 |
| - private final Timer timer = new Timer(HttpResourceScheduler.class.getName()); |
27 |
| - private final Map<String, TimerTask> resourceTimers = new ConcurrentHashMap<>(); |
| 21 | + private final FetchedDataHandler fetchedDataHandler; |
| 22 | + private final Timer timer = new Timer(HttpResourceScheduler.class.getName()); |
| 23 | + private final Map<String, TimerTask> resourceTimers = new ConcurrentHashMap<>(); |
28 | 24 |
|
29 |
| - public HttpResourceScheduler(FetchedDataHandler fetchedDataHandler) { |
30 |
| - this.fetchedDataHandler = fetchedDataHandler; |
31 |
| - } |
| 25 | + public HttpResourceScheduler(FetchedDataHandler fetchedDataHandler,HttpClient httpClient) { |
| 26 | + this.fetchedDataHandler = fetchedDataHandler; |
| 27 | + this.client= httpClient; |
| 28 | + } |
32 | 29 |
|
33 |
| - public void schedule(ConfigSource<HttpSourceConfig> configSource) { |
34 |
| - cancelCurrentTask(configSource); |
35 |
| - final var url = configSource.getSourceConfig().getUrl(); |
36 |
| - final var intervalSeconds = configSource.getSourceConfig().getIntervalSeconds(); |
37 |
| - final var intervalMilliseconds = (int) (intervalSeconds * 1000.0); |
38 |
| - final TimerTask task = new TimerTask() { |
39 |
| - @Override |
40 |
| - public void run() { |
41 |
| - log.debug("Fetching data from {}", url); |
42 |
| - final var request = HttpRequest.newBuilder().uri(URI.create(url)).GET().build(); |
43 |
| - try { |
44 |
| - final HttpResponse<String> response; |
45 |
| - response = client.send(request, HttpResponse.BodyHandlers.ofString()); |
46 |
| - if (response.statusCode() < 200 || response.statusCode() > 299) { |
47 |
| - log.warn("Fetching data from url {} failed with status code {} and body {}", url, response.statusCode(), response.body()); |
48 |
| - return; |
49 |
| - } |
50 |
| - final var body = response.body(); |
| 30 | + public void schedule(ConfigSource<HttpSourceConfig> configSource) { |
| 31 | + cancelCurrentTask(configSource); |
| 32 | + final var url = configSource.getSourceConfig().getUrl(); |
| 33 | + final var intervalSeconds = configSource.getSourceConfig().getIntervalSeconds(); |
| 34 | + final var intervalMilliseconds = (int) (intervalSeconds * 1000.0); |
| 35 | + final TimerTask task = |
| 36 | + new TimerTask() { |
| 37 | + @Override |
| 38 | + public void run() { |
| 39 | + log.debug("Fetching data from {}", url); |
| 40 | + final var request = HttpRequest.newBuilder().uri(URI.create(url)).GET().build(); |
| 41 | + try { |
| 42 | + final HttpResponse<String> response; |
| 43 | + response = client.send(request, HttpResponse.BodyHandlers.ofString()); |
| 44 | + if (response.statusCode() < 200 || response.statusCode() > 299) { |
| 45 | + configSource.getContext().emit("Error"); |
| 46 | + log.warn( |
| 47 | + "Fetching data from url {} failed with status code {} and body {}", |
| 48 | + url, |
| 49 | + response.statusCode(), |
| 50 | + response.body()); |
| 51 | + return; |
| 52 | + } |
| 53 | + final var body = response.body(); |
51 | 54 |
|
52 |
| - fetchedDataHandler.handle(configSource, body); |
53 |
| - } catch (IOException | InterruptedException e) { |
54 |
| - log.error("Error in fetching data from " + url, e); |
55 |
| - } |
| 55 | + fetchedDataHandler.handle(configSource, body); |
| 56 | + } catch (IOException | InterruptedException e) { |
| 57 | + configSource.getContext().emit("Error"); |
| 58 | + log.error("Error in fetching data from " + url, e); |
56 | 59 | }
|
| 60 | + } |
57 | 61 | };
|
58 | 62 |
|
59 |
| - timer.schedule(task, 0, intervalMilliseconds); |
60 |
| - resourceTimers.put(configSource.getUid(), task); |
61 |
| - log.info("Config source {} scheduled to refresh every {}s", configSource.getName(), intervalSeconds); |
62 |
| - } |
| 63 | + timer.schedule(task, 0, intervalMilliseconds); |
| 64 | + resourceTimers.put(configSource.getUid(), task); |
| 65 | + log.info( |
| 66 | + "Config source {} scheduled to refresh every {}s", configSource.getName(), intervalSeconds); |
| 67 | + } |
63 | 68 |
|
64 |
| - private void cancelCurrentTask(ConfigSource<HttpSourceConfig> configSource) { |
65 |
| - final var uid = configSource.getUid(); |
66 |
| - final var currentTimerTask = resourceTimers.get(uid); |
67 |
| - if (currentTimerTask == null) { |
68 |
| - log.info("Resource with UID [{}] has no task yet!", uid); |
69 |
| - return; |
70 |
| - } |
71 |
| - log.info("Canceling task for Resource UID [{}]", uid); |
72 |
| - final var cancellationResult = currentTimerTask.cancel(); |
73 |
| - log.info("Canceling task result [{}]", cancellationResult); |
74 |
| - if (cancellationResult) { |
75 |
| - resourceTimers.remove(uid); |
76 |
| - } |
| 69 | + private void cancelCurrentTask(ConfigSource<HttpSourceConfig> configSource) { |
| 70 | + final var uid = configSource.getUid(); |
| 71 | + final var currentTimerTask = resourceTimers.get(uid); |
| 72 | + if (currentTimerTask == null) { |
| 73 | + log.debug("Resource with UID [{}] has no task yet!", uid); |
| 74 | + return; |
77 | 75 | }
|
| 76 | + log.debug("Canceling task for Resource UID [{}]", uid); |
| 77 | + final var cancellationResult = currentTimerTask.cancel(); |
| 78 | + log.debug("Canceling task result [{}]", cancellationResult); |
| 79 | + if (cancellationResult) { |
| 80 | + resourceTimers.remove(uid); |
| 81 | + } |
| 82 | + } |
78 | 83 |
|
79 |
| - @Override |
80 |
| - public void cancel(RequestContext<ConfigSourceResource> context) { |
81 |
| - final var uid = context.getResource().getMetadata().getUid(); |
82 |
| - final var timerTask = resourceTimers.get(uid); |
83 |
| - if (timerTask != null) { |
84 |
| - log.info("Canceling task for Resource UID [{}]", uid); |
85 |
| - final var cancellationResult = timerTask.cancel(); |
86 |
| - log.info("Canceling task result for UID [{}] is [{}]", uid, cancellationResult); |
87 |
| - if (cancellationResult) { |
88 |
| - resourceTimers.remove(uid); |
89 |
| - } |
90 |
| - } |
| 84 | + @Override |
| 85 | + public void cancel(RequestContext<ConfigSourceResource> context) { |
| 86 | + final var uid = context.getResource().getMetadata().getUid(); |
| 87 | + final var timerTask = resourceTimers.get(uid); |
| 88 | + if (timerTask != null) { |
| 89 | + log.debug("Canceling task for Resource UID [{}]", uid); |
| 90 | + final var cancellationResult = timerTask.cancel(); |
| 91 | + log.debug("Canceling task result for UID [{}] is [{}]", uid, cancellationResult); |
| 92 | + if (cancellationResult) { |
| 93 | + resourceTimers.remove(uid); |
| 94 | + } |
91 | 95 | }
|
| 96 | + } |
92 | 97 | }
|
0 commit comments