spring AOP 自己實(shí)現(xiàn)一個(gè)手機(jī)授權(quán)

2016.7.11 4:32

如今很多后臺(tái)涉及很多手機(jī)登錄授權(quán)管理,很多人使用shiro 或者spring security 解決,其中shiro 接入比較簡單扫沼,但是security 接入入門水平就比較高了。如果不想接入任何的登錄授權(quán)的話,啃官方文檔的話镊掖。那就對(duì)了。接下來我們就結(jié)合spring AOP 來寫一個(gè)簡單的token管理褂痰,自定義注解來解決此授權(quán)問題亩进。主要分為下面的幾個(gè)步驟:

  1. tokenManager 的建立
  2. 自定義注解的建立
  3. 實(shí)現(xiàn)一個(gè)和User一樣的類
  4. 結(jié)合AOP來解決節(jié)省多余的代碼。
  5. 體驗(yàn)一下我們的tokenValid 的使用吧脐恩。

寫一個(gè)具有時(shí)效性的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);
    }

}

然后我們只要在每個(gè)需要登錄的接口處镐侯,判斷獲取userId 并且判斷數(shù)據(jù)庫內(nèi)是否存在該用戶即可。這時(shí)候AOP就派上用場了驶冒。不哆嗦我們來結(jié)合AOP來完成一個(gè)token認(rèn)證吧苟翻。

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

}
實(shí)現(xiàn)一個(gè)和User一樣的類(至于為什么這么干,我是想當(dāng)修改用戶信息的時(shí)候傳遞參數(shù)對(duì)象和我們token管理的user信息區(qū)分開來骗污。)
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, "請(qǐng)重新登錄");
        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);
    }

}
下面我們來體驗(yàn)一下我們的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 是不是很好玩?各位兄弟姐妹們需忿。喜歡的點(diǎn)贊诅炉。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市屋厘,隨后出現(xiàn)的幾起案子涕烧,更是在濱河造成了極大的恐慌,老刑警劉巖汗洒,帶你破解...
    沈念sama閱讀 222,681評(píng)論 6 517
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件议纯,死亡現(xiàn)場離奇詭異,居然都是意外死亡溢谤,警方通過查閱死者的電腦和手機(jī)瞻凤,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,205評(píng)論 3 399
  • 文/潘曉璐 我一進(jìn)店門憨攒,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人阀参,你說我怎么就攤上這事肝集。” “怎么了蛛壳?”我有些...
    開封第一講書人閱讀 169,421評(píng)論 0 362
  • 文/不壞的土叔 我叫張陵杏瞻,是天一觀的道長。 經(jīng)常有香客問我炕吸,道長伐憾,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 60,114評(píng)論 1 300
  • 正文 為了忘掉前任赫模,我火速辦了婚禮树肃,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘瀑罗。我一直安慰自己胸嘴,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 69,116評(píng)論 6 398
  • 文/花漫 我一把揭開白布斩祭。 她就那樣靜靜地躺著劣像,像睡著了一般。 火紅的嫁衣襯著肌膚如雪摧玫。 梳的紋絲不亂的頭發(fā)上耳奕,一...
    開封第一講書人閱讀 52,713評(píng)論 1 312
  • 那天,我揣著相機(jī)與錄音诬像,去河邊找鬼屋群。 笑死,一個(gè)胖子當(dāng)著我的面吹牛坏挠,可吹牛的內(nèi)容都是我干的芍躏。 我是一名探鬼主播,決...
    沈念sama閱讀 41,170評(píng)論 3 422
  • 文/蒼蘭香墨 我猛地睜開眼降狠,長吁一口氣:“原來是場噩夢啊……” “哼对竣!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起榜配,我...
    開封第一講書人閱讀 40,116評(píng)論 0 277
  • 序言:老撾萬榮一對(duì)情侶失蹤否纬,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后蛋褥,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體烦味,經(jīng)...
    沈念sama閱讀 46,651評(píng)論 1 320
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,714評(píng)論 3 342
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了谬俄。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 40,865評(píng)論 1 353
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡弃理,死狀恐怖溃论,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情痘昌,我是刑警寧澤钥勋,帶...
    沈念sama閱讀 36,527評(píng)論 5 351
  • 正文 年R本政府宣布,位于F島的核電站辆苔,受9級(jí)特大地震影響算灸,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜驻啤,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 42,211評(píng)論 3 336
  • 文/蒙蒙 一菲驴、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧骑冗,春花似錦赊瞬、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,699評(píng)論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至遥倦,卻和暖如春谤绳,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背袒哥。 一陣腳步聲響...
    開封第一講書人閱讀 33,814評(píng)論 1 274
  • 我被黑心中介騙來泰國打工缩筛, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人统诺。 一個(gè)月前我還...
    沈念sama閱讀 49,299評(píng)論 3 379
  • 正文 我出身青樓歪脏,卻偏偏與公主長得像,于是被迫代替她去往敵國和親粮呢。 傳聞我的和親對(duì)象是個(gè)殘疾皇子婿失,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,870評(píng)論 2 361

推薦閱讀更多精彩內(nèi)容