RestrictProjectServiceImpl.java 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115
  1. package com.goafanti.business.service.impl;
  2. import com.goafanti.business.bo.InputRestrictProject;
  3. import com.goafanti.business.bo.RestrictProjectPageList;
  4. import com.goafanti.business.service.RestrictProjectService;
  5. import com.goafanti.common.dao.RestrictProjectMapper;
  6. import com.goafanti.common.dao.UserTransferLogMapper;
  7. import com.goafanti.common.error.BusinessException;
  8. import com.goafanti.common.model.RestrictProject;
  9. import com.goafanti.common.model.UserTransferLog;
  10. import com.goafanti.core.mybatis.BaseMybatisDao;
  11. import com.goafanti.core.mybatis.page.Pagination;
  12. import com.goafanti.core.shiro.token.TokenManager;
  13. import org.springframework.stereotype.Service;
  14. import javax.annotation.Resource;
  15. import java.util.Date;
  16. import java.util.HashMap;
  17. import java.util.Map;
  18. @Service
  19. public class RestrictProjectServiceImpl extends BaseMybatisDao<RestrictProjectMapper> implements RestrictProjectService {
  20. @Resource
  21. private RestrictProjectMapper restrictProjectMapper;
  22. @Resource
  23. private UserTransferLogMapper userTransferLogMapper;
  24. @Override
  25. public int add(InputRestrictProject in) {
  26. String aid = TokenManager.getAdminId();
  27. in.setAid(aid);
  28. in.setType(1);
  29. in.setLockTime(new Date());
  30. RestrictProject use = restrictProjectMapper.selectByPid(in);
  31. if (use != null && use.getType() != 0){
  32. throw new BusinessException("该项目已存在");
  33. }
  34. if(use ==null){
  35. restrictProjectMapper.insertSelective(in);
  36. }else if (use.getType() == 0){
  37. in.setId(use.getId());
  38. restrictProjectMapper.updateByPrimaryKeySelective(in);
  39. }
  40. addUserLog(in,0);
  41. return 1;
  42. }
  43. /**
  44. *
  45. * @param in
  46. * @param i 0=新增 1=移除
  47. */
  48. private void addUserLog(RestrictProject in, int i) {
  49. addUserLog(in,i,null);
  50. }
  51. @Override
  52. public Object list(InputRestrictProject in) {
  53. in.setAid(TokenManager.getAdminId());
  54. return restrictProjectMapper.selectByUidAndAid(in);
  55. }
  56. @Override
  57. public Object update(Integer id,Integer type,String takeAid) {
  58. Date date = new Date();
  59. RestrictProject in = new RestrictProject();
  60. RestrictProject use = restrictProjectMapper.selectByPrimaryKey(id);
  61. in.setId(id);
  62. if (type==0){
  63. in.setReleaseTime(date);
  64. in.setType(0);
  65. addUserLog(use,1);
  66. }else if (type==1){
  67. in.setLockTime(date);
  68. in.setAid(takeAid);
  69. in.setPid(use.getPid());
  70. in.setUid(use.getUid());
  71. addUserLog(use,2,takeAid);
  72. }
  73. return restrictProjectMapper.updateByPrimaryKeySelective(in);
  74. }
  75. @Override
  76. public Object pageList(InputRestrictProject in) {
  77. Map<String,Object> param = new HashMap<>();
  78. if (in.getUserName()!=null)param.put("userName",in.getUserName());
  79. if (in.getType()!=null)param.put("type",in.getType());
  80. if (in.getStartTime()!=null)param.put("startTime",in.getStartTime());
  81. if (in.getEndTime()!=null)param.put("endTime",in.getEndTime()+" 23:59:59");
  82. if (in.getProjectName()!=null)param.put("projectName",in.getProjectName());
  83. Pagination<RestrictProjectPageList> page =
  84. (Pagination<RestrictProjectPageList>) findPage("RestrictProjectList", "RestrictProjectCount",
  85. param, in.getPageNo(), in.getPageSize());
  86. return page;
  87. }
  88. private void addUserLog(RestrictProject in, int i, String takeAid) {
  89. UserTransferLog log = new UserTransferLog();
  90. log.setUid(in.getUid());
  91. log.setPid(in.getPid());
  92. if (i == 0){
  93. log.setAid(in.getAid());
  94. log.setType(15);
  95. } else if (i == 1) {
  96. log.setAid(in.getAid());
  97. log.setType(16);
  98. }else if (i == 2) {
  99. log.setAid(TokenManager.getAdminId());
  100. log.setTakeAid(takeAid);
  101. log.setType(17);
  102. }
  103. userTransferLogMapper.insertSelective(log);
  104. }
  105. }