69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
|
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()
|
||
|
|
}
|