spring AOP 自己实现一个手机授权

2016.7.11 4:32

如今很多后台涉及很多手机登录授权管理，很多人使用shiro 或者spring security 解决，其中shiro 接入比较简单，但是security 接入入门水平就比较高了。如果不想接入任何的登录授权的话，啃官方文档的话。那就对了。接下来我们就结合spring AOP 来写一个简单的token管理，自定义注解来解决此授权问题。主要分为下面的几个步骤：

1. tokenManager 的建立
2. 自定义注解的建立
3. 实现一个和User一样的类
4. 结合AOP来解决节省多余的代码。
5. 体验一下我们的tokenValid 的使用吧。

写一个具有时效性的tokenManager

@Component
public class TokenManager {
    public static Map<String, Map<String,Object>> token = new HashMap<>();

    private static Map<String, String> tokenMap = new ConcurrentHashMap<>();

    public String createToken(String username) {
        String token = UUID.randomUUID().toString();
        tokenMap.put(token, username);
        return token;
    }

    public Integer verifyToken(String token) {
        /**
         * 走后门
         */
        Map<String,Object> map = new HashMap<>();
        map.put("user_id", String.valueOf(15));
        map.put("time",new Date().getTime());
        TokenManager.token.put("1",map);
        Map<String,Object> tokenMap = TokenManager.getToken(token);
        if(tokenMap == null){
            return null;
        }
        Assert.notNull(tokenMap.get("time"));
        Assert.notNull(tokenMap.get("user_id"));
        long time = Long.parseLong(tokenMap.get("time").toString()) ;
        if (new Date().getTime() - time > 7 * 24 * 60 * 60 * 1000) {
            TokenManager.token.remove(token);
            return null;
        }
        tokenMap.put("time",new Date().getTime());
        TokenManager.setToken(token,tokenMap);
        return Integer.parseInt(tokenMap.get("user_id").toString());
    }

    public boolean checkToken(String token) {
        return !StringUtils.isEmpty(token) && tokenMap.containsKey(token);
    }

    public synchronized String createToken(){
        return UUID.randomUUID().toString().replace("-","");
    }

    public static Map<String, Map<String,Object>> getToken() {
        return token;
    }

    public static Map<String,Object> getToken(String token) {

        return TokenManager.token.get(token);
    }

    public synchronized static void setToken(String token,Map<String, Object> map) {
        TokenManager.token.put(token,map);
    }

}

然后我们只要在每个需要登录的接口处，判断获取userId 并且判断数据库内是否存在该用户即可。这时候AOP就派上用场了。不哆嗦我们来结合AOP来完成一个token认证吧。

自定义token

/**
 * Created by zengfeng on 16/6/30.
 */
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface TokenValid {

}

实现一个和User一样的类（至于为什么这么干，我是想当修改用户信息的时候传递参数对象和我们token管理的user信息区分开来。）

public class Principle{
  private String userName;
  private String password;
  /** 忽略属性和getter setter 方法 */

}

Spring Aop 的使用

@Aspect
@Order(10)
@Configuration
public class TokenAop {
    @Autowired
    TokenManager tokenManager;

    @Autowired
    AsUserMapper asUserMapper;

    @Autowired
    HttpServletRequest request;
    @Pointcut("execution(* com.zf.controller.*.*(..))&&" + "@annotation(com.zf.config.TokenValid)")
    public void tokenPointCut() {
    }



    @Around(value = "tokenPointCut()")
    public Object addTokenToMethod(ProceedingJoinPoint joinPoint) throws Throwable {
        Object[] args = joinPoint.getArgs();
        Object token = request.getParameter("token");
        AsUser asUser = findUser(args,token);
        Assert.notNull(asUser, "请重新登录");
        injectUserObject(asUser, args,token);
        return joinPoint.proceed(args);
    }

    private void injectUserObject(AsUser asUser, Object[] args,Object token) {
        for (int i=0; i < args.length;i++)
            if (args[i] instanceof Principle) {
                args[i] = userConvertPrinciple(asUser);
                break;
            }
    }

    /**
     * @param u
     * @return
     */
    private Object userConvertPrinciple(AsUser u) {
        Principle principle = new Principle();
        BeanUtils.copyProperties(u,principle);
        return principle;
    }

    private AsUser findUser(Object[] args, Object token) {
        Integer userId = tokenManager.verifyToken((String) token);
        return asUserMapper.selectByPrimaryKey(userId);
    }

}

下面我们来体验一下我们的tokenValid 的使用吧。

@RestController
@RequestMapping("/api/user")
public class UserController {
    @TokenValid
    @RequestMapping(value = "/setDefaultAddress",method = RequestMethod.POST)
    public void setDefaultAddress(Principle principle,@RequestParam Integer uadId) throws ExpireException, NotSamePeopleException {
        usersService.save(principle, uadId);
    };

😄 spring 是不是很好玩？各位兄弟姐妹们。喜欢的点赞。