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.
59 lines
1.3 KiB
59 lines
1.3 KiB
package service |
|
|
|
import ( |
|
"context" |
|
"sync" |
|
|
|
"go-common/app/interface/main/push/conf" |
|
"go-common/app/interface/main/push/dao" |
|
pushrpc "go-common/app/service/main/push/api/grpc/v1" |
|
pushmdl "go-common/app/service/main/push/model" |
|
"go-common/library/cache" |
|
httpx "go-common/library/net/http/blademaster" |
|
) |
|
|
|
// Service push service. |
|
type Service struct { |
|
c *conf.Config |
|
dao *dao.Dao |
|
cache *cache.Cache |
|
pushRPC pushrpc.PushClient |
|
callbackCh chan *pushmdl.Callback |
|
httpClient *httpx.Client |
|
waiter sync.WaitGroup |
|
closed bool |
|
} |
|
|
|
// New creates a push service instance. |
|
func New(c *conf.Config) *Service { |
|
s := &Service{ |
|
c: c, |
|
dao: dao.New(c), |
|
cache: cache.New(1, 10240), |
|
callbackCh: make(chan *pushmdl.Callback, c.Push.CallbackChanLen), |
|
httpClient: httpx.NewClient(c.HTTPClient), |
|
} |
|
var err error |
|
if s.pushRPC, err = pushrpc.NewClient(c.PushRPC); err != nil { |
|
panic(err) |
|
} |
|
for i := 0; i < s.c.Push.CallbackGoroutines; i++ { |
|
s.waiter.Add(1) |
|
go s.callbackproc() |
|
} |
|
return s |
|
} |
|
|
|
// Close closes service. |
|
func (s *Service) Close() { |
|
s.closed = true |
|
close(s.callbackCh) |
|
s.waiter.Wait() |
|
s.dao.Close() |
|
} |
|
|
|
// Ping checks service. |
|
func (s *Service) Ping(c context.Context) (err error) { |
|
err = s.dao.Ping(c) |
|
return |
|
}
|
|
|