aljaz/CTFd
1
0
Fork 0
mirror of https://github.com/aljazceru/CTFd.git synced 2025-12-19 06:54:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

reduce ping events (Fix #1685) (#1699)

Browse Source 
Closes #1685
This commit is contained in:
Frank
2020-11-19 09:27:48 +08:00
committed by GitHub
parent 44a588869e
commit 1883c62cf5
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
CTFd/utils/events/__init__.py
View File

@@ -53,13 +53,12 @@ class EventManager(object):
def subscribe(self, channel="ctf"): def subscribe(self, channel="ctf"):
q = defaultdict(Queue) q = defaultdict(Queue)
self.clients[id(q)] = q self.clients[id(q)] = q
try:
while True:
try: try:
# Immediately yield a ping event to force Response headers to be set # Immediately yield a ping event to force Response headers to be set
# or else some reverse proxies will incorrectly buffer SSE # or else some reverse proxies will incorrectly buffer SSE
yield ServerSentEvent(data="", type="ping") yield ServerSentEvent(data="", type="ping")
while True:
try:
with Timeout(5): with Timeout(5):
message = q[channel].get() message = q[channel].get()
yield ServerSentEvent(**message) yield ServerSentEvent(**message)
@@ -105,13 +104,12 @@ class RedisEventManager(EventManager):
def subscribe(self, channel="ctf"): def subscribe(self, channel="ctf"):
q = defaultdict(Queue) q = defaultdict(Queue)
self.clients[id(q)] = q self.clients[id(q)] = q
try:
while True:
try: try:
# Immediately yield a ping event to force Response headers to be set # Immediately yield a ping event to force Response headers to be set
# or else some reverse proxies will incorrectly buffer SSE # or else some reverse proxies will incorrectly buffer SSE
yield ServerSentEvent(data="", type="ping") yield ServerSentEvent(data="", type="ping")
while True:
try:
with Timeout(5): with Timeout(5):
message = q[channel].get() message = q[channel].get()
yield ServerSentEvent(**message) yield ServerSentEvent(**message)