Changed Exception Code

src/main/java/com/school/project/controller/CourseController.java

@@ -4,8 +4,6 @@ package com.school.project.controller;
 import com.school.project.dto.course.CourseRegisterDTO;
 import com.school.project.service.course.CourseServiceImp;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.http.HttpStatus;
-import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -15,17 +13,13 @@ import org.springframework.web.server.ResponseStatusException;
 @RestController
 @RequestMapping("/course")
 public class CourseController {
+
     @Autowired
     private CourseServiceImp courseServiceImp;
 
     @PostMapping("/save")
-    private ResponseEntity saveCourse(@RequestBody CourseRegisterDTO course) {
-        try {
-            courseServiceImp.saveCourse(course);
-            return ResponseEntity.ok("Course Saved");
-        } catch (Exception e) {
-            throw new ResponseStatusException(HttpStatus.BAD_REQUEST, "Data Is Not Correct", e);
-        }
+    private ResponseStatusException saveCourse(@RequestBody CourseRegisterDTO course) {
+        return courseServiceImp.saveCourse(course);
     }
 
 

src/main/java/com/school/project/service/course/CourseService.java

@@ -2,9 +2,10 @@ package com.school.project.service.course;
 
 
 import com.school.project.dto.course.CourseRegisterDTO;
+import org.springframework.web.server.ResponseStatusException;
 
 public interface CourseService {
 
-    public String saveCourse(CourseRegisterDTO courseDTO);
+    public ResponseStatusException saveCourse(CourseRegisterDTO courseDTO);
 
 }

src/main/java/com/school/project/service/course/CourseServiceImp.java

@@ -4,58 +4,33 @@ import com.school.project.dto.course.CourseRegisterDTO;
 import com.school.project.model.Course;
 import com.school.project.repository.modelRepositery.CourseRepository;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpStatus;
 import org.springframework.stereotype.Service;
+import org.springframework.web.server.ResponseStatusException;
 
 @Service
 public class CourseServiceImp implements CourseService {
 
+    private static final String courseMessage = "Course Saved";
+    private static final String incorrectData = "Data Is Not Correct";
     @Autowired(required = true)
     public CourseRepository courseRepository;
 
     @Override
-    public String saveCourse(CourseRegisterDTO courseDTO) {
+    public ResponseStatusException saveCourse(CourseRegisterDTO courseDTO) {
+
+        try {
+
             Course newCourse = new Course();
 
             newCourse.setCourseCode(courseDTO.getCourseCode());
             newCourse.setCourseName(courseDTO.getCourseName());
 
-        courseRepository.save(newCourse);
-
-        return "Course Saved";
-    }
-
-   /* @Transactional
-    @Override
-    public List<CourseDTO> readAllCourse() {
-        return courseMapper.courseModelToDTO(courseRepository.findAll());
-    }
-
-    @Override
-    public CourseDTO readCourseById(Long id) {
-        return courseMapper.courseModelToDTO(courseRepository.findById(id).get());
-    }
+            return new ResponseStatusException(HttpStatus.OK,courseMessage);
 
-    @Override
-    public CourseDTO updateCourseById(CourseDTO courseDTO, Long id) {
-        CourseDTO dto = readCourseById(id);
+        } catch (Exception e) {
 
-        if (courseDTO.getTitle() != null) {
-            dto.setTitle(courseDTO.getTitle());
-        }
-        if (courseDTO.getTeacherList() != null) {
-            dto.setTeacherList(courseDTO.getTeacherList());
+            return new ResponseStatusException(HttpStatus.BAD_REQUEST,incorrectData,e);
         }
-        if (courseDTO.getStudentList() != null) {
-            dto.setStudentList(courseDTO.getStudentList());
     }
-
-        return courseMapper.courseModelToDTO(
-                courseRepository.save(
-                        courseMapper.courseDTOToModel(dto)));
-    }
-
-    @Override
-    public void deleteCourseById(Long id) {
-        courseRepository.deleteById(id);
-    }*/
 }