AddBook API code with headers updates

Author: sumanmathad

src/main/java/com/learn/SpringBootRESTService/AddResponse.java

@@ -0,0 +1,25 @@
+package com.learn.SpringBootRESTService;
+
+import org.springframework.stereotype.Component;
+
+@Component
+public class AddResponse {
+	
+	
+	String msg;
+	String id;
+	public String getMsg() {
+		return msg;
+	}
+	public void setMsg(String msg) {
+		this.msg = msg;
+	}
+	public String getId() {
+		return id;
+	}
+	public void setId(String id) {
+		this.id = id;
+	}
+	
+
+}

src/main/java/com/learn/SpringBootRESTService/LibraryController.java

@@ -1,6 +1,10 @@
 package com.learn.SpringBootRESTService;
 
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.HttpStatusCode;
+import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RestController;
@@ -13,9 +17,24 @@ public class LibraryController {
 	@Autowired
 	LibraryRepository repository;
 
+	@Autowired
+	AddResponse addResponse;
+	
+	
 	@PostMapping("/addBook")
-	public void addBookImplementation(@RequestBody Library library) {
-		library.setId(library.getIsbn()+library.getAisle());
+	public ResponseEntity addBookImplementation(@RequestBody Library library) {
+		String id= library.getIsbn()+library.getAisle();
+		library.setId(id);
 		repository.save(library);
+		addResponse.setMsg("Success!! Book is added");
+		addResponse.setId(id);
+		HttpHeaders headers = new HttpHeaders();
+		headers.add("Unique-Header", "First");
+		//return addResponse; without response headers!
+		return new ResponseEntity<AddResponse>(addResponse,headers,HttpStatus.CREATED);
+		
 	}
+	
+	
+	
 }