forked from JointCloud/pcm-coordinator
69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
/*
|
|
|
|
Copyright (c) [2023] [pcm]
|
|
[pcm-coordinator] is licensed under Mulan PSL v2.
|
|
You can use this software according to the terms and conditions of the Mulan PSL v2.
|
|
You may obtain a copy of Mulan PSL v2 at:
|
|
http://license.coscl.org.cn/MulanPSL2
|
|
THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
EITHER EXPaRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
See the Mulan PSL v2 for more details.
|
|
|
|
*/
|
|
|
|
package svc
|
|
|
|
import (
|
|
"github.com/go-redis/redis/v8"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/utils"
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/rpc/internal/config"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
"gorm.io/gorm/schema"
|
|
"time"
|
|
)
|
|
|
|
type ServiceContext struct {
|
|
Config config.Config
|
|
DbEngin *gorm.DB
|
|
RedisClient *redis.Client
|
|
}
|
|
|
|
func NewServiceContext(c config.Config) *ServiceContext {
|
|
//启动Gorm支持
|
|
dbEngin, _ := gorm.Open(mysql.Open(c.DB.DataSource), &gorm.Config{
|
|
NamingStrategy: schema.NamingStrategy{
|
|
SingularTable: true, // 使用单数表名，启用该选项，此时，`User` 的表名应该是 `t_user`
|
|
},
|
|
Logger: logger.Default.LogMode(logger.Warn),
|
|
})
|
|
sqlDB, err := dbEngin.DB()
|
|
// SetMaxIdleConns 设置空闲连接池中连接的最大数量
|
|
sqlDB.SetMaxIdleConns(10)
|
|
|
|
// SetMaxOpenConns 设置打开数据库连接的最大数量。
|
|
sqlDB.SetMaxOpenConns(50)
|
|
|
|
// SetConnMaxLifetime 设置了连接可复用的最大时间。
|
|
sqlDB.SetConnMaxLifetime(time.Hour)
|
|
//添加snowflake支持
|
|
err = utils.InitSnowflake(c.SnowflakeConf.MachineId)
|
|
if err != nil {
|
|
logx.Errorf("InitSnowflake err: ", err)
|
|
panic("InitSnowflake err")
|
|
}
|
|
return &ServiceContext{
|
|
Config: c,
|
|
DbEngin: dbEngin,
|
|
RedisClient: redis.NewClient(&redis.Options{
|
|
Addr: c.RedisConf.Host,
|
|
Password: c.RedisConf.Pass,
|
|
DB: 0, // use default DB
|
|
}),
|
|
}
|
|
}
|