Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > Go初始化数据库

Golang初始化MySQL数据库方法浅析

作者:三杯温开水

这篇文章主要介绍了Golang初始化MySQL数据库的方法,数据库的建立第一步即要初始化,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习吧

安装依赖

主要这边还需要安装两个依赖,gormviper ,具体的可以访问他们的官网(Gorm官网地址 Viper Github地址)

初始化配置

在根目录下打开终端执行:

go mod init 项目名称 (例如: go mod init demo )

安装依赖

go get github.com/jinzhu/gorm 或 go get -u gorm.io/gorm

go get -u gorm.io/driver/mysql

go get github.com/spf13/viper

创建文件

创建一个yml文件(application.yml)

#server:
#  post: 8888
datasource:
  driverName: mysql
  host: # 端口地址
  port: # 端口号
  database: # 数据库名称
  username: # 账号
  password: # 密码
  charset: utf8

在目录下创建一个common包,包下再创建一个database.go文件,内部写一个初始化数据库,建议拆分出来,这样拼接的方式有利于项目在不同环境下进行快速的更改配置。

package common
import (
	"fmt"
	// 看清楚引入包,包的差异存在方法差异
	"github.com/jinzhu/gorm" 
	"github.com/spf13/viper"
)
// 用大写声明(可以全局访问)
var DB *gorm.DB
// 创建数据库连接池
func InitDB() *gorm.DB {
	driverName := viper.GetString("datasource.driverName")
	host := viper.GetString("datasource.host")
	port := viper.GetString("datasource.port")
	database := viper.GetString("datasource.database")
	username := viper.GetString("datasource.username")
	password := viper.GetString("datasource.password")
	charset := viper.GetString("datasource.charset")
	args := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s&parseTime=true",
		username,
		password,
		host,
		port,
		database,
		charset)
	db, err := gorm.Open(driverName, args)
	if err != nil {
		fmt.Println("fail err mysql", err.Error())
	}
	// gorm 自动创建表,需要放入model层中的模型,比如 User{}
	db.AutoMigrate(User{})
	// 进行赋值 否则会空指针
	DB = db
	return db
}
// 获取DB的示例
func GetDB() *gorm.DB {
	return DB
}

main函数执行

在根目录下创建一个main文件(唯一),在这个环境我们需要做做一个初始化配置文件的操作,在main中优先加载这个config,然后才是初始化数据库,顺序不能错哈,因为数据库的连接信息在yml文件中,所以不要将顺序搞错。

package main
import (
	"github.com/gin-gonic/gin"
	_ "github.com/go-sql-driver/mysql"
	"github.com/spf13/viper"
	"go-gin-vue/common"
	"os"
)
func main() {
	InitConfig()
	// 尝试连接数据库
	db := common.InitDB()
	// 延迟关闭
	defer db.Close()
	r := gin.Default()
	// 加载路由
	r = CollectRouter(r)
	// 监听的端口
	// 可以在yml中定义server端口并且使用yml文件中定义的端口
	// port := viper.GetString("server.port") 
	if port != "" {
		panic(r.Run(":" + port))
	}
	r.Run(":9000") // 测试端口 9000 上启动服务
}
// 初始化config文件
func InitConfig() {
	workDir, _ := os.Getwd()
	// 读取的文件名
	viper.SetConfigName("application")
	// 读取的文件类型
	viper.SetConfigType("yml")
	// 读取的路径
	viper.AddConfigPath(workDir + "/config")
	err := viper.ReadInConfig()
	if err != nil {
		panic("")
	}
}

到此这篇关于Golang初始化数据库方法浅析的文章就介绍到这了,更多相关Go初始化数据库内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文