spring security oauth2 自定義invalid_token返回體

框架

springboot+security+oauth2

問題

默認(rèn)token過期的返回格式是這樣的

{
  "error": "invalid_token",
  "error_description": "Invalid access token: XXXXXXXXXXXXXXXXXXXXXXX"
}

通常情況下闽巩,這與我們自己定義的數(shù)據(jù)返回的response格式不統(tǒng)一,不方便客戶端做統(tǒng)一處理征炼。如果要改成自己的格式的,需要怎么做呢疆柔?

效果

修改格式前

{
  "error": "invalid_token",
  "error_description": "Invalid access token: XXXXXXXXXXXXXXXXXXXXXXX"
}

修改格式后

{
  "errCode": -1,
  "errorMsg": "登陸憑證過期"
}
處理流程

1.新建或者打開security的繼承自ResourceServerConfigurerAdapter的配置類(一般情況都已經(jīng)創(chuàng)建了這個(gè)類),添加如下代碼

    @Override
    public void configure(ResourceServerSecurityConfigurer resources) {
        OAuth2AuthenticationEntryPoint authenticationEntryPoint = new OAuth2AuthenticationEntryPoint();
        authenticationEntryPoint.setExceptionTranslator(new CustomExceptionTranslator());
        resources.authenticationEntryPoint(authenticationEntryPoint);
    }

2.創(chuàng)建CustomExceptionTranslator類

public class CustomExceptionTranslator extends DefaultWebResponseExceptionTranslator {

    @Override
    public ResponseEntity<OAuth2Exception> translate(Exception e) throws Exception {
        ResponseEntity<OAuth2Exception> translate = super.translate(e);
        OAuth2Exception body = translate.getBody();
        CustomOauthException customOauthException = new CustomOauthException(body.getMessage(),body.getOAuth2ErrorCode(),
                body.getHttpErrorCode());
        ResponseEntity<OAuth2Exception> response = new ResponseEntity<>(customOauthException, translate.getHeaders(),
                translate.getStatusCode());
        return response;
    }

}

3.創(chuàng)建CustomOauthException類

@JsonSerialize(using = CustomOauthExceptionSerializer.class)
public class CustomOauthException extends OAuth2Exception {

    private String oAuth2ErrorCode;

    private int httpErrorCode;

    public CustomOauthException(String msg, String oAuth2ErrorCode, int httpErrorCode) {
        super(msg);
        this.oAuth2ErrorCode = oAuth2ErrorCode;
        this.httpErrorCode = httpErrorCode;
    }

    public String getoAuth2ErrorCode() {
        return oAuth2ErrorCode;
    }

    public int getHttpErrorCode() {
        return httpErrorCode;
    }
}

4.創(chuàng)建CustomOauthException類

public class CustomOauthExceptionSerializer extends StdSerializer<CustomOauthException> {

    public CustomOauthExceptionSerializer() {
        super(CustomOauthException.class);
    }

    @Override
    public void serialize(CustomOauthException value, JsonGenerator gen, SerializerProvider provider) throws IOException {
        //value內(nèi)容適當(dāng)?shù)淖鲆恍╁e(cuò)誤類型判斷
        gen.writeStartObject();
        gen.writeObjectField("errCode",-1);
        gen.writeObjectField("errorMsg","登陸憑證過期");
        gen.writeEndObject();
    }
}

打完收工
參考了這篇文章

參考源代碼分析一波原因

由于我用的生成token的類是JdbcTokenStore

    @Bean
    public JdbcTokenStore tokenStore() {
        return new JdbcTokenStore(dataSource);
    }

所以打開JdbcTokenStore類弊攘,可以看到有很多跟查詢token相關(guān)的方法,逐個(gè)斷點(diǎn)姑曙,嘗試一個(gè)無效的token襟交,最終斷點(diǎn)命中在readAccessToken函數(shù)


JdbcTokenStore.readAccessToken

果不其然,斷點(diǎn)處會(huì)報(bào)錯(cuò)伤靠,并且這個(gè)函數(shù)最終accessToken會(huì)返回null捣域。token_invalid的錯(cuò)誤信息寫入response的代碼這里沒有,說明在這個(gè)函數(shù)更上層宴合,往外找到DefaultTokenServices.loadAuthentication


DefaultTokenServices.loadAuthentication

同理焕梅,經(jīng)過幾層的查看,最終找到OAuth2AuthenticationProcessingFilter.doFilter,可以看到這里在操作response


OAuth2AuthenticationProcessingFilter.doFilter

實(shí)際上這里框出來的authenticationEntryPoint對象就是我們前面提到的解決方案中形纺,被我們替換成自定義ExceptionTranslator的authenticationEntryPoint,再看一下原始的commence方法里跟我們的操作response相關(guān)的代碼做了些什么丘侠,debug進(jìn)去,來到DefaultWebResponseExceptionTranslator.translate

translate+handleOAuth2Exception

translate方法里面一頓操作逐样,最終會(huì)返回一個(gè)ResponseEntity蜗字,看一下ResponseEntity接受的三個(gè)參數(shù),分別是一個(gè)會(huì)被轉(zhuǎn)成json的對象脂新,headers挪捕,http狀態(tài)碼。這三個(gè)東西熟悉http請求的就感覺很親切了争便。也就是說我不管這里這一大坨代碼做了什么级零,只要我按照我自己的要求返回一個(gè)ResponseEntity就可以。于是乎就有了開頭解決方法中的替換成自己定義的ExceptionTranslator的解決思路滞乙。

現(xiàn)在框架有很多這種設(shè)計(jì)模式中的策略模式的設(shè)計(jì)來方便開發(fā)人員自己定義執(zhí)行方法奏纪。類似DefaultXXXXXX的類,基本上都可以通過自己創(chuàng)建自定義對象來實(shí)現(xiàn)多樣化的需求斩启。當(dāng)然我也是先百度了解決方案之后序调,再去看原因的,多看看原理,萬一以后百度沒作業(yè)給你抄的時(shí)候,你自己也能通過閱讀代碼兔簇,找到方法发绢。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市垄琐,隨后出現(xiàn)的幾起案子边酒,更是在濱河造成了極大的恐慌,老刑警劉巖狸窘,帶你破解...
    沈念sama閱讀 217,185評論 6 503
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件墩朦,死亡現(xiàn)場離奇詭異,居然都是意外死亡翻擒,警方通過查閱死者的電腦和手機(jī)氓涣,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,652評論 3 393
  • 文/潘曉璐 我一進(jìn)店門鹃操,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人春哨,你說我怎么就攤上這事《魉牛” “怎么了赴背?”我有些...
    開封第一講書人閱讀 163,524評論 0 353
  • 文/不壞的土叔 我叫張陵,是天一觀的道長晶渠。 經(jīng)常有香客問我凰荚,道長,這世上最難降的妖魔是什么褒脯? 我笑而不...
    開封第一講書人閱讀 58,339評論 1 293
  • 正文 為了忘掉前任便瑟,我火速辦了婚禮,結(jié)果婚禮上番川,老公的妹妹穿的比我還像新娘到涂。我一直安慰自己,他們只是感情好颁督,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,387評論 6 391
  • 文/花漫 我一把揭開白布践啄。 她就那樣靜靜地躺著,像睡著了一般沉御。 火紅的嫁衣襯著肌膚如雪屿讽。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,287評論 1 301
  • 那天吠裆,我揣著相機(jī)與錄音伐谈,去河邊找鬼。 笑死试疙,一個(gè)胖子當(dāng)著我的面吹牛诵棵,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播效斑,決...
    沈念sama閱讀 40,130評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼非春,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了缓屠?” 一聲冷哼從身側(cè)響起奇昙,我...
    開封第一講書人閱讀 38,985評論 0 275
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎敌完,沒想到半個(gè)月后储耐,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,420評論 1 313
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡滨溉,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,617評論 3 334
  • 正文 我和宋清朗相戀三年什湘,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了长赞。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 39,779評論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡闽撤,死狀恐怖得哆,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情哟旗,我是刑警寧澤贩据,帶...
    沈念sama閱讀 35,477評論 5 345
  • 正文 年R本政府宣布,位于F島的核電站闸餐,受9級(jí)特大地震影響饱亮,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜舍沙,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,088評論 3 328
  • 文/蒙蒙 一近上、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧拂铡,春花似錦壹无、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,716評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽留瞳。三九已至,卻和暖如春她倘,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背硬梁。 一陣腳步聲響...
    開封第一講書人閱讀 32,857評論 1 269
  • 我被黑心中介騙來泰國打工荧止, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留屹电,地道東北人危号。 一個(gè)月前我還...
    沈念sama閱讀 47,876評論 2 370
  • 正文 我出身青樓素邪,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個(gè)殘疾皇子磨确,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,700評論 2 354