fineract-provider/src/main/java/org/apache/fineract/infrastructure/core/config/cache/CacheConfig.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you 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.
 */

package org.apache.fineract.infrastructure.core.config.cache;

import java.time.Duration;
import javax.cache.CacheManager;
import javax.cache.Caching;
import javax.cache.spi.CachingProvider;
import org.ehcache.config.builders.CacheConfigurationBuilder;
import org.ehcache.config.builders.ExpiryPolicyBuilder;
import org.ehcache.config.builders.ResourcePoolsBuilder;
import org.ehcache.jsr107.Eh107Configuration;
import org.springframework.cache.jcache.JCacheCacheManager;
import org.springframework.cache.support.NoOpCacheManager;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class CacheConfig {

    public static final String CONFIG_BY_NAME_CACHE_NAME = "configByName";

    @Bean
    public TransactionBoundCacheManager defaultCacheManager(JCacheCacheManager ehCacheManager) {
        SpecifiedCacheSupportingCacheManager cacheManager = new SpecifiedCacheSupportingCacheManager();
        cacheManager.setNoOpCacheManager(new NoOpCacheManager());
        cacheManager.setDelegateCacheManager(ehCacheManager);
        cacheManager.setSupportedCaches(CONFIG_BY_NAME_CACHE_NAME);
        return new TransactionBoundCacheManager(cacheManager);
    }

    @Bean
    public JCacheCacheManager ehCacheManager() {
        JCacheCacheManager jCacheCacheManager = new JCacheCacheManager();
        jCacheCacheManager.setCacheManager(getInternalEhCacheManager());
        return jCacheCacheManager;
    }

    private CacheManager getInternalEhCacheManager() {
        CachingProvider provider = Caching.getCachingProvider();
        CacheManager cacheManager = provider.getCacheManager();

        javax.cache.configuration.Configuration<Object, Object> defaultTemplate = Eh107Configuration.fromEhcacheCacheConfiguration(
                CacheConfigurationBuilder.newCacheConfigurationBuilder(Object.class, Object.class, ResourcePoolsBuilder.heap(10000))
                        .withExpiry(ExpiryPolicyBuilder.noExpiration()).build());

        if (cacheManager.getCache("users") == null) {
            cacheManager.createCache("users", defaultTemplate);
        }
        if (cacheManager.getCache("usersByUsername") == null) {
            cacheManager.createCache("usersByUsername", defaultTemplate);
        }
        if (cacheManager.getCache("tenantsById") == null) {
            cacheManager.createCache("tenantsById", defaultTemplate);
        }
        if (cacheManager.getCache("offices") == null) {
            cacheManager.createCache("offices", defaultTemplate);
        }
        if (cacheManager.getCache("officesForDropdown") == null) {
            cacheManager.createCache("officesForDropdown", defaultTemplate);
        }
        if (cacheManager.getCache("officesById") == null) {
            cacheManager.createCache("officesById", defaultTemplate);
        }
        if (cacheManager.getCache("charges") == null) {
            cacheManager.createCache("charges", defaultTemplate);
        }
        if (cacheManager.getCache("funds") == null) {
            cacheManager.createCache("funds", defaultTemplate);
        }
        if (cacheManager.getCache("code_values") == null) {
            cacheManager.createCache("code_values", defaultTemplate);
        }
        if (cacheManager.getCache("codes") == null) {
            cacheManager.createCache("codes", defaultTemplate);
        }
        if (cacheManager.getCache("hooks") == null) {
            cacheManager.createCache("hooks", defaultTemplate);
        }
        if (cacheManager.getCache("tfConfig") == null) {
            cacheManager.createCache("tfConfig", defaultTemplate);
        }
        if (cacheManager.getCache(CONFIG_BY_NAME_CACHE_NAME) == null) {
            cacheManager.createCache(CONFIG_BY_NAME_CACHE_NAME, defaultTemplate);
        }

        javax.cache.configuration.Configuration<Object, Object> accessTokenTemplate = Eh107Configuration.fromEhcacheCacheConfiguration(
                CacheConfigurationBuilder.newCacheConfigurationBuilder(Object.class, Object.class, ResourcePoolsBuilder.heap(10000))
                        .withExpiry(ExpiryPolicyBuilder.timeToIdleExpiration(Duration.ofHours(2))).build());

        if (cacheManager.getCache("userTFAccessToken") == null) {
            cacheManager.createCache("userTFAccessToken", accessTokenTemplate);
        }

        return cacheManager;
    }
}