SwiftNIO-链接Redis(一)

上星期用SwiftNIO写了一套程序，而后须要用到redis作缓存，而后就去GitHub上找了一下发现有一个开源库，可是用的时候发现有Bug，向做者提了issue以后至今没有修复... 因此就本身动手造了个轮子，顺便学习一下。

要作redis客户端的话有两个东西必需要了解,RESP协议和redis命令。

简单介绍RESP

• 单行字符串(Simple Strings), 开头字符为:'+' "+OK\r\n"
• 错误信息(Errors)，开头字符为:'-' "-Error message\r\n"
• 整形数字(Integers)，开头字符为:':' ":0\r\n"
• 多行字符串(Bulk Strings)，开头字符为:'$' "$6\r\nfoobar\r\n"
• 数组(Arrays)，开头字符为:'*' "*2\r\n$3\r\nfoo\r\n$3\r\nbar\r\n"

须要注意的是无论是发送的命令仍是收到来自redis服务端的回复，都是以\r\n结尾。

Java那边的叫Jedis，用Netty写的叫Nedis，那么我就把这个命名为Sedis了！

首先建立一个struct,用来存储链接信息：

struct SedisOptions {
    let prot: Int
    let host: String
    var password: String?
    var database: Int?
}
复制代码

而后建立SedisClient类：

这个类里面须要根据SedisOptions的信息来建立链接，包括身份验证。

class SedisClient {
    private let options: SedisOptions
    private var bootstrap: ClientBootstrap?
    private var loopGroup: EventLoopGroup!
    
    init(options: SedisOptions) {
        self.options = options
        
        loopGroup = MultiThreadedEventLoopGroup(numThreads: System.coreCount)
        bootstrap = ClientBootstrap(group: loopGroup)
            .channelOption(ChannelOptions.socket(SocketOptionLevel(SOL_SOCKET),
                                                 SO_REUSEADDR), value: 1)
            .channelInitializer({ (channel) -> EventLoopFuture<Void> in
                    channel.pipeline.add(handler: RESPHandler())
            })
        
    }
    
    private func _connect() -> EventLoopFuture<Channel> {
        assert(bootstrap != nil, "init failure")
        
        return bootstrap!.connect(host: options.host, port: options.prot)
    }
}
复制代码

以前写了一遍用SwiftNIO创建UDP通信的，用的是DatagramBootstrap，咱们这里须要看成客户端链接，因此用的是ClientBootstrap。

最后加上一个RESPHandler

class RESPHandler: ChannelDuplexHandler {
    typealias InboundIn = ByteBuffer
    
    func channelRead(ctx: ChannelHandlerContext, data: NIOAny) {
        var value = unwrapInboundIn(data)
        print(value.readString(length: value.writerIndex))
    }
}
复制代码

这里接收到服务端回复后先不作任何操做直接输出。

如今在SedisClient的init方法末尾加上一段测试代码测试是否能正常通信

init(options: SedisOptions) {
        self.options = options
        
        loopGroup = MultiThreadedEventLoopGroup(numThreads: System.coreCount)
        bootstrap = ClientBootstrap(group: loopGroup)
            .channelOption(ChannelOptions.socket(SocketOptionLevel(SOL_SOCKET),
                                                 SO_REUSEADDR), value: 1)
            .channelInitializer({ (channel) -> EventLoopFuture<Void> in
                    channel.pipeline.add(handler: RESPHandler())
            })
        
        let channel = try? _connect().wait()
        let command = "set a 1\r\n".utf8
        var byteBuffer = ByteBufferAllocator().buffer(capacity: command.count)
        byteBuffer.write(bytes: command)
        channel?.writeAndFlush(byteBuffer, promise: nil)
        try? channel?.closeFuture.wait()
    }

复制代码

测试运行

let sdies = SedisClient(options: SedisOptions(prot: 6379, host: "127.0.0.1", password: nil, database: 0))
复制代码

能够看到控制台输出

Optional("+OK\r\n")
复制代码

至此，第一部分就已经finish