fix: empty subscribe message

LilithGames · Jan 29, 2023 · 15262c5 · 15262c5
1 parent f26bbb7
commit 15262c5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/agent/subscriber.go b/pkg/agent/subscriber.go
@@ -137,12 +137,18 @@ func (g *GqlSubscriber) GqlSubscriberWrapHandler(name string, tick Ticker) Messa
 			log.Error("receive event data error", zap.Error(err))
 			return err
 		}
+		if message == nil {
+			return fmt.Errorf("receive raw message nil")
+		}
 		var rawMsg Message
 		err = json.Unmarshal(*message, &rawMsg)
 		if err != nil {
 			log.Error("unmarshal message error", zap.Error(err))
 			return err
 		}
+		if rawMsg.Data == nil {
+			return fmt.Errorf("receive raw message.data nil: %s", string(*message))
+		}
 		if g.reply != nil {
 			err = json.Unmarshal(*rawMsg.Data, g.reply)
 			if err != nil {