利用golang中的chan数据类型来实现简易连接池

chan其实是一个队列，而且是一个带有锁等待功能的队列，因此用它可以做很多任务队列相关的工作，比如说连接池。最近打算用go来写一个小的软件配置管理系统，使用中发现没有连接池，于是就是chan写了一个简单的，使用了一下发现相当好用。

var MySQLPool chan *mysql.MySQL func getMySQL() *mysql.MySQL { if MySQLPool == nil { MySQLPool = make(chan *mysql.MySQL, MAX_POOL_SIZE) } if len(MySQLPool) == 0 { go func() { for i := 0; i < MAX_POOL_SIZE/2; i++ { mysql := mysql.New() err := mysql.Connect("127.0.0.1", "root", "", "wgt", 3306) if err != nil { panic(err.String()) } putMySQL(mysql) } }() } return <-MySQLPool } func putMySQL(conn *mysql.MySQL) { if MySQLPool == nil { MySQLPool = make(chan *mysql.MySQL, MAX_POOL_SIZE) } if len(MySQLPool) == MAX_POOL_SIZE { conn.Close() return } MySQLPool <- conn }

短短几十行代码就可以实现一个较为完美的连接池功能，go语言的表现力确实很令人兴奋。

有疑问加站长微信联系（非本文作者）