All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.swirlds.common.threading.framework.internal.AbstractMultiQueueThreadConfiguration Maven / Gradle / Ivy

Go to download

Swirlds is a software platform designed to build fully-distributed applications that harness the power of the cloud without servers. Now you can develop applications with fairness in decision making, speed, trust and reliability, at a fraction of the cost of traditional server-based platforms.

There is a newer version: 0.56.6
Show newest version
/*
 * Copyright (C) 2023-2024 Hedera Hashgraph, LLC
 *
 * Licensed 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 com.swirlds.common.threading.framework.internal;

import com.swirlds.common.threading.framework.MultiQueueThread;
import com.swirlds.common.threading.manager.ThreadManager;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Consumer;

/**
 * A base class for configurations that build {@link MultiQueueThread}s.
 *
 * @param 
 * 		the type of the concrete class
 */
public abstract class AbstractMultiQueueThreadConfiguration>
        extends AbstractQueueThreadConfiguration {

    /**
     * A map of data type to handler for that type.
     */
    private final Map, Consumer> subHandlers = new HashMap<>();

    /**
     * Construct a new instance.
     *
     * @param threadManager
     * 		the thread manager, responsible for creating new threads
     */
    protected AbstractMultiQueueThreadConfiguration(final ThreadManager threadManager) {
        super(threadManager);
    }

    /**
     * Copy constructor.
     */
    protected AbstractMultiQueueThreadConfiguration(final AbstractMultiQueueThreadConfiguration other) {
        super(other);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public abstract AbstractMultiQueueThreadConfiguration copy();

    /**
     * Build the multi-queue.
     *
     * @param start
     * 		if true then automatically start the queue thread
     * @return the multi queue
     */
    protected MultiQueueThread buildMultiQueue(final boolean start) {
        return new MultiQueueThreadImpl(subHandlers, handler -> {
            setHandler(handler);
            return buildQueueThread(start);
        });
    }

    /**
     * Add a handler for a particular data type and create the corresponding inserter.
     *
     * @param clazz
     * 		the class of the data type
     * @param handler
     * 		the handler for the data type
     * @return this object
     */
    @SuppressWarnings("unchecked")
    protected  C addHandler(final Class clazz, final Consumer handler) {
        final Consumer prev = subHandlers.put(clazz, (Consumer) handler);
        if (prev != null) {
            throw new IllegalStateException("Handler already exists for " + clazz);
        }
        return (C) this;
    }
}