Fix: notify closed client

cmd/api/handler/websocket/client.go

@@ -7,6 +7,7 @@ import (
 	"context"
 	"io"
 	"net"
+	"sync/atomic"
 	"time"
 
 	"github.com/dipdup-io/workerpool"
@@ -43,14 +44,19 @@ type Client struct {
 	filters *Filters
 	ch      chan any
 	g       workerpool.Group
+
+	closed *atomic.Bool
 }
 
 func newClient(id uint64, manager *Manager) *Client {
+	closed := new(atomic.Bool)
+	closed.Store(false)
 	return &Client{
 		id:      id,
 		manager: manager,
 		ch:      make(chan any, 1024),
 		g:       workerpool.NewGroup(),
+		closed:  closed,
 	}
 }
 
@@ -102,11 +108,15 @@ func (c *Client) DetachFilters(msg Unsubscribe) error {
 }
 
 func (c *Client) Notify(msg any) {
+	if c.closed.Load() {
+		return
+	}
 	c.ch <- msg
 }
 
 func (c *Client) Close() error {
 	c.g.Wait()
+	c.closed.Store(true)
 	close(c.ch)
 	return nil
 }

cmd/api/handler/websocket/client_test.go

@@ -0,0 +1,14 @@
+package websocket
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestNotifyClosedClient(t *testing.T) {
+	client := newClient(10, nil)
+	err := client.Close()
+	require.NoError(t, err, "closing client")
+	client.Notify("test")
+}

cmd/api/init.go

@@ -288,7 +288,7 @@ func initHandlers(ctx context.Context, e *echo.Echo, cfg Config, db postgres.Sto
 
 	v1.GET("/swagger/*", echoSwagger.WrapHandler)
 
-	// initWebsocket(ctx, db, v1)
+	initWebsocket(ctx, db, v1)
 
 	log.Info().Msg("API routes:")
 	for _, route := range e.Routes() {
@@ -300,8 +300,8 @@ var (
 	wsManager *websocket.Manager
 )
 
-// func initWebsocket(ctx context.Context, db postgres.Storage, group *echo.Group) {
-// 	wsManager = websocket.NewManager(db, db.Blocks, db.Tx)
-// 	wsManager.Start(ctx)
-// 	group.GET("/ws", wsManager.Handle)
-// }
+func initWebsocket(ctx context.Context, db postgres.Storage, group *echo.Group) {
+	wsManager = websocket.NewManager(db, db.Blocks, db.Tx)
+	wsManager.Start(ctx)
+	group.GET("/ws", wsManager.Handle)
+}