| package org.eclipse.osee.ote.endpoint; |
| |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.nio.channels.AsynchronousCloseException; |
| import java.nio.channels.ClosedByInterruptException; |
| import java.nio.channels.ClosedChannelException; |
| import java.nio.channels.DatagramChannel; |
| import java.util.ArrayList; |
| import java.util.List; |
| import java.util.concurrent.ArrayBlockingQueue; |
| import java.util.concurrent.TimeUnit; |
| import java.util.logging.Level; |
| import java.util.zip.Deflater; |
| import org.eclipse.osee.framework.jdk.core.util.Lib; |
| import org.eclipse.osee.framework.logging.OseeLog; |
| import org.eclipse.osee.ote.collections.ObjectPool; |
| |
| final class OteEndpointSendRunnable implements Runnable { |
| |
| private static final int SEND_BUFFER_SIZE = 1024 * 512; |
| /** |
| * This is the header for a ZIP file |
| */ |
| public static final byte[] MAGIC_NUMBER = {0x04, 0x03, 0x4b, 0x50}; |
| private static final int MAX_DGRAM_SIZE = 65500; |
| |
| private final ArrayBlockingQueue<AddressBuffer> toSend; |
| private final ObjectPool<AddressBuffer> buffers; |
| private final Deflater compressor = new Deflater(); |
| |
| private boolean debug = false; |
| |
| OteEndpointSendRunnable(ArrayBlockingQueue<AddressBuffer> toSend, ObjectPool<AddressBuffer> buffers, boolean debug) { |
| this.toSend = toSend; |
| this.buffers = buffers; |
| this.debug = debug; |
| compressor.setLevel(Deflater.BEST_SPEED); |
| } |
| |
| @Override |
| public void run() { |
| DatagramChannel threadChannel = null; |
| try { |
| threadChannel = openAndInitializeDatagramChannel(); |
| boolean keepRunning = true; |
| final List<AddressBuffer> dataToSend = new ArrayList<>(32); |
| System.setSecurityManager(null); |
| while(keepRunning){ |
| try{ |
| dataToSend.clear(); |
| if (toSend.drainTo(dataToSend) < 1) { |
| try { |
| // block until something is available |
| AddressBuffer addrBuf = toSend.poll(15, TimeUnit.SECONDS); |
| if (addrBuf == null) { |
| // no activity for a while so lets clean ourselves up. Our master will restart |
| // a new thread if another event comes along after we self terminate |
| keepRunning = false; |
| } else { |
| dataToSend.add(addrBuf); |
| } |
| } catch (InterruptedException e) { |
| keepRunning = false; |
| continue; |
| } |
| } |
| int size = dataToSend.size(); |
| for (int i = 0; i < size; i++) { |
| AddressBuffer data = dataToSend.get(i); |
| if (data == OteUdpEndpointSender.POISON_PILL) { |
| keepRunning = false; |
| break; |
| } |
| |
| int bufSize = data.getBuffer().remaining(); |
| if(bufSize > MAX_DGRAM_SIZE) { |
| compressor.reset(); |
| compressor.setInput(data.getBuffer().array(), 0, bufSize); |
| compressor.finish(); |
| |
| ByteArrayOutputStream bos = new ByteArrayOutputStream(bufSize); |
| |
| byte[] buf = new byte[1024]; |
| while (!compressor.finished()) { |
| int count = compressor.deflate(buf); |
| bos.write(buf, 0, count); |
| } |
| try { |
| bos.close(); |
| byte[] dataBytes = new byte[bos.size() + 4]; |
| System.arraycopy(MAGIC_NUMBER, 0, dataBytes, 0, 4); |
| System.arraycopy(bos.toByteArray(), 0, dataBytes, 4, bos.size()); |
| data.setBytes(dataBytes); |
| } |
| catch (IOException e) { |
| OseeLog.log(getClass(), Level.SEVERE, "Error trying to compress data", e); |
| continue; |
| } |
| } |
| |
| threadChannel.send(data.getBuffer(), data.getAddress()); |
| } |
| } catch (ClosedByInterruptException ex){ |
| if(debug){ |
| OseeLog.log(getClass(), Level.SEVERE, "Error trying to send data", ex); |
| } |
| threadChannel = openAndInitializeDatagramChannel(); |
| } catch (AsynchronousCloseException ex){ |
| if(debug){ |
| OseeLog.log(getClass(), Level.SEVERE, "Error trying to send data", ex); |
| } |
| Lib.close(threadChannel); |
| threadChannel = openAndInitializeDatagramChannel(); |
| } catch (ClosedChannelException ex){ |
| if(debug){ |
| OseeLog.log(getClass(), Level.SEVERE, "Error trying to send data", ex); |
| } |
| Lib.close(threadChannel); |
| threadChannel = openAndInitializeDatagramChannel(); |
| } catch (IOException ex){ |
| OseeLog.log(getClass(), Level.SEVERE, "Error trying to send data", ex); |
| } finally { |
| int size = dataToSend.size(); |
| for (int i = 0; i < size; i++) { |
| buffers.returnObj(dataToSend.get(i)); |
| } |
| } |
| } |
| } catch (IOException ex){ |
| OseeLog.log(getClass(), Level.SEVERE, "Error opening DatagramChannel. Ending OteEndpointSendRunnable unexpectedly.", ex); |
| } finally{ |
| Lib.close(threadChannel); |
| } |
| |
| } |
| |
| private DatagramChannel openAndInitializeDatagramChannel() throws IOException { |
| DatagramChannel channel = DatagramChannel.open(); |
| if (channel.socket().getSendBufferSize() < SEND_BUFFER_SIZE) { |
| channel.socket().setSendBufferSize(SEND_BUFFER_SIZE); |
| } |
| channel.socket().setReuseAddress(true); |
| channel.configureBlocking(true); |
| return channel; |
| } |
| } |