admin
2024-01-11 a674a57120c3530151aa7b79d843a84ffb703e1d
middle_server.py
@@ -1,3 +1,4 @@
import builtins
import hashlib
import json
import logging
@@ -14,6 +15,7 @@
from db import mysql_data
from db.redis_manager import RedisUtils, RedisManager
from log import logger_debug, logger_request_debug
from output import push_msg_manager
from utils import socket_util, kpl_api_util, hosting_api_util, kp_client_msg_manager, global_data_cache_util, tool
from utils.juejin_util import JueJinHttpApi
@@ -159,15 +161,18 @@
                                    method = getattr(RedisUtils, cmd)
                                    args_ = [redis, key]
                                    if args is not None:
                                        if type(args) == tuple or type(args) == list:
                                        if builtins.type(args) == tuple or builtins.type(args) == list:
                                            args = list(args)
                                            if cmd == "setex":
                                                args_.append(json.dumps(args))
                                            else:
                                            for a in args:
                                                args_.append(a)
                                        else:
                                            args_.append(args)
                                    args_ = tuple(args_)
                                    result = method(*args_)
                                    if type(result) == set:
                                    if builtins.type(result) == set:
                                        result = list(result)
                                    result_str = json.dumps({"code": 0, "data": result})
                                elif ctype == "cmds":
@@ -182,15 +187,18 @@
                                        method = getattr(RedisUtils, cmd)
                                        args_ = [redis, key]
                                        if args is not None:
                                            if type(args) == tuple or type(args) == list:
                                            if builtins.type(args) == tuple or builtins.type(args) == list:
                                                args = list(args)
                                                if cmd == "setex":
                                                    args_.append(json.dumps(args))
                                                else:
                                                for a in args:
                                                    args_.append(a)
                                            else:
                                                args_.append(args)
                                        args_ = tuple(args_)
                                        result = method(*args_)
                                        if type(result) == set:
                                        if builtins.type(result) == set:
                                            result = list(result)
                                        result_list.append(result)
                                    result_str = json.dumps({"code": 0, "data": result_list})
@@ -212,7 +220,7 @@
                                method = getattr(mysql, cmd)
                                args_ = []
                                if args:
                                    if type(args) == tuple or type(args) == list:
                                    if builtins.type(args) == tuple or builtins.type(args) == list:
                                        args_ = list(args)
                                    else:
                                        args_.append(args)
@@ -261,6 +269,15 @@
                            result_str = json.dumps({"code": 0, "data": {}})
                            sk.sendall(socket_util.load_header(result_str.encode(encoding='utf-8')))
                            pass
                        elif data_json["type"] == "push_msg":
                            data = data_json["data"]
                            data = data["data"]
                            _type = data["type"]
                            data = data.get("data")
                            logger_debug.info(f"推送消息:{data_json}")
                            push_msg_manager.__push_msg(_type, data)
                            result_str = json.dumps({"code": 0, "data": {}})
                            sk.sendall(socket_util.load_header(result_str.encode(encoding='utf-8')))
                    finally:
                        log.request_info("middle_server", f"请求结束")
@@ -311,4 +328,4 @@
if __name__ == "__main__":
    pass
    print(builtins.type("")==str)