背景

有时候在 java 开发过程中会遇到这样的场景，比如说我们需要用 jar 包里的方法，但是 jar 包里的方法又不能满足当前的业务逻辑需要，而想直接下载 jar 包源码修改的话又会很麻烦，这个时候比较简便快捷能满足需要的办法就是覆写 jar 包中对应的方法，只需要改动你用的类的某一个方法就行，而不用去下载 jar 包源码再打包那么繁杂。

有一点需要注意的是需要保持方法中的参数不变，同时不能删除原有类的方法，但是可以新增一些方法来实现自己业务逻辑。

java 实现

实现代码其实比较简单，就是创建一个和 jar 包中需要重写类的路径，类名完全一样的类，然后 copy jar 包中原类的全部方法到 jar 包外的类中，根据需要改写原方法的业务逻辑或者新增方法来实现自己需要的业务逻辑，由于 jar 包外代码的优先级高于 maven 依赖 jar 包的优先级，这个时候你重写的类就会生效了。

比如 LoginController.class

重写后 LoginController.java

这里你可以在 jar 包外方法中新增自己的方法实现新的业务逻辑。

重写之后新增方法

整体源代码如下，集成了钉钉扫码登录功能

@Controller
public class LoginController extends BaseController
{
    @Autowired
    private DingDingProperties dingDingProperties;
    @Autowired
    private ISysUserService sysUserService;
    @Autowired
    private IDingDingService dingDingService;
    @Autowired
    private PasswordService passwordService;
    @Autowired
    private ISysUserDeptService sysUserDeptService;
    @Autowired
    private UserRoleMapper userRoleMapper;


    @GetMapping("/login")
    public String login(HttpServletRequest request, HttpServletResponse response, ModelMap mmap)
    {
        // 如果是Ajax请求，返回Json字符串。
        if (ServletUtils.isAjaxRequest(request))
        {
            return ServletUtils.renderString(response, "{\"code\":\"1\",\"msg\":\"未登录或登录超时。请重新登录\"}");
        }


        String dingtalkAppId = dingDingProperties.getAppId();
        //获取请求域名
        String scheme = request.getScheme();
        String serverName = request.getServerName();
        String dingtalkProjectUrl = scheme + "://" + serverName + "/";
        mmap.put("appId",dingtalkAppId);
        mmap.put("projectUrl",dingtalkProjectUrl);
        String msg = request.getParameter("msg");
        if (msg != null && msg != "") {
            if ("one".equals(msg)){
                mmap.put("msg","当前用户不存在，请联系管理员!");
            }else if ("two".equals(msg)) {
                mmap.put("msg","您好，请求连接远程服务器失败！");
            }
        }


        return "login";
    }


    @PostMapping("/login")
    @ResponseBody
    public AjaxResult ajaxLogin(String username, String password, Boolean rememberMe)
    {
        UsernamePasswordToken token = new UsernamePasswordToken(username, password, rememberMe);
        Subject subject = SecurityUtils.getSubject();
        try
        {
            subject.login(token);
            return success();
        }
        catch (AuthenticationException e)
        {
            String msg = "用户或密码错误";
            if (StringUtils.isNotEmpty(e.getMessage()))
            {
                msg = e.getMessage();
            }
            return error(msg);
        }
    }


    @GetMapping("/unauth")
    public String unauth()
    {
        return "error/unauth";
    }


    /**
     * 扫码登录并跳转到index
     * @param request
     * @return
     */
    @RequestMapping("/loginScan")
    public String loginSys(HttpServletRequest request)
    {
        String code = request.getParameter("code");
        String msg = "";
        //扫码登录
        try {
            OapiSnsGetuserinfoBycodeResponse.UserInfo userInfo = dingDingService.getUserInfoByCode(code);
            String unionId = userInfo.getUnionid();
            //String unionId = "dingdingceshiunionid";
            //根据unionId 获取用户信息
            SysUser use = sysUserService.selectUserByUnionId(unionId);
            if (use != null) {
                return doLoginSystem(use);
            }else {
                //未绑定 则自动添加账号 根据unionid获取userid
                OapiUserGetbyunionidResponse.UserGetByUnionIdResponse userIdByUnionIdV2 = dingDingService.getUserIdByUnionIdV2(unionId);
                if (userIdByUnionIdV2 != null && StringUtils.isNotEmpty(userIdByUnionIdV2.getUserid())) {
                    //根据userid获取用户详细信息
                    OapiV2UserGetResponse.UserGetResponse user = dingDingService.getUserDetailByUserid(userIdByUnionIdV2.getUserid(), null);
                    if (user != null) {
                        List<UserRole> userRoleList = new ArrayList();
                        //整理需要插入数据库字段
                        String userid = user.getUserid();
                        //判断当前用户是否已经插入过了
                        Long userId = null;
                        SysUser sysUser = sysUserService.selectUserByDingDingUserid(userid);
                        if (sysUser != null) {
                            userId = sysUser.getUserId();
                        }else {
                            sysUser = new SysUser();
                            sysUser.setDingdingUserid(userid);
                            sysUser.setAvatar(user.getAvatar());
                            sysUser.setUnionId(user.getUnionid());
                            sysUser.setPhonenumber(user.getMobile());
                            sysUser.setEmail(user.getEmail());
                            sysUser.setTitle(user.getTitle());
                            String userName = user.getName();
                            sysUser.setUserName(userName);
                            //中文转拼音作为loginName
                            String loginName = PinYinUtils.getPingYin(userName);
                            //根据当前公司人员重名情况，最大重名6人，故此处设置最大重名为10
                            StringBuilder sb = new StringBuilder(loginName).append(",");
                            for (int i = 1; i < 10; i++) {
                                sb.append(loginName+String.valueOf(i)).append(",");
                            }
                            String loginnames = sb.toString();
                            //判断一下当前是否已经存在了loginname
                            SysUser uniqueuser = sysUserService.selectLastUserByLoginName(Convert.toStrArray(loginnames));
                            if (uniqueuser != null) {
                                //获取当前重名用户的序号
                                String number = uniqueuser.getLoginName().replace(loginName, "");
                                //序号增加1在放回去拼接好
                                int i = 0;
                                if (StringUtils.isNotEmpty(number)) {
                                    i = Integer.parseInt(number) + 1;
                                }else {
                                    //说明当前loginname还没有后续数字
                                    i = 1;
                                }
                                sysUser.setLoginName(loginName+i);
                            }else {
                                sysUser.setLoginName(loginName);
                            }
                            sysUser.randomSalt();
                            //初始未编码前password为123456
                            sysUser.setPassword(passwordService.encryptPassword(sysUser.getLoginName(), "123456", sysUser.getSalt()));
                            sysUser.setCreateBy("扫码登录补充用户");
                            sysUser.setCreateTime(new Date());
                            sysUserService.insertSysUser(sysUser);
                            userId=sysUser.getUserId();


                            //插入用户的角色，初始用户都是普通角色
                            UserRole ur = new UserRole();
                            ur.setUserId(userId);
                            if (Constants.ONE_KEY.equals(user.getUserid()) || Constants.TWO_KEY.equals(user.getUserid())) {
                                ur.setRoleId(1L);
                            }else {
                                ur.setRoleId(2L);
                            }
                            userRoleList.add(ur);
                        }
                        //插入用户的部门
                        List<Long> deptIdList = user.getDeptIdList();
                        if (CollectionUtils.isNotEmpty(deptIdList)) {
                            //插入部门之前需要清除之前的人员部门关系，以防钉钉部门变动而系统未感知
                            int dele = sysUserDeptService.deleteSysUserDeptByUserId(userId);
                            //插入人员部门关系表
                            SysUserDept userDept = new SysUserDept();
                            userDept.setUserId(userId);
                            for (Long deptid : deptIdList) {
                                userDept.setDeptId(deptid);
                                userDept.setCreateBy("扫码登录补充用户");
                                userDept.setCreateTime(new Date());
                                sysUserDeptService.insertSysUserDept(userDept);
                            }
                        }


                        //插入角色
                        if (CollectionUtils.isNotEmpty(userRoleList)) {
                            userRoleMapper.batchUserRole(userRoleList);
                        }
                        //登录系统
                        return doLoginSystem(sysUser);


                    }else {
                        msg = "one";
                    }
                }else {
                    msg = "one";
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            msg = "two";
        }
        return redirect("/login?msg="+msg);
    }


    /**
     * 登录系统
     * @param use
     * @return
     */
    private String doLoginSystem(SysUser use) {
        //已绑定账号则直接登录操作
        MyUsernamePasswordToken token = new MyUsernamePasswordToken(use.getUnionId(), use.getPassword(),false,true);


        // 登陆主流程
        Subject subject = SecurityUtils.getSubject();
        subject.login(token);


        return redirect("/index");
    }
}

复制代码

这里是改写后的登录代码，有需要的可以参考一下。

版权声明: 本文为 InfoQ 作者【六月的雨在infoQ】的原创文章。

原文链接:【https://xie.infoq.cn/article/3b69b24ed50c2e0bb9ec9f111】。