package com.nju.edu.pay.controller.api; import cn.dev33.satoken.annotation.SaCheckLogin; import cn.dev33.satoken.annotation.SaCheckPermission; import cn.dev33.satoken.stp.StpUtil; import com.nju.edu.pay.controller.dto.CheckOrderDTO; import com.nju.edu.pay.service.OrderService; import com.nju.edu.pay.web.annotation.RootAllowed; import com.nju.edu.pay.web.response.ErrorResponse; import com.nju.edu.pay.web.response.Response; import com.nju.edu.pay.web.response.SuccessResponse; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; @RestController @RequestMapping("/order") public class OrderController { @Autowired private OrderService orderService; @PostMapping("/check") public Response checkOrder(@RequestBody CheckOrderDTO checkOrderDTO) { if (orderService.hasOrder(checkOrderDTO.getUserId(), checkOrderDTO.getProductId(), checkOrderDTO.getPlatformType())) { return new SuccessResponse("success"); } else return new ErrorResponse(500, "failed"); } @GetMapping("/orderList") @SaCheckLogin() @RootAllowed public Response getOrderList() { System.out.println(StpUtil.getLoginId()); if (StpUtil.hasPermission("PAY_MANAGE")) { return new SuccessResponse(orderService.retrieveAll()); } return new SuccessResponse(orderService.getByUserId(Long.parseLong((String) StpUtil.getLoginId()))); } }