潘钊 / bsth_control · Commits

Commit fa6bc2697ff069fb6d760a962d10d479d6d175b2

Authored by 徐烜
1 parent 9652aed9
build: skipped

Update

Inline Side-by-side
Showing 1 changed file with 15 additions and 1 deletions
src/main/java/com/bsth/controller/schedule/core/SchedulePlanController.java
  View file @fa6bc26
@@ -5,7 +5,9 @@ import com.bsth.common.ResponseCode; @@ -5,7 +5,9 @@ import com.bsth.common.ResponseCode;
5 import com.bsth.controller.schedule.BController; 5 import com.bsth.controller.schedule.BController;
6 import com.bsth.entity.schedule.SchedulePlan; 6 import com.bsth.entity.schedule.SchedulePlan;
7 import com.bsth.entity.sys.CompanyAuthority; 7 import com.bsth.entity.sys.CompanyAuthority;
  8 +import com.bsth.entity.sys.SysUser;
8 import com.bsth.service.schedule.SchedulePlanService; 9 import com.bsth.service.schedule.SchedulePlanService;
  10 +import com.bsth.service.sys.SysUserService;
9 import org.springframework.beans.factory.annotation.Autowired; 11 import org.springframework.beans.factory.annotation.Autowired;
10 import org.springframework.web.bind.annotation.*; 12 import org.springframework.web.bind.annotation.*;
11 13
@@ -23,11 +25,23 @@ import java.util.Map; @@ -23,11 +25,23 @@ import java.util.Map;
23 public class SchedulePlanController extends BController<SchedulePlan, Long> { 25 public class SchedulePlanController extends BController<SchedulePlan, Long> {
24 @Autowired 26 @Autowired
25 private SchedulePlanService schedulePlanService; 27 private SchedulePlanService schedulePlanService;
  28 + @Autowired
  29 + private SysUserService sysUserService;
26 30
27 @Override 31 @Override
28 public Map<String, Object> save(@RequestBody SchedulePlan schedulePlan, HttpSession httpSession) { 32 public Map<String, Object> save(@RequestBody SchedulePlan schedulePlan, HttpSession httpSession) {
  33 + // 用户信息
  34 + String userName = String.valueOf(httpSession.getAttribute(Constants.SESSION_USERNAME));
  35 + SysUser sysUser = sysUserService.findByUserName(userName);
  36 +
  37 + Date cdate = new Date();
  38 + schedulePlan.setCreateBy(sysUser);
  39 + schedulePlan.setCreateDate(cdate);
  40 + schedulePlan.setUpdateBy(sysUser);
  41 + schedulePlan.setUpdateDate(cdate);
  42 +
  43 + // 如果多个公司，选第一个，以后改成页面控制
29 List<CompanyAuthority> cmyAuths = (List<CompanyAuthority>) httpSession.getAttribute(Constants.COMPANY_AUTHORITYS); 44 List<CompanyAuthority> cmyAuths = (List<CompanyAuthority>) httpSession.getAttribute(Constants.COMPANY_AUTHORITYS);
30 - // TODO：如果多个公司，选第一个，以后改成页面控制  
31 if (cmyAuths == null || cmyAuths.size() == 0) 45 if (cmyAuths == null || cmyAuths.size() == 0)
32 schedulePlanService.save(schedulePlan, new CompanyAuthority()); 46 schedulePlanService.save(schedulePlan, new CompanyAuthority());
33 else 47 else