|
@@ -1,8 +1,10 @@
|
|
|
package org.jeecg.modules.ctop.service.impl;
|
|
|
|
|
|
-import cn.com.ctop.common.module.constant.CtopRoleCodeConstant;
|
|
|
+import cn.com.ctop.common.module.entity.UserAllocation;
|
|
|
+import cn.com.ctop.common.module.mapper.UserAllocationMapper;
|
|
|
import cn.com.ctop.common.module.service.ISysRoleService;
|
|
|
import cn.com.ctop.common.module.utils.Check;
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
|
import org.apache.shiro.SecurityUtils;
|
|
|
import org.jeecg.common.system.vo.LoginUser;
|
|
@@ -14,6 +16,7 @@ import org.jeecg.modules.system.entity.SysUser;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
+import java.util.ArrayList;
|
|
|
import java.util.HashMap;
|
|
|
import java.util.List;
|
|
|
import java.util.Map;
|
|
@@ -27,10 +30,13 @@ import java.util.Map;
|
|
|
*/
|
|
|
@Service
|
|
|
public class ProjectMemberServiceImpl extends ServiceImpl<ProjectMemberMapper, ProjectMember> implements IProjectMemberService {
|
|
|
+
|
|
|
@Autowired
|
|
|
private ProjectMemberMapper memberMapper;
|
|
|
@Autowired
|
|
|
private ISysRoleService sysRoleService;
|
|
|
+ @Autowired
|
|
|
+ private UserAllocationMapper userAllocationMapper;
|
|
|
|
|
|
@Override
|
|
|
public void addMember(SysUser sysUser, Project project) {
|
|
@@ -63,30 +69,39 @@ public class ProjectMemberServiceImpl extends ServiceImpl<ProjectMemberMapper, P
|
|
|
}
|
|
|
|
|
|
//视频报表
|
|
|
- public List<ProjectMember> getProjects(List<Integer> mediaIds){
|
|
|
+ public List<ProjectMember> getProjects(List<Integer> mediaIds) {
|
|
|
LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
|
|
|
String userId = sysUser.getId();
|
|
|
String roleCode = memberMapper.getRoleCodeByUserId(userId);
|
|
|
String companyId = memberMapper.getCompanyIdByUserId(userId);
|
|
|
|
|
|
- //List<Integer> mediaIds = new ArrayList<>();
|
|
|
- //mediaIds.add(1);
|
|
|
- //mediaIds.add(3);
|
|
|
//角色权限判断
|
|
|
List<ProjectMember> projectMemberList = null;
|
|
|
- if(roleCode.equals("plan") || roleCode.equals("clip") || roleCode.equals("shot") || roleCode.equals("plane")|| roleCode.equals("planeLeader")){
|
|
|
+ if (roleCode.equals("plan") || roleCode.equals("clip") || roleCode.equals("shot") || roleCode.equals("plane") || roleCode.equals("planeLeader")) {
|
|
|
projectMemberList = memberMapper.designProjects(userId, mediaIds);
|
|
|
- }else if( roleCode.equals("designTeamLeader")){
|
|
|
+ } else if (roleCode.equals("designTeamLeader")) {
|
|
|
projectMemberList = memberMapper.designLeaderProjects(companyId, mediaIds);
|
|
|
- }else if(roleCode.equals(CtopRoleCodeConstant.COMMON_ROLE_CODE_ADMIN)){
|
|
|
+ } else if (roleCode.equals("admin")) {
|
|
|
projectMemberList = memberMapper.adminProjects(mediaIds);
|
|
|
- }else if(roleCode.equals("operator") || roleCode.equals("operationAssistant") || roleCode.equals("touTiaoOperationManager")){
|
|
|
+ } else if (roleCode.equals("operator") || roleCode.equals("operationAssistant") || roleCode.equals("touTiaoOperationManager")) {
|
|
|
projectMemberList = memberMapper.operationProjects(userId, mediaIds);
|
|
|
}
|
|
|
|
|
|
return projectMemberList;
|
|
|
}
|
|
|
|
|
|
-
|
|
|
-
|
|
|
+ public Map<String, Object> queryAccountByProjectId(ProjectMember projectMember) {
|
|
|
+ Map<String, Object> temp = new HashMap<>();
|
|
|
+ temp.put("project", projectMember);
|
|
|
+ QueryWrapper<UserAllocation> queryWrapper = new QueryWrapper<>();
|
|
|
+ queryWrapper.eq("project_id", projectMember.getProjectId());
|
|
|
+ queryWrapper.isNotNull("account_id");
|
|
|
+ List<UserAllocation> userAllocations = userAllocationMapper.selectList(queryWrapper);
|
|
|
+ if(userAllocations.isEmpty()){
|
|
|
+ temp=null;
|
|
|
+ }else {
|
|
|
+ temp.put("account", userAllocations);
|
|
|
+ }
|
|
|
+ return temp;
|
|
|
+ }
|
|
|
}
|