Golang 通用连接池

墨宜人
2023-12-01

在使用之前我们需要先了解清楚连接池的概念,总结下来连接池主要解决以下几类问题:

  • 减少连接创建时间 不论是与数据库还是Thrift等程序建立连接都是有开销的。如果这类连接是“循环”使用的,使用该方式这些花销都是可以避免的。
  • 简化了编程模式  当使用连接池时,具体的操作都是统一的不用关心初始化等各项关心点。
  • 受控的资源使用  如果用户不使用连接池,而是每当需要时创建一个新的连接,那么用户的应用程序的资源使用会产生非常大的浪费并且可能会导致高负载下的异常发生。

 

go-commons-pool 使用

package main

import (
        "github.com/jolestar/go-commons-pool"
        "fmt"
        "time"
)

var pCommonPool *pool.ObjectPool

type PoolTest struct{}

func (this *PoolTest) Test() string {
        return "PoolTest"
}

func init() {
        // 初始化连接池配置项
        PoolConfig := pool.NewDefaultPoolConfig()
        // 连接池最大容量设置
        PoolConfig.MaxTotal = 1000
        WithAbandonedConfig := pool.NewDefaultAbandonedConfig()
        // 注册连接池初始化链接方式
        pCommonPool = pool.NewObjectPoolWithAbandonedConfig(pool.NewPooledObjectFactorySimple(
                func() (interface{}, error) {
                        return Link()
                }), PoolConfig, WithAbandonedConfig)
}

// 初始化链接类
func Link() (*PoolTest, error) {
        fmt.Println("初始化PoolTest类!!!")
        return &PoolTest{}, nil
}

func main() {

        //----------------------------------第一次使用将会调用初始化方法---------------------------------
        fmt.Println("第一次使用将会调用初始化方法")
        Test()

        //----------------------------------第二次使用将会复用初始化好的对象---------------------------------
        fmt.Println("第二次使用将会复用初始化好的实例")
        Test()

        //----------------------------------连续多次并发调用当连接池不够用的会扩充连接池---------------------------
        fmt.Println("连续多次并发调用当连接池不够用的会扩充连接池")
        go Test()
        go Test()
        go Test()
        go Test()
        go Test()

        time.Sleep(1 * time.Second)
}

func Test() {
        var client *PoolTest
        // 从连接池中获取一个实例
        obj, _ := pCommonPool.BorrowObject()
        // 转换为对应实体
        if obj != nil {
                client = obj.(*PoolTest)
        }
        // 调用需要的方法
        fmt.Println(client.Test())
        // 交还连接池
        pCommonPool.ReturnObject(client)
}

输出如下:

第一次使用将会调用初始化方法
初始化PoolTest类!!!PoolTest
第二次使用将会复用初始化好的实例
PoolTest
连续多次并发调用当连接池不够用的会扩充连接池
PoolTest
初始化PoolTest类!!!PoolTest
PoolTest
初始化PoolTest类!!!PoolTest
PoolTest

参考:

https://www.yuque.com/docs/share/f6ce9dd8-eed0-41b7-9eb1-f109f3306632

 类似资料: