[大数据]golang初始化数据库(MySQL)

Golang初始化MySQL数据

安装依赖

主要这边还需要安装两个依赖，gorm、viper ,具体的可以访问他们的官网(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("")
	}
}