4dkankan-manage-jp/src/main/java/com/fdkankan/manage_jp/controller/SceneProController.java

package com.fdkankan.manage_jp.controller;


import com.fdkankan.common.util.JwtUtil;
import com.fdkankan.manage_jp.common.Result;
import com.fdkankan.manage_jp.entity.User;
import com.fdkankan.manage_jp.entity.UserRole;
import com.fdkankan.manage_jp.httpClient.client.FdKKClient;
import com.fdkankan.manage_jp.service.IDownService;
import com.fdkankan.manage_jp.service.ISceneProService;
import com.fdkankan.manage_jp.service.IUserRoleService;
import com.fdkankan.manage_jp.service.IUserService;
import com.fdkankan.manage_jp.vo.request.SceneParam;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * <p>
 * pro场景表 前端控制器
 * </p>
 *
 * @author 
 * @since 2022-12-30
 */
@RestController
@RequestMapping("/manage_jp/scene")
public class SceneProController extends BaseController{

    @Autowired
    ISceneProService sceneProService;
    @Autowired
    IUserService userService;
    @Autowired
    IUserRoleService userRoleService;
    @Autowired
    IDownService downService;
    @Autowired
    FdKKClient fdKKClient;

    @PostMapping("/list")
    public Result list(@RequestBody SceneParam param){
        String username = JwtUtil.getUsername(getToken());
        User user = userService.getByUserName(username);

        Set<Long> roleIds = userRoleService.getByUser(user);
        if(!roleIds.contains(5L)){      //平台管理员
            param.setCompanyId(user.getCompanyId());
            if(roleIds.contains(6L)){
                param.setUserId(user.getId());
            }
        }
        param.setToken(getToken());
        return Result.success(sceneProService.pageList(param));
    }


    /**
     * 增加场景下载次数
     */
    @RequestMapping(value = "/addDownloadNum", method = RequestMethod.GET)
    public Result addDownloadNum(@RequestParam(required = false) String sceneNum) throws Exception{
        //sceneProService.addDownloadNum(sceneNum);
        return Result.success();
    }



    @GetMapping("/upgradeToV4")
    public Result upgradeToV4(@RequestParam(required = false) String num){
        HashMap<String,Object> map = new HashMap<>();
        map.put("num",num);
        return Result.success(fdKKClient.upgradeToV4(map));
    }

}