4dkankan-smart-site/src/main/java/com/fdkankan/site/controller/TestController.java

package com.fdkankan.site.controller;

import cn.dev33.satoken.annotation.SaCheckLogin;
import cn.dev33.satoken.annotation.SaIgnore;
import cn.hutool.core.util.ObjectUtil;
import com.alibaba.fastjson.JSONObject;
import com.fdkankan.site.common.BaseController;
import com.fdkankan.site.common.ResultCode;
import com.fdkankan.site.common.ResultData;
import com.fdkankan.site.entity.ProjectTeam;
import com.fdkankan.site.entity.User;
import com.fdkankan.site.entity.UserRole;
import com.fdkankan.site.exception.BusinessException;
import com.fdkankan.site.httpClient.bim.BimCallBackDTO;
import com.fdkankan.site.httpClient.bim.BimFaceVO;
import com.fdkankan.site.httpClient.client.BimClient;
import com.fdkankan.site.httpClient.bim.BimUploadParam;
import com.fdkankan.site.httpClient.client.FdKKClient;
import com.fdkankan.site.service.IProjectTeamService;
import com.fdkankan.site.service.IUserRoleService;
import com.fdkankan.site.service.IUserService;
import com.fdkankan.sms.SmsService;
import com.fdkankan.sms.SmsServiceV2;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;

@RestController
@RequestMapping("/test")
@SaCheckLogin
public class TestController extends BaseController {

    @Resource
    FdKKClient fdKKClient;
    @Autowired
    IProjectTeamService projectTeaml;
    @Autowired
    IUserService userService;

    @Autowired
    IUserRoleService userRoleService;
    @Autowired
    SmsServiceV2 smsService;

    @Value("${phone.code.cn}")
    private String cnCode;
    @PostMapping("/test")
    @SaIgnore
    public ResultData test() throws Exception {
        smsService.sendSms("18666186515", "{\"projectname\":\"" +"测试短信" + "\"}", cnCode);
        return ResultData.ok( );
    }

    @PostMapping("/buildUserRole")
    @SaIgnore
    public ResultData buildUserRole(){
        for (ProjectTeam projectTeam : projectTeaml.list()) {
            Integer isCreater = projectTeam.getIsCreater();
            UserRole userRole=new UserRole();
            userRole.setProjectId(projectTeam.getProjectId());
            User byUserName = userService.getById(projectTeam.getUserId());
            if (isCreater==1){
                userRole.setUserId(byUserName.getUserId());
                userRole.setRoleId(1);
            }else {
                userRole.setUserId(byUserName.getUserId());
                userRole.setRoleId(2);
            }
            UserRole byUserIdAndPorId = userRoleService.findByUserIdAndPorId(byUserName.getUserId(), projectTeam.getProjectId());
            if (ObjectUtil.isNull(byUserIdAndPorId)){
                userRoleService.save(userRole);
            }

        }
        return ResultData.ok( );
    }
}