fish.payara.nucleus.hazelcast.contextproxy.CompleteConfigurationProxy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of payara-micro Show documentation
Micro Distribution of the Payara Project for IBM JDK
The newest version!
/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) [2016-2017] Payara Foundation and/or its affiliates. All rights reserved.
 *
 * The contents of this file are subject to the terms of either the GNU
 * General Public License Version 2 only ("GPL") or the Common Development
 * and Distribution License("CDDL") (collectively, the "License").  You
 * may not use this file except in compliance with the License.  You can
 * obtain a copy of the License at
 * https://github.com/payara/Payara/blob/master/LICENSE.txt
 * See the License for the specific
 * language governing permissions and limitations under the License.
 *
 * When distributing the software, include this License Header Notice in each
 * file and include the License file at glassfish/legal/LICENSE.txt.
 *
 * GPL Classpath Exception:
 * The Payara Foundation designates this particular file as subject to the "Classpath"
 * exception as provided by the Payara Foundation in the GPL Version 2 section of the License
 * file that accompanied this code.
 *
 * Modifications:
 * If applicable, add the following below the License Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyright [year] [name of copyright owner]"
 *
 * Contributor(s):
 * If you wish your version of this file to be governed by only the CDDL or
 * only the GPL Version 2, indicate your decision by adding "[Contributor]
 * elects to include this software in this distribution under the [CDDL or GPL
 * Version 2] license."  If you don't indicate a single choice of license, a
 * recipient has the option to distribute your version of this file under
 * either the CDDL, the GPL Version 2 or to extend the choice of license to
 * its licensees as provided above.  However, if you add GPL Version 2 code
 * and therefore, elected the GPL Version 2 license, then the option applies
 * only if the new code is made subject to such option by the copyright
 * holder.
 */
package fish.payara.nucleus.hazelcast.contextproxy;

import org.glassfish.internal.api.JavaEEContextUtil;
import org.glassfish.internal.api.JavaEEContextUtil.Context;
import java.util.Collection;
import java.util.Map;
import javax.cache.Cache;
import javax.cache.configuration.CompleteConfiguration;
import javax.cache.configuration.Factory;
import javax.cache.configuration.MutableConfiguration;
import javax.cache.integration.CacheLoader;
import javax.cache.integration.CacheLoaderException;
import javax.cache.integration.CacheWriter;
import javax.cache.integration.CacheWriterException;
import lombok.Cleanup;
import lombok.RequiredArgsConstructor;

/**
 * Proxy all applicable factory calls
 * so Java EE context is propagated from within Hazelcast threads
 *
 * @author lprimak
 */
class CompleteConfigurationProxy extends MutableConfiguration {
    public CompleteConfigurationProxy(CompleteConfiguration config, JavaEEContextUtil ctxUtil) {
        super(config);
        this.ctxUtil = ctxUtil;
        init();
    }

    private void init() {
        if(cacheLoaderFactory != null) {
            cacheLoaderFactory = proxyLoader(cacheLoaderFactory);
        }
        if(cacheWriterFactory != null) {
            cacheWriterFactory = proxyWriter(cacheWriterFactory);
        }
    }

    private Factory> proxyLoader(final Factory> fact) {
        return new Factory>() {
            @Override
            public CacheLoader create() {
                @Cleanup Context ctx = ctxUtil.pushContext();
                final CacheLoader loader = fact.create();
                return new CacheLoaderImpl(loader);
            }
            
            class CacheLoaderImpl implements CacheLoader {
                public CacheLoaderImpl(CacheLoader loader) {
                    this.loader = loader;
                }

                @Override
                public V load(K k) throws CacheLoaderException {
                    @Cleanup Context context = ctxUtil.pushRequestContext();
                    return loader.load(k);
                }

                @Override
                public Map loadAll(Iterable itrbl) throws CacheLoaderException {
                    @Cleanup Context context = ctxUtil.pushRequestContext();
                    return loader.loadAll(itrbl);
                }

                private final CacheLoader loader;
            }

            private static final long serialVersionUID = 1L;
        };
    }

    private Factory> proxyWriter(final Factory> fact) {
        return new Factory>() {
            @Override
            public CacheWriter create() {
                @Cleanup Context ctx = ctxUtil.pushContext();
                @SuppressWarnings("unchecked")
                final CacheWriter delegate = (CacheWriter) fact.create();
                return new CacheWriterImpl(delegate);
            }

            @RequiredArgsConstructor
            class CacheWriterImpl implements CacheWriter {
                @Override
                public void write(Cache.Entry entry) throws CacheWriterException {
                    @Cleanup Context context = ctxUtil.pushRequestContext();
                    delegate.write(entry);
                }

                @Override
                public void writeAll(Collection> clctn) throws CacheWriterException {
                    @Cleanup Context context = ctxUtil.pushRequestContext();
                    delegate.writeAll(clctn);
                }

                @Override
                public void delete(Object o) throws CacheWriterException {
                    @Cleanup Context context = ctxUtil.pushRequestContext();
                    delegate.delete(o);
                }

                @Override
                public void deleteAll(Collection clctn) throws CacheWriterException {
                    @Cleanup Context context = ctxUtil.pushRequestContext();
                    delegate.deleteAll(clctn);
                }

                private final CacheWriter delegate;
            }

            private static final long serialVersionUID = 1L;
        };
    }

    private final JavaEEContextUtil ctxUtil;
    private static final long serialVersionUID = 1L;
}