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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
/*-
* ============LICENSE_START=======================================================
* org.onap.dcaegen2.restconfcollector
* ================================================================================
* Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
* Copyright (C) 2018 Nokia. All rights reserved.
* Copyright (C) 2018-2019 Huawei. All rights reserved.
* ================================================================================
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* ============LICENSE_END=========================================================
*/
package org.onap.dcae.common.publishing;
import com.att.nsa.cambria.client.CambriaBatchingPublisher;
import com.google.common.cache.*;
import io.vavr.collection.Map;
import io.vavr.control.Option;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.annotation.Nonnull;
import java.io.IOException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;
import static io.vavr.API.Option;
import static org.onap.dcae.common.publishing.VavrUtils.f;
/**
* @author Pawel Szalapski (pawel.szalapski@nokia.com)
*/
class DMaaPPublishersCache {
private static final Logger log = LoggerFactory.getLogger(DMaaPPublishersCache.class);
private final LoadingCache<String, CambriaBatchingPublisher> publishersCache;
private AtomicReference<Map<String, PublisherConfig>> dMaaPConfiguration;
DMaaPPublishersCache(Map<String, PublisherConfig> dMaaPConfiguration) {
this.dMaaPConfiguration = new AtomicReference<>(dMaaPConfiguration);
this.publishersCache = CacheBuilder.newBuilder()
.removalListener(new OnPublisherRemovalListener())
.build(new CambriaPublishersCacheLoader());
}
DMaaPPublishersCache(CambriaPublishersCacheLoader dMaaPPublishersCacheLoader,
OnPublisherRemovalListener onPublisherRemovalListener,
Map<String, PublisherConfig> dMaaPConfiguration) {
this.dMaaPConfiguration = new AtomicReference<>(dMaaPConfiguration);
this.publishersCache = CacheBuilder.newBuilder()
.removalListener(onPublisherRemovalListener)
.build(dMaaPPublishersCacheLoader);
}
Option<CambriaBatchingPublisher> getPublisher(String streamID) {
try {
return Option(publishersCache.getUnchecked(streamID));
} catch (Exception e) {
log.warn("Could not create / load Cambria Publisher for streamID", e);
return Option.none();
}
}
void closePublisherFor(String streamId) {
publishersCache.invalidate(streamId);
}
synchronized void reconfigure(Map<String, PublisherConfig> newConfig) {
log.info("reconfigure in DMaaPPublishersCache");
Map<String, PublisherConfig> currentConfig = dMaaPConfiguration.get();
Map<String, PublisherConfig> removedConfigurations = currentConfig
.filterKeys(domain -> !newConfig.containsKey(domain));
Map<String, PublisherConfig> changedConfigurations = newConfig
.filterKeys(e -> currentConfig.containsKey(e) && !currentConfig.get(e).equals(newConfig.get(e)));
dMaaPConfiguration.set(newConfig);
removedConfigurations.merge(changedConfigurations).forEach(e -> publishersCache.invalidate(e._1));
}
static class OnPublisherRemovalListener implements RemovalListener<String, CambriaBatchingPublisher> {
@Override
public void onRemoval(@Nonnull RemovalNotification<String, CambriaBatchingPublisher> notification) {
CambriaBatchingPublisher publisher = notification.getValue();
if (publisher != null) { // The value might get Garbage Collected at this moment, regardless of @Nonnull
try {
int timeout = 20;
TimeUnit unit = TimeUnit.SECONDS;
java.util.List<?> stuck = publisher.close(timeout, unit);
if (!stuck.isEmpty()) {
log.error(f("Publisher got stuck and did not manage to close in '%s' '%s', "
+ "%s messages were dropped", stuck.size(), timeout, unit));
}
} catch (InterruptedException | IOException e) {
log.error("Could not close Cambria publisher, some messages might have been dropped", e);
Thread.currentThread().interrupt();
}
}
}
}
class CambriaPublishersCacheLoader extends CacheLoader<String, CambriaBatchingPublisher> {
@Override
public CambriaBatchingPublisher load(@Nonnull String domain) {
return dMaaPConfiguration.get()
.get(domain)
.toTry(() -> new RuntimeException(
f("DMaaP configuration contains no configuration for domain: '%s'", domain)))
.flatMap(DMaaPPublishersBuilder::buildPublisher)
.get();
}
}
}
|