31 lines
		
	
	
		
			1023 B
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			31 lines
		
	
	
		
			1023 B
		
	
	
	
		
			Go
		
	
	
	
	
	
| package svc
 | |
| 
 | |
| import (
 | |
| 	"github.com/zeromicro/go-zero/rest"
 | |
| 	"github.com/zeromicro/go-zero/zrpc"
 | |
| 	"tianyuan-api/apps/admin/internal/config"
 | |
| 	"tianyuan-api/apps/admin/internal/middleware"
 | |
| 	"tianyuan-api/apps/sentinel/sentinel"
 | |
| 	"tianyuan-api/apps/user/user"
 | |
| )
 | |
| 
 | |
| type ServiceContext struct {
 | |
| 	Config          config.Config
 | |
| 	AuthInterceptor rest.Middleware
 | |
| 	EntRpc          user.EnterpriseClient
 | |
| 	UserRpc         user.UserClient
 | |
| 	ProductRpc      sentinel.ProductClient
 | |
| 	WalletsRpc      user.WalletServiceClient
 | |
| }
 | |
| 
 | |
| func NewServiceContext(c config.Config) *ServiceContext {
 | |
| 	return &ServiceContext{
 | |
| 		Config:          c,
 | |
| 		AuthInterceptor: middleware.NewAuthInterceptorMiddleware(c).Handle,
 | |
| 		EntRpc:          user.NewEnterpriseClient(zrpc.MustNewClient(c.UserRpc).Conn()),
 | |
| 		UserRpc:         user.NewUserClient(zrpc.MustNewClient(c.UserRpc).Conn()),
 | |
| 		ProductRpc:      sentinel.NewProductClient(zrpc.MustNewClient(c.SentinelRpc).Conn()),
 | |
| 		WalletsRpc:      user.NewWalletServiceClient(zrpc.MustNewClient(c.UserRpc).Conn()),
 | |
| 	}
 | |
| }
 |