retrieving autogenerated primary key

Author: gowrishankar95

UserManagement/src/main/java/com/hms/management/dao/impl/UserRepositoryImpl.java

@@ -5,8 +5,14 @@
 import com.hms.management.dao.UserRepository;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.jdbc.core.PreparedStatementCreator;
+import org.springframework.jdbc.support.GeneratedKeyHolder;
+import org.springframework.jdbc.support.KeyHolder;
 import org.springframework.stereotype.Repository;
 
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.SQLException;
 import java.util.List;
 
 @Repository
@@ -17,7 +23,17 @@ public class UserRepositoryImpl implements UserRepository {
     @Override
     public User create(User user) {
         String sql = "INSERT INTO users (userName,passowrd) VALUES (?, ?)";
-        jdbcTemplate.update(sql,user.getUsername(),user.getPassword());
+        KeyHolder keyHolder = new GeneratedKeyHolder();
+        jdbcTemplate.update(new PreparedStatementCreator() {
+            @Override
+            public PreparedStatement createPreparedStatement(Connection connection) throws SQLException {
+                PreparedStatement ps = connection.prepareStatement(sql, new String[]{"UserId"});
+                ps.setString(1, user.getUsername());
+                ps.setString(2, user.getPassword());
+                return ps;
+            }
+        }, keyHolder);
+        user.setUserId( Integer.parseInt(keyHolder.getKey()+""));
         return user;
 
     }

UserManagement/src/main/java/com/hms/management/rest/UserManagementRestControllerExceptionHandler.java

@@ -2,9 +2,7 @@
 
 import com.hms.management.rest.Exceptions.InvalidNameException;
 import com.hms.management.rest.Exceptions.InvalidPasswordException;
-import org.aspectj.lang.annotation.Before;
 import org.springframework.http.HttpStatus;
-import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.ControllerAdvice;
 import org.springframework.web.bind.annotation.ExceptionHandler;
 import org.springframework.web.bind.annotation.ResponseBody;
@@ -14,23 +12,24 @@
 
 @ControllerAdvice
 public class UserManagementRestControllerExceptionHandler {
+
     @ResponseBody
     @ExceptionHandler(value = InvalidNameException.class)
-    public Map<String,String> invalidUserNameException(){
-        Map<String,String> map = new HashMap<String, String>() ;
-        map.put("status",HttpStatus.NOT_FOUND.toString());
-        map.put("error","invalid user name. please enter a valid user name");
+    public Map<String, String> invalidUserNameException() {
+        Map<String, String> map = new HashMap<String, String>();
+        map.put("status", HttpStatus.NOT_FOUND.toString());
+        map.put("error", "invalid user name. please enter a valid user name");
 
-         return map;
+        return map;
 
     }
 
     @ResponseBody
     @ExceptionHandler(value = InvalidPasswordException.class)
-    public Map<String,String> invalidUserPasswordException(){
-        Map<String,String> map = new HashMap<String, String>() ;
-        map.put("status",HttpStatus.NOT_FOUND.toString());
-        map.put("error","invalid password. please enter a valid password less than 20 characters");
+    public Map<String, String> invalidUserPasswordException() {
+        Map<String, String> map = new HashMap<String, String>();
+        map.put("status", HttpStatus.NOT_FOUND.toString());
+        map.put("error", "invalid password. please enter a valid password less than 20 characters");
 
         return map;