org.nustaq.fastcast.api.FastCast Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fast-cast Show documentation
a fast brokerless messaging library based on reliable UDP multicast
The newest version!
/**
 * Copyright (c) 2014, Ruediger Moeller. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301  USA
 *
 * Date: 03.01.14
 * Time: 21:19
 * To change this template use File | Settings | File Templates.
 */
package org.nustaq.fastcast.api;

import org.nustaq.fastcast.config.*;
import org.nustaq.fastcast.impl.TransportDriver;
import org.nustaq.fastcast.util.FCLog;
import org.nustaq.fastcast.util.FCUtils;
import org.nustaq.fastcast.impl.*;
import org.nustaq.fastcast.transport.*;
import org.nustaq.offheap.structs.unsafeimpl.FSTStructFactory;

import java.io.IOException;
import java.util.*;

/**
 * Created with IntelliJ IDEA.
 * User: moelrue
 * Date: 8/6/13
 * Time: 3:00 PM
 * To change this template use File | Settings | File Templates.
 */
public class FastCast {

    static {
        FSTStructFactory.getInstance().registerSystemClz((byte)127, Packet.class, DataPacket.class, RetransPacket.class, RetransEntry.class, ControlPacket.class);
    }

    static FastCast fc;
    private ClusterConf config;

    public static FastCast getFastCast() {
        synchronized (FastCast.class) {
            if ( fc != null ) {
                return fc;
            }
            fc = new FastCast();
            FCLog.get().info(
                                "____ ____ ____ ___ ____ ____ ____ ___\n" +
                                    "|--- |--| ====  |  |___ |--| ====  |  \n" + "> v3"
                            );
            return fc;
        }
    }

    protected HashMap transports = new HashMap();
    protected HashMap drivers = new HashMap();
    String nodeId;

    public void setNodeId(String nodeName) {
        if ( nodeId != null )
            throw new RuntimeException("Node Id can only be set once per process");
        nodeId = FCUtils.createNodeId(nodeName);
    }

    public String getNodeId() {
        return nodeId;
    }

    public PhysicalTransport getTransport(String name) {
        PhysicalTransport physicalTransport = transports.get(name);
        if ( physicalTransport == null ) {
            FCLog.log("could not find transport '" + name + "'. Falling back to transport 'default'");
            return transports.get("default");
        }
        return physicalTransport;
    }

    /**
     * same as getTransportDriver
     * @param transName
     * @return
     */
    public TransportDriver onTransport(String transName) {
        return getTransportDriver(transName);
    }

    public TransportDriver getTransportDriver(String transName) {
        TransportDriver res = drivers.get(transName);
        if ( res == null ) {
            PhysicalTransport physicalTransport = getTransport(transName);
            res = new TransportDriver( physicalTransport, nodeId );
            drivers.put(transName, res);
        }
        return res;
    }

    public FastCast loadConfig(String filePath) throws Exception {
        setConfig(ClusterConf.readFrom(filePath));
        return this;
    }

    public void setConfig(ClusterConf config) {
        this.config = config;
        addTransportsFrom(config);
    }

    /**
     * only avaiable if initialized with setConfig or loadConfig
     * @param name
     * @return
     */
    public SubscriberConf getSubscriberConf(String name) {
        TopicConf topic = getConfig().getTopic(name);
        if ( topic != null )
            return topic.getSub();
        return null;
    }

    /**
     * only avaiable if initialized with setConfig or loadConfig
     * @param name
     * @return
     */
    public PublisherConf getPublisherConf(String name) {
        TopicConf topic = getConfig().getTopic(name);
        if ( topic != null )
            return topic.getPublisher();
        return null;
    }

    public ClusterConf getConfig() {
        return config;
    }

    public static class ConfigurationAlreadyDefinedException extends RuntimeException {
        public ConfigurationAlreadyDefinedException(String message) {
            super(message);
        }
    }

    public void addTransportsFrom(ClusterConf config) {
        PhysicalTransportConf[] trs = config.transports;
        for (int i = 0; i < trs.length; i++) {
            PhysicalTransportConf tr = trs[i];
            addTransport(tr);
        }
    }

    /**
     *
     * @param tconf
     */
    public void addTransport(PhysicalTransportConf tconf) {
        if ( nodeId == null )
            throw new RuntimeException("define nodeId first");
        if ( transports.get(tconf.getName()) != null ) {
            throw new ConfigurationAlreadyDefinedException("transport "+tconf.getName()+" already initialized ");
        }
        try {
            FCLog.log("Connecting transport " + tconf.getName() + " as "+getNodeId());
            MulticastChannelPhysicalTransport tr = new MulticastChannelPhysicalTransport(tconf,tconf.getSpinLoopMicros()==0);
            tr.join();
            transports.put(tconf.getName(), tr);
        } catch (IOException e) {
            FCLog.log(e);
        }
    }

}