Java语言 Springboot或者ssm框架餐厅管理系统

package com.example.service; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.date.DateUtil; import cn.hutool.core.util.RandomUtil; import com.example.common.ResultCode; import com.example.dao.OrderGoodsRelDao; import com.example.dao.OrderInfoDao; import com.example.entity.*; import com.example.exception.CustomException; import com.github.pagehelper.PageHelper; import com.github.pagehelper.PageInfo; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import javax.annotation.Resource; import javax.servlet.http.HttpServletRequest; import java.util.ArrayList; import java.util.Date; import java.util.List; import java.util.Map; import java.util.stream.Collectors; @Service public class OrderInfoService { @Resource private OrderInfoDao orderInfoDao; @Resource private GoodsInfoService goodsInfoService; @Resource private AdminInfoService adminInfoService; @Resource private SaleInfoService saleInfoService; @Resource private UserInfoService userInfoService; @Resource private OrderGoodsRelDao orderGoodsRelDao; @Resource private CartInfoService cartInfoService; /** * 根据id查询订单信息 */ public OrderInfo findById(Long id) { OrderInfo orderInfo = orderInfoDao.selectByPrimaryKey(id); packOrder(orderInfo); return orderInfo; } public List<OrderInfo> findAll(Long userId, Integer level) { List<OrderInfo> orderInfos = orderInfoDao.findByUserId(userId, level); for (OrderInfo orderInfo : orderInfos) { packOrder(orderInfo); } return orderInfos; } /** * 包装订单的用户和菜品信息 */ private void packOrder(OrderInfo orderInfo) { Long orderId = orderInfo.getId(); List<OrderGoodsRel> rels = orderGoodsRelDao.findByOrderId(orderId); orderInfo.setUserInfo(getUserInfo(orderInfo.getUserId(), orderInfo.getLevel())); List<GoodsInfo> goodsList = CollUtil.newArrayList(); orderInfo.setGoodsList(goodsList); for (OrderGoodsRel rel : rels) { GoodsInfo goodsDetailInfo = goodsInfoService.findById(rel.getGoodsId()); if (goodsDetailInfo != null) { // 注意这里返回的count是用户加入菜品的数量，而不是菜品的库存 goodsDetailInfo.setCount(rel.getCount()); goodsList.add(goodsDetailInfo); } } } /** * 分页查询订单信息 */ public PageInfo<OrderInfo> findEndPages(Integer pageNum, Integer pageSize, HttpServletRequest request) { Account user = (Account) request.getSession().getAttribute("user"); if (user == null) { throw new CustomException("1001", "session已失效，请重新登录"); } Integer level = user.getLevel(); Long userId = user.getId(); PageHelper.startPage(pageNum, pageSize); List<OrderInfo> orderInfos; if (1 == level) { orderInfos = orderInfoDao.selectAll(); } else if (userId != null){ orderInfos = orderInfoDao.findByEndUserId(userId, null, level); } else { orderInfos = new ArrayList<>(); } for (OrderInfo orderInfo : orderInfos) { packOrder(orderInfo); } return PageInfo.of(orderInfos); } public PageInfo<OrderInfo> findFrontPages(Long userId, Integer level, Integer pageNum, Integer pageSize) { PageHelper.startPage(pageNum, pageSize); List<OrderInfo> orderInfos; if (userId != null){ orderInfos = orderInfoDao.findByUserId(userId, level); } else { orderInfos = new ArrayList<>(); } for (OrderInfo orderInfo : orderInfos) { packOrder(orderInfo); } return PageInfo.of(orderInfos); } /** * 分页查询订单信息 */ public PageInfo<OrderInfo> findPages(Long userId, Integer level, Integer pageNum, Integer pageSize) { PageHelper.startPage(pageNum, pageSize); List<OrderInfo> orderInfos; if (userId != null) { orderInfos = orderInfoDao.findByUserId(userId, level); } else { orderInfos = orderInfoDao.selectAll(); } for (OrderInfo orderInfo : orderInfos) { packOrder(orderInfo); } return PageInfo.of(orderInfos); } /** * 下单 */ @Transactional public OrderInfo add(OrderInfo info) { Long userId = info.getUserId(); Integer level = info.getLevel(); Account userInfo = getUserInfo(userId, level); List<GoodsInfo> goodsList = info.getGoodsList(); // 将菜品分分类，根据上传用户id_用户level来确定唯一的key Map<String, List<GoodsInfo>> collectMap = goodsList.stream() .collect(Collectors.groupingBy(x -> x.getUserId() + "_" + x.getLevel())); for (String key : collectMap.keySet()) { List<GoodsInfo> list = collectMap.get(key); // 这里面创建一个订单 OrderInfo orderInfo = new OrderInfo(); orderInfo.setUserId(userId); orderInfo.setLevel(level); orderInfo.setStatus("待付款"); orderInfo.setLinkAddress(userInfo.getAddress()); orderInfo.setLinkMan(userInfo.getNickName()); orderInfo.setLinkPhone(userInfo.getPhone()); orderInfo.setCreateTime(DateUtil.formatDateTime(new Date())); // 订单id：用户id + 当前年月日时分 + 4位流水号 String orderId = userId + DateUtil.format(new Date(), "yyyyMMddHHmm") + RandomUtil.randomNumbers(4); orderInfo.setOrderId(orderId); // 生成订单 orderInfoDao.insertSelective(orderInfo); double totalPrice = 0; for (GoodsInfo orderGoodsVO : list) { Long goodsId = orderGoodsVO.getId(); // 查询菜品信息 GoodsInfo goodsDetail = goodsInfoService.findById(goodsId); if (goodsDetail != null) { Integer orderCount = orderGoodsVO.getCount() == null ? 0 : orderGoodsVO.getCount(); Integer goodsCount = goodsDetail.getCount() == null ? 0 : goodsDetail.getCount(); // 扣库存 if (orderCount <= goodsCount) { goodsDetail.setCount(goodsCount - orderCount); // 销量 +count int sales = goodsDetail.getSales() == null ? 0 : goodsDetail.getSales(); goodsDetail.setSales(sales + orderCount); goodsInfoService.update(goodsDetail); // 建立关系 OrderGoodsRel orderGoodsRel = new OrderGoodsRel(); orderGoodsRel.setGoodsId(goodsId); orderGoodsRel.setOrderId(orderInfo.getId()); orderGoodsRel.setCount(orderCount); orderGoodsRelDao.insertSelective(orderGoodsRel); totalPrice += goodsDetail.getPrice() * goodsDetail.getDiscount() * orderCount; } } } orderInfo.setTotalPrice(totalPrice); // 更新订单信息 orderInfoDao.updateByPrimaryKeySelective(orderInfo); // 下单 清空购物车 cartInfoService.empty(userId, level); } return info; } @Transactional(rollbackFor = Exception.class) public void add(Long userId, Integer level, List<GoodsInfo> goodsList) { Account userInfo = ge