hust-epi/whepi-web/src/main/java/com/bofeng/wx/controller/UserOpenController.java

package com.bofeng.wx.controller;

import com.bofeng.dao.*;
import com.bofeng.entity.*;
import com.bofeng.service.HomeService;
import com.bofeng.service.QzTaskService;
import com.bofeng.service.RbService;
import com.bofeng.service.UserOpenService;
import com.yvan.platform.YvanUtil;
import com.yvan.springmvc.ResultModel;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.List;

@RestController
public class UserOpenController {

    @Autowired
    private HomeService homeService;
    @Autowired
    private UserOpenService userOpenService;
    @Autowired
    private UserOpenMapper userOpenMapper;
    @Autowired
    private UptownMapper uptownMapper;
    @Autowired
    private UptownHouseMapper uptownHouseMapper;
    @Autowired
    private UptownHomeMapper uptownHomeMapper;
    @Autowired
    private UserRoleMapper userRoleMapper;
    @Autowired
    private UptownUnitMapper uptownUnitMapper;
    @Autowired
    private QzTaskService qzTaskService;
    @Autowired
    private RbService rbService;

    @GetMapping("/user/home.html")
    public ModelAndView home(ModelMap model, @RequestParam(value = "userId", required = false,defaultValue = "0") Long userId) {
        List<UserRole> list = new ArrayList<>();
        if (userId == 0) {
            UserOpen userOpen = homeService.getUserOpen();
            list = userRoleMapper.getUserRoleByUserId(userOpen.getUserId());
//            list = userRoleMapper.getUserRoleByUserId(1224706869998772226L);
//            UserOpen userOpen = userOpenMapper.selectByUserId(1224706869998772226L);
            userId = userOpen.getUserId();
            model.put("user", userOpen);
            model.put("user_id", "\""+userOpen.getUserId()+"\"");
        } else {
            model.put("user", userOpenMapper.selectByUserId(userId));
            list = userRoleMapper.getUserRoleByUserId(userId);
            model.put("user_id", "\""+userId+"\"");
        }
        if (list.size() == 0) {
            return new ModelAndView("/user/home.ftl", model);
        } else if (list.size()>0) {
            if (list.get(0).getRoleId() == 1) {
                return new ModelAndView("/home/home.ftl", model);
            } else if(list.get(0).getRoleId() == 2) {
                List<QzTask> taskList = qzTaskService.selectAll();
                model.put("taskList", YvanUtil.toJsonPretty(taskList));
                List<sysUptownUnit> rbList = rbService.selectAll(userId);
                model.put("rbList", YvanUtil.toJsonPretty(rbList));
                return new ModelAndView("/yeweihui/home.ftl", model);
            }
        }
        return null;
    }

    @GetMapping("/home/getUptownData.json")
    public List<Uptown> getUptownData(HttpServletRequest request, @RequestParam(value = "area") Integer area) {
        return uptownMapper.getAllUptown(area);
    }

    @GetMapping("/home/getUptownUnitData.json")
    public List<UptownUnit> getUptownUnitData(HttpServletRequest request, @RequestParam(value = "uptownId") Long uptownId) {
        return uptownUnitMapper.getUptownUnitData(uptownId);
    }

    @PostMapping("/home/checkHouseId.json")
    public ResultModel checkHouseId(HttpServletRequest request) {
        ResultModel resultModel = new ResultModel();
        Long unitId = Long.parseLong(request.getParameter("unitId"));
        String doorplate = request.getParameter("doorplate");
        List<UptownHome> list = uptownHouseMapper.getUptownHouseBy(unitId, doorplate);
        if (list.size() > 0) {
            resultModel.put("success", false);
            resultModel.put("msg", "该门牌已有人入住！");
            return resultModel;
        }
        resultModel.put("success", true);
        resultModel.put("msg", "操作成功");
        return resultModel;
    }

    @PostMapping("/home/saveUser.json")
    public void saveUser(ModelMap model, HttpServletRequest request) {
        Long userId = Long.parseLong(request.getParameter("userId"));
        Integer userType = Integer.parseInt(request.getParameter("userType"));
        String linkman = request.getParameter("linkman");
        String phone = request.getParameter("phone");
        Integer area = Integer.parseInt(request.getParameter("area"));
        Long uptownId = Long.parseLong(request.getParameter("uptownId"));
        Long unitId = 0L;
        String doorplate = "";
        if (userType == 1) {
            unitId = Long.parseLong(request.getParameter("unitId"));
            doorplate = request.getParameter("doorplate");
        }
        userOpenService.saveUserInfo(userId,userType,linkman,phone,area,uptownId,unitId,doorplate);
//        UserOpen userOpen = userOpenMapper.selectByUserId(userId);
//        model.put("user", userOpen);
//        if (userType == 1) {
//            return new ModelAndView("redirect:/home/home.html");
//        } else if (userType == 2) {
//            List<QzTask> taskList = qzTaskService.selectAll();
//            model.put("taskList", YvanUtil.toJsonPretty(taskList));
//            return new ModelAndView("redirect:/yeweihui/home.html");
//        } else {
//            return null;
//        }

    }
}