redigo源码解析

前言

redigo是用Go语言开发的Redis客户端，受到Redis官方的推荐。

redigo的源码量比较少，也比较清晰易读。redigo主要做了以下事项

• 与Redis server建立连接
• 按照RESP协议组装指令
• 向Redis server发用指令
• 接收Redis server返回的数据
• 将返回数据解析成Go的数据类型
• 连接池（Pool），支持设置最大的活动连接数，最大的空闲连接数

源码概述

redigo主要使用了Go官方的net、io和bufio包，这些包是Go对网络连接，I/O，Buffered I/O的抽象实现。

在看Pool代码的过程中，对于如何控制最大连接数，笔者多花了一些时间才看明白。Pool使用buffered channel (可缓冲的管道，以下简称 buff chan) 来控制最大连接数。

首先看一下活动连接和空闲连接的定义

• 活动连接：经Pool创建，未被关闭。这里的关闭是指关闭网络连接，也就是断开与Redis server的连接
• 空闲连接：活动连接使用完之后，需交还给Pool，Pool会将连接放入空闲连接列表，列表中的连接数量就是空闲连接数

可以看出，活动连接是包含空闲连接的。

Pool原理解读

概述部分提到，Pool使用buff chan来控制最大连接数，这里也是源码中比较难以理解的地方，需要对Go的channel有较深的理解。

Go有两种channel，chan和 buff chan，它们的特性如下

• chan，读写都是同步、阻塞操作。写操作阻塞，等待写入的数据被读取；读操作阻塞，等待数据写入。
• buff chan，读操作和chan类似；写操作在缓冲区未满时非阻塞，否则阻塞等待缓冲区有空位。

Pool利用buff chan的缓冲区长度固定，及读操作的阻塞等待来控制最大连接数。

源码解析

redis/
| – redis.go // 定义接口
| – conn.go // 实现redis.go中定义的接口，用于与Redis server建立连接
| – -- type conn struct // 链接结构体，实现了 redis.go中定义的Conn接口
| – -- type dialOptions struct // 拨号选项，包括拨号器、拨号函数，超时配置，TLS配置等
| – -- type DialOption struct // 拨号选项设置结构体，只包含函数f，用于修改拨号选项
| – -- func Dial // 拨号函数，参数：网络协议，网络地址，拨号选项设置结构体，返回 conn
| – -- func write* // 写命令函数，用于将命Redis指令写入buffer io
| – reply.go // 将redis返回数据解析成go的数据类型

连接池如何用channel控制打开的连接数

redis/
| – pool.go // Get()用于获取连接，Close()用于归还连接
| – -- Pool.active // 活动连接数，每建立一个连接（Dial）active加1，每关闭（Close）一个连接，active减1
| – -- Pool.ch chan struct{} // 用来控制最大的活动连接数
| – -- Pool.lazyInit() // 将Pool.ch 初始化为bu