108 lines
4.6 KiB
Java
108 lines
4.6 KiB
Java
package com.wabestway.recall.web;
|
|
|
|
import com.alibaba.fastjson.JSON;
|
|
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 lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.web.bind.annotation.*;
|
|
import reactor.core.publisher.Flux;
|
|
import reactor.core.publisher.Mono;
|
|
|
|
import java.util.UUID;
|
|
|
|
@Slf4j
|
|
@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");
|
|
// forder.setHolderName("王五");
|
|
// forder.setHolderPhone("18515064530");
|
|
// forder.setSupplierName("阳光保险");
|
|
// forder.setStartDate("2023-01-01");
|
|
// forder.setEndDate("2035-01-01");
|
|
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");
|
|
// order.setHolderName("王五");
|
|
// order.setHolderPhone("18515064530");
|
|
// order.setSupplierName("阳光保险");
|
|
// order.setStartDate("2023-01-01");
|
|
// order.setEndDate("2035-01-01");
|
|
if (record.isLast()) {
|
|
order.setComplete("1");
|
|
}
|
|
orderRepository.save(order).subscribe();
|
|
}
|
|
}
|
|
record.setCreateAt(System.currentTimeMillis());
|
|
log.info("保存数据：{}", JSON.toJSONString(record));
|
|
return recordRepository.save(record).flatMap(ss -> {
|
|
ss.setEvents(null);
|
|
return Mono.just(ResObj.ok(ss));
|
|
});
|
|
}
|
|
}
|