Golangdatabasesql源码分析

　　简介
　　Gorm是Go语言开发用的比较多的一个ORM。它的功能比较全：增删改查关联（包含一个，包含多个，属于，多对多，多种包含）CallBacks（创建、保存、更新、删除、查询找）之前之后都可以有callback函数预加载事务复合主键日志databasesql包
　　但是这篇文章中并不会直接看Gorm的源码，我们会先从databasesql分析。原因是Gorm也是基于这个包来封装的一些功能。所以只有先了解了databasesql包才能更加好的理解Gorm源码。
　　databasesql其实也是一个对于mysql驱动的上层封装。github。comgosqldrivermysql就是一个对于mysql的驱动，databasesql就是在这个基础上做的基本封装包含连接池的使用使用例子
　　下面这个是最基本的增删改查操作
　　操作分下面几个步骤：引入github。comgosqldrivermysql包（包中的init方法会初始化mysql驱动的注册）使用sql。Open初始化一个sql。DB结构调用PrepareExec执行sql语句
　　使用Exec函数无需释放调用完毕之后会自动释放，把连接放入连接池中packagemainimport（databasesqlfmtgithub。comgosqldrivermysqlstrconv）funcmain（）｛打开连接db，err：sql。Open（mysql，root：feg125800tcp（47。100。245。167：3306）artifact？charsetutf8locAsia2FShanghaiparseTimeTrue）iferr！nil｛fmt。Println（err：，err）｝设置最大空闲连接数db。SetMaxIdleConns（1）设置最大链接数db。SetMaxOpenConns（1）query（db，3）｝修改funcupdate（dbsql。DB，idint，userstring）｛stmt，err：db。Prepare（updateusersetUserName？whereId？）iferr！nil｛fmt。Println（err）｝res，err：stmt。Exec（user，id）updateId，err：res。LastInsertId（）fmt。Println（updateId）｝删除funcdelete（dbsql。DB，idint）｛stmt，err：db。Prepare（deletefromuserwhereid？）iferr！nil｛fmt。Println（err）｝res，err：stmt。Exec（1）updateId，err：res。LastInsertId（）fmt。Println（updateId）｝查询funcquery（dbsql。DB，idint）｛rows，err：db。Query（selectfromuserwhereidstrconv。Itoa（id））iferr！nil｛fmt。Println（err）return｝forrows。Next（）｛varidintvaruserstringvarpwdstringrows。Scan（id，user，pwd）fmt。Println（id：，id，user：，user，pwd：，pwd）｝rows。Close（）｝插入funcinsert（dbsql。DB，user，pwdstring）｛stmt，err：db。Prepare（insertintousersetUserName？，Password？）iferr！nil｛fmt。Println（err）｝res，err：stmt。Exec（peter，panlei）id，err：res。LastInsertId（）fmt。Println（id）｝连接池
　　因为Gorm的连接池就是使用databasesql包中的连接池，所以这里我们需要学习一下包里的连接池的源码实现。其实所有连接池最重要的就是连接池对象、获取函数、释放函数下面来看一下databasesql中的连接池。
　　DB对象typeDBstruct｛数据库实现驱动driverdriver。DriverdsnstringnumCloseduint64锁musync。Mutexprotectsfollowingfields空闲连接freeConn〔〕driverConn阻塞请求队列，等连接数达到最大限制时，后续请求将插入此队列等待可用连接connRequestsmap〔uint64〕chanconnRequest记录下一个key用于connRequestsmap的keynextRequestuint64NextkeytouseinconnRequests。numOpenintnumberofopenedandpendingopenconnectionsopenerChchanstruct｛｝closedbooldepmap〔finalCloser〕depSetlastPutmap〔driverConn〕string最大空闲连接数maxIdleint最大打开连接数maxOpenint连接最大存活时间maxLifetimetime。DurationcleanerChchanstruct｛｝｝
　　获取方法func（dbDB）conn（ctxcontext。Context，strategyconnReuseStrategy）（driverConn，error）｛db。mu。Lock（）ifdb。closed｛db。mu。Unlock（）returnnil，errDBClosed｝Checkifthecontextisexpired。select｛default：casectx。Done（）：db。mu。Unlock（）returnnil，ctx。Err（）｝lifetime：db。maxLifetime查看是否有空闲的连接如果有则直接使用空闲连接numFree：len（db。freeConn）ifstrategycachedOrNewConnnumFree0｛取出数据第一个conn：db。freeConn〔0〕复制数组，去除第一个连接copy（db。freeConn，db。freeConn〔1：〕）db。freeConndb。freeConn〔：numFree1〕conn。inUsetruedb。mu。Unlock（）ifconn。expired（lifetime）｛conn。Close（）returnnil，driver。ErrBadConn｝returnconn，nil｝判断是否超出最大连接数ifdb。maxOpen0db。numOpendb。maxOpen｛创建一个chanreq：make（chanconnRequest，1）获取下一个request作为map中的keyreqKey：db。nextRequestKeyLocked（）db。connRequests〔reqKey〕reqdb。mu。Unlock（）Timeouttheconnectionrequestwiththecontext。select｛casectx。Done（）：Removetheconnectionrequestandensurenovaluehasbeensentonitafterremoving。db。mu。Lock（）delete（db。connRequests，reqKey）db。mu。Unlock（）select｛default：caseret，ok：req：ifok｛db。putConn（ret。conn，ret。err）｝｝returnnil，ctx。Err（）如果没有取消则从reqchan中获取数据阻塞主一直等待有conn数据传入caseret，ok：req：if！ok｛returnnil，errDBClosed｝判断超时ifret。errnilret。conn。expired（lifetime）｛ret。conn。Close（）returnnil，driver。ErrBadConn｝returnret。conn，ret。err｝｝db。numOpenoptimisticallydb。mu。Unlock（）调用driver的Open方法建立连接ci，err：db。driver。Open（db。dsn）iferr！nil｛db。mu。Lock（）db。numOpencorrectforearlieroptimismdb。maybeOpenNewConnections（）db。mu。Unlock（）returnnil，err｝db。mu。Lock（）dc：driverConn｛db：db，createdAt：nowFunc（），ci：ci，inUse：true，｝db。addDepLocked（dc，dc）db。mu。Unlock（）returndc，nil｝
　　释放连接方法释放连接func（dbDB）putConn（dcdriverConn，errerror）｛db。mu。Lock（）if！dc。inUse｛ifdebugGetPut｛fmt。Printf（putConn（v）DUPLICATEwas：sPREVIOUSwas：s，dc，stack（），db。lastPut〔dc〕）｝panic（sql：connectionreturnedthatwasneverout）｝ifdebugGetPut｛db。lastPut〔dc〕stack（）｝设置已经在使用中dc。inUsefalsefor，fn：rangedc。onPut｛fn（）｝dc。onPutnil判断连接是否有错误iferrdriver。ErrBadConn｛db。maybeOpenNewConnections（）db。mu。Unlock（）dc。Close（）return｝ifputConnHook！nil｛putConnHook（db，dc）｝调用方法释放连接added：db。putConnDBLocked（dc，nil）db。mu。Unlock（）判断如果没有加到了空闲列表中dc关闭if！added｛dc。Close（）｝｝func（dbDB）putConnDBLocked（dcdriverConn，errerror）bool｛ifdb。closed｛returnfalse｝ifdb。maxOpen0db。numOpendb。maxOpen｛returnfalse｝如果等待chan列表大于0ifc：len（db。connRequests）；c0｛varreqchanconnRequestvarreqKeyuint64获取map中chan和keyforreqKey，reqrangedb。connRequests｛break｝从列表中删除chandelete（db。connRequests，reqKey）Removefrompendingrequests。iferrnil｛dc。inUsetrue｝把连接传入chan中让之前获取连接被阻塞的获取函数继续reqconnRequest｛conn：dc，err：err，｝returntrue｝elseiferrnil！db。closeddb。maxIdleConnsLocked（）len（db。freeConn）｛如果没有等待列表，则把连接放到空闲列表中db。freeConnappend（db。freeConn，dc）db。startCleanerLocked（）returntrue｝returnfalse｝
　　连接池的实现有很多方法，在databasesql包中使用的是chan阻塞使用map记录等待列表，等到有连接释放的时候再把连接传入等待列表中的chan不在阻塞返回连接。
　　之前我们看到的Redigo是使用一个chan来阻塞，然后释放的时候放入空闲列表，在往这一个chan中传入struct｛｝｛｝，让程序继续获取的时候再从空闲列表中获取。并且使用的是链表的结构来存储空闲列表。总结
　　databasesql是对于mysql驱动的封装，然而Gorm则是对于databasesql的再次封装。让我们可以更加简单的实现对于mysql数据库的操作。