Merge pull request #120 from mattn/fix-goroutine-leak

fix possibly goroutine leak
fiatjaf · Jul 22, 2024 · 8d1bd80 · 8d1bd80
2 parents 3d04ee4 + 01d5005
commit 8d1bd80
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/handlers.go b/handlers.go
@@ -83,6 +83,9 @@ func (s *Server) doEvent(ctx context.Context, ws *WebSocket, request []json.RawM
 		// event deletion -- nip09
 		for _, tag := range evt.Tags {
 			if len(tag) >= 2 && tag[0] == "e" {
+				ctx, cancel := context.WithTimeout(ctx, time.Millisecond*200)
+				defer cancel()
+
 				// fetch event to be deleted
 				res, err := s.relay.Storage(ctx).QueryEvents(ctx, nostr.Filter{IDs: []string{tag[1]}})
 				if err != nil {
@@ -94,7 +97,7 @@ func (s *Server) doEvent(ctx context.Context, ws *WebSocket, request []json.RawM
 				exists := false
 				select {
 				case target, exists = <-res:
-				case <-time.After(time.Millisecond * 200):
+				case <-ctx.Done():
 				}
 				if !exists {
 					// this will happen if event is not in the database