@@ -16,15 +16,18 @@ import (
16
16
"github.com/frain-dev/convoy"
17
17
"github.com/frain-dev/convoy/internal/pkg/socket"
18
18
convoyNet "github.com/frain-dev/convoy/net"
19
+ "github.com/frain-dev/convoy/pkg/log"
19
20
"github.com/frain-dev/convoy/util"
20
21
"github.com/gorilla/websocket"
21
- log "github.com/sirupsen/logrus"
22
+
22
23
"github.com/spf13/cobra"
23
24
"gopkg.in/yaml.v3"
24
25
)
25
26
26
- var done chan interface {}
27
- var interrupt chan os.Signal
27
+ var (
28
+ done chan interface {}
29
+ interrupt chan os.Signal
30
+ )
28
31
29
32
const (
30
33
// Time allowed to write a message to the server.
@@ -103,7 +106,6 @@ func addListenCommand(a *app) *cobra.Command {
103
106
"Authorization" : []string {"Bearer " + c .ActiveApiKey },
104
107
"Body" : []string {string (body )},
105
108
})
106
-
107
109
if err != nil {
108
110
if response != nil {
109
111
buf , e := io .ReadAll (response .Body )
@@ -131,7 +133,7 @@ func addListenCommand(a *app) *cobra.Command {
131
133
// Send a message to the server to resend unsuccessful events to the device
132
134
err := conn .WriteMessage (websocket .TextMessage , []byte (since ))
133
135
if err != nil {
134
- log .Println ( "an error occured sending 'since' message" , err )
136
+ log .WithError ( err ). Errorln ( "an error occurred sending 'since' message" )
135
137
}
136
138
}
137
139
@@ -142,7 +144,7 @@ func addListenCommand(a *app) *cobra.Command {
142
144
case <- ticker .C :
143
145
err := conn .SetWriteDeadline (time .Now ().Add (writeWait ))
144
146
if err != nil {
145
- log .Println (err )
147
+ log .WithError (err ). Errorln ( "failed to set write deadline" )
146
148
}
147
149
148
150
if err := conn .WriteMessage (websocket .PingMessage , nil ); err != nil {
@@ -159,14 +161,14 @@ func addListenCommand(a *app) *cobra.Command {
159
161
// Send a message to set the device to offline
160
162
err := conn .WriteMessage (websocket .TextMessage , []byte ("disconnect" ))
161
163
if err != nil {
162
- log .Println ( "Error during closing websocket:" , err )
164
+ log .WithError ( err ). Errorln ( "error during closing websocket" )
163
165
return
164
166
}
165
167
166
168
// Close our websocket connection
167
169
err = conn .WriteMessage (websocket .CloseMessage , websocket .FormatCloseMessage (websocket .CloseNormalClosure , "" ))
168
170
if err != nil {
169
- log .Println ( "Error during closing websocket:" , err )
171
+ log .WithError ( err ). Errorln ( "error during closing websocket" )
170
172
return
171
173
}
172
174
0 commit comments