From b832880e1fcf3307387d53106078498f68a95853 Mon Sep 17 00:00:00 2001
From: jack <jack.su@pollex.com.tw>
Date: 星期二, 05 三月 2024 16:59:02 +0800
Subject: [PATCH] 移除TestLoginResource

---
 pamapi/src/main/java/com/pollex/pam/web/rest/ConsultantResource.java |   59 ++++++++++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 44 insertions(+), 15 deletions(-)

diff --git a/pamapi/src/main/java/com/pollex/pam/web/rest/ConsultantResource.java b/pamapi/src/main/java/com/pollex/pam/web/rest/ConsultantResource.java
index d672590..7c8bc6f 100644
--- a/pamapi/src/main/java/com/pollex/pam/web/rest/ConsultantResource.java
+++ b/pamapi/src/main/java/com/pollex/pam/web/rest/ConsultantResource.java
@@ -1,21 +1,23 @@
 package com.pollex.pam.web.rest;
 
-import com.pollex.pam.service.AppointmentService;
-import com.pollex.pam.service.ConsultantService;
-import com.pollex.pam.service.dto.*;
+import com.pollex.pam.business.aop.logging.audit.AuditLoggingInject;
+import com.pollex.pam.business.domain.Consultant;
+import com.pollex.pam.business.security.SecurityUtils;
+import com.pollex.pam.business.service.AppointmentService;
+import com.pollex.pam.business.service.ConsultantService;
+import com.pollex.pam.business.service.dto.*;
 import org.apache.commons.compress.utils.IOUtils;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.core.io.ClassPathResource;
-import org.springframework.core.io.Resource;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 
-import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.io.InputStream;
 import java.util.List;
+
+import static com.pollex.pam.business.aop.logging.audit.AuditLoggingType.EDIT_CONSULTANT_DATA;
 
 @RestController
 @RequestMapping("/api/consultant")
@@ -24,10 +26,19 @@
     @Autowired
     ConsultantService consultantService;
 
+    @Autowired
+    AppointmentService appointmentService;
+
     @GetMapping("/favorite")
     public ResponseEntity<List<CustomerFavoriteConsultantDTO>> getMyConsultantList() {
         List<CustomerFavoriteConsultantDTO> myConsultants = consultantService.getMyConsultantList();
         return new ResponseEntity<>(myConsultants, HttpStatus.OK);
+    }
+
+    @PostMapping("/favorite/view")
+    public ResponseEntity<Void> recordMyConsultantListView() {
+        consultantService.recordMyConsultantListView();
+        return ResponseEntity.noContent().build();
     }
 
     @GetMapping("/recommend")
@@ -72,15 +83,23 @@
     	return consultantService.getMyAppointment();
     }
 
-    @GetMapping(value = "/avatar/{fileName}", produces = MediaType.IMAGE_JPEG_VALUE)
-    public ResponseEntity<byte[]> getAvatarImage(@PathVariable String fileName) throws IOException {
-        try {
-            Resource resource = new ClassPathResource("static/consultant/" + fileName);
-            InputStream in = resource.getInputStream();
-            return new ResponseEntity<>(IOUtils.toByteArray(in), HttpStatus.OK);
-        } catch (FileNotFoundException e) {
-            return new ResponseEntity<>(HttpStatus.NOT_FOUND);
-        }
+    @GetMapping(value = "/avatar/{agentNo}", produces = MediaType.IMAGE_JPEG_VALUE)
+    public ResponseEntity<byte[]> getAvatarImage(@PathVariable String agentNo) throws IOException {
+//        try {
+//            Resource resource = new ClassPathResource("static/consultant/" + fileName);
+//            InputStream in = resource.getInputStream();
+//            return new ResponseEntity<>(IOUtils.toByteArray(in), HttpStatus.OK);
+//        } catch (FileNotFoundException e) {
+//            return new ResponseEntity<>(HttpStatus.NOT_FOUND);
+//        }
+    	InputStream in = consultantService.getAvatarImage(agentNo);
+    	if(in!=null) {
+    		return new ResponseEntity<>(IOUtils.toByteArray(in), HttpStatus.OK);
+    	}else {
+    		return new ResponseEntity<>(HttpStatus.NOT_FOUND);
+    	}
+
+
     }
 
     @PostMapping("/record/allAppointmentsView")
@@ -88,4 +107,14 @@
         consultantService.recordAllAppointmentsView();
         return ResponseEntity.noContent().build();
     }
+
+    @AuditLoggingInject(type = EDIT_CONSULTANT_DATA)
+    @PostMapping("/edit")
+    public ResponseEntity<Consultant> editConsultant(@RequestBody ConsultantEditDTO editDTO) {
+    	if(!editDTO.getAgentNo().equals(SecurityUtils.getAgentNo())) {
+    		throw new IllegalAccessError();
+    	}
+    	Consultant editResult = consultantService.editConsultant(editDTO);
+        return new ResponseEntity<>(editResult, HttpStatus.OK);
+    }
 }

--
Gitblit v1.8.0