这个示例代码是一个简单的订单撮合交易系统,基于Spring Boot和Spring Data JPA,演示了订单的创建、撮合和交易记录的生成:
// 实体类 @Entity public class Order { @Id @GeneratedValue(strategy = GenerationType.IDENTITY) private Long id; private Long userId; private String type; // "BUY" or "SELL" private BigDecimal price; private int quantity; private boolean matched; // ...其他字段、Getter和Setter } @Entity public class Trade { @Id @GeneratedValue(strategy = GenerationType.IDENTITY) private Long id; private Long buyerId; private Long sellerId; private BigDecimal price; private int quantity; private LocalDateTime tradeTime; // ...其他字段、Getter和Setter } // 仓库接口 public interface OrderRepository extends JpaRepository<Order, Long> { List<Order> findByTypeAndMatchedOrderByPriceAsc(String type, boolean matched); } public interface TradeRepository extends JpaRepository<Trade, Long> { } // 服务类 @Service public class TradingService { @Autowired private OrderRepository orderRepository; @Autowired private TradeRepository tradeRepository; @Transactional public void createOrder(Order order) { orderRepository.save(order); matchOrders(order); } private void matchOrders(Order newOrder) { List<Order> oppositeOrders; if ("BUY".equals(newOrder.getType())) { oppositeOrders = orderRepository.findByTypeAndMatchedOrderByPriceAsc("SELL", false); } else { oppositeOrders = orderRepository.findByTypeAndMatchedOrderByPriceDesc("BUY", false); } for (Order oppositeOrder : oppositeOrders) { if (oppositeOrder.getPrice().compareTo(newOrder.getPrice()) == 0) { int quantity = Math.min(oppositeOrder.getQuantity(), newOrder.getQuantity()); createTrade(oppositeOrder.getUserId(), newOrder.getUserId(), oppositeOrder.getPrice(), quantity); updateOrdersMatchedStatus(oppositeOrder, newOrder, quantity); } } } private void createTrade(Long buyerId, Long sellerId, BigDecimal price, int quantity) { Trade trade = new Trade(); trade.setBuyerId(buyerId); trade.setSellerId(sellerId); trade.setPrice(price); trade.setQuantity(quantity); trade.setTradeTime(LocalDateTime.now()); tradeRepository.save(trade); } private void updateOrdersMatchedStatus(Order... orders) { for (Order order : orders) { order.setMatched(true); orderRepository.save(order); } } }
网友回复