DRMPN
diff --git a/‎IntensiveCourse/Lesson4/app/src/main/java/crud/servlets/StudentsServlet.java‎
Lines changed: 15 additions & 7 deletions b/‎IntensiveCourse/Lesson4/app/src/main/java/crud/servlets/StudentsServlet.java‎
Lines changed: 15 additions & 7 deletions
@@ -41,17 +41,25 @@ protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws Se
     @Override
     protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
 
-        Student student = new Student(req.getParameter("firstName"),
-                req.getParameter("lastName"),
-                Integer.parseInt(req.getParameter("age")),
-                req.getParameter("program"));
+        boolean update = true;
+        Student student = StudentsRepo.findStudentById(Integer.parseInt(req.getParameter("studentId")));
 
-        if (req.getParameter("studentId") == null ) {
-            StudentsRepo.addStudent(student);
+        if (student == null) {
+            student = new Student();
+            update = false;
         }
-        else {
+
+        student.setFirstName(req.getParameter("firstName"));
+        student.setFirstName(req.getParameter("lastName"));
+        student.setAge(Integer.parseInt(req.getParameter("age")));
+        student.setProgram(req.getParameter("program"));
+
+        if (update) {
             StudentsRepo.updateStudent(student);
         }
+        else {
+            StudentsRepo.addStudent(student);
+        }
 
         doGet(req, resp);
     }