jyInfo/rpc/manager/init/init.go

package init

import (
	"flag"
	"fmt"
	"github.com/zeromicro/go-zero/core/conf"
	"jyInfo/rpc/manager/internal/config"
	"jyInfo/rpc/model"
	"log"

	_ "github.com/go-sql-driver/mysql"
	"github.com/jinzhu/gorm"
)

var configF = flag.String("ff", "etc/manager.yaml", "the config file")

var C config.Config
var DB *gorm.DB
var err error

func init() {
	conf.MustLoad(*configF, &C)
	//初始化msyql
	mm := C.MysqlMain
	data_source := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8&parseTime=True&loc=Local",
		mm.UserName,
		mm.PassWord,
		mm.Address,
		mm.DbName,
	)
	DB, err = gorm.Open("mysql", data_source)
	if err != nil {
		panic(err)
	}
	DB.DB().SetMaxOpenConns(mm.MaxOpenConns) // 最大连接数
	DB.DB().SetMaxIdleConns(mm.MaxIdleConns) // 最大空闲数
	log.Println("初始化 mysql")

	//初始redis
	rm := C.Redis
	model.InitRedis(&rm)
	model.InitMysql(&mm)
	//初始NSQ
	nm := C.Nsq
	model.InitNsqConfig(&nm)

	//初始Sensitive
	sm := C.Sensitive
	model.InitSensitive(&sm)

}