| | |
| | | import cn.gistack.system.user.service.IUserService; |
| | | import cn.gistack.system.user.sync.dto.AccountBean; |
| | | import cn.gistack.system.user.sync.dto.ApiPushDTO; |
| | | import cn.gistack.system.user.sync.util.GenerateUtil; |
| | | import cn.gistack.system.user.sync.util.SecurityUtil; |
| | | import cn.gistack.system.user.sync.exception.ErrorCode; |
| | | import cn.gistack.system.user.sync.util.*; |
| | | import com.alibaba.fastjson.JSON; |
| | | import com.alibaba.fastjson.JSONArray; |
| | | import lombok.AllArgsConstructor; |
| | | import org.springblade.core.mp.support.Condition; |
| | | import org.springblade.core.tenant.annotation.NonDS; |
| | | import org.springblade.core.tool.api.R; |
| | | import org.springframework.scheduling.annotation.Async; |
| | | import org.springframework.web.bind.annotation.PostMapping; |
| | | import org.springframework.web.bind.annotation.RequestBody; |
| | | import org.springframework.web.bind.annotation.RequestMapping; |
| | | import org.springframework.web.bind.annotation.RestController; |
| | | |
| | | import javax.servlet.http.HttpServletRequest; |
| | | import java.util.HashMap; |
| | | import java.util.List; |
| | | import java.util.Map; |
| | | import java.util.concurrent.*; |
| | | |
| | | /** |
| | | * @PROJECT_NAME: skjcmanager |
| | |
| | | private final static String method = "post"; |
| | | |
| | | @PostMapping("/userpush") |
| | | public R userPush(ApiPushDTO apiPushDTO) { |
| | | String syncData = SecurityUtil.decryptAES(apiPushDTO.getSyncData(), appSecret); |
| | | public Object userPush(@RequestBody ApiPushDTO apiPushDTO, HttpServletRequest request) { |
| | | Map<String, Object> ret = new HashMap<>(); |
| | | |
| | | String clientId = request.getParameter("clientId"); |
| | | String timestamp = request.getParameter("timestamp"); |
| | | String sign = request.getParameter("sign"); |
| | | String salt = request.getParameter("salt"); |
| | | |
| | | // 校验签名防止篡改参数 |
| | | String sign2 = SignBuilder.create(appId, MD5Util.getMD5String(appSecret)) |
| | | .requestPath("/api/blade-user/sync/userpush") |
| | | .requestMethod("POST") |
| | | .salt(salt) |
| | | .timestamp(Long.valueOf(timestamp)) |
| | | .sign(); |
| | | |
| | | if (!clientId.equals(appId) || !sign.equals(sign2)) { |
| | | ret.put("code", ErrorCode.SIGN_CHECK_ERROR.getCode()); |
| | | ret.put("msg", ErrorCode.SIGN_CHECK_ERROR.getMsg()); |
| | | ret.put("data", ""); |
| | | return ret; |
| | | } |
| | | |
| | | // 解密syncData |
| | | String syncData = SecurityUtil.decryptAES(apiPushDTO.getSyncData(), MD5Util.getMD5String(appSecret)); |
| | | List<AccountBean> accountBeanList = JSONArray.parseArray(syncData, AccountBean.class); |
| | | |
| | | // 异步执行用户处理 |
| | | |
| | | int pcount = Runtime.getRuntime().availableProcessors(); |
| | | //最大线程数控制 |
| | | int maxthreadNum = accountBeanList.size(); |
| | | ExecutorService executor = new ThreadPoolExecutor(pcount, maxthreadNum, 10, TimeUnit.SECONDS, |
| | | new ArrayBlockingQueue<>(1000), Executors.defaultThreadFactory(), new ThreadPoolExecutor.CallerRunsPolicy()); |
| | | System.out.println("开始执行用户处理!!!!!!!!!!!!"); |
| | | //循环list |
| | | accountBeanList.forEach(accountBean -> { |
| | | //判断手机号是否在库中 |
| | | User params = new User(); |
| | | params.setPhone(accountBean.getPrincipalId()); |
| | | User one = userService.getOne(Condition.getQueryWrapper(params)); |
| | | for (int i = 0; i < maxthreadNum; i++) { |
| | | final int index = i; |
| | | final AccountBean accountBean = accountBeanList.get(i); |
| | | executor.execute(() -> { |
| | | //业务处理 |
| | | System.out.println("当前线程:" + Thread.currentThread().getName() + " " + index); |
| | | System.out.println("当前处理对象:" + JSON.toJSONString(accountBean)); |
| | | asyncUserGo(accountBean); |
| | | }); |
| | | } |
| | | // 关闭线程池 |
| | | executor.shutdown(); |
| | | |
| | | if (one == null){ |
| | | //不在,看status状态,若为1则不操作。0添加进库 |
| | | if (accountBean.getStatus().equals("0")){ |
| | | ret.put("code", 1000); |
| | | ret.put("msg", "成功"); |
| | | ret.put("data", apiPushDTO.getSign()); |
| | | System.out.println("处理返回了~~~~"); |
| | | return ret; |
| | | } |
| | | |
| | | User saveUser = new User(); |
| | | //设置id为手机号 |
| | | public void asyncUserGo(AccountBean accountBean) { |
| | | //判断手机号是否在库中 |
| | | User params = new User(); |
| | | params.setPhone(accountBean.getPrincipalId()); |
| | | User one = userService.getOne(Condition.getQueryWrapper(params)); |
| | | |
| | | if (one == null) { |
| | | //不在,看status状态,若为1则不操作。0添加进库 |
| | | if (accountBean.getStatus().equals("0")) { |
| | | |
| | | User saveUser = new User(); |
| | | //设置id为手机号 |
| | | // saveUser.setId(Long.parseLong(accountBean.getPrincipalId())); |
| | | //账号为手机号 |
| | | saveUser.setAccount(accountBean.getPrincipalId()); |
| | | saveUser.setPhone(accountBean.getPrincipalId()); |
| | | //设置姓名 |
| | | saveUser.setRealName(accountBean.getAttributes().get("realName")); |
| | | //设置部门(默认给省) |
| | | saveUser.setDeptId("420000000000"); |
| | | //设置角色(权限与市州水库管理业务人员一致) |
| | | saveUser.setRoleId("1688732764750766082"); |
| | | //设置租户id |
| | | saveUser.setTenantId("000000"); |
| | | userService.save(saveUser); |
| | | } |
| | | }else { |
| | | //在,看status状态,若为1则删除用户。0更新? |
| | | if (accountBean.getStatus().equals("1")){ |
| | | userService.removeUser(accountBean.getPrincipalId()); |
| | | } else { //更新 |
| | | //设置姓名 |
| | | one.setRealName(accountBean.getAttributes().get("realName")); |
| | | userService.updateUser(one); |
| | | } |
| | | //账号为手机号 |
| | | saveUser.setAccount(accountBean.getPrincipalId()); |
| | | saveUser.setPhone(accountBean.getPrincipalId()); |
| | | //设置姓名 |
| | | saveUser.setRealName(accountBean.getAttributes().get("realName")); |
| | | //设置部门(默认给省) |
| | | saveUser.setDeptId("420000000000"); |
| | | //设置角色(权限与市州水库管理业务人员一致) |
| | | saveUser.setRoleId("1688732764750766082"); |
| | | //设置租户id |
| | | saveUser.setTenantId("000000"); |
| | | userService.save(saveUser); |
| | | } |
| | | }); |
| | | return R.data(apiPushDTO.getTimestamp()); |
| | | } else { |
| | | //在,看status状态,若为1则删除用户。0更新? |
| | | if (accountBean.getStatus().equals("1")) { |
| | | userService.removeUser(accountBean.getPrincipalId()); |
| | | } else { //更新 |
| | | //设置姓名 |
| | | one.setRealName(accountBean.getAttributes().get("realName")); |
| | | userService.updateUser(one); |
| | | } |
| | | } |
| | | |
| | | } |
| | | |
| | | } |