plugins/plugins-context/plugins-context-distribution/plugins-context-distribution-infinispan/src/main/java/org/onap/policy/apex/plugins/context/distribution/infinispan/InfinispanContextDistributor.java

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