coopgo-platform/gorush
7
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

refactor(worker): remove context from notification structure (#503)

Browse Source
This commit is contained in:
Bo-Yi Wu
2020-04-25 15:23:43 +08:00
committed by GitHub
parent 685a87c930
commit 7dbb5c98e7
3 changed files with 9 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
rpc/server.go
View File

@@ -50,8 +50,6 @@ func (s *Server) Check(ctx context.Context, in *proto.HealthCheckRequest) (*prot
func (s *Server) Send(ctx context.Context, in *proto.NotificationRequest) (*proto.NotificationReply, error) {
var badge = int(in.Badge)
notification := gorush.PushNotification{
Ctx: ctx,
Platform: int(in.Platform),
Tokens: in.Tokens,
Message: in.Message,
@@ -92,7 +90,7 @@ func (s *Server) Send(ctx context.Context, in *proto.NotificationRequest) (*prot
}
}
go gorush.SendNotification(notification)
go gorush.SendNotification(ctx, notification)
return &proto.NotificationReply{
Success: true,
@@ -125,6 +123,7 @@ func RunGRPCServer(ctx context.Context) error {
select {
case <-ctx.Done():
s.GracefulStop() // graceful shutdown
gorush.LogAccess.Info("shutdown the gRPC server")
}
}()
if err = s.Serve(lis); err != nil {