java NIO实现UDP通讯

NIO Udp通讯工具类

java 复制代码
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.DatagramChannel;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.util.Iterator;

import lombok.extern.slf4j.Slf4j;

/**
 * NIO Udp工具类
 */
@Slf4j
public class MyUdpSocket {

	//获取报文的通道
	private DatagramChannel channel;
	
	//多路复用选择器
    private Selector selector;

	private final int MAX_SIZE = 1280;

	//当前获取到版本的端口
	private int client_port;

	//记录当前获取到报文的ip
	private String client_ip;

	ByteBuffer receive_buffer = ByteBuffer.allocate(MAX_SIZE);
  
	public MyUdpSocket(String ip, int port) throws IOException {
		channel=DatagramChannel.open();
		selector=Selector.open();
		try{
			//调整此通道为非阻塞模式  
			channel.configureBlocking(false);  
			//获取与套接字通道关联的套接字,并将该套接字绑定到本机指定端口  
			channel.socket().bind(new InetSocketAddress(port));
			//为通道选择器注册通道,并指定操作的选择键集  
			channel.register(selector, SelectionKey.OP_READ);  
		}catch (IOException e) {
			log.error("MyUdpSocket init error", e);
		}
	}

	/**
	 * 接收数据
	 * @return
	 */
	public final int receive() {
		try{
			receive_buffer.clear();
			if(selector.select(1000)!=0) {
				Iterator<SelectionKey> itr = selector.selectedKeys().iterator();
				while(itr.hasNext()){  
					SelectionKey key = itr.next( );
					itr.remove(); 
					if(key.isReadable()){
						DatagramChannel dc = (DatagramChannel)key.channel();  
						InetSocketAddress client = (InetSocketAddress)dc.receive(receive_buffer); //接收来自任意一个Client的数据报
						if (client!=null) {
							client_ip=client.getAddress().getHostAddress();
							client_port=client.getPort();
							//System.out.println(client_ip+":"+client_port+" size="+receive_buffer.position());
							key.interestOps(SelectionKey.OP_READ);
							return receive_buffer.position();
						} else {
							return 0;
						}
					} else {
						return 0;
					}
				}
			}
		}catch (IOException e) {
			log.error(String.format("udp_receive出错:%s", e.getMessage()), e);
		}
		return 0;
	}
		
	public final void close() throws IOException {
		selector.close();
		channel.close();
	}
	
	public final int get_client_port(){
		return client_port;
	}

	public final String get_client_ip(){
		return client_ip;
	}
	
	public final byte[] get_receive_packet(int size){
		byte[] packet = new byte[size];
		for (int i=0; i<size; i++){
			packet[i]=receive_buffer.array()[i];
		}
		return packet;
	}

	public final void get_receive_buffer(byte[] buffer){
		receive_buffer.get(buffer);
	}
}

用法:

java 复制代码
import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetSocketAddress;

public class UdpSocketTest {
    public static void main(String[] args) throws IOException {
        MyUdpSocket socket = new MyUdpSocket("127.0.0.1", 1234);
        while(true) {
            int size = socket.receive();
            if (size>0) {
                byte[] data = socket.get_receive_packet(size);
                System.out.println("ip:" + socket.get_client_ip());

                DatagramPacket packet = new DatagramPacket(data, 0, data.length);
                packet.setSocketAddress(new InetSocketAddress(socket.get_client_ip(), socket.get_client_port()));
                DatagramSocket datagramSocket = new DatagramSocket(null);
                datagramSocket.setReuseAddress(true);
                datagramSocket.send(packet);
                datagramSocket.close();
            }
        }
    }
}
相关推荐
好家伙VCC几秒前
**发散创新:探索群体智能编程中的新境界**随着科技的飞速发展,群体智能逐渐成为编程领域的一大研究热点。本文将深入探讨群体智能的概念、优
java·python·科技
秉承初心14 分钟前
Java 23种设计模式的详细解析
java·设计模式
千码君201619 分钟前
Go语言:记录一下Go语言系统学习的第一天
java·开发语言·学习·golang·gin·并发编程·编译语言
聪明的笨猪猪21 分钟前
Java 面试清单(含超通俗生活案例与深度理解)
java·经验分享·笔记·面试
MrSYJ39 分钟前
学完涨工资的技巧2:Spring Authorization Server如何签发JWTToken
java·spring boot·微服务
珹洺41 分钟前
Java-Spring入门指南(二十六)Android Studio下载与安装
java·spring·android studio
JAVA学习通41 分钟前
JDK高版本特性总结与ZGC实践
java·jvm·算法
cxyxiaokui0011 小时前
JDK 动态代理 vs CGLIB:原理、区别与 Spring AOP 底层揭秘
java·后端·spring
代码充电宝1 小时前
LeetCode 算法题【中等】189. 轮转数组
java·算法·leetcode·职场和发展·数组
我命由我123451 小时前
PDFBox - PDDocument 与 byte 数组、PDF 加密
java·服务器·前端·后端·学习·java-ee·pdf