From 3f4ed4dc3c7a02436018fc45ad8e4bbeb29fd121 Mon Sep 17 00:00:00 2001 From: RockChinQ <1010553892@qq.com> Date: Thu, 28 Sep 2023 12:16:30 +0000 Subject: [PATCH] =?UTF-8?q?fix:=20=E9=98=BB=E5=A1=9E=E5=9C=B0=E5=8F=91?= =?UTF-8?q?=E9=80=81=E5=AE=A1=E8=AE=A1=E6=8A=A5=E5=91=8A=E6=95=B0=E6=8D=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/audit/gatherer.py | 22 ++++++++++++++-------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/pkg/audit/gatherer.py b/pkg/audit/gatherer.py index ceef49e2..699bbaee 100644 --- a/pkg/audit/gatherer.py +++ b/pkg/audit/gatherer.py @@ -5,6 +5,7 @@ import hashlib import json import logging +import threading import requests @@ -42,15 +43,20 @@ class DataGatherer: 只会报告此次请求的使用量，不会报告总量。 不包含除版本号、使用类型、使用量以外的任何信息，仅供开发者分析使用情况。 """ - try: - config = pkg.utils.context.get_config() - if not config.report_usage: + + def thread_func(): + + try: + config = pkg.utils.context.get_config() + if not config.report_usage: + return + res = requests.get("http://reports.rockchin.top:18989/usage?service_name=qchatgpt.{}&version={}&count={}&msg_source={}".format(subservice_name, self.version_str, count, config.msg_source_adapter)) + if res.status_code != 200 or res.text != "ok": + logging.warning("report to server failed, status_code: {}, text: {}".format(res.status_code, res.text)) + except: return - res = requests.get("http://reports.rockchin.top:18989/usage?service_name=qchatgpt.{}&version={}&count={}&msg_source={}".format(subservice_name, self.version_str, count, config.msg_source_adapter)) - if res.status_code != 200 or res.text != "ok": - logging.warning("report to server failed, status_code: {}, text: {}".format(res.status_code, res.text)) - except: - return + + threading.Thread(target=thread_func).start() def get_usage(self, key_md5): return self.usage[key_md5] if key_md5 in self.usage else {}