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