Yadciel/Titan
1
0
Fork 0
mirror of https://github.com/TitanEmbeds/Titan.git synced 2024-12-24 14:07:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

Use envvar TITAN_GATEWAY_ID to name each gateway workers

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
webapp/titanembeds/blueprints/gateway/gateway.py
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"]

8
webapp/titanembeds/static/js/embed.js
View File

@ -2122,6 +2122,14 @@
socket.emit('identify', {"guild_id": guild_id, "visitor_mode": visitor_mode});
process_message_users_cache();
});
socket.on('hello', function (msg) {
var gateway_identifier = msg.gateway_identifier;
if (!gateway_identifier) {
gateway_identifier = "null";
}
console.log("%c[TitanEmbeds]%cConnected to gateway via%c" + gateway_identifier, 'color:aqua;background-color:black;border:1px solid black;padding: 3px;', 'color:white;background-color:black;border:1px solid black;padding: 3px;', 'color:white;background-color:black;border:1px solid black;font-family: Courier;padding: 3px;');
});
socket.on("disconnect", function () {