阅读 1602

SwiftNIO初探-简单UDP通讯

最近苹果开源了Swift版的Netty

SwiftNIO是一个跨平台异步事件驱动的网络应用程序框架,用于快速开发可维护的高性能协议服务器和客户端。简单来说就是可以用来实现各种高性能服务端和客户端,如http、tcp。

因为最近项目刚好要用到udp,所以趁机把平常用Netty实现的该用Swift。

首先简单看一下几个用到的类

MultiThreadedEventLoopGroup

顾名思义,这东西是一个线程池。每个EventLoopGroup里面有多个EventLoop,而每个EventLoop都与一个线程绑定。

DatagramBootstrap

Bootstrap是开发Netty程序的基础,SwiftNIO也是一样。通过Bootstrap的bind方法来创建连接,我们也可以通过该方法返回的Channel来判断是否创建成功。

ChannelHandler

ChannelHandler 是用来处理数据,如客户端向服务端发送数据,服务端的数据处理就是在ChannelHandler中完成。ChannelHandler 本身是一个protocol,我们用到的有ChannelInboundHandlerChannelOutboundHandler这两个,ChannlPipeline会从头到尾顺序调用ChannelInboundHandler处理数据,从尾到头调用ChannelOutboundHandler数据。

接下来看服务端代码:

class UDPServer {
    private let loopGroup = MultiThreadedEventLoopGroup(numberOfThreads: System.coreCount)
    
    func listen(on port: Int) {
        let bootstrap = DatagramBootstrap(group: loopGroup)
            .channelOption(ChannelOptions.socket(SocketOptionLevel(SOL_SOCKET), SO_REUSEADDR), value: 1)
            .channelInitializer { (channel) -> EventLoopFuture<Void> in
                channel.pipeline.add(handler: UDPServerHandler())
        }
        
        do {
            let channel = try bootstrap.bind(host: "127.0.0.1", port: port).wait()
            print("listen on \(channel.localAddress!)")
            try channel.closeFuture.wait()
        } catch {
            print(error)
        }
    }
    
    final class UDPServerHandler: ChannelInboundHandler {
        typealias InboundIn = AddressedEnvelope<ByteBuffer>
        typealias OutboundOut = AddressedEnvelope<ByteBuffer>
        
        func channelRead(ctx: ChannelHandlerContext, data: NIOAny) {
            let getData = unwrapInboundIn(data)
            print(getData.data)
        }
    }
}

let server = UDPServer()
server.listen(on: 5656)
复制代码

运行代码就会可以看到控制台输出: listen on [IPv4]127.0.0.1:5656

我们通过内部类的形式实现了一个 ChannelInboundHandler ,并把它添加到 ChannlPipelineHandler 需要设置两个东西 InboundInOutboundOut

InboundIn: 是入站数据的类型,就是接收客户端发过来的数据类型。 OutboundOut: 是出站数据的类型,就是返回给客户端的数据类型,同时也是传递给下一个ChannelOutboundHandler的类型。

我们这里用的是 AddressedEnvelope<ByteBuffer> 它里面是一个 SocketAddress 加上 ByteBuffer

当客户端发来数据的时候会调用HandlerchannelRead(ctx: , data: ) 这里进来的是NIOAny类型,需要调用 HandlerunwrapInboundIn() 方法把 NIOAny 转成 InboundIn 类型。

就这样一个简单的UDP服务端就完成了,可以通过各种UDP工具进行测试,或者用SwiftNIO再写一个客户端:

class UDPClient {
    private let loopGroup = MultiThreadedEventLoopGroup(numberOfThreads: System.coreCount)
    private var channel: Channel!
    
    init(port: Int) {
        let bootstrap = DatagramBootstrap(group: loopGroup)
            .channelOption(ChannelOptions.socket(SocketOptionLevel(SOL_SOCKET), SO_REUSEADDR), value: 1)
            .channelInitializer { (channel) -> EventLoopFuture<Void> in
                channel.pipeline.add(handler: UDPClientHandler())
        }
        
        do {
            channel = try bootstrap.bind(host: "0.0.0.0", port: port).wait()
            sent(with: "test".data(using: .utf8)!)
        } catch {
            print(error)
        }
    }
    
    func sent(with data: Data) {
        var byteBuffer = ByteBufferAllocator().buffer(capacity: data.count)
        byteBuffer.write(bytes: data)
        let address = try! SocketAddress(ipAddress: "127.0.0.1", port: 5656)
        channel.writeAndFlush(NIOAny(AddressedEnvelope<ByteBuffer>(remoteAddress: address, data: byteBuffer)), promise: nil)
    }
    
    final class UDPClientHandler: ChannelInboundHandler {
        typealias InboundIn = AddressedEnvelope<ByteBuffer>
        
        func channelRead(ctx: ChannelHandlerContext, data: NIOAny) {
            print(data)
        }
    }
}

let client = UDPClient(port: 22222)
复制代码

到这里,用SwiftNIO构建简单的UDP通讯已经OK了。

文章分类
iOS