feat(websocket): add deduplication support to PeriodicWrite function and introduce DeepEqual utility

This commit is contained in:
yusing
2025-09-13 22:37:51 +08:00
parent a483e15a20
commit 99c1922342
3 changed files with 285 additions and 4 deletions

View File

@@ -7,14 +7,16 @@ import (
apitypes "github.com/yusing/go-proxy/internal/api/types"
)
func PeriodicWrite(c *gin.Context, interval time.Duration, get func() (any, error)) {
type DeduplicateFunc func(last, current any) bool
func PeriodicWrite(c *gin.Context, interval time.Duration, get func() (any, error), deduplicate ...DeduplicateFunc) {
manager, err := NewManagerWithUpgrade(c)
if err != nil {
c.Error(apitypes.InternalServerError(err, "failed to upgrade to websocket"))
return
}
defer manager.Close()
err = manager.PeriodicWrite(interval, get)
err = manager.PeriodicWrite(interval, get, deduplicate...)
if err != nil {
c.Error(apitypes.InternalServerError(err, "failed to write to websocket"))
}