org.apache.openejb.client.MulticastTool Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of openejb-client Show documentation
The newest version!
/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 org.apache.openejb.client;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.MulticastSocket;
import java.net.SocketAddress;
import java.net.SocketTimeoutException;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.Iterator;
import java.util.Timer;
import java.util.TimerTask;

/**
 * @version $Rev$ $Date$
 */
public class MulticastTool {

    private static final int BUFF_SIZE = 8192;

    public static void main(String[] array) throws Exception {

        SimpleDateFormat format = new SimpleDateFormat("HH:mm:ss");

        String send = null;

        long rate = 1000;

        String host = "239.255.3.2";
        int port = 6142;
        Integer ttl = null;
        Boolean loopbackmode = null;
        Integer socketTimeout = null;

        Iterator args = Arrays.asList(array).iterator();
        while (args.hasNext()) {
            String arg = args.next();

            if (arg.equals("--host") || arg.equals("-h")) {
                host = args.next();
            } else if (arg.equals("--port") || arg.equals("-p")) {
                port = Integer.parseInt(args.next());
            } else if (arg.equals("--rate") || arg.equals("-r")) {
                rate = new Long(args.next());
            } else if (arg.equals("--ttl")) {
                ttl = new Integer(args.next());
            } else if (arg.equals("--send") || arg.equals("-s")) {
                send = args.next();
            } else if (arg.equals("--timeout") || arg.equals("-t")) {
                socketTimeout = new Integer(args.next());
            } else if (arg.equals("--loopback") || arg.equals("-l")) {
                loopbackmode = new Boolean(args.next());
            } else {
                throw new IllegalArgumentException(arg);
            }
        }

        InetAddress inetAddress = InetAddress.getByName(host);

        InetSocketAddress address = new InetSocketAddress(inetAddress, port);

        MulticastSocket multicast = new MulticastSocket(port);
        multicast.joinGroup(inetAddress);

        if (ttl != null) {
            multicast.setTimeToLive(ttl);
        }

        if (socketTimeout != null) {
            multicast.setSoTimeout(socketTimeout);
        }

        if (loopbackmode != null) {
            multicast.setLoopbackMode(loopbackmode);
        }

        System.out.print("Connecting to multicast group: ");
        System.out.print(host);
        System.out.print(":");
        System.out.println(multicast.getLocalPort());

        print("LoopbackMode", multicast.getLoopbackMode());
        print("TimeToLive", multicast.getTimeToLive());
        print("SoTimeout", multicast.getSoTimeout());

        System.out.println("-------------------------------");

        if (send != null) {
            Timer timer = new Timer("Multicast Send", true);
            timer.scheduleAtFixedRate(new Send(address, multicast, send), 0, rate);
        }

        byte[] buf = new byte[BUFF_SIZE];
        DatagramPacket packet = new DatagramPacket(buf, 0, buf.length);

        while (true) {
            try {
                multicast.receive(packet);
                if (packet.getLength() > 0) {
                    InetAddress a = packet.getAddress();
                    System.out.print(format.format(new Date()));
                    System.out.print(" - ");
                    System.out.print(a.getHostAddress());
                    System.out.print(" - ");
                    String str = new String(packet.getData(), packet.getOffset(), packet.getLength());
                    System.out.println(str);
                }
            } catch (SocketTimeoutException e) {
            }
        }
    }

    private static void print(String name, Object value) {
        System.out.print(name);
        System.out.print(":");
        System.out.println(value);
    }

    static class Send extends TimerTask {
        private final MulticastSocket multicast;
        private final String text;
        private final SocketAddress address;

        public Send(SocketAddress address, MulticastSocket multicast, String text) {
            this.address = address;
            this.multicast = multicast;
            this.text = text;
        }

        public void run() {
            try {
                byte[] data = text.getBytes();
                DatagramPacket packet = new DatagramPacket(data, 0, data.length, address);
                multicast.send(packet);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

    }

}