pangolin_mirror/newt
1
0
Fork 0
mirror of https://github.com/fosrl/newt.git synced 2026-03-04 01:36:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Limit the amount of times the send message sends

Browse Source 
Fixes #115
This commit is contained in:
Owen
2025-09-01 11:53:46 -07:00
parent 0ba44206b1
commit 288413fd15
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
websocket/client.go
View File

@@ -175,19 +175,29 @@ func (c *Client) SendMessage(messageType string, data interface{}) error {
func (c *Client) SendMessageInterval(messageType string, data interface{}, interval time.Duration) (stop func()) { func (c *Client) SendMessageInterval(messageType string, data interface{}, interval time.Duration) (stop func()) {
stopChan := make(chan struct{}) stopChan := make(chan struct{})
go func() { go func() {
count := 0
maxAttempts := 10
err := c.SendMessage(messageType, data) // Send immediately err := c.SendMessage(messageType, data) // Send immediately
if err != nil { if err != nil {
logger.Error("Failed to send initial message: %v", err) logger.Error("Failed to send initial message: %v", err)
} }
count++
ticker := time.NewTicker(interval) ticker := time.NewTicker(interval)
defer ticker.Stop() defer ticker.Stop()
for { for {
select { select {
case <-ticker.C: case <-ticker.C:
if count >= maxAttempts {
logger.Info("SendMessageInterval timed out after %d attempts for message type: %s", maxAttempts, messageType)
return
}
err = c.SendMessage(messageType, data) err = c.SendMessage(messageType, data)
if err != nil { if err != nil {
logger.Error("Failed to send message: %v", err) logger.Error("Failed to send message: %v", err)
} }
count++
case <-stopChan: case <-stopChan:
return return
} }