tyc-server/app/user/cmd/api/internal/svc/servicecontext.go

package svc

import (
	"github.com/hibiken/asynq"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/stores/redis"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
	"github.com/zeromicro/go-zero/rest"
	"qnc-server/app/user/cmd/api/internal/config"
	"qnc-server/app/user/cmd/api/internal/middleware"
	"qnc-server/app/user/cmd/api/internal/service"
	"qnc-server/app/user/model"
)

type ServiceContext struct {
	Config              config.Config
	Redis               *redis.Redis
	SourceInterceptor   rest.Middleware
	UserModel           model.UserModel
	UserAuthModel       model.UserAuthModel
	ProductModel        model.ProductModel
	FeatureModel        model.FeatureModel
	ProductFeatureModel model.ProductFeatureModel
	OrderModel          model.OrderModel
	QueryModel          model.QueryModel
	AlipayService       *service.AliPayService
	WechatPayService    *service.WechatPayService
	ApplePayService     *service.ApplePayService
	WestDexService      *service.WestDexService
	ApiRequestService   *service.ApiRequestService
	AsynqServer         *asynq.Server         // 服务端
	AsynqService        *service.AsynqService // 客户端
	VerificationService *service.VerificationService
}

func NewServiceContext(c config.Config) *ServiceContext {
	db := sqlx.NewMysql(c.DataSource)
	redisConf := redis.RedisConf{
		Host: c.CacheRedis[0].Host,
		Pass: c.CacheRedis[0].Pass,
		Type: c.CacheRedis[0].Type,
	}
	asynqServer := asynq.NewServer(
		asynq.RedisClientOpt{Addr: c.CacheRedis[0].Host, Password: c.CacheRedis[0].Pass},
		asynq.Config{
			IsFailure: func(err error) bool {
				logx.Errorf("异步任务失败: %+v \n", err)
				return true
			},
			Concurrency: 10,
		},
	)
	westDexService := service.NewWestDexService(c)
	productFeatureModel := model.NewProductFeatureModel(db, c.CacheRedis)
	featureModel := model.NewFeatureModel(db, c.CacheRedis)
	return &ServiceContext{
		Config:              c,
		Redis:               redis.MustNewRedis(redisConf),
		SourceInterceptor:   middleware.NewSourceInterceptorMiddleware().Handle,
		AlipayService:       service.NewAliPayService(c),
		WechatPayService:    service.NewWechatPayService(c),
		ApplePayService:     service.NewApplePayService(c),
		WestDexService:      westDexService,
		VerificationService: service.NewVerificationService(c, westDexService),
		AsynqServer:         asynqServer,
		ApiRequestService:   service.NewApiRequestService(c, westDexService, featureModel, productFeatureModel),
		AsynqService:        service.NewAsynqService(c),
		UserModel:           model.NewUserModel(db, c.CacheRedis),
		UserAuthModel:       model.NewUserAuthModel(db, c.CacheRedis),
		ProductModel:        model.NewProductModel(db, c.CacheRedis),
		OrderModel:          model.NewOrderModel(db, c.CacheRedis),
		QueryModel:          model.NewQueryModel(db, c.CacheRedis),
		FeatureModel:        featureModel,
		ProductFeatureModel: productFeatureModel,
	}
}
func (s *ServiceContext) Close() {
	if s.AsynqService != nil {
		s.AsynqService.Close()
	}
}