com.avaje.ebeaninternal.server.cluster.mcast.McastSender Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of ebean Show documentation
Ebean
The newest version!
/**
 * Copyright (C) 2009 Authors
 * 
 * This file is part of Ebean.
 * 
 * Ebean 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.
 *  
 * Ebean 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 Ebean; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA  
 */
package com.avaje.ebeaninternal.server.cluster.mcast;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

import com.avaje.ebeaninternal.server.cluster.Packet;

/**
 * Handles the sending of Packets via DatagramPacket.
 * 
 * @author rbygrave
 */
public class McastSender {

    private static final Logger logger = Logger.getLogger(McastSender.class.getName());

    private final int port;

    private final InetAddress inetAddress;

    private final DatagramSocket sock;

    private final InetSocketAddress sendAddr;

    private final String senderHostPort;

    
    public McastSender(int port, String address, int sendPort, String sendAddress) {

        try {
            this.port = port;
            this.inetAddress = InetAddress.getByName(address);

            InetAddress sendInetAddress = null;
            if (sendAddress != null) {
                sendInetAddress = InetAddress.getByName(sendAddress);
            } else {
                sendInetAddress = InetAddress.getLocalHost();
            }

            if (sendPort > 0) {
                this.sock = new DatagramSocket(sendPort, sendInetAddress);
            } else {
                this.sock = new DatagramSocket(new InetSocketAddress(sendInetAddress, 0));
            }

            String msg = "Cluster Multicast Sender on["+sendInetAddress.getHostAddress()+":"+sock.getLocalPort()+"]";
            logger.info(msg);

            this.sendAddr = new InetSocketAddress(sendInetAddress, sock.getLocalPort());
            this.senderHostPort = sendInetAddress.getHostAddress()+":"+sock.getLocalPort();
            
        } catch (Exception e) {
            String msg = "McastSender port:" + port + " sendPort:" + sendPort + " " + address;
            throw new RuntimeException(msg, e);
        }
    }

    /**
     * Return the send Address so that if we have loopback messages we can
     * detect if they where sent by this local sender and hence should be
     * ignored.
     */
    public InetSocketAddress getAddress() {
        return sendAddr;
    }

    /**
     * Return the Host and Port of the sender. This is used to uniquely identify
     * this instance in the cluster.
     */
    public String getSenderHostPort() {
        return senderHostPort;
    }

    /**
     * Send the packet.
     */
    public int sendPacket(Packet packet) throws IOException {

        byte[] pktBytes = packet.getBytes();

        if (logger.isLoggable(Level.FINE)){
            logger.fine("OUTGOING packet: " + packet.getPacketId() + " size:" + pktBytes.length);
        }

        if (pktBytes.length > 65507){
            logger.warning("OUTGOING packet: " + packet.getPacketId() + " size:" + pktBytes.length
                    +" likely to be truncated using UDP with a MAXIMUM length of 65507");
        }
        
        DatagramPacket pack = new DatagramPacket(pktBytes, pktBytes.length, inetAddress, port);
        sock.send(pack);
        
        return pktBytes.length;
    }

    /**
     * Send the list of Packets.
     */
    public int sendPackets(List packets) throws IOException {

        int totalBytes = 0;
        for (int i = 0; i < packets.size(); i++) {
            totalBytes += sendPacket(packets.get(i));
        }
        return totalBytes;
    }
    
}