cps-service/src/main/java/org/onap/cps/cache/HazelcastCacheConfig.java

   1 /*
   2  * ============LICENSE_START========================================================
   3  *  Copyright (C) 2023 Nordix Foundation
   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.cps.cache;
  22
  23 import com.hazelcast.config.Config;
  24 import com.hazelcast.config.MapConfig;
  25 import com.hazelcast.config.NamedConfig;
  26 import com.hazelcast.config.QueueConfig;
  27 import com.hazelcast.config.SetConfig;
  28 import com.hazelcast.core.Hazelcast;
  29 import com.hazelcast.core.HazelcastInstance;
  30 import lombok.extern.slf4j.Slf4j;
  31 import org.springframework.beans.factory.annotation.Value;
  32
  33 /**
  34  * Core infrastructure of the hazelcast distributed cache.
  35  */
  36 @Slf4j
  37 public class HazelcastCacheConfig {
  38
  39     @Value("${hazelcast.cluster-name}")
  40     protected String clusterName;
  41
  42     @Value("${hazelcast.mode.kubernetes.enabled}")
  43     protected boolean cacheKubernetesEnabled;
  44
  45     @Value("${hazelcast.mode.kubernetes.service-name}")
  46     protected String cacheKubernetesServiceName;
  47
  48     protected HazelcastInstance createHazelcastInstance(final String hazelcastInstanceName,
  49                                                         final NamedConfig namedConfig) {
  50         return Hazelcast.newHazelcastInstance(initializeConfig(hazelcastInstanceName, namedConfig));
  51     }
  52
  53     private Config initializeConfig(final String instanceName, final NamedConfig namedConfig) {
  54         final Config config = new Config(instanceName);
  55         if (namedConfig instanceof MapConfig) {
  56             config.addMapConfig((MapConfig) namedConfig);
  57         }
  58         if (namedConfig instanceof QueueConfig) {
  59             config.addQueueConfig((QueueConfig) namedConfig);
  60         }
  61         if (namedConfig instanceof SetConfig) {
  62             config.addSetConfig((SetConfig) namedConfig);
  63         }
  64
  65         config.setClusterName(clusterName);
  66         config.setClassLoader(org.onap.cps.spi.model.Dataspace.class.getClassLoader());
  67         updateDiscoveryMode(config);
  68         return config;
  69     }
  70
  71     protected static MapConfig createMapConfig(final String configName) {
  72         final MapConfig mapConfig = new MapConfig(configName);
  73         mapConfig.setBackupCount(3);
  74         mapConfig.setAsyncBackupCount(3);
  75         return mapConfig;
  76     }
  77
  78     protected static QueueConfig createQueueConfig(final String configName) {
  79         final QueueConfig commonQueueConfig = new QueueConfig(configName);
  80         commonQueueConfig.setBackupCount(3);
  81         commonQueueConfig.setAsyncBackupCount(3);
  82         return commonQueueConfig;
  83     }
  84
  85     protected static SetConfig createSetConfig(final String configName) {
  86         final SetConfig commonSetConfig = new SetConfig(configName);
  87         commonSetConfig.setBackupCount(1);
  88         commonSetConfig.setAsyncBackupCount(1);
  89         return commonSetConfig;
  90     }
  91
  92     protected void updateDiscoveryMode(final Config config) {
  93         if (cacheKubernetesEnabled) {
  94             log.info("Enabling kubernetes mode with service-name : {}", cacheKubernetesServiceName);
  95             config.getNetworkConfig().getJoin().getKubernetesConfig().setEnabled(true)
  96                 .setProperty("service-name", cacheKubernetesServiceName);
  97         }
  98     }
  99
 100 }