346d23db8bcefc4ab0acf5db6dd8f8bd90bd1b3b
[policy/apex-pdp.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
4  *  Modifications Copyright (C) 2021 AT&T Intellectual Property. All rights reserved.
5  * ================================================================================
6  * Licensed under the Apache License, Version 2.0 (the "License");
7  * you may not use this file except in compliance with the License.
8  * You may obtain a copy of the License at
9  *
10  *      http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing, software
13  * distributed under the License is distributed on an "AS IS" BASIS,
14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  * See the License for the specific language governing permissions and
16  * limitations under the License.
17  *
18  * SPDX-License-Identifier: Apache-2.0
19  * ============LICENSE_END=========================================================
20  */
21
22 package org.onap.policy.apex.plugins.context.distribution.infinispan;
23
24 import java.util.Map;
25 import lombok.AccessLevel;
26 import lombok.Setter;
27 import org.onap.policy.apex.context.ContextException;
28 import org.onap.policy.apex.context.impl.distribution.AbstractDistributor;
29 import org.onap.policy.apex.context.parameters.ContextParameterConstants;
30 import org.onap.policy.apex.model.basicmodel.concepts.AxArtifactKey;
31 import org.onap.policy.common.parameters.ParameterService;
32 import org.slf4j.ext.XLogger;
33 import org.slf4j.ext.XLoggerFactory;
34
35 /**
36  * This context distributor distributes context across threads in multiple JVMs on multiple hosts. It uses Infinispan to
37  * distribute maps.
38  *
39  * @author Liam Fallon (liam.fallon@ericsson.com)
40  */
41 public class InfinispanContextDistributor extends AbstractDistributor {
42     // Logger for this class
43     private static final XLogger LOGGER = XLoggerFactory.getXLogger(InfinispanContextDistributor.class);
44
45     // The infinispan manager for distributing context for this JVM
46     @Setter(AccessLevel.PRIVATE)
47     private static InfinispanManager infinispanManager = null;
48
49     /**
50      * Create an instance of an Infinispan Context Distributor.
51      *
52      * @throws ContextException On errors creating the context distributor
53      */
54     public InfinispanContextDistributor() throws ContextException {
55         LOGGER.entry("InfinispanContextDistributor()");
56
57         LOGGER.exit("InfinispanContextDistributor()");
58     }
59
60     /**
61      * {@inheritDoc}.
62      */
63     @Override
64     public void init(final AxArtifactKey key) throws ContextException {
65         LOGGER.entry("init(" + key + ")");
66
67         super.init(key);
68
69         // Create the infinispan manager if it does not already exist
70         if (infinispanManager == null) {
71             // Get the parameters from the parameter service
72             final InfinispanDistributorParameters parameters = ParameterService
73                             .get(ContextParameterConstants.DISTRIBUTOR_GROUP_NAME);
74
75             LOGGER.debug("initiating Infinispan with the parameters: {}", parameters);
76
77             // Create the manager
78             setInfinispanManager(new InfinispanManager(parameters));
79         }
80
81         LOGGER.exit("init(" + key + ")");
82     }
83
84     /**
85      * {@inheritDoc}.
86      */
87     @Override
88     public Map<String, Object> getContextAlbumMap(final AxArtifactKey contextAlbumKey) {
89         LOGGER.info("InfinispanContextDistributor: create album: " + contextAlbumKey.getId());
90
91         // Get the Cache from Infinispan
92         return infinispanManager.getCacheManager().getCache(contextAlbumKey.getId().replace(':', '_'));
93     }
94
95     /**
96      * {@inheritDoc}.
97      */
98     @Override
99     public void shutdown() {
100         // Shut down the infinispan manager
101         if (infinispanManager != null) {
102             infinispanManager.shutdown();
103         }
104         setInfinispanManager(null);
105     }
106 }