org.redkale.net.UdpBioProtocolServer Maven / Gradle / Ivy

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.redkale.net;

import java.io.IOException;
import java.net.*;
import java.nio.ByteBuffer;
import java.nio.channels.DatagramChannel;
import java.util.Set;
import java.util.concurrent.CountDownLatch;
import org.redkale.util.AnyValue;

/**
 * 协议底层Server
 *
 * 
 * 详情见: https://redkale.org
 *
 * @author zhangjx
 */
public class UdpBioProtocolServer extends ProtocolServer {

    private boolean running;

    private DatagramChannel serverChannel;

    public UdpBioProtocolServer(Context context) {
        super(context);
    }

    @Override
    public void open(AnyValue config) throws IOException {
        DatagramChannel ch = DatagramChannel.open();
        ch.configureBlocking(true);
        this.serverChannel = ch;
        final Set> options = this.serverChannel.supportedOptions();
        if (options.contains(StandardSocketOptions.TCP_NODELAY)) {
            this.serverChannel.setOption(StandardSocketOptions.TCP_NODELAY, true);
        }
        if (options.contains(StandardSocketOptions.SO_KEEPALIVE)) {
            this.serverChannel.setOption(StandardSocketOptions.SO_KEEPALIVE, true);
        }
        if (options.contains(StandardSocketOptions.SO_REUSEADDR)) {
            this.serverChannel.setOption(StandardSocketOptions.SO_REUSEADDR, true);
        }
        if (options.contains(StandardSocketOptions.SO_RCVBUF)) {
            this.serverChannel.setOption(StandardSocketOptions.SO_RCVBUF, 16 * 1024);
        }
        if (options.contains(StandardSocketOptions.SO_SNDBUF)) {
            this.serverChannel.setOption(StandardSocketOptions.SO_SNDBUF, 16 * 1024);
        }
    }

    @Override
    public void bind(SocketAddress local, int backlog) throws IOException {
        this.serverChannel.bind(local);
    }

    @Override
    public  void setOption(SocketOption name, T value) throws IOException {
        this.serverChannel.setOption(name, value);
    }

    @Override
    public  Set> supportedOptions() {
        return this.serverChannel.supportedOptions();
    }

    @Override
    public void accept() throws IOException {
        final DatagramChannel serchannel = this.serverChannel;
        final int readTimeoutSeconds = this.context.readTimeoutSeconds;
        final int writeTimeoutSeconds = this.context.writeTimeoutSeconds;
        final CountDownLatch cdl = new CountDownLatch(1);
        this.running = true;
        new Thread() {
            @Override
            public void run() {
                cdl.countDown();
                while (running) {
                    final ByteBuffer buffer = context.pollBuffer();
                    try {
                        SocketAddress address = serchannel.receive(buffer);
                        buffer.flip();
                        AsyncConnection conn = new UdpBioAsyncConnection(serchannel, address, false, readTimeoutSeconds, writeTimeoutSeconds, null, null);
                        context.runAsync(new PrepareRunner(context, conn, buffer, null));
                    } catch (Exception e) {
                        context.offerBuffer(buffer);
                    }
                }
            }
        }.start();
        try {
            cdl.await();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @Override
    public void close() throws IOException {
        this.running = false;
        this.serverChannel.close();
    }

    @Override
    public long getCreateCount() {
        return -1;
    }

    @Override
    public long getClosedCount() {
        return -1;
    }

    @Override
    public long getLivingCount() {
        return -1;
    }
}