You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
1.7 KiB
68 lines
1.7 KiB
package service |
|
|
|
import ( |
|
"context" |
|
|
|
"go-common/app/admin/main/mcn/conf" |
|
bfs "go-common/app/admin/main/mcn/dao/bfs" |
|
msg "go-common/app/admin/main/mcn/dao/msg" |
|
dao "go-common/app/admin/main/mcn/dao/up" |
|
videoup "go-common/app/admin/main/mcn/dao/videoup" |
|
"go-common/app/admin/main/mcn/model" |
|
"go-common/app/interface/main/mcn/tool/worker" |
|
accgrpc "go-common/app/service/main/account/api" |
|
arcgrpc "go-common/app/service/main/archive/api" |
|
memgrpc "go-common/app/service/main/member/api" |
|
|
|
"github.com/pkg/errors" |
|
) |
|
|
|
// Service struct |
|
type Service struct { |
|
c *conf.Config |
|
dao *dao.Dao |
|
bfs *bfs.Dao |
|
msg *msg.Dao |
|
videoup *videoup.Dao |
|
msgMap map[model.MSGType]*model.MSG |
|
memGRPC memgrpc.MemberClient |
|
accGRPC accgrpc.AccountClient |
|
arcGRPC arcgrpc.ArchiveClient |
|
worker *worker.Pool |
|
} |
|
|
|
// New init |
|
func New(c *conf.Config) (s *Service) { |
|
s = &Service{ |
|
c: c, |
|
dao: dao.New(c), |
|
bfs: bfs.New(c), |
|
msg: msg.New(c), |
|
videoup: videoup.New(c), |
|
worker: worker.New(nil), |
|
} |
|
var err error |
|
if s.accGRPC, err = accgrpc.NewClient(c.GRPCClient.Account); err != nil { |
|
panic(errors.WithMessage(err, "Failed to dial account service")) |
|
} |
|
if s.arcGRPC, err = arcgrpc.NewClient(c.GRPCClient.Archive); err != nil { |
|
panic(errors.WithMessage(err, "Failed to dial archive service")) |
|
} |
|
if s.memGRPC, err = memgrpc.NewClient(c.GRPCClient.Member); err != nil { |
|
panic(errors.WithMessage(err, "Failed to dial member service")) |
|
} |
|
s.setMsgTypeMap() |
|
return s |
|
} |
|
|
|
// Ping Service |
|
func (s *Service) Ping(c context.Context) (err error) { |
|
return s.dao.Ping(c) |
|
} |
|
|
|
// Close Service |
|
func (s *Service) Close() { |
|
s.dao.Close() |
|
s.worker.Close() |
|
s.worker.Wait() |
|
}
|
|
|