com.flybotix.hfr.io.sender.UDPSender Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of highfrequencyrobots Show documentation

A way for wifi robots and IoT devices to quickly send compressed data arrays across a network.

There is a newer version: 2019.1.11

package com.flybotix.hfr.io.sender;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.net.SocketException;
import java.nio.ByteBuffer;
import java.util.Arrays;
import java.util.concurrent.Executor;
import java.util.concurrent.Executors;

import com.flybotix.hfr.util.log.ILog;
import com.flybotix.hfr.util.log.Logger;

public class UDPSender extends ADataSender{
  
  private final Executor mThreads = Executors.newFixedThreadPool(1);
  private DatagramSocket socket = null;

  private ILog mLog = Logger.createLog(UDPSender.class);

  @Override
  protected void establishConnection(InetAddress addr) {
    try {
      socket = new DatagramSocket(mHostPort);
      mLog.debug("Established local socket at ", mHostPort);
      update(mStatus.connectionEstablished());
    } catch (SocketException e) {
      e.printStackTrace();
    }
    startSendThread(addr);
  }

  private void startSendThread(final InetAddress addr) {
    mThreads.execute(() -> {
      while(mStatus.isConnected()) {
        ByteBuffer msg = null;
        try {
          Thread.sleep(10);
          msg = mMessageQ.removeFirst();
          if(msg != null) {
            byte[] arr = msg.array();
            mLog.debug("Sending  to " + addr + ":" + mDestPort + "\t" + arr.length + " raw bytes ", Arrays.toString(arr));
            DatagramPacket packet = new DatagramPacket(arr, arr.length, addr, mDestPort);
            socket.send(packet);
          }
        } catch (InterruptedException e) {
          mLog.warn("Interrupted while pulling data off the message queue");
        } catch (IOException e) {
          if(msg != null) {
            try {
              mMessageQ.addFirst(msg);
            } catch (InterruptedException e1) {
              mLog.error("Interrupted while restoring a message to the queue after connection error.  Message lost.");
            }
          }
          update(mStatus.unexpectedDisconnect());
          mLog.exception(e);
        }
      }
    });
  }
}