com.netflix.zuul.context.RequestContext#getResponse

Source File: ZuulFilter.java From fw-spring-cloud with Apache License 2.0

6 votes

@Override
public Object run() {
    RequestContext ctx = RequestContext.getCurrentContext();
    HttpServletRequest request = ctx.getRequest();
    log.info("我是ZuulFilter");
    String token = request.getHeader("name");// 获取请求的参数

    // 如果有name参数并且token值为zuul，才进行路由
    if (StringUtils.isNotBlank(token) && token.equals("zuul")) {
        ctx.setSendZuulResponse(true); //对请求进行路由
        ctx.setResponseStatusCode(200);
        ctx.set("code", 1);
    } else {
        ctx.setSendZuulResponse(false); //不对其进行路由
        ctx.setResponseStatusCode(401);
        HttpServletResponse response = ctx.getResponse();
        response.setHeader("content-type", "text/html;charset=utf8");
        ctx.setResponseBody("请携带网关必须参数");
        ctx.set("code", 0);
    }
    return null;
}

Source File: ValidateCodeFilter.java From Taroco with Apache License 2.0

6 votes

@Override
public Object run() {
    try {
        checkCode(RequestContext.getCurrentContext().getRequest());
    } catch (ValidateCodeException e) {
        final RequestContext ctx = RequestContext.getCurrentContext();
        final Response result = Response.failure(e.getMessage());
        final HttpServletResponse response = ctx.getResponse();

        response.setCharacterEncoding(Charset.defaultCharset().name());
        response.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
        response.setStatus(478);
        try {
            response.getWriter().print(JsonUtils.toJsonString(result));
        } catch (IOException e1) {
            log.error("response io异常");
            e1.printStackTrace();
        }
        ctx.setSendZuulResponse(false);
        ctx.setResponse(response);
    }
    return null;
}

Source File: PreviewFilter.java From Taroco with Apache License 2.0

6 votes

@Override
public Object run() {
    final RequestContext ctx = RequestContext.getCurrentContext();
    final Response result = Response.failure(DefaultError.SHOW_AUTH_CONTROL);
    final HttpServletResponse response = ctx.getResponse();

    response.setCharacterEncoding(Charset.defaultCharset().name());
    response.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
    response.setStatus(479);
    try {
        response.getWriter().print(JSONObject.toJSONString(result));
    } catch (IOException e1) {
        log.error("response io异常");
        e1.printStackTrace();
    }
    ctx.setSendZuulResponse(false);
    ctx.setResponse(response);
    return null;
}

Source File: BaseFilter.java From convergent-ui with Apache License 2.0

6 votes

protected void writeResponse(String responseBody, MimeType contentType) throws Exception {
    RequestContext context = RequestContext.getCurrentContext();
    // there is no body to send
    if (responseBody == null || responseBody.isEmpty()) {
        return;
    }
    HttpServletResponse servletResponse = context.getResponse();
    servletResponse.setCharacterEncoding("UTF-8");
    servletResponse.setContentType(contentType.toString());
    OutputStream outStream = servletResponse.getOutputStream();
    InputStream is = null;
    try {
        writeResponse(new ByteArrayInputStream(responseBody.getBytes()), outStream);
    } finally {
        try {
            if (is != null) {
                is.close();
            }
            outStream.flush();
            outStream.close();
        } catch (IOException ex) {
        }
    }
}

Source File: TokenFilter.java From xmfcn-spring-cloud with Apache License 2.0

6 votes

/**
 * 处理逻辑
 *
 * @return
 */
@Override
public Object run() {
    RequestContext requestContext = RequestContext.getCurrentContext();
    HttpServletRequest request = requestContext.getRequest();
    HttpServletResponse response = requestContext.getResponse();
    String sourceCode = request.getParameter("sourceCode");
    String dbCode = sysCommonService.getDictValue(ConstantUtil.DICT_TYPE_BASE_CONFIG, "sourceCode");
    if (StringUtil.isBlank(dbCode)||!dbCode.equals(sourceCode)) {
        requestContext.setSendZuulResponse(false);// 对该请求不路由
        requestContext.set("isSuccess", false);// 设值，让下一个Filter看到上一个Filter的状态
        // 构建返回信息
        RetData retData = new RetData();
        retData.setCode(ResultCodeMessage.UNAUTHORIZED);
        retData.setMessage(ResultCodeMessage.UNAUTHORIZED_MESSAGE);
        String jsonString = JSON.toJSONString(retData, SerializerFeature.WriteMapNullValue);
        requestContext.setResponseBody(jsonString);
        requestContext.addZuulResponseHeader("content-type", MediaType.APPLICATION_JSON_UTF8_VALUE);
        return response;
    }
    requestContext.set("isSuccess", true);// 设值，让下一个Filter看到上一个Filter的状态
    return null;
}

Source File: AddResponseIDHeaderFilter.java From Mastering-Spring-Cloud with MIT License

5 votes

@Override
public Object run() {
    RequestContext context = RequestContext.getCurrentContext();
    HttpServletResponse servletResponse = context.getResponse();
    servletResponse.addHeader("X-Response-ID", String.valueOf(id++));
    return null;
}

Source File: ZuulResponseFilter.java From open-cloud with MIT License

5 votes

/**
 *
 */
@Override
public Object run() {
    // 代理后响应日志处理
    RequestContext ctx = RequestContext.getCurrentContext();
    HttpServletRequest request = ctx.getRequest();
    HttpServletResponse response = ctx.getResponse();
    // 把路由的serviceId放入作用域
    request.setAttribute(FilterConstants.SERVICE_ID_KEY, ctx.get(FilterConstants.SERVICE_ID_KEY));
    accessLogService.sendLog(request, response,null);
    return null;
}

Source File: AddResponseIDHeaderFilter.java From Mastering-Spring-Cloud with MIT License

5 votes

@Override
public Object run() {
    RequestContext context = RequestContext.getCurrentContext();
    HttpServletResponse servletResponse = context.getResponse();
    servletResponse.addHeader("X-Response-ID", String.valueOf(id++));
    return null;
}

Source File: BaseFilter.java From convergent-ui with Apache License 2.0

5 votes

protected void writeResponse(BufferedImage image, MediaType type) throws Exception {
        RequestContext context = RequestContext.getCurrentContext();
        // there is no body to send
        if (image == null) {
            return;
        }
        HttpServletResponse servletResponse = context.getResponse();
//        servletResponse.setCharacterEncoding("UTF-8");
        servletResponse.setContentType(type.toString());
        
        ByteArrayOutputStream tmp = new ByteArrayOutputStream();
        ImageIO.write(image, type.getSubtype(), tmp);
        tmp.close();
        Integer contentLength = tmp.size();
        
        servletResponse.setContentLength(contentLength);
        
        OutputStream outStream = servletResponse.getOutputStream();
        ByteArrayOutputStream os = new ByteArrayOutputStream();

        ImageIO.write(image, type.getSubtype(), os);
        InputStream is = new ByteArrayInputStream(os.toByteArray());

        try {
            writeResponse(is, outStream);
        } finally {
            try {
                if (is != null) {
                    is.close();
                }
                outStream.flush();
                outStream.close();
            } catch (IOException ex) {
            }
        }
    }

Source File: RateLimitFilter.java From springcloud-course with GNU General Public License v3.0

5 votes

@Override
    public Object run() throws ZuulException {
        try {
            RequestContext ctx = RequestContext.getCurrentContext();
            HttpServletRequest request = ctx.getRequest();
            log.info(request.getRequestURI());
            HttpServletResponse response = ctx.getResponse();
            if (!rateLimiter.tryAcquire()) {
//                HttpStatus httpStatus = HttpStatus.TOO_MANY_REQUESTS;
//                response.setContentType(MediaType.TEXT_PLAIN_VALUE);
//                response.setStatus(httpStatus.value());
//                response.getWriter().append(httpStatus.getReasonPhrase());
//                ctx.setSendZuulResponse(false);
//                throw new ZuulException(
//                        httpStatus.getReasonPhrase(),
//                        httpStatus.value(),
//                        httpStatus.getReasonPhrase()
//                );
                ctx.setSendZuulResponse(false);
                response.setStatus(200);
                response.getWriter().write("{\"code\": 99999,\"message\": \"too many requests.\"}");
            }
        } catch (Exception e) {
            log.warning(e.getMessage());
            ReflectionUtils.rethrowRuntimeException(e);
        }
        return null;
    }

Source File: RenewFilter.java From paascloud-master with Apache License 2.0

5 votes

private void doSomething(RequestContext requestContext) {
	HttpServletRequest request = requestContext.getRequest();
	String token = StringUtils.substringAfter(request.getHeader(HttpHeaders.AUTHORIZATION), "bearer ");
	if (StringUtils.isEmpty(token)) {
		return;
	}
	OAuth2AccessToken oAuth2AccessToken = jwtTokenStore.readAccessToken(token);
	int expiresIn = oAuth2AccessToken.getExpiresIn();

	if (expiresIn < EXPIRES_IN) {
		HttpServletResponse servletResponse = requestContext.getResponse();
		servletResponse.addHeader("Renew-Header", "true");
	}
}

Source File: AddResponseHeaderFilter.java From sample-zuul-filters with Apache License 2.0

5 votes

public Object run() {
	RequestContext context = RequestContext.getCurrentContext();
	HttpServletResponse servletResponse = context.getResponse();
	servletResponse.addHeader("X-Foo",
			UUID.randomUUID().toString());
	return null;
}

Source File: ResponseLoggingFilter.java From Showcase with Apache License 2.0

5 votes

@Override
public Object run() {
    RequestContext ctx = RequestContext.getCurrentContext();
    HttpServletRequest request = ctx.getRequest();
    HttpServletResponse response = ctx.getResponse();
    String headers = response.getHeaderNames()
        .stream()
        .map(headername -> headername + "=" + response.getHeader(headername))
        .collect(Collectors.joining("; "));
    LOGGER.info(String.format("Response for %s request to %s has status %s and headers %s", request.getMethod(),
        request.getRequestURL().toString(), response.getStatus(), headers));
    return null;
}

Source File: AddResponseIDHeaderFilter.java From Mastering-Spring-Cloud with MIT License

5 votes

@Override
public Object run() {
    RequestContext context = RequestContext.getCurrentContext();
    HttpServletResponse servletResponse = context.getResponse();
    servletResponse.addHeader("X-Response-ID", String.valueOf(id++));
    return null;
}

Source File: PostResponseHeaderFilter.java From code with Apache License 2.0

5 votes

@Override
public Object run() throws ZuulException {
    RequestContext requestContext = RequestContext.getCurrentContext();
    HttpServletResponse response = requestContext.getResponse();

    response.setHeader("X-Foo", UUID.randomUUID().toString());
    return null;
}

Source File: ZuulErrorFilter.java From open-cloud with MIT License

5 votes

@Override
public Object run() {
    // 代理错误日志记录
    RequestContext ctx = RequestContext.getCurrentContext();
    HttpServletRequest request = ctx.getRequest();
    HttpServletResponse response = ctx.getResponse();
    Exception exception = new Exception(ctx.getThrowable());
    if(StringUtils.toBoolean(ctx.get("rateLimitExceeded"))){
        exception = new Exception(HttpStatus.TOO_MANY_REQUESTS.name());
    }
    accessLogService.sendLog(request, response,exception);
    ResultBody resultBody = OpenGlobalExceptionHandler.resolveException(exception,request.getRequestURI());
    WebUtils.writeJson(response, resultBody);
    return null;
}

Source File: RateLimitPreFilter.java From spring-cloud-zuul-ratelimit with Apache License 2.0

4 votes

@Override
public Object run() {
    final RequestContext ctx = RequestContext.getCurrentContext();
    final HttpServletResponse response = ctx.getResponse();
    final HttpServletRequest request = ctx.getRequest();
    final Route route = route(request);

    policy(route, request).forEach(policy -> {
        Map<String, String> responseHeaders = Maps.newHashMap();

        final String key = rateLimitKeyGenerator.key(request, route, policy);
        final Rate rate = rateLimiter.consume(policy, key, null);

        final Long limit = policy.getLimit();
        final Long remaining = rate.getRemaining();
        if (limit != null) {
            responseHeaders.put(HEADER_LIMIT, String.valueOf(limit));
            responseHeaders.put(HEADER_REMAINING, String.valueOf(Math.max(remaining, 0)));
        }

        final Duration quota = policy.getQuota();
        final Long remainingQuota = rate.getRemainingQuota();
        if (quota != null) {
            request.setAttribute(REQUEST_START_TIME, System.currentTimeMillis());
            responseHeaders.put(HEADER_QUOTA, String.valueOf(quota.getSeconds()));
            responseHeaders.put(HEADER_REMAINING_QUOTA, String.valueOf(MILLISECONDS.toSeconds(Math.max(remainingQuota, 0))));
        }

        responseHeaders.put(HEADER_RESET, String.valueOf(rate.getReset()));

        if (VERBOSE.equals(properties.getResponseHeaders())) {
            final String httpHeaderKey = key.replaceAll("[^A-Za-z0-9-.]", "_").replaceAll("__", "_");
            for (Map.Entry<String, String> headersEntry : responseHeaders.entrySet()) {
                String header = VERBOSE.equals(properties.getResponseHeaders()) ? headersEntry.getKey() + "-" + httpHeaderKey : headersEntry.getKey();
                response.setHeader(header, headersEntry.getValue());
            }
        }

        if ((limit != null && remaining < 0) || (quota != null && remainingQuota < 0)) {
            ctx.setResponseStatusCode(HttpStatus.TOO_MANY_REQUESTS.value());
            ctx.put(RATE_LIMIT_EXCEEDED, "true");
            ctx.setSendZuulResponse(false);

            eventPublisher.publishEvent(new RateLimitExceededEvent(this, policy, rateLimitUtils.getRemoteAddress(request)));

            throw new RateLimitExceededException();
        }
    });

    return null;
}

Source File: ResponseHelper.java From heimdall with Apache License 2.0

4 votes

public static Map<String, String> getResponseHeaders(RequestContext context) {

        Map<String, String> headers = new HashMap<>();

        final HttpServletResponse response = context.getResponse();

        context.getZuulResponseHeaders().forEach(pair -> headers.put(pair.first(), pair.second()));

        final Collection<String> headerNames = response.getHeaderNames();

        headerNames.forEach(s -> headers.putIfAbsent(s, response.getHeader(s)));

        if (!headers.containsKey(HttpHeaders.CONTENT_TYPE)) {
            headers.put(HttpHeaders.CONTENT_TYPE, context.getResponse().getContentType());
        }

        headers.remove("X-Application-Context");

        return headers;
    }

Source File: CacheWriterFilter.java From heimdall with Apache License 2.0

4 votes

private void process() throws Throwable {
    RequestContext context = RequestContext.getCurrentContext();

    RBucket<ApiResponse> rBucket = (RBucket<ApiResponse>) context.get(CACHE_BUCKET);

    HttpServletResponse response = context.getResponse();

    Map<String, String> headers = ResponseHelper.getResponseHeaders(context);

    ApiResponse apiResponse = new ApiResponseImpl();
    apiResponse.setHeaders(headers);
    apiResponse.setBody(ResponseHelper.getResponseBody(context, headers));
    apiResponse.setStatus(response.getStatus());

    Long timeToLive = (Long) context.get(CACHE_TIME_TO_LIVE);

    if (timeToLive != null && timeToLive > 0)
        rBucket.set(apiResponse, timeToLive, TimeUnit.MILLISECONDS);
    else
        rBucket.set(apiResponse);

}

Source File: ErrorFilter.java From mini-platform with MIT License

4 votes

@Override
public Object run() {
    try {
        RequestContext ctx = RequestContext.getCurrentContext();

        Enumeration<String> headerIt = ctx.getRequest().getHeaderNames();
        StringBuilder sb = new StringBuilder();
        while (headerIt.hasMoreElements()) {
            String name = headerIt.nextElement();
            String value = ctx.getRequest().getHeader(name);
            sb.append("REQUEST:: > " + name + ":" + value+"\n");
        }
        log.debug(sb.toString());

        Throwable throwable = getOriginException(ctx.getThrowable());
        ExceptionResult result = getExceptionResult(ctx.getRequest(), throwable);

        ctx.setSendZuulResponse(false);

        String json = JsonUtils.toJson(result);
        HttpServletResponse response = ctx.getResponse();
        response.setContentType("application/json;charset=UTF-8");
        //response.getOutputStream().write(json.getBytes());

        PrintWriter writer = null;
        try {
            writer = response.getWriter();
            writer.print(json);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if(writer!=null){
                writer.close();
            }
        }
    } catch (Exception ex) {
        log.error("-------------->Exception filtering in custom error filter", ex);
        ReflectionUtils.rethrowRuntimeException(ex);
    }
    return null;
}

Java Code Examples for com.netflix.zuul.context.RequestContext#getResponse()