From 17d495c73a431e46851ccc77c5718c95d00a7ba6 Mon Sep 17 00:00:00 2001 From: HuangHai <10402852@qq.com> Date: Fri, 28 Feb 2025 15:11:00 +0800 Subject: [PATCH] 'commit' --- .../Config/__pycache__/Config.cpython-310.pyc | Bin 217 -> 349 bytes .../__pycache__/__init__.cpython-310.pyc | Bin 143 -> 143 bytes AI/AiService/Model/TaskModel.py | 30 ++++++++++++++++++ AI/AiService/Model/__init__.py | 0 .../__pycache__/TaskModel.cpython-310.pyc | Bin 0 -> 1305 bytes .../__pycache__/__init__.cpython-310.pyc | Bin 0 -> 142 bytes AI/AiService/Start.py | 28 +++------------- .../__pycache__/__init__.cpython-310.pyc | Bin 0 -> 136 bytes 8 files changed, 34 insertions(+), 24 deletions(-) create mode 100644 AI/AiService/Model/TaskModel.py create mode 100644 AI/AiService/Model/__init__.py create mode 100644 AI/AiService/Model/__pycache__/TaskModel.cpython-310.pyc create mode 100644 AI/AiService/Model/__pycache__/__init__.cpython-310.pyc create mode 100644 AI/AiService/__pycache__/__init__.cpython-310.pyc diff --git a/AI/AiService/Config/__pycache__/Config.cpython-310.pyc b/AI/AiService/Config/__pycache__/Config.cpython-310.pyc index d0a94ccbe37e7740b7d1b41bdfd6153867573f49..0b1451a52c09582d0a9b9e4b08e260171c5da22e 100644 GIT binary patch delta 229 zcmcb~c$dj1pO=@50SI=+9ZY9oWMFs<;vfSyAjbiSi)DaB3PTh_3S$&w3R4Pm3QG!W z3R?>M9M&kN6pmm9P0oon^08Gsh6Z{dWTIzeXpm_X#=vlkr6@nYBuc=gI5Q@PvADQAzbNGvM@nKzVp3vp zs-GtF#5gtaTRgsz!GS*U&i;OGp6)9dia^dU;+oj&A|e1}u`seQGO;i*LLk>~0BnOc AIsgCw delta 98 zcmcc1bd%90pO=@50SH*#52i-~>Bk@rGGGL99Dul(2S}taL@}f=Mlq%^1v6+ePxO&z g_S0mVctVYHB|{N2P`-$HGApAC2MdtN!pOl00My$J<^TWy diff --git a/AI/AiService/Config/__pycache__/__init__.cpython-310.pyc b/AI/AiService/Config/__pycache__/__init__.cpython-310.pyc index 372515ba9df64e5a0bbb49d0581c6635708cd0fa..f1bce1e7ffbf425379ebde3e06c9ff8a0d24a7cd 100644 GIT binary patch delta 19 YcmeBY>}TZ8=jG*M0D_prgA=)1045m(T>t<8 delta 19 YcmeBY>}TZ8=jG*M0D^ddJrlWG03?D03IG5A diff --git a/AI/AiService/Model/TaskModel.py b/AI/AiService/Model/TaskModel.py new file mode 100644 index 00000000..195e2309 --- /dev/null +++ b/AI/AiService/Model/TaskModel.py @@ -0,0 +1,30 @@ +import mysql.connector +from AiService.Config.Config import * + +class TaskModel: + def __init__(self): + # 初始化 MySQL 连接 + self.connection = mysql.connector.connect(**MYSQL_CONFIG) + self.cursor = self.connection.cursor() + + def insert_task(self, task_id: str, keyword: str): + """ + 插入任务记录到 t_gen_tasks 表 + """ + try: + query = """ + INSERT INTO t_gen_tasks (task_id, keyword, status) + VALUES (%s, %s, %s) + """ + self.cursor.execute(query, (task_id, keyword, 'pending')) + self.connection.commit() + except Exception as e: + print(f"Failed to insert task: {str(e)}") + raise + + def close(self): + """ + 关闭 MySQL 连接 + """ + self.cursor.close() + self.connection.close() diff --git a/AI/AiService/Model/__init__.py b/AI/AiService/Model/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/AI/AiService/Model/__pycache__/TaskModel.cpython-310.pyc b/AI/AiService/Model/__pycache__/TaskModel.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..f369f80748d882880e43219e4982e8fcd8b245a3 GIT binary patch literal 1305 zcmZt`U29ZFbY||{{W2dCBDQZAso+Z3hrSd<#1Nwdk`>)m>9TNH?o5`O+`TuMnKkaR zk+z8-K3FJ}J`^Pj74)G+=%bY4-h@JjknG3 z9YWsWAw~Bi$dPi9T6!@+@;Gp~ zT!o`cfMjZ@#sn2?pMbg}x!{tobK#22KCihVD{{ceh&&iB*nG92H`fv&qkw|`7?4K8 zOK|j8fDGB78(=mh#`%ptr9I;F9@*drZl3_{3IgU>`4uTe!s`hjD0p-^B#b);v2+yK_fEV*#5rPys&_HoGZ1 zJBbnjwr32ekIhUAaTkd0yr8=dq)rZK&*{+~6Kvh-GYVzcUr5*E8d0kdchl>lYE=qlCgkJbkAOIgpm}UNxU%DQW+0DT=L4rtja?=HY*>-80TUrTh zj-CU=IH&XcG@YSyl+oOG?4RzL>Ug+I$29&o=nS;e{(b?(u(R03w4W5IbO~AeJg_H~ z8a(>*<1fE^YaN(p@9^*EAD+IL@XAGJn7a2m8=iDR3w(xlXOBWo zb)l3v%yFY7ecu**zmg`kf_Q;F=^*+sh(HF6K#l_t7qb9~6oz01O-8?!3`HPe1o6wz#VRJHI6S{7 zJ0>tQFWo0UFFnT5GsZD9IJKxOGdVTJH$NpcCni2VGcU6wK3=b&@)n0pZhlH>PO2Tq LsA47{!NLFl)^Hy6 literal 0 HcmV?d00001 diff --git a/AI/AiService/Start.py b/AI/AiService/Start.py index 0794aea8..b6934169 100644 --- a/AI/AiService/Start.py +++ b/AI/AiService/Start.py @@ -5,8 +5,7 @@ from kafka import KafkaProducer, KafkaAdminClient from kafka.admin import NewTopic import uuid import json -import mysql.connector -from Config.Config import KAFKA_BOOTSTRAP_SERVERS, KAFKA_TOPIC, MYSQL_CONFIG +from AiService.Model.TaskModel import * # 定义请求体模型 class TaskRequest(BaseModel): @@ -56,27 +55,6 @@ async def lifespan(app: FastAPI): # 应用关闭时清理资源（可选） print("Application shutdown.") -# 写入 MySQL 日志 -def log_to_mysql(task_id: str, keyword: str): - try: - # 连接 MySQL - connection = mysql.connector.connect(**MYSQL_CONFIG) - cursor = connection.cursor() - - # 插入日志 - query = """ - INSERT INTO t_gen_tasks (task_id, keyword, status) - VALUES (%s, %s, %s) - """ - cursor.execute(query, (task_id, keyword, 'pending')) - connection.commit() - - # 关闭连接 - cursor.close() - connection.close() - except Exception as e: - print(f"Failed to log to MySQL: {str(e)}") - # 定义接口 @app.post("/create-task") def create_task(task_request: TaskRequest): @@ -95,7 +73,9 @@ def create_task(task_request: TaskRequest): producer.flush() # 确保消息发送完成 # 将任务信息写入 MySQL - log_to_mysql(task_id, task_request.prompt) + tm = TaskModel() + tm.insert_task(task_id, task_request.prompt) + tm.close() # 返回 task_id return {"task_id": task_id} diff --git a/AI/AiService/__pycache__/__init__.cpython-310.pyc b/AI/AiService/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..00007bb72c197d86cf15f6cdbfeb4a31d7317ede GIT binary patch literal 136 zcmd1j<>g`kf_Q;F=^*+sh(HF6K#l_t7qb9~6oz01O-8?!3`HPe1o2DT#VRJHI6S{7 zJ0>tQFWo0UFFnT5GsZD9IJKxOGdVRTK0Y%qvm`!Vub}c4hfQvNN@-529mtSkCLqDW F000X>9HIaK literal 0 HcmV?d00001