erlang mysql driver_erlang_mysql_driver 源码分析1
erlang_mysql_driver 是个mysql的数据库驱动
源码主要包含 mysql mysql_conn mysql_recv mysql_auth 这几个模块
mysql模块提供给外部调用的接口，包括启动、添加连接、执行sql语句。
mysql模块的另一主要功能是维护mysql_conn连接池，在执行sql语句时，选择合适的mysql_conn进程进行sql处理。
mysql_conn 和 mysql_recv 负责具体的sql执行逻辑和获取返回结果。
一、mysql模块的启动
1、mysql提供了两种启动方式 start 和start_link
start1(PoolId, Host, Port, User, Password, Database, LogFun, Encoding,
StartFunc) ->
crypto:start(),
gen_server:StartFunc(
{local, ?SERVER}, ?MODULE,
[PoolId, Host, Port, User, Password, Database, LogFun, Encoding], []).
我们可以看上面代码，通过传入StartFunc值的不同(start_link或start)，最终分别调用gen_server:start和gen_server:start_link
关于gen_server:start和gen_server:start_link的异同，我们可以查看官方文档：
gen_server:start启动一个独立的gen_server进程:
Creates a stand-alone gen_server process, i.e. a gen_server which is not part of a supervision tree and thus has no supervisor.
gen_server:start_link在监督树下启动一个gen_server进程:
Creates a gen_server process as part of a supervision tree. The function should be called, directly or indirectly, by the supervisor. It will, among other things, ensure that the gen_server is linked to the supervisor.
回到正题，当我们调用mysql:start/start_link后
这里启动的进程名是 mysql_dispatcher，并不是mysql，回调模块才是mysql。
启动成功后，通过sys:get_status(mysql_dispatcher) 可以查看mysql_dispatcher的state状态。
2、启动时的初始化准备
init([PoolId, Host, Port, User, Password, Database, LogFun, Encoding]) ->
LogFun1 = if LogFun == undefined -> fun log/4; true -> LogFun end,
case mysql_conn:start(Host, Port, User, Password, Database, LogFun1,
Encoding, PoolId) of
{ok, ConnPid} ->
Conn = new_conn(PoolId, ConnPid, true, Host, Port, User, Password,
Database, Encoding),
State = #state{log_fun = LogFun1},
{ok, add_conn(Conn, State)};
{error, _Reason} ->
?Log(LogFun1, error,
"failed starting first MySQL connection handler, "
"exiting"),
C = #conn{pool_id = PoolId,
reconnect = true,
host = Host,
port = Port,
user = User,
password = Password,
database = Database,
encoding = Encoding},
start_reconnect(C, LogFun),
{ok, #state{log_fun = LogFun1}}
end.
调用gen_server:start或gen_server:start_link时，init方法启动一个mysql_conn进程，并且在mysql_conn成功启动后将mysql_conn的进程信息记录在gen_server state的结构中，方便下次使用时，直接在state中找到需要的mysql_conn进程。
二、添加mysql_conn进程
1、mysql_conn进程 与 mysql:connect
mysql:connect 启动一个mysql_conn进程，(mysql_conn进程会建立一个和指定mysql数据库的连接)，启动成功后记录下mysql_conn的进程ID。
这里在调用mysql:connect方法的时候，需要提供pool_id(atom)，用来表示将这个进程添加到具体 哪个连接池中。
另外需要注意的是，mysql_conn进程与mysql_dispacher之间并不是父子进程关系，mysql_conn与调用这个方法(mysql:connect)的进程才具有监督关系(如果start_link)的话，mysql_dispacher只是在state上记录mysql_conn的pid。
connect(PoolId, Host, Port, User, Password, Database, Encoding, Reconnect,
LinkConnection) ->
Port1 = if Port == undefined -> ?PORT; true -> Port end,
Fun = if LinkConnection ->
fun mysql_conn:start_link/8;
true ->
fun mysql_conn:start/8
end,
{ok, LogFun} = gen_server:call(?SERVER, get_logfun),
case Fun(Host, Port1, User, Password, Database, LogFun,
Encoding, PoolId) of
{ok, ConnPid} ->
Conn = new_conn(PoolId, ConnPid, Reconnect, Host, Port1, User,
Password, Database, Encoding),
case gen_server:call(
?SERVER, {add_conn, Conn}) of
ok ->
{ok, ConnPid};
Res ->
Res
end;
Err->
Err
end.
2、 mysql:connect
1、启动mysql_conn
2、如果顺利启动了mysql_conn，则call add_conn消息给mysql_dispacher
发送add_conn消息给mysql_dispacher的时候，我们可以注意下，record #conn，在mysql里使用一个#conn来表示一个mysql连接。
#conn里也包含了一个mysql连接所需的数据(host ip port database encoding等信息)，还有就是 pool_id和本身的pid。
三、mysql_dispacher是怎么处理的 add_conn 的消息的？
mysql_dispacher 对于这个消息的处理，都封装在 mysql:add_conn里
add_conn(Conn, State) ->
Pid = Conn#conn.pid,
erlang:monitor(process, Conn#conn.pid),
PoolId = Conn#conn.pool_id,
ConnPools = State#state.conn_pools,
NewPool =
case gb_trees:lookup(PoolId, ConnPools) of
none ->
{[Conn],[]};
{value, {Unused, Used}} ->
{[Conn | Unused], Used}
end,
State#state{conn_pools =
gb_trees:enter(PoolId, NewPool,
ConnPools),
pids_pools = gb_trees:enter(Pid, PoolId,
State#state.pids_pools)}.
1、monitor 传过来的pid，这个monitor的目的是可以在mysql_conn挂掉的时候，收到一条通知消息{‘DOWN’…}，具体可以查看erlang:monitor文档
2、修改mysql_dispacher中的两个重要的数据 conn_pools和pid_pools
这两个数据结构都是使用的gb_trees，事实上state中有3个元素是使用gb_trees的。gb_trees就是二叉平衡树，使用二叉平衡树的目的是为了快速查找。
conn_pools ：连接池，如果添加连接的时候，发现这个连接对应的连接池PoolId并不存在，则新建一个PoolId
pid_pools: 进程池，添加mysql_conn对应的进程到进程池中
3、返回最终的State
四、conn_pools 的维护
mysql_dispatcher进程的主要作用就是维护多个连接池，管理连接池的连接。具体的业务就是，当用户请求使用连接时选择合适的连接，并记录该连接的使用情况。再到用户使用完连接收到消息，重新记录下连接的使用情况。
这个业务过程就是很常见的池的处理了，如emysql在处理这样情况的时候就是对连接进行分组，分为可用队列和已在使用队列，然后玩家请求则从可用组取，放入使用队列，用后再放回可用队列。
然而erlang_mysql_driver并不是这样处理的，但是他的处理也是一种不错的手段，对以后(进程池、连接池等)池的处理可以作为参考，所以我们重点关心mysql_dispatcher对于conn_pools的维护。
(注：conn_pools的存储位置在mysql_dispatcher中state中)
1、conn_pools与pool_id的关系
gen_server进程mysql_dispatcher的state里有一个元素是conn_pools，conn_pools以gb_trees的形式维护所有的pool_id。
一个pool_id对应gb_trees上的一个节点，每次新添加一个关于pool_id_1的conn的时候。会现在conn_pools中寻找是否存在pool_id_1，
{value, ConnList} = gb_trees:lookup(pool_id_1, ConnPools)，如果已经存在则已添加列表的形式，将Conn添加到ConnList的列表头上。
none = gb_trees:lookup(pool_id_1, ConnPools)，如果不存在pool_id_1，则新建一个，最终gb_trees:enter到原先的conn_pools中。
也就是说，conn_pools是一个gb_trees的数据结构，而pool_id为这个结构上的key。
而这个key对应的value是ConnList。ConnList是一个列表，该列表的形式:
[{#conn{}, Num}, {#conn{}, Num}, {#conn{}, Num}…]，一个#conn表示一个mysql_conn进程，Num表示该进程当前执行的sql数。
2、为什么要使用gb_trees ?
这里我们可以看到，在添加连接到连接池的操作中，pool_id的查询操作是需要经常用到的。
频繁地查询使用二叉查找树可以提高查询的效率。
不过在实际使用中，并不会有太多的连接池，一般来说有一个连接池专门负责select操作，一个连接池负责insert、delete和update操作，就可以了。
3、当我们调用mysql:fetch(PoolId, Query)的时候，mysql_dispatcher是怎么选择mysql_conn进程来执行的？
查看代码 mysql:get_next_conn(PoolId, State)
1、 在mysql_dispatcher的state里的conn_pools中找到PoolId
如果这个时候连PoolId都找不到的话，就失败了，证明不存在这个连接池。
或者说找到后，发现这个连接池里的conn数为空，那也失败了。这种情况正常逻辑下不会出现，因为在一开始的时候都是通过启动mysql_conn才添加pool的。
2、lookup PoolId后，会获得一个列表，列表里面就是所有可用的conn，[{conn1, num1}, {conn2, num2}, {conn3, num3}…]。
每个conn后面都会有一个num，num表示当前这个conn被调用的次数。
这里我们会直接选择列表里的第一个Conn，选择后再把这个Conn对应的Num值加1。
最后根据Num值从小到大排序一遍，那么列表开头的conn又是被调用次数最少的一个。
下次调用的时候，有继续上面的过程。
所以回顾我们之前说的add_connect的过程，新添加的conn，其num为0，直接添加到列表头。
这里面还是有一个效率问题，就是每一次执行sql语句，其实都是需要对连接池里的连接进行重新排序的。
4、sql执行完成后，Num数据减一
在上面，我们可以看到，当mysql_dispatcher发消息给mysql_conn的时候，该Conn对应的Num是会加一的。那么当mysql_conn执行完sql后，他又会发消息回来给mysql_dispatcher，将Num减一。
具体的这个处理过程，可以查看mysql_dispatcher：query_done。