X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=provider%2Fsrc%2Fmain%2Fjava%2Fio%2Fpactflow%2Fexample%2Fsirenprovider%2Fcontrollers%2FOrderController.java;h=d08e4432f3239c01fb3fb714fb356d25c0ccdf35;hb=HEAD;hp=68810690b50deeb71d9aa0e246a8b9107aa855c2;hpb=1931500aaefbe7fe63e775a39a5408d17672ccbf;p=demos%2Fexample-siren

diff --git a/provider/src/main/java/io/pactflow/example/sirenprovider/controllers/OrderController.java b/provider/src/main/java/io/pactflow/example/sirenprovider/controllers/OrderController.java
index 6881069..d08e443 100644
--- a/provider/src/main/java/io/pactflow/example/sirenprovider/controllers/OrderController.java
+++ b/provider/src/main/java/io/pactflow/example/sirenprovider/controllers/OrderController.java
@@ -14,47 +14,57 @@ import org.springframework.web.bind.annotation.PutMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
+import java.util.Random;
+
 import static org.springframework.hateoas.server.mvc.WebMvcLinkBuilder.afford;
 import static org.springframework.hateoas.server.mvc.WebMvcLinkBuilder.linkTo;
 import static org.springframework.hateoas.server.mvc.WebMvcLinkBuilder.methodOn;
 
 @RestController
 @ExposesResourceFor(Order.class)
-@RequestMapping("/orders")
+@RequestMapping(value = "/orders")
 public class OrderController {
 
-  @GetMapping("/orders")
-  public RepresentationModel<?> orders() {
-    Order order = new Order(1234L);
-    Link selfLink = linkTo(methodOn(OrderController.class).order(order.getId())).withSelfRel()
-      .andAffordance(afford(methodOn(OrderController.class).update(order.getId(), null)))
-      .andAffordance(afford(methodOn(OrderController.class).delete(order.getId())));
+  @GetMapping
+  public ResponseEntity<RepresentationModel<?>> orders() {
+    Long id = Math.abs(new Random().nextLong());
+    Order order = new Order(id);
+    Link selfLink = actions(order);
     EntityModel<Order> model = EntityModel.of(order, selfLink);
     RepresentationModel<?> orders = CollectionModel.of(model);
     orders.add(linkTo(methodOn(OrderController.class).orders()).withSelfRel());
-    return orders;
+    return ResponseEntity.ok(orders);
   }
 
-  @GetMapping("/orders/{id}")
-  public EntityModel<Order> order(@PathVariable(value = "id", required = true) Long id) {
+  @GetMapping(value = "/{id}")
+  public ResponseEntity<EntityModel<Order>> order(@PathVariable(value = "id", required = true) Long id) {
     Order order = new Order(id);
-    Link selfLink = linkTo(methodOn(OrderController.class).order(order.getId())).withSelfRel()
-      .andAffordance(afford(methodOn(OrderController.class).update(order.getId(), null)))
-      .andAffordance(afford(methodOn(OrderController.class).delete(order.getId())));
+    Link selfLink = actions(order);
     EntityModel<Order> model = EntityModel.of(order, selfLink);
-    return model;
+    return ResponseEntity.ok(model);
   }
 
-  @PutMapping("/orders/{id}")
+  @PutMapping("/{id}")
   public EntityModel<Order> update(@PathVariable(value = "id", required = true) Long id, Order order) {
-    Link selfLink = linkTo(methodOn(OrderController.class).order(order.getId())).withSelfRel()
+    Link selfLink = actions(order);
+    return EntityModel.of(order, selfLink);
+  }
+
+  private Link actions(Order order) {
+    return linkTo(methodOn(OrderController.class).order(order.getId())).withSelfRel()
       .andAffordance(afford(methodOn(OrderController.class).update(order.getId(), null)))
-      .andAffordance(afford(methodOn(OrderController.class).delete(order.getId())));
-    EntityModel<Order> model = EntityModel.of(order, selfLink);
-    return model;
+      .andAffordance(afford(methodOn(OrderController.class).delete(order.getId())))
+      .andAffordance(afford(methodOn(OrderController.class).changeStatus(order.getId(), null)));
+  }
+
+  @PutMapping("/{id}/status")
+  public EntityModel<Order> changeStatus(@PathVariable(value = "id", required = true) Long id, String status) {
+    Order order = new Order(id);
+    Link selfLink = actions(order);
+    return EntityModel.of(order, selfLink);
   }
 
-  @DeleteMapping("/orders/{id}")
+  @DeleteMapping("/{id}")
   public ResponseEntity<Void> delete(@PathVariable(value = "id", required = true) Long id) {
     return ResponseEntity.ok().build();
   }