Added model mapper class.

Author: skumar3006

pom.xml

@@ -70,6 +70,11 @@
 			<artifactId>spring-security-oauth2</artifactId>
 			<version>2.3.3.RELEASE</version>
 		</dependency>
+		<dependency>
+			<groupId>org.modelmapper</groupId>
+			<artifactId>modelmapper</artifactId>
+			<version>2.3.5</version>
+		</dependency>
 	</dependencies>
 
 </project>

src/main/java/sandeep/demo/controller/OwnerController.java

@@ -1,12 +1,10 @@
 package sandeep.demo.controller;
 
 import java.util.List;
-import java.util.Optional;
 
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpStatus;
-import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.DeleteMapping;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -15,8 +13,8 @@
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
-import sandeep.demo.entity.Owner;
 import sandeep.demo.manager.OwnerManager;
+import sandeep.demo.model.OwnerBO;
 
 @RestController
 @RequestMapping("/owners")
@@ -26,28 +24,34 @@ public class OwnerController {
 	OwnerManager ownerManager;
 
 	@GetMapping
-	public ResponseEntity<List<Owner>> getAllRecords() {
-		List<Owner> list = ownerManager.getAllRecords();
-
-		return new ResponseEntity<List<Owner>>(list, new HttpHeaders(), HttpStatus.OK);
+	public List<OwnerBO> getAllRecords() {
+		List<OwnerBO> list = ownerManager.getAllRecords();
+		return list;
 	}
 
 	@GetMapping(value = "/{ownerId}")
-	public Optional<Owner> getById(@PathVariable Integer ownerId) {
-		Optional<Owner> owner = ownerManager.getById(ownerId);
+	public OwnerBO getById(@PathVariable Integer ownerId) {
+		OwnerBO owner = ownerManager.getById(ownerId);
 		return owner;
 	}
 
 	@PostMapping
-	public ResponseEntity<Owner> createOrUpdateEmployee(@RequestBody Owner owner) {
-		Owner response = ownerManager.createOwner(owner);
-		return new ResponseEntity<Owner>(response, new HttpHeaders(), HttpStatus.OK);
+	public OwnerBO createOrUpdateEmployee(@RequestBody OwnerBO owner) {
+		OwnerBO ownerBO = ownerManager.createOwner(owner);
+		return ownerBO;
 	}
 
 	@PutMapping(value = "{/ownerId}")
-	public ResponseEntity<Owner> update(Owner owner) {
-		Owner response = ownerManager.updateOwner(owner);
-		return new ResponseEntity<Owner>(response, new HttpHeaders(), HttpStatus.OK);
+	public OwnerBO update(@RequestBody OwnerBO owner) {
+		OwnerBO ownerBO = ownerManager.updateOwner(owner);
+		return ownerBO;
+	}
+	
+	@DeleteMapping(value = "/{ownerId}")
+	public HttpStatus deleteById(@PathVariable Integer ownerId) {
+		 ownerManager.deleteById(ownerId);
+		return HttpStatus.ACCEPTED;
 	}
 
+
 }

src/main/java/sandeep/demo/manager/OwnerManager.java

@@ -1,12 +1,15 @@
 package sandeep.demo.manager;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Optional;
 
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
 import sandeep.demo.entity.Owner;
+import sandeep.demo.model.ModelEntityMapper;
+import sandeep.demo.model.OwnerBO;
 import sandeep.demo.repository.OwnerRepository;
 
 @Component
@@ -15,22 +18,32 @@ public class OwnerManager {
 	@Autowired
 	OwnerRepository ownerRepository;
 
-	public Owner createOwner(Owner owner) {
-		ownerRepository.save(owner);
-		return owner;
+	public OwnerBO createOwner(OwnerBO ownerBO) {
+		Owner owner = ModelEntityMapper.convertToEntity(ownerBO);
+		owner = ownerRepository.save(owner);
+		return ModelEntityMapper.convertToBO(owner);
 	}
 
-	public List<Owner> getAllRecords() {
-		return (List<Owner>) ownerRepository.findAll();
+	public List<OwnerBO> getAllRecords() {
+		List<Owner> owners = (List<Owner>) ownerRepository.findAll();
+		List<OwnerBO> list = new ArrayList<>();
+		owners.forEach(owner -> list.add(ModelEntityMapper.convertToBO(owner)));
+		return list;
 	}
 
-	public Owner updateOwner(Owner owner) {
-		ownerRepository.save(owner);
-		return owner;
+	public OwnerBO updateOwner(OwnerBO ownerBO) {
+		Owner owner = ModelEntityMapper.convertToEntity(ownerBO);
+		owner = ownerRepository.save(owner);
+		return ModelEntityMapper.convertToBO(owner);
 	}
 
-	public Optional<Owner> getById(Integer ownerId) {
-		return ownerRepository.findById(ownerId);
+	public OwnerBO getById(Integer ownerId) {
+		Optional<Owner> owner = ownerRepository.findById(ownerId);
+		return ModelEntityMapper.convertToBO(owner.get());
+	}
+
+	public void deleteById(Integer ownerId) {
+		ownerRepository.deleteById(ownerId);
 	}
 
 }

src/main/java/sandeep/demo/model/ModelEntityMapper.java

@@ -0,0 +1,24 @@
+package sandeep.demo.model;
+
+import org.modelmapper.ModelMapper;
+
+import sandeep.demo.entity.Owner;
+
+public class ModelEntityMapper {
+
+	public static OwnerBO convertToBO(Owner owner) {
+
+		ModelMapper modelMapper = new ModelMapper();
+		OwnerBO ownerBO = modelMapper.map(owner, OwnerBO.class);
+		return ownerBO;
+
+	}
+
+	public static Owner convertToEntity(OwnerBO ownerBO) {
+		ModelMapper modelMapper = new ModelMapper();
+		Owner owner = modelMapper.map(ownerBO, Owner.class);
+		return owner;
+
+	}
+
+}