highperfocused/go-nostr
1
0
Fork 0
mirror of https://github.com/nbd-wtf/go-nostr.git synced 2025-11-15 08:37:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

problem: connection can be nil when writing and we're panic'ing.

Browse Source
This commit is contained in:
fiatjaf
2024-06-21 09:32:53 -03:00
parent 3862333dfa
commit a35cdff6c8
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
connection.go
View File

@@ -105,7 +105,13 @@ func NewConnection(ctx context.Context, url string, requestHeader http.Header, t
}, nil
}
func (c *Connection) WriteMessage(data []byte) error {
func (c *Connection) WriteMessage(ctx context.Context, data []byte) error {
select {
case <-ctx.Done():
return errors.New("context canceled")
default:
}
if c.msgStateW.IsCompressed() && c.enableCompression {
c.flateWriter.Reset(c.writer)
if _, err := io.Copy(c.flateWriter, bytes.NewReader(data)); err != nil {