org.wildfly.clustering.singleton.server.DistributedSingletonService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of wildfly-clustering-singleton-server Show documentation

Contains implementations of the clustering singleton services.

There is a newer version: 35.0.0.Final

Show newest version

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

package org.wildfly.clustering.singleton.server;

import java.util.List;
import java.util.Map;
import java.util.function.Consumer;
import java.util.function.Function;

import org.jboss.msc.service.ServiceBuilder;
import org.jboss.msc.service.ServiceController;
import org.jboss.msc.service.ServiceName;
import org.jboss.msc.service.ServiceTarget;
import org.wildfly.clustering.server.manager.Service;
import org.wildfly.clustering.singleton.Singleton;

/**
 * Distributed singleton service implementation that uses JBoss MSC 1.4.x service installation.
 * @author Paul Ferraro
 */
public class DistributedSingletonService extends AbstractSingletonService {

    public DistributedSingletonService(SingletonServiceContext context, Function> builderFactory, List>>> injectors, Consumer singleton) {
        super(context, new ServiceFactory(context.getService(), builderFactory, injectors), DefaultSingletonContext::new, singleton);
    }

    private static class ServiceFactory implements Function {
        private final org.jboss.msc.Service service;
        private final Function> builderFactory;
        private final List>>> injectors;

        ServiceFactory(org.jboss.msc.Service service, Function> builderFactory, List>>> injectors) {
            this.service = service;
            this.builderFactory = builderFactory;
            this.injectors = injectors;
        }

        @Override
        public Service apply(ServiceTarget target) {
            ServiceBuilder builder = this.builderFactory.apply(target);
            for (Map.Entry>> entry : this.injectors) {
                entry.getValue().accept(builder.provides(entry.getKey()));
            }
            return new ServiceControllerService(builder.setInstance(this.service).setInitialMode(ServiceController.Mode.NEVER).install());
        }
    }
}