BJ_ZTEITS-Developers / zteits-oa

Blame view

src/main/java/com/zteits/oa/report/web/OAuthController.java 7.39 KB
Edit Raw Normal View History
900781d3   王富生   提交 
1
2
 
  package com.zteits.oa.report.web;
b1704d7c   王富生   提交 
3
5e8d9026   王富生   提交 
4
5
6
 
  import java.util.ArrayList;
  import java.util.List;
900781d3   王富生   提交 
7
8
9
 
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpSession;
900781d3   王富生   提交 
10
11
12
13
14
15
16
 
  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   王富生   提交 
17
18
 
  import org.springframework.web.bind.annotation.RestController;
5e8d9026   王富生   提交 
19
 
  import com.alibaba.fastjson.JSONObject;
b1704d7c   王富生   提交 
20
 
  import com.xiaoleilu.hutool.util.CollectionUtil;
900781d3   王富生   提交 
21
22
23
24
25
26
27
 
  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   王富生   提交 
28
 
  import com.zteits.oa.api.service.report.query.AsraOpQueryService;
0b156ed7   王富生   提交 
29
 
  import com.zteits.oa.report.vo.OAuthResult;
900781d3   王富生   提交 
30
31
32
33
34
35
36
37
38
39
40
41
 
  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   员工管理 
42
 
  	 private AsraOpQueryService asraOpQueryService;
900781d3   王富生   提交 
43
44
45
46
47
48
 
  	 
  	 @Autowired
  	 private HttpServletRequest request;
  	 
  	 @ApiOperation("用户登录")
  	 @PostMapping("/login")
0b156ed7   王富生   提交 
49
50
 
  	 public OAuthResult<LoginOathRes> login(@RequestBody LoginOauthReq req ) throws Exception {
  		 OAuthResult<LoginOathRes> result =  this._login(req);
900781d3   王富生   提交 
51
52
53
54
55
56
57
58
 
  		 return result;
  	 }
  	 /**
  	  * 登陆验证.<br/>
  	  * @param req
  	  * @return
  	  * 2018年7月31日  wangfs.<br/>
  	  */
0b156ed7   王富生   提交 
59
60
 
  	 private OAuthResult<LoginOathRes> _login(LoginOauthReq req ){
  		 OAuthResult<LoginOathRes> result = new OAuthResult<LoginOathRes>(false);
900781d3   王富生   提交 
61
62
63
64
65
 
  		 LoginOathRes loginOathRes = new  LoginOathRes();
  		 String loginCode = req.getLoginCode();
  		 String passWord = req.getPassWord();
  		 if(StringUtils.isEmpty(loginCode) || StringUtils.isEmpty(passWord)) {
          	  logger.info("校验登录信息,用户名 或者 登录密码为空!");
0b156ed7   王富生   提交 
66
 
          	  result.setErrorType(ErrorType.PARAMM_NULL, "用户名 或者 登录密码为空");
900781d3   王富生   提交 
67
68
69
70
71
72
 
  	      }
  		 AsraOpQueryReq asraOpQueryRe = new AsraOpQueryReq();
  		 AsraOpDTO asraOpDTO = new AsraOpDTO();
  		 boolean isCheckSuccess = false;
  		 //1.判断登录账号/密码
  		 asraOpQueryRe.setLoginCode(loginCode);
9ce77292   xiejianpeng   员工管理 
73
 
  		 BizResult<AsraOpDTO> asraOpReult = asraOpQueryService.queryAsraOp(asraOpQueryRe);
900781d3   王富生   提交 
74
75
76
77
 
  		 if(asraOpReult != null && asraOpReult.getData() != null){
  			 asraOpDTO = asraOpReult.getData();
  			 if(StringUtils.isEmpty(asraOpDTO.getLoginCode())){
  				 logger.info("{}登录账号不存在",loginCode);
0b156ed7   王富生   提交 
78
 
  	        	  result.setErrorType(ErrorType.AUTH_LOGIN_ERROR, "登录账号不存在!");
900781d3   王富生   提交 
79
80
81
 
  			 }else{
  				 if(!asraOpDTO.getLoginPassword().equalsIgnoreCase(MD5Utils.enMD5(passWord))){
  					 logger.info("{}登录账号输入的密码不正确",loginCode);
0b156ed7   王富生   提交 
82
 
  		        	  result.setErrorType(ErrorType.AUTH_PASS_ERROR, "登录密码不匹配!");
900781d3   王富生   提交 
83
84
85
86
87
88
 
  				 }else{
  					 isCheckSuccess = true;
  				 }
  			 }
  			 
  		 }else{
0b156ed7   王富生   提交 
89
 
  			 result.setErrorType(ErrorType.BIZ_ERROR, "用户登录失败");
900781d3   王富生   提交 
90
91
92
93
 
  		 }
  		 
  		 
  		 if(isCheckSuccess){
5e8d9026   王富生   提交 
94
95
96
97
 
  			 
  			 List<Long> opIds = new ArrayList<>();
  			 opIds.add(asraOpDTO.getId());
  			 /**递归查询员工ids.*/
b1704d7c   王富生   提交 
98
 
  			 opIds = queryOpTreeByOpId(asraOpDTO.getId());
5e8d9026   王富生   提交 
99
100
 
  			 logger.info("---获取到的opIds={}",JSONObject.toJSON(opIds));
  			 asraOpDTO.setOpIds(opIds);
900781d3   王富生   提交 
101
102
103
104
105
106
107
108
109
 
  			 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   王富生   提交 
110
 
  			 loginOathRes.setRoleId(asraOpDTO.getRoleId());
900781d3   王富生   提交 
111
 
  			 result.setData(loginOathRes);
0b156ed7   王富生   提交 
112
 
  			 result.setErrorType(ErrorType.BIZ_SUCCESS, "登录成功");
900781d3   王富生   提交 
113
114
115
116
117
 
  		 }
  		 
  		
  		 return result;
  	 }
d365b3fa   王富生   提交 
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
 
  	 /**
  	  * 退出登录
  	  * @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   王富生   提交 
137
138
139
140
141
142
143
144
 
      
      /**
       * 递归查询员工id.<br/>
       * @param parentOpId
       * @param opList
       * @return
       * 2018年8月2日  wangfs.<br/>
       */
b1704d7c   王富生   提交 
145
146
147
148
149
 
      private List<Long> queryOpTreeByOpId(Long opId){
      	List<Long> rootAllList = new ArrayList<>();
      	rootAllList.add(opId);
      	//1.查找第一级下的所有用户
      	List<Long> rootOpIds = this.getRootOpIds(opId);
1ab8179c   王富生   斯蒂芬斯蒂芬 
150
 
      	rootAllList.addAll(rootOpIds);
b1704d7c   王富生   提交 
151
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
 
      	//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   王富生   提交 
197
 
      }
b1704d7c   王富生   提交 
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
 
      
      /**
       * 查找登陆用户下一级用户
       * @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   王富生   提交 
223
224
225
 
  	 
  
  }