PowLu/RuoYi-Vue3-FastAPI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf: 优化通用模块service层及异常处理

Browse Source
This commit is contained in:
insistence
2024-07-10 21:14:41 +08:00
parent bbd4a4cd95
commit f255595b30
2 changed files with 21 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
ruoyi-fastapi-backend/module_admin/controller/common_controller.py
View File

@@ -10,44 +10,23 @@ commonController = APIRouter(prefix='/common', dependencies=[Depends(LoginServic
@commonController.post("/upload")
async def common_upload(request: Request, file: UploadFile = File(...)):
try:
upload_result = await CommonService.upload_service(request, file)
if upload_result.is_success:
logger.info('上传成功')
return ResponseUtil.success(model_content=upload_result.result)
else:
logger.warning('上传失败')
return ResponseUtil.failure(msg=upload_result.message)
except Exception as e:
logger.exception(e)
return ResponseUtil.error(msg=str(e))
upload_result = await CommonService.upload_service(request, file)
logger.info('上传成功')
return ResponseUtil.success(model_content=upload_result.result)
@commonController.get("/download")
async def common_download(request: Request, background_tasks: BackgroundTasks, file_name: str = Query(alias='fileName'), delete: bool = Query()):
try:
download_result = await CommonService.download_services(background_tasks, file_name, delete)
if download_result.is_success:
logger.info(download_result.message)
return ResponseUtil.streaming(data=download_result.result)
else:
logger.warning(download_result.message)
return ResponseUtil.failure(msg=download_result.message)
except Exception as e:
logger.exception(e)
return ResponseUtil.error(msg=str(e))
download_result = await CommonService.download_services(background_tasks, file_name, delete)
logger.info(download_result.message)
return ResponseUtil.streaming(data=download_result.result)
@commonController.get("/download/resource")
async def common_download(request: Request, resource: str = Query()):
try:
download_resource_result = await CommonService.download_resource_services(resource)
if download_resource_result.is_success:
logger.info(download_resource_result.message)
return ResponseUtil.streaming(data=download_resource_result.result)
else:
logger.warning(download_resource_result.message)
return ResponseUtil.failure(msg=download_resource_result.message)
except Exception as e:
logger.exception(e)
return ResponseUtil.error(msg=str(e))
download_resource_result = await CommonService.download_resource_services(resource)
logger.info(download_resource_result.message)
return ResponseUtil.streaming(data=download_resource_result.result)