package com.pollex.pam.web.rest; import java.util.ArrayList; import java.util.List; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.http.HttpStatus; import org.springframework.http.ResponseEntity; import org.springframework.util.StringUtils; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import com.pollex.pam.domain.Consultant; import com.pollex.pam.domain.PersonalNotification; import com.pollex.pam.enums.NotificationTypeEnum; import com.pollex.pam.enums.PersonalNotificationRoleEnum; import com.pollex.pam.repository.PersonalNotificationRepository; import com.pollex.pam.security.SecurityUtils; import com.pollex.pam.service.ConsultantService; import com.pollex.pam.service.PersonalNotificationService; @RestController @RequestMapping("/api/personal_notification") public class PersonalNotificationResource { @Autowired PersonalNotificationService personalNotificationService; @Autowired PersonalNotificationRepository personalNotificationRepository; @Autowired ConsultantService consultantService; @GetMapping("/getMyPersonalNotification") public ResponseEntity> getMyPersonalNotification() { List personalNotificationList = new ArrayList<>(); if(StringUtils.hasText(SecurityUtils.getAgentNo())) { Consultant consultant = consultantService.findByAgentNo(SecurityUtils.getAgentNo()); personalNotificationList = personalNotificationService.getMyPersonalNotification(consultant.getId(), PersonalNotificationRoleEnum.CONSULTANT); }else if(SecurityUtils.getCustomerDBId()!=null){ personalNotificationList = personalNotificationService.getMyPersonalNotification(SecurityUtils.getCustomerDBId(), PersonalNotificationRoleEnum.CUSTOMER); } return new ResponseEntity<>(personalNotificationList, HttpStatus.OK); } @PostMapping("/readAllMyNotification") public void readAll() { personalNotificationService.readAllMyNotification(); } }