Java Web 學習筆記之十四：RestEasy添加Filter過濾器預處理請求

RestEasy添加Filter過濾器預處理請求

前提

• 定義filter過濾器，預處理http請求
• 在resteasy框架下配置filter

實現功能

• 攔截http請求，獲取請求頭中的Cookie信息
• 統一處理Cookie中的token，將token信息與用戶信息映射(後端業務需要)
• 將獲取到的用戶信息重新放置到請求頭中，作爲HeaderParam傳遞給rest接口使用

實現步驟

編寫filter代碼

package xuyihao.*;

import xuyihao.*.UserBindCache;

import java.io.IOException;
import java.util.Map;

import javax.ws.rs.container.ContainerRequestContext;
import javax.ws.rs.container.ContainerRequestFilter;
import javax.ws.rs.core.Cookie;

/**
 * 全局過濾器
 * 
 * Created by xuyh at 2017年12月11日 下午4:07:39.
 */
public class GlobalFilter implements ContainerRequestFilter {
    @Override
    public void filter(ContainerRequestContext requestContext) throws IOException {

        Map<String, Cookie> cookieMap = requestContext.getCookies();
        if (cookieMap == null || cookieMap.isEmpty())
            return;

        Cookie tokenCookie = cookieMap.get("token");

        if (tokenCookie != null) {

            String token = tokenCookie.getValue();

            if (token != null && !token.isEmpty()) {
                String userId = UserBindCache.getBoundUser(token);
                requestContext.getHeaders().add("userId", userId);
            }

        }

    }
}

web.xml配置添加Filter註冊

使用web.xml註冊restEasy的，在web.xml中添加如下內容

<!-- 配置過濾器,異常處理器等 -->
<context-param>
    <param-name>resteasy.providers</param-name>
    <param-value>
        xuyihao.*.GlobalFilter
    </param-value>
</context-param>

rest接口方法中使用filter傳遞的請求頭參數

/**
 * 用戶登出
 * 
 * <pre>
 * 需要在Cookie中設置token
 * </pre>
 *
 * @param userId
 * @return
 */
@GET
@Path("/logout")
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
public boolean logout(@HeaderParam(RestConstants.REQUEST_HEADER_NAME_USERID) String userId) throws Exception {
    if (userId != null) {
        getUserLogic().removeBoundUserByUserId(userId);
        return true;
    } else {
        return false;
    }
}