aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat : design 新增 callback 模式
All checks were successful
git commit AiDA python develop 分支构建部署 / scheduled_deploy (push) Has been skipped
Details

Browse Source
This commit is contained in:
zcr
2025-12-16 11:59:28 +08:00
parent 832ca6fd05
commit fac1eab1bc
2 changed files with 5 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app/schemas/design.py
View File

@@ -10,6 +10,7 @@ class DesignStreamModel(BaseModel):
objects: list[dict] objects: list[dict]
process_id: str process_id: str
requestId: str requestId: str
callback_url: str
class DesignProgressModel(BaseModel): class DesignProgressModel(BaseModel):

20
app/service/design_fast/design_generate.py
View File

@@ -145,10 +145,11 @@ def design_generate(request_data):
@RunTime @RunTime
def design_generate_v2(request_data): def design_generate_v2(request_data):
objects_data = request_data.dict()['objects'] objects_data = request_data.dict()['objects']
callback_url = request_data.callback_url
request_id = request_data.requestId request_id = request_data.requestId
threads = [] threads = []
def process_object(step, object): def process_object(step, object, callback_url):
basic = object['basic'] basic = object['basic']
items_response = { items_response = {
'layers': [], 'layers': [],
@@ -211,12 +212,8 @@ def design_generate_v2(request_data):
}) })
items_response['synthesis_url'] = synthesis_single(item_result['front_image'], item_result['back_image']) items_response['synthesis_url'] = synthesis_single(item_result['front_image'], item_result['back_image'])
# 发送结果给java端 # 发送结果给java端
url = JAVA_STREAM_API_URL url = callback_url
xu_pei_test_url = "https://137f6b5c3490.ngrok-free.app/api/third/party/receiveDesignResults"
tianxaing_test_url = "https://d7436c137fb9.ngrok-free.app/api/third/party/receiveDesignResults"
logger.info(f"java 回调 -> {url}") logger.info(f"java 回调 -> {url}")
logger.info(f"xupei java 回调 -> {xu_pei_test_url}")
logger.info(f"tianxiang java 回调 -> {tianxaing_test_url}")
headers = { headers = {
'Accept': "*/*", 'Accept': "*/*",
@@ -231,17 +228,8 @@ def design_generate_v2(request_data):
# 打印结果 # 打印结果
logger.info(response.text) logger.info(response.text)
test_xp_response = post_request(xu_pei_test_url, json_data=items_response, headers=headers)
test_response = post_request(tianxaing_test_url, json_data=items_response, headers=headers)
if test_response:
# 打印结果
logger.info(f"tianxiang test response : {test_response.text}")
if test_xp_response:
logger.info(f"xupei test response : {test_xp_response.text}")
for step, object in enumerate(objects_data): for step, object in enumerate(objects_data):
t = threading.Thread(target=process_object, args=(step, object)) t = threading.Thread(target=process_object, args=(step, object, callback_url))
threads.append(t) threads.append(t)
t.start() t.start()