Service to create order finished, needs to handle multi-thread issues, Will use Redis Lock later

Author: csjihe

src/main/java/com/csjihe/springbootwechatordersystem/dataobject/OrderMaster.java

@@ -11,6 +11,7 @@
 import java.math.BigDecimal;
 import java.util.Date;
 
+
 @Entity
 @Data
 @DynamicUpdate
@@ -49,4 +50,5 @@ public class OrderMaster {
     /** Order updated time */
     /** Add @DynamicUpdate annotation */
     private Date updateTime;
+
 }

src/main/java/com/csjihe/springbootwechatordersystem/dto/CartDTO.java

@@ -0,0 +1,24 @@
+package com.csjihe.springbootwechatordersystem.dto;
+
+
+import lombok.Data;
+
+/**
+ * Shopping cart
+ */
+@Data
+public class CartDTO {
+
+    /** Product Id */
+    private String productId;
+
+
+    /** Product Quantity */
+    private Integer productQuantity;
+
+
+    public CartDTO(String productId, Integer productQuantity) {
+        this.productId = productId;
+        this.productQuantity = productQuantity;
+    }
+}

src/main/java/com/csjihe/springbootwechatordersystem/dto/OrderDTO.java

@@ -0,0 +1,51 @@
+package com.csjihe.springbootwechatordersystem.dto;
+
+
+import com.csjihe.springbootwechatordersystem.dataobject.OrderDetail;
+import com.csjihe.springbootwechatordersystem.enums.OrderStatusEnum;
+import com.csjihe.springbootwechatordersystem.enums.PayStatusEnum;
+import lombok.Data;
+
+import javax.persistence.Id;
+import java.math.BigDecimal;
+import java.util.Date;
+import java.util.List;
+
+@Data
+public class OrderDTO {
+
+    /** Order ID */
+    private String orderId;
+
+    /** Buyer Name */
+    private String buyerName;
+
+    /** Buyer Phone */
+    private String buyerPhone;
+
+    /** Buyer Address */
+    private String buyerAddress;
+
+    /** Buyer Wechat OpenId */
+    private String buyerOpenid;
+
+    /**  Buyer Order Amount */
+    private BigDecimal orderAmount;
+
+    /** Order status, default as new order */
+    private Integer orderStatus;
+
+    /** Payment status, default as unpaid denoted as 0 */
+    private Integer payStatus;
+
+    /**  Order created time */
+    /** Add @DynamicUpdate annotation */
+    private Date createTime;
+
+
+    /** Order updated time */
+    /** Add @DynamicUpdate annotation */
+    private Date updateTime;
+
+    List<OrderDetail> orderDetailList;
+}

src/main/java/com/csjihe/springbootwechatordersystem/enums/ResultEnum.java

@@ -0,0 +1,21 @@
+package com.csjihe.springbootwechatordersystem.enums;
+
+
+import lombok.Getter;
+
+@Getter
+public enum ResultEnum {
+
+    PRODUCT_NOT_EXIST(10, "Product Not Exist"),
+
+    PRODUCT_STOCK_ERROR(11, "Stock Error"),
+    ;
+
+    private Integer code;
+    private String message;
+
+    ResultEnum(Integer code, String message) {
+        this.code = code;
+        this.message = message;
+    }
+}

src/main/java/com/csjihe/springbootwechatordersystem/exception/SellException.java

@@ -0,0 +1,13 @@
+package com.csjihe.springbootwechatordersystem.exception;
+
+import com.csjihe.springbootwechatordersystem.enums.ResultEnum;
+
+public class SellException extends RuntimeException{
+
+    private Integer code;
+
+    public SellException(ResultEnum resultEnum) {
+        super(resultEnum.getMessage());
+        this.code = resultEnum.getCode();
+    }
+}

src/main/java/com/csjihe/springbootwechatordersystem/service/OrderService.java

@@ -0,0 +1,26 @@
+package com.csjihe.springbootwechatordersystem.service;
+
+import com.csjihe.springbootwechatordersystem.dto.OrderDTO;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.Pageable;
+
+public interface OrderService {
+
+    /** Set up an order */
+    OrderDTO create(OrderDTO orderDTO);
+
+    /** Query a single order */
+    OrderDTO findOne(String OrderId);
+
+    /** Query a list of orders */
+    Page<OrderDTO> findList(String buyerOpenid, Pageable pageable);
+
+    /** Cancel an order */
+    OrderDTO cancel(OrderDTO orderDTO);
+
+    /** Finish an order */
+    OrderDTO finish(OrderDTO orderDTO);
+
+    /** Pay an order */
+    OrderDTO paid(OrderDTO orderDTO);
+}

src/main/java/com/csjihe/springbootwechatordersystem/service/ProductService.java

@@ -1,6 +1,7 @@
 package com.csjihe.springbootwechatordersystem.service;
 
 import com.csjihe.springbootwechatordersystem.dataobject.ProductInfo;
+import com.csjihe.springbootwechatordersystem.dto.CartDTO;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.Pageable;
 
@@ -19,7 +20,8 @@ public interface ProductService {
     ProductInfo save(ProductInfo productInfo);
 
     // Increase stock
-
+    void increaseStock(List<CartDTO> cartDTOList);
 
     // Decrease stock
+    void decreaseStock(List<CartDTO> cartDTOList);
 }

src/main/java/com/csjihe/springbootwechatordersystem/service/implementation/OrderServiceImpl.java

@@ -0,0 +1,123 @@
+package com.csjihe.springbootwechatordersystem.service.implementation;
+
+import com.csjihe.springbootwechatordersystem.dataobject.OrderDetail;
+import com.csjihe.springbootwechatordersystem.dataobject.OrderMaster;
+import com.csjihe.springbootwechatordersystem.dataobject.ProductInfo;
+import com.csjihe.springbootwechatordersystem.dto.CartDTO;
+import com.csjihe.springbootwechatordersystem.dto.OrderDTO;
+import com.csjihe.springbootwechatordersystem.enums.OrderStatusEnum;
+import com.csjihe.springbootwechatordersystem.enums.PayStatusEnum;
+import com.csjihe.springbootwechatordersystem.enums.ResultEnum;
+import com.csjihe.springbootwechatordersystem.exception.SellException;
+import com.csjihe.springbootwechatordersystem.repository.OrderDetailRepository;
+import com.csjihe.springbootwechatordersystem.repository.OrderMasterRepository;
+import com.csjihe.springbootwechatordersystem.service.OrderService;
+import com.csjihe.springbootwechatordersystem.service.ProductService;
+import com.csjihe.springbootwechatordersystem.utils.KeyUtil;
+import org.springframework.beans.BeanUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.Pageable;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+
+import java.math.BigDecimal;
+import java.math.BigInteger;
+import java.util.List;
+import java.util.stream.Collectors;
+
+
+@Service
+public class OrderServiceImpl implements OrderService {
+
+    @Autowired
+    private ProductService productService;
+
+    @Autowired
+    private OrderDetailRepository orderDetailRepository;
+
+    @Autowired
+    private OrderMasterRepository orderMasterRepository;
+
+
+//    List<CartDTO> cartDTOS = new ArrayList<>();
+
+    @Override
+    @Transactional
+    public OrderDTO create(OrderDTO orderDTO) {
+
+        String orderId = KeyUtil.genUniqueKey();
+
+        BigDecimal orderAmount = new BigDecimal(BigInteger.ZERO);
+
+        // 1. 查询商品 数量 价格 库存 等
+        for (OrderDetail orderDetail : orderDTO.getOrderDetailList()) {
+            ProductInfo productInfo = productService.findOne(orderDetail.getProductId());
+            if (productInfo == null) {
+                throw new SellException(ResultEnum.PRODUCT_NOT_EXIST);
+            }
+
+            // 2. 计算总价
+            orderAmount = productInfo.getProductPrice()
+                    .multiply(new BigDecimal(orderDetail.getProductQuantity()))
+            .add(orderAmount);
+
+            //  订单详情入库 OrderDetail
+            orderDetail.setDetailId(KeyUtil.genUniqueKey());
+            orderDetail.setOrderId(orderId);
+
+            // copy other Product properties from productInfo to orderDetail
+            BeanUtils.copyProperties(productInfo, orderDetail);
+
+            orderDetailRepository.save(orderDetail);
+
+        }
+
+
+        // 3. 写入订单数据库 OrderMaster
+        OrderMaster orderMaster = new OrderMaster();
+        BeanUtils.copyProperties(orderDTO, orderMaster);
+        orderMaster.setOrderId(orderId);
+        orderMaster.setOrderAmount(orderAmount);
+        orderMaster.setOrderStatus(OrderStatusEnum.NEW.getCode());
+        orderMaster.setPayStatus(PayStatusEnum.WAIT.getCode());
+
+
+        orderMasterRepository.save(orderMaster);
+
+        // 4. Update 库存
+        List<CartDTO> cartDTOS = orderDTO.getOrderDetailList().stream().map(e ->
+            new CartDTO(e.getProductId(), e.getProductQuantity())
+        ).collect(Collectors.toList());
+
+
+        productService.decreaseStock(cartDTOS);
+        return orderDTO;
+    }
+
+    @Override
+    public OrderDTO findOne(String OrderId) {
+        return null;
+    }
+
+    @Override
+    public Page<OrderDTO> findList(String buyerOpenid, Pageable pageable) {
+        return null;
+    }
+
+    @Override
+    public OrderDTO cancel(OrderDTO orderDTO) {
+        return null;
+    }
+
+    @Override
+    public OrderDTO finish(OrderDTO orderDTO) {
+        return null;
+    }
+
+    @Override
+    public OrderDTO paid(OrderDTO orderDTO) {
+        return null;
+    }
+}

src/main/java/com/csjihe/springbootwechatordersystem/service/implementation/ProductServiceImpl.java

@@ -1,13 +1,17 @@
 package com.csjihe.springbootwechatordersystem.service.implementation;
 
 import com.csjihe.springbootwechatordersystem.dataobject.ProductInfo;
+import com.csjihe.springbootwechatordersystem.dto.CartDTO;
 import com.csjihe.springbootwechatordersystem.enums.ProductStatusEnum;
+import com.csjihe.springbootwechatordersystem.enums.ResultEnum;
+import com.csjihe.springbootwechatordersystem.exception.SellException;
 import com.csjihe.springbootwechatordersystem.repository.ProductInfoRepository;
 import com.csjihe.springbootwechatordersystem.service.ProductService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.Pageable;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
 import java.util.List;
 
@@ -37,4 +41,29 @@ public Page<ProductInfo> findAll(Pageable pageable) {
     public ProductInfo save(ProductInfo productInfo) {
         return repository.save(productInfo);
     }
+
+    @Override
+    public void increaseStock(List<CartDTO> cartDTOList) {
+
+    }
+
+    @Override
+    @Transactional
+    public void decreaseStock(List<CartDTO> cartDTOList) {
+        for (CartDTO cartDTO : cartDTOList) {
+            ProductInfo productInfo = repository.findOne(cartDTO.getProductId());
+            if (productInfo == null) {
+                throw new SellException(ResultEnum.PRODUCT_NOT_EXIST);
+            }
+
+            Integer result = productInfo.getProductStock() - cartDTO.getProductQuantity();
+            if (result < 0) {
+                throw new SellException(ResultEnum.PRODUCT_STOCK_ERROR);
+            }
+
+            productInfo.setProductStock(result);
+
+            repository.save(productInfo);
+        }
+    }
 }

src/main/java/com/csjihe/springbootwechatordersystem/utils/KeyUtil.java

@@ -0,0 +1,25 @@
+package com.csjihe.springbootwechatordersystem.utils;
+
+
+import java.util.Random;
+
+/**
+ * Utility class generating Database key
+ */
+public class KeyUtil {
+
+    /**
+     * 声称唯一primary key
+     * format: time + randomNumber
+     * @return
+     */
+    public static synchronized String genUniqueKey() {
+
+
+        Random random = new Random();
+
+        Integer number = random.nextInt(900000) + 100000; // 6-bit random number
+
+        return System.currentTimeMillis() + String.valueOf(number);
+    }
+}