Skip to content
This repository was archived by the owner on Dec 17, 2025. It is now read-only.

Commit e4da4d0

Browse files
authored
fix: fix client 'retry' tag (#16)
* fix: fix client 'retry' tag * feat: optimize client_test
1 parent 800cc5a commit e4da4d0

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

client.go

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,9 @@ import (
2020
"bytes"
2121
"context"
2222
"encoding/base64"
23-
"encoding/binary"
2423
"fmt"
2524
"io"
25+
"strconv"
2626
"sync/atomic"
2727

2828
"github.com/cloudwego/hertz/pkg/network/standard"
@@ -288,7 +288,10 @@ func (c *Client) processEvent(msg []byte) (event *Event, err error) {
288288
case bytes.HasPrefix(line, headerEvent):
289289
e.Event = string(append([]byte(nil), trimHeader(len(headerEvent), line)...))
290290
case bytes.HasPrefix(line, headerRetry):
291-
e.Retry = binary.BigEndian.Uint64(append([]byte(nil), trimHeader(len(headerRetry), line)...))
291+
e.Retry, err = strconv.ParseUint(b2s(append([]byte(nil), trimHeader(len(headerRetry), line)...)), 10, 64)
292+
if err != nil {
293+
return nil, fmt.Errorf("process message `retry` failed, err is %s", err)
294+
}
292295
default:
293296
// Ignore any garbage that doesn't match what we're looking for.
294297
}

client_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ func publishMsgs(s *Stream, empty bool, count int) {
148148
if empty {
149149
s.Publish(&Event{Data: []byte("\n")})
150150
} else {
151-
s.Publish(&Event{Data: []byte("ping")})
151+
s.Publish(&Event{Data: []byte("ping"), Retry: uint64(3000), Event: "test", ID: "1111"})
152152
}
153153
}
154154
}

0 commit comments

Comments
 (0)