aljaz/CTFd
1
0
Fork 0
mirror of https://github.com/aljazceru/CTFd.git synced 2025-12-18 22:44:24 +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

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

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