Merge pull request #1 from digma-ai/add-feedback-endpoint

Add feedback endpoint

Author: asafchen-dig

src/main/java/org/springframework/samples/petclinic/clinicfeedback/FeedbackController.java

@@ -1,6 +1,11 @@
 package org.springframework.samples.petclinic.clinicfeedback;
 
+import io.opentelemetry.api.trace.Span;
+import jakarta.validation.Valid;
+import org.springframework.http.ResponseEntity;
 import org.springframework.samples.petclinic.model.ClinicFeedback;
+import org.springframework.samples.petclinic.owner.Owner;
+import org.springframework.validation.BindingResult;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
@@ -36,8 +41,30 @@ public String clear() {
 	}
 
 	@PostMapping("populate")
-	public String populateFeedbacks(@RequestParam(name = "count", defaultValue = "10000") int count) {
-		service.populate(count);
-		return "done";
+	public ResponseEntity<String> populateFeedbacks(@RequestParam(name = "count", defaultValue = "10000") int count) {
+		try{
+			service.populate(count);
+			return ResponseEntity.ok("Populated");
+		}
+		catch (Exception ex){
+			Span.current().recordException(ex);
+			return ResponseEntity.internalServerError().body(ex.getMessage());
+		}
+	}
+
+	@PostMapping("add")
+	public ResponseEntity<String> addFeedback(@RequestBody ClinicFeedback newFeedback, BindingResult result) {
+		if(result.hasErrors())
+			return ResponseEntity.badRequest().body("Failed to parsed request");
+
+		try{
+			service.add(newFeedback);
+			return ResponseEntity.ok("Inserted");
+		}
+		catch (Exception ex){
+			Span.current().recordException(ex);
+			return ResponseEntity.internalServerError().body(ex.getMessage());
+		}
+
 	}
 }

src/main/java/org/springframework/samples/petclinic/clinicfeedback/FeedbackRepository.java

@@ -9,7 +9,6 @@
 import org.springframework.data.domain.Pageable;
 import org.springframework.samples.petclinic.model.ClinicFeedback;
 import org.springframework.stereotype.Component;
-import org.springframework.stereotype.Service;
 
 import java.net.URI;
 import java.net.http.HttpClient;
@@ -32,7 +31,30 @@ public FeedbackRepository(@Value("${spring.data.flaskdb.uri}") String apiBaseUrl
 		this.objectMapper.disable(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS);
 	}
 
-	public void save(ClinicFeedback feedback) {
+	public void saveMultiple(List<ClinicFeedback> feedbacks) {
+		for (ClinicFeedback feedback : feedbacks){
+			HttpResponse<String> response;
+			try {
+				var json = objectMapper.writeValueAsString(feedback);
+				var request = HttpRequest.newBuilder()
+					.uri(URI.create(this.apiBaseUrl))
+					.header("Content-Type", "application/json")
+					.POST(HttpRequest.BodyPublishers.ofString(json))
+					.build();
+
+				response = httpClient.send(request, HttpResponse.BodyHandlers.ofString());
+
+			} catch (Exception e) {
+				throw new RuntimeException("Error during saving feedbacks", e);
+			}
+
+			if (response.statusCode() != 200 && response.statusCode() != 201) {
+				throw new RuntimeException("Failed to save feedbacks to flaskdb: " + response.body());
+			}
+		}
+	}
+
+	public void saveSingle(ClinicFeedback feedback) {
 		HttpResponse<String> response;
 		try {
 			String json = objectMapper.writeValueAsString(feedback);
@@ -44,11 +66,11 @@ public void save(ClinicFeedback feedback) {
 
 			response = httpClient.send(request, HttpResponse.BodyHandlers.ofString());
 		} catch (Exception e) {
-			throw new RuntimeException("Error during saving feedbacks", e);
+			throw new RuntimeException("Error during saving feedback", e);
 		}
 
 		if (response.statusCode() != 200 && response.statusCode() != 201) {
-			throw new RuntimeException("Failed to save feedbacks to flaskdb: " + response.body());
+			throw new RuntimeException("Failed to save feedback to flaskdb: " + response.body());
 		}
 	}
 

src/main/java/org/springframework/samples/petclinic/clinicfeedback/FeedbackService.java

@@ -21,13 +21,15 @@ public FeedbackService(FeedbackRepository repository) {
 
 	@WithSpan
 	public void populate(int numberOfDocs){
+		var feedbacks = new ArrayList<ClinicFeedback>();
 		for (int i = 0; i < numberOfDocs; i++) {
 			var feedback = new ClinicFeedback(
 				lorem.getEmail(),
 				lorem.getParagraphs(1, 3)
 			);
-			repository.save(feedback);
+			feedbacks.add(feedback);
 		}
+		repository.saveMultiple(feedbacks);
 	}
 
 	public long count() {
@@ -41,4 +43,8 @@ public long clear() {
 	public List<ClinicFeedback> list(int page, int pageSize) {
 		return repository.findAll(Pageable.ofSize(pageSize).withPage(page));
 	}
+
+	public void add(ClinicFeedback newFeedback) {
+		repository.saveSingle(newFeedback);
+	}
 }

src/main/java/org/springframework/samples/petclinic/model/ClinicFeedback.java

@@ -22,4 +22,20 @@ public ClinicFeedback(String userEmail, String comment) {
 		this.comment = comment;
 		this.submittedAt = Instant.now();
 	}
+
+	public void setId(String id) {
+		this.id = id;
+	}
+
+	public void setUserEmail(String userEmail) {
+		this.userEmail = userEmail;
+	}
+
+	public void setComment(String comment) {
+		this.comment = comment;
+	}
+
+	public void setSubmittedAt(Instant submittedAt) {
+		this.submittedAt = submittedAt;
+	}
 }