jy_docs/rpc/stdlib/internal/svc/servicecontext.go

package svc

import (
	"log"

	"app.yhyue.com/moapp/jybase/mysql"
	"app.yhyue.com/moapp/jydocs/rpc/stdlib/internal/config"
	"gorm.io/gorm"
)

type ServiceContext struct {
	Config config.Config
	DB     *gorm.DB
}

func NewServiceContext(c config.Config) *ServiceContext {
	db := initDB(c.DriverName, c.DbUrl)
	return &ServiceContext{
		Config: c,
		DB:     db,
	}
}

func initDB(driverName, url string) *gorm.DB {
	log.Println("----------->【jy_docs】 rpc base init start<--------------")
	docDB := mysql.GormMysql(driverName, url, 10, 10, nil)
	if docDB != nil {
		log.Println("----------->【jy_docs】 DB初始化成功!<--------------")
		return docDB
	}
	return nil
}