Pu Zhibing
2 天以前 5dacdee9b54c78372b68140e2b068d03a620eab9
ManagementQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/specialTrain/TDispatchController.java
@@ -6,22 +6,23 @@
import com.stylefeng.guns.core.base.tips.ErrorTip;
import com.stylefeng.guns.core.beetl.ShiroExtUtil;
import com.stylefeng.guns.core.common.constant.factory.PageFactory;
import com.stylefeng.guns.core.log.LogObjectHolder;
import com.stylefeng.guns.core.shiro.ShiroKit;
import com.stylefeng.guns.core.util.SinataUtil;
import com.stylefeng.guns.modular.system.model.TCompany;
import com.stylefeng.guns.modular.system.model.TDispatch;
import com.stylefeng.guns.modular.system.service.ITCompanyService;
import com.stylefeng.guns.modular.system.service.ITDispatchService;
import com.stylefeng.guns.modular.system.service.IUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.beans.factory.annotation.Autowired;
import com.stylefeng.guns.core.log.LogObjectHolder;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import com.stylefeng.guns.modular.system.model.TDispatch;
import com.stylefeng.guns.modular.system.service.ITDispatchService;
import org.springframework.web.bind.annotation.ResponseBody;
import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
@@ -36,46 +37,48 @@
@Controller
@RequestMapping("/tDispatch")
public class TDispatchController extends BaseController {
    private String PREFIX = "/system/tDispatch/";
    @Autowired
    private ITDispatchService tDispatchService;
    @Autowired
    private ITCompanyService tCompanyService;
    @Autowired
    private IUserService userService;
    /**
     * 跳转到调度管理首页
     */
    @RequestMapping("")
    public String index() {
        return PREFIX + "tDispatch.html";
    }
    /**
   private String PREFIX = "/system/tDispatch/";
   @Autowired
   private ITDispatchService tDispatchService;
   @Autowired
   private ITCompanyService tCompanyService;
   @Autowired
   private IUserService userService;
   @Resource
   private ShiroExtUtil shiroExtUtil;
   /**
    * 跳转到调度管理首页
    */
   @RequestMapping("")
   public String index() {
      return PREFIX + "tDispatch.html";
   }
   /**
     * 跳转到添加调度管理
     */
    @RequestMapping("/tDispatch_add")
    public String tDispatchAdd(Model model) {
        model.addAttribute("roleType", ShiroExtUtil.getUser().getRoleType());
        List<TCompany> companyList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 2)
                .eq("isCross", 1)
                .last(" and state != 1 and FIND_IN_SET(flag,'1,2')"));
        model.addAttribute("companyList",companyList);
        if (2 == ShiroExtUtil.getUser().getRoleType()) {
            List<TCompany> franchiseeList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 3)
                    .eq("superiorId", ShiroExtUtil.getUser().getObjectId()));
            model.addAttribute("franchiseeList", franchiseeList);
        } else {
            model.addAttribute("franchiseeList", null);
        }
        return PREFIX + "tDispatch_add.html";
       model.addAttribute("roleType", shiroExtUtil.getUser().getRoleType());
       List<TCompany> companyList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 2)
             .eq("isCross", 1)
             .last(" and state != 1 and FIND_IN_SET(flag,'1,2')"));
       model.addAttribute("companyList", companyList);
       if (2 == shiroExtUtil.getUser().getRoleType()) {
          List<TCompany> franchiseeList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 3)
                .eq("superiorId", shiroExtUtil.getUser().getObjectId()));
          model.addAttribute("franchiseeList", franchiseeList);
       } else {
          model.addAttribute("franchiseeList", null);
       }
       return PREFIX + "tDispatch_add.html";
    }
    /**
@@ -102,26 +105,26 @@
     */
    @RequestMapping("/tDispatch_update/{tDispatchId}")
    public String tDispatchUpdate(@PathVariable Integer tDispatchId, Model model) {
        TDispatch tDispatch = tDispatchService.selectById(tDispatchId);
        model.addAttribute("item",tDispatch);
        LogObjectHolder.me().set(tDispatch);
        Integer roleType = ShiroExtUtil.getUser().getRoleType();
        model.addAttribute("roleType",roleType);
        if (1 == roleType){
            List<TCompany> companyList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 2)
                    .eq("isCross", 1)
                    .last(" and state != 1 and FIND_IN_SET(flag,'1,2')"));
            model.addAttribute("companyList",companyList);
            List<TCompany> franchiseeList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 3)
                    .eq("superiorId",tDispatch.getCompanyId())
                    .eq("isCross", 1)
       TDispatch tDispatch = tDispatchService.selectById(tDispatchId);
       model.addAttribute("item", tDispatch);
       LogObjectHolder.me().set(tDispatch);
       Integer roleType = shiroExtUtil.getUser().getRoleType();
       model.addAttribute("roleType", roleType);
       if (1 == roleType) {
          List<TCompany> companyList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 2)
                .eq("isCross", 1)
                .last(" and state != 1 and FIND_IN_SET(flag,'1,2')"));
          model.addAttribute("companyList", companyList);
          List<TCompany> franchiseeList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 3)
                .eq("superiorId", tDispatch.getCompanyId())
                .eq("isCross", 1)
                    .last(" and state != 1 and FIND_IN_SET(flag,'1,2')"));
            model.addAttribute("franchiseeList",franchiseeList);
        }else if (2 == roleType){
            List<TCompany> franchiseeList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 3)
                    .eq("superiorId", ShiroExtUtil.getUser().getObjectId())
                  .eq("superiorId", shiroExtUtil.getUser().getObjectId())
                    .eq("isCross", 1)
                    .last(" and state != 1 and FIND_IN_SET(flag,'1,2')"));
            model.addAttribute("franchiseeList",franchiseeList);
@@ -143,7 +146,7 @@
            endTime = timeArray[1];
        }
        Page<Map<String, Object>> page = new PageFactory<Map<String, Object>>().defaultPage();
        page.setRecords(tDispatchService.getDispatchList(page, ShiroExtUtil.getUser().getRoleType(), ShiroExtUtil.getUser().getObjectId(), beginTime, endTime, name, companyName, franchiseeName, account, state));
       page.setRecords(tDispatchService.getDispatchList(page, shiroExtUtil.getUser().getRoleType(), shiroExtUtil.getUser().getObjectId(), beginTime, endTime, name, companyName, franchiseeName, account, state));
        return super.packForBT(page);
    }
@@ -153,31 +156,31 @@
    @RequestMapping(value = "/add")
    @ResponseBody
    public Object add(TDispatch tDispatch) {
        //判断账号是否存在
        int count = tDispatchService.selectCount(new EntityWrapper<TDispatch>().eq("account", tDispatch.getAccount()).last(" and FIND_IN_SET(state,'1,2')"));
        if (count > 0) {
            return new ErrorTip(500, "账号已存在,请重新输入");
        }
        Integer roleType = ShiroExtUtil.getUser().getRoleType();
        Integer objectId = ShiroExtUtil.getUser().getObjectId();
        if (2 == roleType) {
            tDispatch.setCompanyId(objectId);
        } else if (3 == roleType) {
            TCompany company = tCompanyService.selectById(objectId);
            tDispatch.setCompanyId(company.getSuperiorId());
            tDispatch.setFranchiseeId(objectId);
        }
        tDispatch.setInsertTime(new Date());
        tDispatch.setInsertUserId(ShiroExtUtil.getUser().getId());
        tDispatch.setInsertUserRole(roleType);
        tDispatch.setState(1);
        if(tDispatch.getCompanyId() == null){
            Integer objectId1 = userService.selectById(ShiroExtUtil.getUser().getId()).getObjectId();
            tDispatch.setCompanyId(objectId1);
        }
        tDispatch.setPassword(ShiroKit.md5(tDispatch.getPassword(), "WL:x9#"));
        tDispatchService.insert(tDispatch);
        return SUCCESS_TIP;
       //判断账号是否存在
       int count = tDispatchService.selectCount(new EntityWrapper<TDispatch>().eq("account", tDispatch.getAccount()).last(" and FIND_IN_SET(state,'1,2')"));
       if (count > 0) {
          return new ErrorTip(500, "账号已存在,请重新输入");
       }
       Integer roleType = shiroExtUtil.getUser().getRoleType();
       Integer objectId = shiroExtUtil.getUser().getObjectId();
       if (2 == roleType) {
          tDispatch.setCompanyId(objectId);
       } else if (3 == roleType) {
          TCompany company = tCompanyService.selectById(objectId);
          tDispatch.setCompanyId(company.getSuperiorId());
          tDispatch.setFranchiseeId(objectId);
       }
       tDispatch.setInsertTime(new Date());
       tDispatch.setInsertUserId(shiroExtUtil.getUser().getId());
       tDispatch.setInsertUserRole(roleType);
       tDispatch.setState(1);
       if (tDispatch.getCompanyId() == null) {
          Integer objectId1 = userService.selectById(shiroExtUtil.getUser().getId()).getObjectId();
          tDispatch.setCompanyId(objectId1);
       }
       tDispatch.setPassword(ShiroKit.md5(tDispatch.getPassword(), "WL:x9#"));
       tDispatchService.insert(tDispatch);
       return SUCCESS_TIP;
    }
    /**
@@ -207,28 +210,28 @@
    @RequestMapping(value = "/update")
    @ResponseBody
    public Object update(TDispatch tDispatch) {
        TDispatch obj = tDispatchService.selectById(tDispatch.getId());
        if (!obj.getAccount().equals(tDispatch.getAccount())) {
            //判断账号是否存在
            int count = tDispatchService.selectCount(new EntityWrapper<TDispatch>().eq("account", tDispatch.getAccount()).last(" and FIND_IN_SET(state,'1,2')"));
            if (count > 0) {
                return new ErrorTip(500, "账号已存在,请重新输入");
            }
        }
        Integer roleType = ShiroExtUtil.getUser().getRoleType();
        Integer objectId = ShiroExtUtil.getUser().getObjectId();
        if (2 == roleType) {
            tDispatch.setCompanyId(objectId);
        } else if (3 == roleType) {
            TCompany company = tCompanyService.selectById(objectId);
            tDispatch.setCompanyId(company.getSuperiorId());
            tDispatch.setFranchiseeId(objectId);
        }
        if (SinataUtil.isNotEmpty(tDispatch.getPassword())) {
            tDispatch.setPassword(ShiroKit.md5(tDispatch.getPassword(), "WL:x9#"));
        }
        tDispatchService.updateById(tDispatch);
        return SUCCESS_TIP;
       TDispatch obj = tDispatchService.selectById(tDispatch.getId());
       if (!obj.getAccount().equals(tDispatch.getAccount())) {
          //判断账号是否存在
          int count = tDispatchService.selectCount(new EntityWrapper<TDispatch>().eq("account", tDispatch.getAccount()).last(" and FIND_IN_SET(state,'1,2')"));
          if (count > 0) {
             return new ErrorTip(500, "账号已存在,请重新输入");
          }
       }
       Integer roleType = shiroExtUtil.getUser().getRoleType();
       Integer objectId = shiroExtUtil.getUser().getObjectId();
       if (2 == roleType) {
          tDispatch.setCompanyId(objectId);
       } else if (3 == roleType) {
          TCompany company = tCompanyService.selectById(objectId);
          tDispatch.setCompanyId(company.getSuperiorId());
          tDispatch.setFranchiseeId(objectId);
       }
       if (SinataUtil.isNotEmpty(tDispatch.getPassword())) {
          tDispatch.setPassword(ShiroKit.md5(tDispatch.getPassword(), "WL:x9#"));
       }
       tDispatchService.updateById(tDispatch);
       return SUCCESS_TIP;
    }
}