SEEC-Pay/src/main/java/com/nju/edu/pay/controller/api/OrderController.java

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())));
    }
}