Added a remove all shifts by employee id method to the Schedule class. Now...

Added a remove all shifts by employee id method to the Schedule class. Now when an employee is removed there shifts are deleted first from the database then they're deleted after.

SchedulerApp/src/main/java/com/example/schedulerapp/Schedule.java

@@ -90,6 +90,23 @@ public class Schedule {
         return editedShift;
     }
 
+    /*
+        Name: removeAllShiftsByID
+        Parameters:
+            intI ID: Employee ID
+        Description: Removes all shifts in the database for a specific employee
+        Return: void
+         */
+    protected void removeAllShiftsByID(int ID) {
+        try {
+            Statement myStatement = dbConnection.createStatement();
+            myStatement.executeUpdate("DELETE FROM shifts WHERE (employee_ID =" + ID + ")");
+        } catch (SQLException exception) {
+            System.out.println("Error removing all shifts by id.");
+            exception.printStackTrace();
+        }
+    }
+
     /*
         Name: allShifts
         Parameters:

SchedulerApp/src/main/java/com/example/schedulerapp/Staff.java

@@ -60,7 +60,6 @@ public class Staff {
     protected int removeEmployee(int ID){
         try {
             Statement myStatement = dbConnection.createStatement();
-            myStatement.executeUpdate("DELETE FROM shifts WHERE (employee_ID =" + ID + ")");
             myStatement.executeUpdate("delete from Employees where employee_id = " + ID);
         } catch (Exception exception) {
             exception.printStackTrace();

SchedulerApp/src/main/java/com/example/schedulerapp/UserThread.java

@@ -53,6 +53,7 @@ public class UserThread extends Thread {
                     case "removeEmployee":
                         try {
                             int employeeID = Integer.parseInt(args[1]);
+                            rafisWeirdClass.schedule.removeAllShiftsByID(employeeID);
                             if (employeeID == rafisWeirdClass.staff.removeEmployee(employeeID)) {
                                 server.broadcast("removeEmployee/" + employeeID);
                             }