Use envvar TITAN_GATEWAY_ID to name each gateway workers

This commit is contained in:
Jeremy Zhang
2019-10-30 21:59:07 -07:00
parent 5c446935bc
commit 9002841933
2 changed files with 11 additions and 1 deletions

View File

@ -5,6 +5,7 @@ import functools
from flask import request, session
import time
import json
import os
class Gateway(Namespace):
def teardown_db_session(self):
@ -12,7 +13,8 @@ class Gateway(Namespace):
db.session.remove()
def on_connect(self):
emit('hello')
gateway_identifier = os.environ.get("TITAN_GATEWAY_ID", None)
emit('hello', {"gateway_identifier": gateway_identifier})
def on_identify(self, data):
guild_id = data["guild_id"]