From fce42e455793da9cc1e2ac7495e06fabf0567bbb Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Sun, 17 Nov 2024 20:39:04 +0800
Subject: [PATCH] :loud_sound: Add log in pushing websocket command

---
 pkg/internal/services/websocket.go | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/pkg/internal/services/websocket.go b/pkg/internal/services/websocket.go
index ddc0072..2f6fd01 100644
--- a/pkg/internal/services/websocket.go
+++ b/pkg/internal/services/websocket.go
@@ -3,6 +3,7 @@ package services
 import (
 	"context"
 	"git.solsynth.dev/hypernet/nexus/pkg/nex"
+	"github.com/rs/zerolog/log"
 	"time"
 
 	"github.com/samber/lo"
@@ -16,10 +17,14 @@ func PushCommand(userId uint, task nex.WebSocketPackage) {
 	defer cancel()
 
 	pc := gap.Nx.GetNexusGrpcConn()
-	_, _ = proto.NewStreamServiceClient(pc).PushStream(ctx, &proto.PushStreamRequest{
+	_, err := proto.NewStreamServiceClient(pc).PushStream(ctx, &proto.PushStreamRequest{
 		UserId: lo.ToPtr(uint64(userId)),
 		Body:   task.Marshal(),
 	})
+
+	if err != nil {
+		log.Warn().Err(err).Msg("Failed to push websocket command to nexus...")
+	}
 }
 
 func PushCommandBatch(userId []uint64, task nex.WebSocketPackage) {
@@ -27,8 +32,12 @@ func PushCommandBatch(userId []uint64, task nex.WebSocketPackage) {
 	defer cancel()
 
 	pc := gap.Nx.GetNexusGrpcConn()
-	_, _ = proto.NewStreamServiceClient(pc).PushStreamBatch(ctx, &proto.PushStreamBatchRequest{
+	_, err := proto.NewStreamServiceClient(pc).PushStreamBatch(ctx, &proto.PushStreamBatchRequest{
 		UserId: userId,
 		Body:   task.Marshal(),
 	})
+
+	if err != nil {
+		log.Warn().Err(err).Msg("Failed to push websocket command to nexus in batches...")
+	}
 }