feat(可回溯): 可回溯

初始化
2023-03-23 18:05:14 +08:00
commit f8286861cd
12 changed files with 538 additions and 0 deletions
@@ -0,0 +1,53 @@
+package com.wabestway.recall.web;
+
+import com.wabestway.recall.model.Order;
+import com.wabestway.recall.model.Record;
+import com.wabestway.recall.model.ResObj;
+import com.wabestway.recall.repository.OrderRepository;
+import com.wabestway.recall.repository.RecordRepository;
+import org.springframework.data.domain.Example;
+import org.springframework.data.domain.Sort;
+import org.springframework.web.bind.annotation.*;
+import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
+
+import java.util.List;
+
+
+@RestController
+@RequestMapping("/order")
+public class OrderController {
+    private final RecordRepository recordRepository;
+    private final OrderRepository orderRepository;
+
+    public OrderController(RecordRepository recordRepository, OrderRepository orderRepository) {
+        this.recordRepository = recordRepository;
+        this.orderRepository = orderRepository;
+    }
+
+    @GetMapping
+    public Flux<Order> getAllOrders() {
+        return orderRepository.findAll();
+    }
+
+    @PostMapping("/list")
+    public Flux<Order> list(@RequestBody Order order) {
+        Example<Order> ep = Example.of(order);
+        Sort s = Sort.by("createAt");
+        return orderRepository.findAll(ep, s);
+    }
+
+    @PostMapping("/info/{orderId}")
+    public Mono<ResObj> findByOrderId(@PathVariable String orderId) {
+        Mono<Order> forder = orderRepository.findByOrderId(orderId);
+        return forder.flatMap(fo -> {
+            Flux<Record> rs = recordRepository.findAllByTraceIdOrderByCreateAtAsc(fo.getTraceId());
+            Mono<List<List<String>>> es = rs.map(rr -> rr.getEvents()).collectList();
+            Mono<List<String>> events = es.flatMapIterable(lists -> lists).flatMapIterable(list -> list).collectList();
+            return events.flatMap(e -> {
+                fo.setEvents(e);
+                return Mono.just(ResObj.ok(fo));
+            });
+        });
+    }
+}
@@ -0,0 +1,93 @@
+package com.wabestway.recall.web;
+
+import com.wabestway.recall.model.Order;
+import com.wabestway.recall.model.Record;
+import com.wabestway.recall.model.ResObj;
+import com.wabestway.recall.repository.OrderRepository;
+import com.wabestway.recall.repository.RecordRepository;
+import org.springframework.web.bind.annotation.*;
+import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
+
+import java.util.UUID;
+
+@RestController
+@RequestMapping("/track")
+public class RecallController {
+    private final RecordRepository recordRepository;
+    private final OrderRepository orderRepository;
+
+    public RecallController(RecordRepository recordRepository, OrderRepository orderRepository) {
+        this.recordRepository = recordRepository;
+        this.orderRepository = orderRepository;
+    }
+
+    @GetMapping
+    public Flux<Record> getAllUsers() {
+        return recordRepository.findAll();
+    }
+
+    @PostMapping("/save")
+    public Mono<ResObj> save(@RequestBody Record record) {
+        if (record.getOrderId() != null && !record.getOrderId().isEmpty()) {
+            Mono<Order> order = orderRepository.findByOrderId(record.getOrderId());
+            order.hasElement().subscribe(exit -> {
+                if (!exit) {
+                    if (record.getTraceId() != null && !record.getTraceId().isEmpty()) {
+                        Mono<Order> tcOrder = orderRepository.findByTraceId(record.getTraceId());
+                        tcOrder.filter(to -> to.getId() != null).flatMap(to -> {
+                            to.setOrderId(record.getOrderId());
+                            if (record.isLast()) {
+                                to.setComplete("1");
+                            }
+                            return orderRepository.save(to);
+                        }).subscribe();
+                    } else {
+                        record.setTraceId(UUID.randomUUID().toString().replace("-", ""));
+                        Order forder = new Order();
+                        forder.setOrderId(record.getOrderId());
+                        forder.setTraceId(record.getTraceId());
+                        forder.setProductCode(record.getProductCode());
+                        forder.setProductName(record.getProductName());
+                        forder.setCreateAt(System.currentTimeMillis());
+                        forder.setAppKey(record.getAppKey());
+                        forder.setArchived("0");
+                        forder.setComplete("0");
+                        if (record.isLast()) {
+                            forder.setComplete("1");
+                        }
+                        orderRepository.save(forder).subscribe();
+                    }
+                } else {
+                    order.subscribe(to -> {
+                        if (record.isLast()) {
+                            to.setComplete("1");
+                            orderRepository.save(to).subscribe();
+                        }
+                    });
+                }
+            });
+        } else {
+            if (record.getTraceId() == null || record.getTraceId().isEmpty()) {
+                record.setTraceId(UUID.randomUUID().toString().replace("-", ""));
+                Order order = new Order();
+                order.setTraceId(record.getTraceId());
+                order.setProductCode(record.getProductCode());
+                order.setProductName(record.getProductName());
+                order.setCreateAt(System.currentTimeMillis());
+                order.setAppKey(record.getAppKey());
+                order.setArchived("0");
+                order.setComplete("0");
+                if (record.isLast()) {
+                    order.setComplete("1");
+                }
+                orderRepository.save(order).subscribe();
+            }
+        }
+        record.setCreateAt(System.currentTimeMillis());
+
+        return recordRepository.save(record).flatMap(ss -> {
+            return Mono.just(ResObj.ok(ss));
+        });
+    }
+}