serviceContext.go 663 B

12345678910111213141516171819202122232425
  1. package svc
  2. import (
  3. "GtDataStore/app/cmd/dtgateway/internal/config"
  4. "GtDataStore/app/cmd/dtgateway/internal/middleware"
  5. "GtDataStore/app/cmd/organization/organization"
  6. "github.com/zeromicro/go-zero/rest"
  7. "github.com/zeromicro/go-zero/zrpc"
  8. )
  9. type ServiceContext struct {
  10. Config config.Config
  11. OrganizationRpc organization.Organization
  12. DataSign rest.Middleware
  13. }
  14. func NewServiceContext(c config.Config) *ServiceContext {
  15. svc := &ServiceContext{
  16. Config: c,
  17. OrganizationRpc: organization.NewOrganization(zrpc.MustNewClient(c.OrganizationRpcConf)),
  18. }
  19. svc.DataSign = middleware.NewDataSignMiddleware(c).Handle
  20. return svc
  21. }