30 lines
969 B
Go
30 lines
969 B
Go
package svc
|
|
|
|
import (
|
|
"github.com/Kagami/go-face"
|
|
"github.com/redis/go-redis/v9"
|
|
"schisandra-album-cloud-microservices/app/aisvc/model/mysql"
|
|
"schisandra-album-cloud-microservices/app/aisvc/model/mysql/query"
|
|
"schisandra-album-cloud-microservices/app/aisvc/rpc/internal/config"
|
|
"schisandra-album-cloud-microservices/common/face_recognizer"
|
|
"schisandra-album-cloud-microservices/common/redisx"
|
|
)
|
|
|
|
type ServiceContext struct {
|
|
Config config.Config
|
|
FaceRecognizer *face.Recognizer
|
|
DB *query.Query
|
|
RedisClient *redis.Client
|
|
}
|
|
|
|
func NewServiceContext(c config.Config) *ServiceContext {
|
|
redisClient := redisx.NewRedis(c.RedisConf.Host, c.RedisConf.Pass, c.RedisConf.DB)
|
|
_, queryDB := mysql.NewMySQL(c.Mysql.DataSource, c.Mysql.MaxOpenConn, c.Mysql.MaxIdleConn, redisClient)
|
|
return &ServiceContext{
|
|
Config: c,
|
|
FaceRecognizer: face_recognizer.NewFaceRecognition(),
|
|
DB: queryDB,
|
|
RedisClient: redisClient,
|
|
}
|
|
}
|