为啥要使用连接池:

因为TCP的三只握手等等原因,建立一个连接是一件成本比较高的行为。所以在一个需要多次与特定实体交互的程序中,就需要维持一个连接池,里面有可以复用的连接可供重复使用

 

mysql

下载依赖包 

go get -u github.com/go-sql-driver/mysql

 

func ConnectMysql() *sql.DB {
    dataSourceName := user + ":" + password + "@tcp(" + host + ":" + port + ")/" + database + "?charset=utf8&parseTime=True"
    conn, err := sql.Open("mysql", dataSourceName)
    if err != nil {
        return nil
    }
    conn.SetConnMaxLifetime(time.Minute * 30) //最大连接超时时间,应小于数据库本身的链接超时时间
    conn.SetMaxIdleConns(10)                  //最大闲置连接数(并发时可以同时获取的连接,也是用完后放回池里面的互用的连接, 从而提升性能)
    conn.SetMaxOpenConns(100)                 //最大打开连接数,0为不限制

    //记得结束后关闭连接
    defer conn.Close()

    //验证连接池是否可以,使用ping初始化一个连接
    err = conn.Ping()
    if err != nil {
        fmt.Println("database init failed, err: , err")
        return nil
    }
    
    /*简单查询
    row := conn.QueryRow("select * from table_name")
    fmt.Println(row)/**/
    
    return conn
}

 

 

redis

下载依赖包

go get -u github.com/go-redis/redis

 

func ConnRedis() *redis.Client {
    rd := redis.NewClient(&redis.Options{
        //连接信息
        Network:  "tcp",                 //网络类型:tcp,unix 默认tcp
        Addr:     "10.176.246.204:9736", //单机连接
        Password: "",                    //密码
        DB:       0,                     //数据库,默认0

        //连接池容量及闲置连接数量
        PoolSize:     8,  //连接池最大socket数,默认为4倍cpu核数
        MinIdleConns: 10, //启动阶段创建指定数量的idle连接,并长期维持idle状态的连接数不少于指定数量

        //超时
        DialTimeout:  5 * time.Second, //连接建立超时时间,默认5秒
        ReadTimeout:  5 * time.Second, //读超时时间,默认3秒,-1表示取消读超时
        WriteTimeout: 5 * time.Second, //写超时时间,默认等于读超时
        PoolTimeout:  5 * time.Second, //当所有连接都处在繁忙状态时,客户端等待可用连接的最大等待时长,默认读超时+1秒

        //闲置连接检查
        IdleCheckFrequency: 60 * time.Second, //闲置连接检查的周期,默认为1分钟,-1表示不作周期性检查,只在客户端获取连接时对闲置连接进行处理
        IdleTimeout:        5 * time.Minute,  //闲置超时,默认5分钟,-1表示取消闲置超时检查
        MaxConnAge:         0 * time.Second,  //连接存活时长,从创建开始计时,超过指定时长则关闭连接,默认为0,即不关闭存活时长较长的连接

        //命令执行失败时的重试策略
        MaxRetries:      0,                      //命令执行失败时,最多重试多少次,默认为0,即不重试
        MinRetryBackoff: 8 * time.Millisecond,   //每次计算重试间隔时间的下限,默认8毫秒,-1表示取消间隔
        MaxRetryBackoff: 512 * time.Millisecond, //每次计算重试间隔时间的上限,默认512毫秒,-1表示取消间隔

        //可自定义连接函数
        Dialer: func() (net.Conn, error) {
            netDialer := &net.Dialer{
                Timeout:   5 * time.Second,
                KeepAlive: 5 * time.Second,
            }
            return netDialer.Dial("tcp", "127.0.0.1:6379")
        },

        //钩子函数
        OnConnect: func(conn *redis.Conn) error { //仅当客户端执行命令需要从连接池获取连接时,且连接池需要新建连接时,才会调用此函数
            fmt.Printf("conn=%v\n", conn)
            return nil
        },
    })

    //集群模式连接
    /*rd := redis.NewClusterClient(&redis.ClusterOptions{
        Addrs: []string{"rec-ops-dev.cn-s01-dgdev01.redis.oppo.dev:6379"},
    })/**/

    //哨兵模式连接
    /*rd := redis.NewFailoverClient(&redis.FailoverOptions{
        MasterName: "name",
        SentinelAddrs: []string{"10.x.x.x:9736","10.x.x.x:9736"},
    })/**/

    defer rd.Close() //记得关闭连接

    result, err := rd.Ping().Result()
    if err != nil {
        fmt.Println("ping err:", err)
        return nil
    }
    fmt.Println(result)

    return rd
}

实例:

package main

import (
    "errors"
    "fmt"
    "github.com/go-redis/redis"
    "time"
)

func ConnRedis() {
    rd := redis.NewClient(&redis.Options{
        Addr:     "10.x.x.x:9736", //单机连接
        Password: "",                    //密码
        DB:       0,                     //数据库
    })

    result, err := rd.Ping().Result()
    if err != nil {
        fmt.Println("ping err:", err)
        return
    }
    fmt.Println(result)

    //新建一个key-value,第三个参数为过期时间(0为不过期)
    err = rd.Set("k1", "v1", 3*time.Second).Err()
    if err != nil {
        fmt.Println("set err:", err)
        return
    }

    //获取key-value
    val, err := rd.Get("k1").Result()
    if err != nil {
        //如果返回的错误是key不存在
        if errors.Is(err, redis.Nil) { //go-redis 库提供了一个 redis.Nil 错误来表示 Key 不存在的错误,以此区别处理 redis.Nil 和其他不为 nil 的错误
            fmt.Println("key error:", nil)
            return
        }
        //其他错误
        fmt.Println(err)
        return
    } else { //打印value
        fmt.Println("k1 == ", val)
    }

    //程序等待5秒钟
    time.Sleep(5 * time.Second)

    v2, err := rd.Get("k1").Result()
    if err != nil {
        //如果返回的错误是key不存在
        if errors.Is(err, redis.Nil) {
            fmt.Println("key error:", nil)
            return
        }
        //其他错误
        fmt.Println(err)
        return
    } else {
        fmt.Println("k1 == ", v2)
    }
}

func main() {
    ConnRedis()
}