From fbae6faadf3e119402ae281c294597120e68e80c Mon Sep 17 00:00:00 2001 From: Kay Date: Fri, 10 Jan 2025 15:01:19 +0000 Subject: [PATCH] feat(ws): white/black list pubkey. --- delivery/websocket/event_handler.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/delivery/websocket/event_handler.go b/delivery/websocket/event_handler.go index 1d269a9..4373691 100644 --- a/delivery/websocket/event_handler.go +++ b/delivery/websocket/event_handler.go @@ -60,7 +60,7 @@ func (s *Server) handleEvent(conn *websocket.Conn, m message.Message) { bloomCheckCmd := pipe.BFExists(qCtx, s.redis.BloomFilterName, eID[:]) - var whiteListCheckCmd *gredis.BoolCmd = nil + var whiteListCheckCmd *gredis.BoolCmd if s.config.Limitation.RestrictedWrites { whiteListCheckCmd = pipe.CFExists(qCtx, s.redis.WhiteListFilterName, msg.Event.PublicKey) @@ -208,7 +208,7 @@ func (s *Server) handleEvent(conn *websocket.Conn, m message.Message) { } } -func checkLimitations(c clientState, redis *redis.Redis, +func checkLimitations(c clientState, r *redis.Redis, limits Limitation, msg message.Event) (accepted bool, isAuthFail bool, failType string, resp string, ) { @@ -233,7 +233,7 @@ func checkLimitations(c clientState, redis *redis.Redis, time.Unix(expiration, 0).String()) } - if err := redis.AddDelayedTask(expirationTaskListName, + if err := r.AddDelayedTask(expirationTaskListName, fmt.Sprintf("%s:%d", msg.Event.ID, msg.Event.Kind), time.Until(time.Unix(expiration, 0))); err != nil { return false, false, serverFail, "error: can't add event to expiration queue." }