BJ_ZTEITS-Developers / zteits-oa

Blame view

src/main/java/com/zteits/oa/report/web/OAuthController.java 7.4 KB
Edit Raw Normal View History
900781d3   王富生   提交 
1
2
 
  package com.zteits.oa.report.web;
b1704d7c   王富生   提交 
3
4
 
  import static org.mockito.Matchers.contains;
5e8d9026   王富生   提交 
5
6
7
 
  import java.util.ArrayList;
  import java.util.List;
900781d3   王富生   提交 
8
9
10
 
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpSession;
900781d3   王富生   提交 
11
12
13
14
15
16
17
 
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.util.StringUtils;
  import org.springframework.web.bind.annotation.PostMapping;
  import org.springframework.web.bind.annotation.RequestBody;
  import org.springframework.web.bind.annotation.RequestMapping;
900781d3   王富生   提交 
18
19
 
  import org.springframework.web.bind.annotation.RestController;
5e8d9026   王富生   提交 
20
 
  import com.alibaba.fastjson.JSONObject;
b1704d7c   王富生   提交 
21
 
  import com.xiaoleilu.hutool.util.CollectionUtil;
900781d3   王富生   提交 
22
23
24
25
26
27
28
 
  import com.zteits.oa.api.base.bean.BizResult;
  import com.zteits.oa.api.base.constants.ErrorType;
  import com.zteits.oa.api.base.constants.SessionEnum;
  import com.zteits.oa.api.dto.asraop.AsraOpDTO;
  import com.zteits.oa.api.dto.asraop.LoginOathRes;
  import com.zteits.oa.api.dto.asraop.param.AsraOpQueryReq;
  import com.zteits.oa.api.dto.asraop.param.LoginOauthReq;
5e8d9026   王富生   提交 
29
 
  import com.zteits.oa.api.service.report.query.AsraOpQueryService;
0b156ed7   王富生   提交 
30
 
  import com.zteits.oa.report.vo.OAuthResult;
900781d3   王富生   提交 
31
32
33
34
35
36
37
38
39
40
41
42
 
  import com.zteits.oa.util.MD5Utils;
  
  import io.swagger.annotations.Api;
  import io.swagger.annotations.ApiOperation;
  
  @Api("用户登录授权")
  @RestController
  @RequestMapping("/oauth")
  public class OAuthController {
  	 private static final Logger logger = LoggerFactory.getLogger(OAuthController.class);
  	 
  	 @Autowired
9ce77292   xiejianpeng   员工管理 
43
 
  	 private AsraOpQueryService asraOpQueryService;
900781d3   王富生   提交 
44
45
46
47
48
49
 
  	 
  	 @Autowired
  	 private HttpServletRequest request;
  	 
  	 @ApiOperation("用户登录")
  	 @PostMapping("/login")
0b156ed7   王富生   提交 
50
51
 
  	 public OAuthResult<LoginOathRes> login(@RequestBody LoginOauthReq req ) throws Exception {
  		 OAuthResult<LoginOathRes> result =  this._login(req);
900781d3   王富生   提交 
52
53
54
55
56
57
58
59
 
  		 return result;
  	 }
  	 /**
  	  * 登陆验证.<br/>
  	  * @param req
  	  * @return
  	  * 2018年7月31日  wangfs.<br/>
  	  */
0b156ed7   王富生   提交 
60
61
 
  	 private OAuthResult<LoginOathRes> _login(LoginOauthReq req ){
  		 OAuthResult<LoginOathRes> result = new OAuthResult<LoginOathRes>(false);
900781d3   王富生   提交 
62
63
64
65
66
 
  		 LoginOathRes loginOathRes = new  LoginOathRes();
  		 String loginCode = req.getLoginCode();
  		 String passWord = req.getPassWord();
  		 if(StringUtils.isEmpty(loginCode) || StringUtils.isEmpty(passWord)) {
          	  logger.info("校验登录信息,用户名 或者 登录密码为空!");
0b156ed7   王富生   提交 
67
 
          	  result.setErrorType(ErrorType.PARAMM_NULL, "用户名 或者 登录密码为空");
900781d3   王富生   提交 
68
69
70
71
72
73
 
  	      }
  		 AsraOpQueryReq asraOpQueryRe = new AsraOpQueryReq();
  		 AsraOpDTO asraOpDTO = new AsraOpDTO();
  		 boolean isCheckSuccess = false;
  		 //1.判断登录账号/密码
  		 asraOpQueryRe.setLoginCode(loginCode);
9ce77292   xiejianpeng   员工管理 
74
 
  		 BizResult<AsraOpDTO> asraOpReult = asraOpQueryService.queryAsraOp(asraOpQueryRe);
900781d3   王富生   提交 
75
76
77
78
 
  		 if(asraOpReult != null && asraOpReult.getData() != null){
  			 asraOpDTO = asraOpReult.getData();
  			 if(StringUtils.isEmpty(asraOpDTO.getLoginCode())){
  				 logger.info("{}登录账号不存在",loginCode);
0b156ed7   王富生   提交 
79
 
  	        	  result.setErrorType(ErrorType.AUTH_LOGIN_ERROR, "登录账号不存在!");
900781d3   王富生   提交 
80
81
82
 
  			 }else{
  				 if(!asraOpDTO.getLoginPassword().equalsIgnoreCase(MD5Utils.enMD5(passWord))){
  					 logger.info("{}登录账号输入的密码不正确",loginCode);
0b156ed7   王富生   提交 
83
 
  		        	  result.setErrorType(ErrorType.AUTH_PASS_ERROR, "登录密码不匹配!");
900781d3   王富生   提交 
84
85
86
87
88
89
 
  				 }else{
  					 isCheckSuccess = true;
  				 }
  			 }
  			 
  		 }else{
0b156ed7   王富生   提交 
90
 
  			 result.setErrorType(ErrorType.BIZ_ERROR, "用户登录失败");
900781d3   王富生   提交 
91
92
93
94
 
  		 }
  		 
  		 
  		 if(isCheckSuccess){
5e8d9026   王富生   提交 
95
96
97
98
 
  			 
  			 List<Long> opIds = new ArrayList<>();
  			 opIds.add(asraOpDTO.getId());
  			 /**递归查询员工ids.*/
b1704d7c   王富生   提交 
99
 
  			 opIds = queryOpTreeByOpId(asraOpDTO.getId());
5e8d9026   王富生   提交 
100
101
 
  			 logger.info("---获取到的opIds={}",JSONObject.toJSON(opIds));
  			 asraOpDTO.setOpIds(opIds);
900781d3   王富生   提交 
102
103
104
105
106
107
108
109
110
 
  			 HttpSession session = request.getSession();
  			 session.setAttribute(SessionEnum.USER_INFO.key(), asraOpDTO);
  			 logger.info("---获取到的session_id={}",session.getId());
  			 loginOathRes.setOpId(asraOpDTO.getId());
  			 loginOathRes.setLoginCode(loginCode);
  			 loginOathRes.setUserName(asraOpDTO.getOpName());
  			 loginOathRes.setCityId(asraOpDTO.getCityId());
  			 loginOathRes.setCityName(asraOpDTO.getCityName());
  			 loginOathRes.setAccessToken(session.getId());
0b156ed7   王富生   提交 
111
 
  			 loginOathRes.setRoleId(asraOpDTO.getRoleId());
900781d3   王富生   提交 
112
 
  			 result.setData(loginOathRes);
0b156ed7   王富生   提交 
113
 
  			 result.setErrorType(ErrorType.BIZ_SUCCESS, "登录成功");
900781d3   王富生   提交 
114
115
116
117
118
 
  		 }
  		 
  		
  		 return result;
  	 }
d365b3fa   王富生   提交 
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
 
  	 /**
  	  * 退出登录
  	  * @param req
  	  * @return
  	  * 2018年7月31日  wangfs.<br/>
  	  */
      @ApiOperation("用户登出")
      @RequestMapping("/loginout")
      public OAuthResult<LoginOathRes> loginOut() {
          HttpSession session = request.getSession();
          AsraOpDTO userInfo = (AsraOpDTO)request.getSession().getAttribute(SessionEnum.USER_INFO.key());
          //登出
          session.invalidate();
          if(userInfo==null){
          	return new OAuthResult<>(true);
          }
          logger.info("end用户登出..");
      	return new OAuthResult<>(true);
      }
5e8d9026   王富生   提交 
138
139
140
141
142
143
144
145
 
      
      /**
       * 递归查询员工id.<br/>
       * @param parentOpId
       * @param opList
       * @return
       * 2018年8月2日  wangfs.<br/>
       */
b1704d7c   王富生   提交 
146
147
148
149
150
 
      private List<Long> queryOpTreeByOpId(Long opId){
      	List<Long> rootAllList = new ArrayList<>();
      	rootAllList.add(opId);
      	//1.查找第一级下的所有用户
      	List<Long> rootOpIds = this.getRootOpIds(opId);
5e8d9026   王富生   提交 
151
b1704d7c   王富生   提交 
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
 
      	//2.查找第二级以下所有用户
      	if(CollectionUtil.isNotEmpty(rootOpIds)){
      		rootAllList.addAll(rootOpIds);
      		for(Long getOpId:rootOpIds ){
      			List<Long> childOpIds = new ArrayList<>();
      			this.getChildOpIds(getOpId,childOpIds);
      			rootAllList.addAll(childOpIds);
      		}
      		
      	}
      	return rootAllList;
      }
      
      /**
       * 递归查找一级以下的用户
       * @param parentId
       * @return
       * 2018年8月2日  wangfs.<br/>
       */
      private List<Long>  getChildOpIds(Long parentId,List<Long> childOpIds){
      	  AsraOpQueryReq asraOpQueryRe = new AsraOpQueryReq();
  		  asraOpQueryRe.setParentId(parentId);
  	      BizResult<List<AsraOpDTO>> asraOpReult = asraOpQueryService.queryAsraOpByParentId(asraOpQueryRe);
  	      if(asraOpReult != null && CollectionUtil.isNotEmpty(asraOpReult.getData())){
  	    	  List<AsraOpDTO> list = asraOpReult.getData();
      		  for(AsraOpDTO asraOpDTO :list){
      			  if(asraOpDTO == null){
      				 continue;
      			  }
      			  childOpIds.add(asraOpDTO.getId());
      		  }
  
      	      //遍历下一级
      		  if(CollectionUtil.isNotEmpty(childOpIds)){
      			  for(Long opId:childOpIds){
      				  if(opId != null && !opId.equals(parentId)){
      					  getChildOpIds(opId,childOpIds);
      				  }
      				  break;
      				 
      			  }
      			 
      		  }
  	      }
  	     
      	return childOpIds;
5e8d9026   王富生   提交 
198
 
      }
b1704d7c   王富生   提交 
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
 
      
      /**
       * 查找登陆用户下一级用户
       * @param opId
       * @param opList
       * @return
       * 2018年8月2日  wangfs.<br/>
       */
      private List<Long> getRootOpIds(Long parentId){
      	List<Long> opList = new ArrayList<>();
      	  AsraOpQueryReq asraOpQueryRe = new AsraOpQueryReq();
  		  asraOpQueryRe.setParentId(parentId);
  	      BizResult<List<AsraOpDTO>> asraOpReult = asraOpQueryService.queryAsraOpByParentId(asraOpQueryRe);
  	      if(asraOpReult != null && CollectionUtil.isNotEmpty(asraOpReult.getData())){
  	    	  List<AsraOpDTO> list = asraOpReult.getData();
  	    		  for(AsraOpDTO asraOpDTO :list){
  	    			  if(asraOpDTO == null){
  	    				 continue;
  	    			  }
  	    			  opList.add(asraOpDTO.getId());
  	    		  }
  	      }  
  	      return opList;
      }
900781d3   王富生   提交 
224
225
226
 
  	 
  
  }