[TriNT][PBS-106] feat: get assignment by StaffId

src/main/java/com/swp/PodBookingSystem/controller/AssignmentController.java

@@ -48,5 +48,22 @@ ApiResponse<AssignmentResponse> updateAssignment(@PathVariable("assignmentId") S
                 .message("Update Assignment successfully")
                 .build();
     }
+
+    @DeleteMapping("/{assignmentId}")
+    ApiResponse<AssignmentResponse> deleteAssignment(@PathVariable("assignmentId") String assignmentId){
+        return ApiResponse.<AssignmentResponse>builder()
+                .message(assignmentService.deleteAssignment(assignmentId))
+                .build();
     }
 
+        @GetMapping("/{staffId}")
+        public ApiResponse<List<AssignmentResponse>> getAssignmentByStaffId(@PathVariable("staffId") String staffId) {
+            List<AssignmentResponse> assignments = assignmentService.getAssignmentsByStaffId(staffId);
+            return ApiResponse.<List<AssignmentResponse>>builder()
+                    .data(assignments)
+                    .message("Assignments retrieved successfully")
+                    .build();
+        }
+
+}
+

src/main/java/com/swp/PodBookingSystem/entity/Assignment.java

@@ -17,7 +17,7 @@
 @Table(name = "assignment")
 public class Assignment {
     @Id
-    String Id;
+    String id;
 
     String staffId;
 

src/main/java/com/swp/PodBookingSystem/mapper/AssignmentMapper.java

@@ -10,6 +10,8 @@
 import org.mapstruct.Mapping;
 import org.mapstruct.MappingTarget;
 
+import java.util.List;
+
 @Mapper(componentModel = "spring")
 
 public interface AssignmentMapper {
@@ -19,4 +21,5 @@ public interface AssignmentMapper {
 
     @Mapping(target = "id", ignore = true)
     Assignment toUpdateAssignment(AssignmentRequest request, @MappingTarget Assignment assignment);
+
 }

src/main/java/com/swp/PodBookingSystem/repository/AssignmentRepository.java

@@ -4,6 +4,10 @@
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.stereotype.Repository;
 
+import java.util.List;
+
 @Repository
 public interface AssignmentRepository extends JpaRepository<Assignment, String> {
+    List<Assignment> findByStaffId(String staffId);
+
 }

src/main/java/com/swp/PodBookingSystem/service/AssignmentService.java

@@ -44,7 +44,20 @@ public AssignmentResponse updateAssignment(String id, AssignmentRequest request)
                 .orElseThrow(() -> new EntityNotFoundException("Assignment not found"));
         Assignment updateAssignment = assignmentMapper.toUpdateAssignment(request, existingAssignment);
         return assignmentMapper.toAssignmentResponse(assignmentRepository.save(updateAssignment));
+    }
 
+    public String deleteAssignment(String id){
+        Assignment existingAssignment = assignmentRepository.findById(id)
+                .orElseThrow(() -> new EntityNotFoundException("Assignment not found"));
+        assignmentRepository.deleteById(id);
+        return "Assignment deleted successfully";
+    }
 
+    public List<AssignmentResponse> getAssignmentsByStaffId(String staffId) {
+        List<Assignment> assignments = assignmentRepository.findByStaffId(staffId);
+        return assignments.stream()
+                .map(assignmentMapper::toAssignmentResponse)
+                .collect(Collectors.toList());
     }
+
 }