UDP协议开发

import io.netty.bootstrap.Bootstrap;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioDatagramChannel;

public class ChineseProverbServer {
    public void run(int port) throws Exception {
        EventLoopGroup group = new NioEventLoopGroup();
        try {
            Bootstrap b = new Bootstrap();
            //由于使用UDP通信，在创建Channel的时候需要通过NioDatagramChannel来创建
            b.group(group).channel(NioDatagramChannel.class)
                    //随后设置Socket参数支持广播，
                    .option(ChannelOption.SO_BROADCAST, true)
                    //最后设置业务处理handler。
                    //相比于TCP通信，UDP不存在客户端和服务端的实际连接，
                    //因此不需要为连接（ChannelPipeline）设置handler，
                    //对于服务端，只需要设置启动辅助类的handler即可。
                    .handler(new ChineseProverbServerHandler());
            b.bind(port).sync().channel().closeFuture().await();
        } finally {
            group.shutdownGracefully();
        }
    }

    public static void main(String[] args) throws Exception {
        int port = 8080;
        if (args.length > 0) {
            try {
                port = Integer.parseInt(args[0]);
            } catch (NumberFormatException e) {
                e.printStackTrace();
            }
        }
        new ChineseProverbServer().run(port);
    }
}

import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.socket.DatagramPacket;
import io.netty.util.CharsetUtil;
import io.netty.util.internal.ThreadLocalRandom;

public class ChineseProverbServerHandler extends SimpleChannelInboundHandler {
    // 谚语列表
    private static final String[] DICTIOnARY= {"只要功夫深，铁棒磨成针。",
            "旧时王谢堂前燕，飞入寻常百姓家。", "洛阳亲友如相问，一片冰心在玉壶。", "一寸光阴一寸金，寸金难买寸光阴。",
            "老骥伏枥，志在千里。烈士暮年，壮心不已!"};

    private String nextQuote() {
        //由于ChineseProverbServerHandler存在多线程并发操作的可能，
        //所以使用了Netty的线程安全随机类ThreadLocalRandom。
        // 如果使用的是JDK7，可以直接使用JDK7的java.util.concurrent.ThreadLocalRandom。
        int quoteId = ThreadLocalRandom.current().nextInt(DICTIONARY.length);
        return DICTIONARY[quoteId];
    }

    @Override
    public void messageReceived(ChannelHandlerContext ctx, Object msg)throws Exception {
        //Netty对UDP进行了封装，因此，接收到的是Netty封装后的io.netty. channel.socket.DatagramPacket对象。
        DatagramPacket packet = (DatagramPacket) msg;
        //将packet内容转换为字符串（利用ByteBuf的toString(Charset)方法），
        String req = packet.content().toString(CharsetUtil.UTF_8);
        System.out.println(req);
        // 然后对请求消息进行合法性判断：如果是“谚语字典查询?”，则构造应答消息返回。
        // DatagramPacket有两个参数：第一个是需要发送的内容，为ByteBuf；
        // 另一个是目的地址，包括IP和端口，可以直接从发送的报文DatagramPacket中获取。
        if ("谚语字典查询?".equals(req)) {
            ctx.writeAndFlush(
                    new DatagramPacket(Unpooled.copiedBuffer("谚语查询结果: " + nextQuote(), CharsetUtil.UTF_8),
                    packet.sender()));
        }
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause)
            throws Exception {
        ctx.close();
        cause.printStackTrace();
    }
}

import io.netty.bootstrap.Bootstrap;
import io.netty.buffer.Unpooled;
import io.netty.channel.Channel;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.DatagramPacket;
import io.netty.channel.socket.nio.NioDatagramChannel;
import io.netty.util.CharsetUtil;

import java.net.InetSocketAddress;

public class ChineseProverbClient {

    public void run(int port) throws Exception {
        EventLoopGroup group = new NioEventLoopGroup();
        try {
            Bootstrap b = new Bootstrap();
            //创建UDP Channel和设置支持广播属性等与服务端完全一致。
            // 由于不需要和服务端建立链路，UDP Channel创建完成之后，客户端就要主动发送广播消息；
            // TCP客户端是在客户端和服务端链路建立成功之后由客户端的业务handler发送消息，这就是两者最大的区别。
            b.group(group).channel(NioDatagramChannel.class)
                    .option(ChannelOption.SO_BROADCAST, true)
                    .handler(new ChineseProverbClientHandler());
            Channel ch = b.bind(0).sync().channel();
            // 向网段内的所有机器广播UDP消息
            // 用于构造DatagramPacket发送广播消息，
            // 注意，广播消息的IP设置为“255.255.255.255”。
            // 消息广播之后，客户端等待15s用于接收服务端的应答消息，然后退出并释放资源。
            ch.writeAndFlush(
                    new DatagramPacket(Unpooled.copiedBuffer("谚语字典查询?",CharsetUtil.UTF_8),
                            new InetSocketAddress("255.255.255.255", port))
            ).sync();
            if (!ch.closeFuture().await(15000)) {
                System.out.println("查询超时!");
            }
        } finally {
            group.shutdownGracefully();
        }
    }

    public static void main(String[] args) throws Exception {
        int port = 8080;
        if (args.length > 0) {
            try {
                port = Integer.parseInt(args[0]);
            } catch (NumberFormatException e) {
                e.printStackTrace();
            }
        }
        new ChineseProverbClient().run(port);
    }
}
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.socket.DatagramPacket;
import io.netty.util.CharsetUtil;

public class ChineseProverbClientHandler extends SimpleChannelInboundHandler {

    @Override
    public void messageReceived(ChannelHandlerContext ctx, Object o)
            throws Exception {
        //接收到服务端的消息之后将其转成字符串，然后判断是否以“谚语查询结果：”开头，
        //如果没有发生丢包等问题，数据是完整的，就打印查询结果，然后释放资源。
        DatagramPacket msg = (DatagramPacket)o;
        String response = msg.content().toString(CharsetUtil.UTF_8);
        if (response.startsWith("谚语查询结果：")) {
            System.out.println(response);
            ctx.close();
        }
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause)
            throws Exception {
        cause.printStackTrace();
        ctx.close();
    }
}