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

feat design stream 部署

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-12-01 20:17:39 +08:00
parent 0f17d8b654
commit b7736553ff
2 changed files with 2 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2,6 +2,7 @@ import logging
import threading import threading
import time import time
import requests
from minio import Minio from minio import Minio
from app.core.config import * from app.core.config import *
@@ -138,16 +139,9 @@ 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']
# process_id = request_data.dict()['process_id']
# object_response = {}
threads = [] threads = []
active_threads = 0
lock = threading.Lock()
# total = len(objects_data)
def process_object(step, object): def process_object(step, object):
nonlocal active_threads
basic = object['basic'] basic = object['basic']
items_response = { items_response = {
'layers': [], 'layers': [],
@@ -219,26 +213,10 @@ def design_generate_v2(request_data):
logger.info(response.text) logger.info(response.text)
logger.info(items_response) logger.info(items_response)
# update_progress(process_id, total)
# with lock:
# object_response[step] = items_response
# active_threads -= 1
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))
threads.append(t) threads.append(t)
t.start() t.start()
# with lock:
# active_threads += 1
# for t in threads:
# t.join()
# final_progress(process_id)
# return object_response
import requests
def post_request(url, data=None, json_data=None, headers=None, auth=None, timeout=5): def post_request(url, data=None, json_data=None, headers=None, auth=None, timeout=5):

2
app/service/utils/new_oss_client.py
View File

@@ -82,7 +82,7 @@ if __name__ == '__main__':
# url = "aida-users/89/sketchboard/female/Dress/e6724ab7-8d3f-4677-abe0-c3e42ab7af85.jpeg" # url = "aida-users/89/sketchboard/female/Dress/e6724ab7-8d3f-4677-abe0-c3e42ab7af85.jpeg"
# url = "aida-users/87/print/956614a2-7e75-4fbe-9ed0-c1831e37a2c9-4-87.png" # url = "aida-users/87/print/956614a2-7e75-4fbe-9ed0-c1831e37a2c9-4-87.png"
# url = "aida-users/89/single_logo/123-89.png" # url = "aida-users/89/single_logo/123-89.png"
url ="aida-users/89/test/123-89.png" url = "aida-users/89/test/123-89.png"
# url = "aida-collection-element/12148/Sketchboard/95ea577b-305b-4a62-b30a-39c0dd3ddb3f.png" # url = "aida-collection-element/12148/Sketchboard/95ea577b-305b-4a62-b30a-39c0dd3ddb3f.png"
read_type = "2" read_type = "2"