83 lines
2.3 KiB
Go
83 lines
2.3 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
|
|||
|
|
"go-common/app/admin/main/reply/conf"
|
|||
|
|
"go-common/app/admin/main/reply/dao"
|
|||
|
|
artrpc "go-common/app/interface/openplatform/article/rpc/client"
|
|||
|
|
accrpc "go-common/app/service/main/account/api"
|
|||
|
|
arcrpc "go-common/app/service/main/archive/api/gorpc"
|
|||
|
|
rlrpc "go-common/app/service/main/relation/rpc/client"
|
|||
|
|
thumbup "go-common/app/service/main/thumbup/api"
|
|||
|
|
bm "go-common/library/net/http/blademaster"
|
|||
|
|
"go-common/library/net/rpc/warden"
|
|||
|
|
"go-common/library/sync/pipeline/fanout"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Service is a service.
|
|||
|
|
type Service struct {
|
|||
|
|
conf *conf.Config
|
|||
|
|
dao *dao.Dao
|
|||
|
|
httpClient *bm.Client
|
|||
|
|
accSrv accrpc.AccountClient
|
|||
|
|
arcSrv *arcrpc.Service2
|
|||
|
|
articleSrv *artrpc.Service
|
|||
|
|
cache *fanout.Fanout
|
|||
|
|
thumbupClient thumbup.ThumbupClient
|
|||
|
|
|
|||
|
|
relationSvc *rlrpc.Service
|
|||
|
|
|
|||
|
|
// 特殊admin，针对大忽悠事件消息通知
|
|||
|
|
ads map[string]struct{}
|
|||
|
|
// 特殊稿件，针对大忽悠时间 不让删评论
|
|||
|
|
oids map[int64]int32
|
|||
|
|
|
|||
|
|
// mark folded or unmark folded worker
|
|||
|
|
marker *fanout.Fanout
|
|||
|
|
// del cache or add cache worker
|
|||
|
|
cacheOperater *fanout.Fanout
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// New new a service and return.
|
|||
|
|
func New(c *conf.Config) (s *Service) {
|
|||
|
|
s = &Service{
|
|||
|
|
conf: c,
|
|||
|
|
dao: dao.New(c),
|
|||
|
|
httpClient: bm.NewClient(c.HTTPClient),
|
|||
|
|
arcSrv: arcrpc.New2(c.RPCClient2.Archive),
|
|||
|
|
articleSrv: artrpc.New(c.RPCClient2.Article),
|
|||
|
|
cache: fanout.New("cache", fanout.Worker(1), fanout.Buffer(1024*10)),
|
|||
|
|
relationSvc: rlrpc.New(c.RPCClient2.Relation),
|
|||
|
|
marker: fanout.New("marker", fanout.Worker(1), fanout.Buffer(1024*10)),
|
|||
|
|
cacheOperater: fanout.New("cacheOp", fanout.Worker(1), fanout.Buffer(1024*10)),
|
|||
|
|
}
|
|||
|
|
accSrv, err := accrpc.NewClient(c.AccountClient)
|
|||
|
|
if err != nil {
|
|||
|
|
panic(err)
|
|||
|
|
}
|
|||
|
|
s.accSrv = accSrv
|
|||
|
|
cc, err := warden.NewConn(fmt.Sprintf("discovery://default/%s", thumbup.AppID))
|
|||
|
|
if err != nil {
|
|||
|
|
panic(err)
|
|||
|
|
}
|
|||
|
|
s.thumbupClient = thumbup.NewThumbupClient(cc)
|
|||
|
|
ads := make(map[string]struct{})
|
|||
|
|
for _, ID := range c.Reply.AdminName {
|
|||
|
|
ads[ID] = struct{}{}
|
|||
|
|
}
|
|||
|
|
s.ads = ads
|
|||
|
|
oids := make(map[int64]int32)
|
|||
|
|
for i, oid := range c.Reply.Oids {
|
|||
|
|
oids[oid] = c.Reply.Tps[i]
|
|||
|
|
}
|
|||
|
|
s.oids = oids
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Ping check service is ok.
|
|||
|
|
func (s *Service) Ping(c context.Context) (err error) {
|
|||
|
|
return s.dao.Ping(c)
|
|||
|
|
}
|