go使用mysql数据库

需要使用的库

• gorm.io/driver/mysql

• gorm.io/gorm

gorm是一个orm，相当于java中使用的mybatis，这里不使用原生的连接方式，直接用框架比较简单。

go get -u gorm.io/gorm
go get -u gorm.io/driver/mysql

开始编写代码

• 定义接口

type Repo interface {
	i()
	// 获取读数据库连接
	GetDbR() *gorm.DB
	// 获取写数据库连接
	GetDbW() *gorm.DB
	// 关闭读数据库连接
	DbRClose() error
	// 关闭写数据库连接
	DbWClose() error
}

• 定义一个结构体，并实现接口中的方法

type dbRepo struct {
	DbR *gorm.DB
	DbW *gorm.DB
}

func (d *dbRepo) i() {}

func (d *dbRepo) GetDbR() *gorm.DB {
	return d.DbR
}

func (d *dbRepo) GetDbW() *gorm.DB {
	return d.DbW
}

func (d *dbRepo) DbRClose() error {
	sqlDB, err := d.DbR.DB()
	if err != nil {
		return err
	}
	return sqlDB.Close()
}

func (d *dbRepo) DbWClose() error {
	sqlDB, err := d.DbW.DB()
	if err != nil {
		return err
	}
	return sqlDB.Close()
}

• 创建一个数据库连接并返回连接对象

// New 创建连接
func New() (Repo, error) {
	dbr, err := dbConnect("root", "root", "3306", "student")
	if err != nil {
		return nil, err
	}

	dbw, err := dbConnect("root", "root", "3306", "student")
	if err != nil {
		return nil, err
	}

	return &dbRepo{
		DbR: dbr,
		DbW: dbw,
	}, nil
}

// dbConnect 数据库连接
func dbConnect(user, pass, addr, dbName string) (*gorm.DB, error) {
	dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=%t&loc=%s",
		user,
		pass,
		addr,
		dbName,
		true,
		"Local")

	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true,
		},
		//Logger: logger.Default.LogMode(logger.Info), // 日志配置
	})

	if err != nil {
		return nil, err
	}

	db.Set("gorm:table_options", "CHARSET=utf8mb4")

	cfg := config.Get().MySql.Base

	sqlDB, err := db.DB()
	if err != nil {
		return nil, err
	}

	// 设置连接池 用于设置最大打开的连接数，默认值为0表示不限制.设置最大的连接数，可以避免并发太高导致连接mysql出现too many connections的错误。
	sqlDB.SetMaxOpenConns(60)

	// 设置最大连接数 用于设置闲置的连接数.设置闲置的连接数则当开启的一个连接使用完成后可以放在池里等候下一次使用。
	sqlDB.SetMaxIdleConns(60)

	// 设置最大连接超时
	sqlDB.SetConnMaxLifetime(time.Minute * time.Duration(10))

	return db, nil
}

• 测试

func TestTable(t *testing.T) {
	repo, _ := db.New()
	dbR := repo.GetDbR()
	var result string
	dbR.Raw(`show databases`).Scan(&result)
	fmt.Println(result)
}

交流群