From 8b38b46e1dfd34b775534200e6ace2a4f8dd96e6 Mon Sep 17 00:00:00 2001
From: s2042968 <s2042968@ed.ac.uk>
Date: Wed, 8 Jan 2025 14:49:37 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BB=BA=E7=AB=8B=E5=85=B3=E7=B3=BB=EF=BC=8C?=
 =?UTF-8?q?=E6=A0=A1=E9=AA=8C=E5=A4=B1=E8=B4=A5=EF=BC=8C=E8=BF=94=E5=9B=9E?=
 =?UTF-8?q?=E7=BB=93=E6=9E=9C=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../manage/controller/EdFileRelationController.java      | 9 ++++++++-
 .../service/serviceimpl/EdFileRelationServiceImpl.java   | 2 +-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/EdFileRelationController.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/EdFileRelationController.java
index 6bbcf6f..a1a39fe 100644
--- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/EdFileRelationController.java
+++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/EdFileRelationController.java
@@ -1,10 +1,12 @@
 package com.electromagnetic.industry.software.manage.controller;
 
+import com.electromagnetic.industry.software.common.exception.BizException;
 import com.electromagnetic.industry.software.common.resp.ElectromagneticResult;
 import com.electromagnetic.industry.software.common.util.ElectromagneticResultUtil;
 import com.electromagnetic.industry.software.manage.pojo.models.EdFileRelation;
 import com.electromagnetic.industry.software.manage.pojo.req.CheckNameUniqueRequest;
 import com.electromagnetic.industry.software.manage.service.EdFileRelationService;
+import org.springframework.http.HttpStatus;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
@@ -25,7 +27,12 @@ public class EdFileRelationController {
      */
     @RequestMapping(value = "/create", method = RequestMethod.POST)
     public ElectromagneticResult<?> createRelation(@RequestBody EdFileRelation relation) {
-        return ElectromagneticResultUtil.success(edFileRelationService.createRelation(relation));
+        try {
+            Boolean res = edFileRelationService.createRelation(relation);
+            return ElectromagneticResultUtil.success(res);
+        } catch (BizException e) {
+            return ElectromagneticResultUtil.fail("500", e.getMsg());
+        }
     }
 
     /**
diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileRelationServiceImpl.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileRelationServiceImpl.java
index 4a7a082..674aaf7 100644
--- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileRelationServiceImpl.java
+++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileRelationServiceImpl.java
@@ -51,7 +51,7 @@ public class EdFileRelationServiceImpl extends ServiceImpl<EdFileRelationMapper,
      * @return
      */
     @Override
-    public Boolean createRelation(EdFileRelation edFileRelation) {
+    public Boolean createRelation(EdFileRelation edFileRelation) throws BizException {
         // 无法建立已建立的关系
         String queryId1 = edFileRelation.getId1();
         String queryId2 = edFileRelation.getId2();