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