Made the User entity bi directional to support cascading

Author: Chamal-Peiris

src/main/java/lk/ijse/dep/note/config/WebRootConfig.java

@@ -16,15 +16,16 @@
 @Import(JpaConfig.class)
 public class WebRootConfig {
 
-    public ModelMapper
+    @Bean
+    public ModelMapper modelMapper(){
+        return new ModelMapper();
+    }
     @Bean
     public static YamlPropertiesFactoryBean yamlPropertiesFactoryBean(ConfigurableEnvironment env ){
         YamlPropertiesFactoryBean yamlPropertiesFactoryBean = new YamlPropertiesFactoryBean();
         yamlPropertiesFactoryBean.setResources( new ClassPathResource("application.yml"));
         Properties yamlProps = yamlPropertiesFactoryBean.getObject();
         env.getPropertySources().addLast(new PropertiesPropertySource("yaml",yamlProps));
         return yamlPropertiesFactoryBean;
-
-
     }
 }

src/main/java/lk/ijse/dep/note/entity/User.java

@@ -1,17 +1,17 @@
 package lk.ijse.dep.note.entity;
 
-import lombok.AllArgsConstructor;
-import lombok.Data;
-import lombok.NoArgsConstructor;
+import jdk.internal.dynalink.linker.LinkerServices;
+import lombok.*;
 
-import javax.persistence.Column;
-import javax.persistence.Entity;
-import javax.persistence.Id;
+import javax.persistence.*;
+import java.util.ArrayList;
+import java.util.List;
 
 @NoArgsConstructor
 @AllArgsConstructor
 @Entity
 @Data
+
 public class User implements SuperEntity {
     @Id
     private String id;
@@ -21,4 +21,13 @@ public class User implements SuperEntity {
     private String password;
     @Column(name="full_name",unique = true)
     private String fullName;
+
+    @Setter(AccessLevel.NONE)
+    @OneToMany(mappedBy = "user",cascade = {CascadeType.REMOVE})
+    private List<Note>  notes=new ArrayList<>();
+
+    @PrePersist
+    private void beforePersist(){
+        notes.forEach(note->note.setUser(this));
+    }
 }

src/main/java/lk/ijse/dep/note/service/impl/UserServiceImpl.java

@@ -1,39 +1,54 @@
 package lk.ijse.dep.note.service.impl;
 
 import lk.ijse.dep.note.dto.UserDTO;
+import lk.ijse.dep.note.entity.User;
 import lk.ijse.dep.note.repository.custom.UserRepository;
 import lk.ijse.dep.note.service.UserService;
 import lk.ijse.dep.note.service.exception.DuplicateEmailException;
 import lk.ijse.dep.note.service.exception.NotFoundException;
+import lk.ijse.dep.note.service.util.EntityDTOConverter;
+import org.modelmapper.ModelMapper;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.util.Optional;
+
 @Service
 @Transactional
 public class UserServiceImpl implements UserService {
 
     @Autowired
     private  UserRepository userRepository;
+    @Autowired
+    private EntityDTOConverter transformer;
+
 
     @Override
     public UserDTO registerUser(UserDTO user) throws DuplicateEmailException {
         if(userRepository.existUserByEmail(user.getEmail()) ) throw new DuplicateEmailException("Email already exists");
-        return userRepository
+        return transformer.getUserDTO(userRepository.save(transformer.getUserEntity(user)));
     }
 
     @Override
     public void updateUser(UserDTO user) throws NotFoundException {
+        Optional<User> optUser = userRepository.findById(user.getId());
+        if(!optUser.isPresent()) throw new NotFoundException("invalid user id");
+        optUser.get().setFullName(user.getFullName());
+        optUser.get().setPassword(user.getPassword());
 
     }
 
     @Override
     public UserDTO getUserInfo(String userId) throws NotFoundException {
-        return null;
+       return userRepository.findById(userId).map(transformer::getUserDTO).orElseThrow(()->new NotFoundException("Invalid user Id"));
+
     }
 
     @Override
     public void deleteUser(String userId) throws NotFoundException {
+        if(!userRepository.existById(userId)) throw new NotFoundException("Invalid User id");
+
 
     }
 }

src/main/java/lk/ijse/dep/note/service/util/EntityDTOConverter.java

@@ -0,0 +1,24 @@
+package lk.ijse.dep.note.service.util;
+
+import lk.ijse.dep.note.dto.UserDTO;
+import lk.ijse.dep.note.entity.User;
+import org.modelmapper.ModelMapper;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+@Component
+public class EntityDTOConverter {
+    @Autowired
+    private ModelMapper mapper;
+
+    public  EntityDTOConverter(ModelMapper mapper){
+        this.mapper=mapper;
+    }
+
+    public User getUserEntity(UserDTO user){
+        return mapper.map(user,User.class);
+    }
+    public UserDTO getUserDTO(User user){
+        return mapper.map(user,UserDTO.class);
+    }
+}