zero-downtime example

thombergs · thombergs · commit 375928575c79 · 2021-08-04T06:33:14.000+10:00
diff --git a/spring-boot/zero-downtime/src/main/java/io/reflectoring/zerodowntime/CustomerController.java b/spring-boot/zero-downtime/src/main/java/io/reflectoring/zerodowntime/CustomerController.java
@@ -1,8 +1,11 @@
 package io.reflectoring.zerodowntime;
 
 import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.RestController;
 
+import java.util.Optional;
+
 @RestController
 public class CustomerController {
 
@@ -17,7 +20,7 @@ public CustomerController(CustomerRepository oldCustomerRepository, NewCustomerR
     }
 
     @GetMapping("/customers/create")
-    String createUser() {
+    String createCustomer() {
         if (featureFlagService.writeToNewCustomerSchema()) {
             NewCustomer customer = new NewCustomer("Bob", "Builder", "Build Street", "21");
             newCustomerRepository.save(customer);
@@ -28,8 +31,19 @@ String createUser() {
         return "customer created";
     }
 
+    @GetMapping("/customers/{id}}")
+    String getCustomer(@PathVariable("id") Long id) {
+        if (featureFlagService.readFromNewCustomerSchema()) {
+            Optional<NewCustomer> customer = newCustomerRepository.findById(id);
+            return customer.get().toString();
+        } else {
+            Optional<OldCustomer> customer = oldCustomerRepository.findById(id);
+            return customer.get().toString();
+        }
+    }
+
     @GetMapping("/customers/list")
-    String showUser() {
+    String getCustomer() {
         StringBuffer buffer = new StringBuffer();
         if (featureFlagService.readFromNewCustomerSchema()) {
             Iterable<NewCustomer> customers = newCustomerRepository.findAll();
