org.wildfly.clustering.ejb.infinispan.bean.InfinispanBeanGroupManager Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of wildfly-clustering-ejb-infinispan Show documentation

An implementation of wildfly-clustering-ejb-spi based on an embedded Infinispan cache.

There is a newer version: 33.0.2.Final

Show newest version

/*
 * Copyright The WildFly Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.wildfly.clustering.ejb.infinispan.bean;

import java.util.Map;

import org.infinispan.Cache;
import org.wildfly.clustering.ee.Creator;
import org.wildfly.clustering.ee.Mutator;
import org.wildfly.clustering.ee.MutatorFactory;
import org.wildfly.clustering.ee.Remover;
import org.wildfly.clustering.ee.infinispan.InfinispanConfiguration;
import org.wildfly.clustering.ee.infinispan.CacheMutatorFactory;
import org.wildfly.clustering.ejb.bean.BeanInstance;
import org.wildfly.clustering.ejb.cache.bean.BeanGroupKey;
import org.wildfly.clustering.marshalling.spi.MarshalledValue;

/**
 * Manages the cache entry for a bean group.
 * @author Paul Ferraro
 * @param  the bean identifier type
 * @param  the bean instance type
 * @param  the marshalled value context type
 */
public class InfinispanBeanGroupManager, C> implements Creator, C>, MarshalledValue, C>>, Remover, MutatorFactory, C>> {

    private final Cache, MarshalledValue, C>> cache;
    private final Cache, MarshalledValue, C>> removeCache;
    private final MutatorFactory, MarshalledValue, C>> mutatorFactory;

    public InfinispanBeanGroupManager(InfinispanConfiguration configuration) {
        this.cache = configuration.getCache();
        this.removeCache = configuration.getWriteOnlyCache();
        this.mutatorFactory = new CacheMutatorFactory<>(configuration.getCache());
    }

    @Override
    public MarshalledValue, C> createValue(K id, MarshalledValue, C> defaultValue) {
        MarshalledValue, C> value = this.cache.putIfAbsent(new InfinispanBeanGroupKey<>(id), defaultValue);
        return (value != null) ? value : defaultValue;
    }

    @Override
    public boolean remove(K id) {
        this.removeCache.remove(new InfinispanBeanGroupKey<>(id));
        return true;
    }

    @Override
    public Mutator createMutator(K id, MarshalledValue, C> value) {
        return this.mutatorFactory.createMutator(new InfinispanBeanGroupKey<>(id), value);
    }
}