Golang 建立TCP时使用连接池

Posted 2023-04-16

参考技术A 程序输出如下，相比不用连接池，单次操作时间少了一个数量级。

Golang 通用连接池

资源链接 

1. pool
   

2. 
   

3. GoDoc
   

4. 
   

5. Golang 实现的连接池
   

6. 
   

7. 功能：
   

8. 
   

9. 连接池中连接类型为interface{}，使得更加通用
   

10. 链接的最大空闲时间，超时的链接将关闭丢弃，可避免空闲时链接自动失效问题
    

11. 使用channel处理池中的链接，高效
    

12. 基本用法
    

13. 
    

14. //factory 创建连接的方法
    

15. factory := func() (interface{}, error) { return net.Dial("tcp", "127.0.0.1:4000") }
    

16. 
    

17. //close 关闭链接的方法
    

18. close := func(v interface{}) error { return v.(net.Conn).Close() }
    

19. 
    

20. //创建一个连接池： 初始化5，最大链接30
    

21. poolConfig := &pool.PoolConfig{
    

22.     InitialCap: 5,
    

23.     MaxCap:     30,
    

24.     Factory:    factory,
    

25.     Close:      close,
    

26.     //链接最大空闲时间，超过该时间的链接 将会关闭，可避免空闲时链接EOF，自动失效的问题
    

27.     IdleTimeout: 15 * time.Second,
    

28. }
    

29. p, err := pool.NewChannelPool(poolConfig)
    

30. if err != nil {
    

31.     fmt.Println("err=", err)
    

32. }
    

33. 
    

34. //从连接池中取得一个链接
    

35. v, err := p.Get()
    

36. 
    

37. //do something
    

38. //conn=v.(net.Conn)
    

39. 
    

40. //将链接放回连接池中
    

41. p.Put(v)
    

42. 
    

43. //释放连接池中的所有链接
    

44. p.Release()
    

45. 
    

46. //查看当前链接中的数量
    

47. current := p.Len()
    

48. 
    

49. 注:
    

50. 
    

51. 该连接池参考 https://github.com/fatih/pool 实现，改变以及增加原有的一些功能