spring cloud zuul的用法(token,限流)

一、pom配置文件

<groupId>com.java</groupId>
<artifactId>zuul</artifactId>
<version>1.0-SNAPSHOT</version>
<packaging>jar</packaging>

<parent>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-parent</artifactId>
    <version>2.0.0.RELEASE</version>
    <relativePath/> <!-- lookup parent from repository -->
</parent>

<properties>
    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
    <jjwt.version>0.9.1</jjwt.version>
    <java.version>1.8</java.version>
    <spring-cloud.version>Finchley.RELEASE</spring-cloud.version>
</properties>

<dependencies>
    <!--基础包 start-->
    <dependency>
        <groupId>org.springframework.cloud</groupId>
        <artifactId>spring-cloud-starter-netflix-eureka-client</artifactId>
    </dependency>

    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>

    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-test</artifactId>
        <scope>test</scope>
    </dependency>
    <!--基础包 end-->
    <!--zuul start-->
    <dependency>
        <groupId>org.springframework.cloud</groupId>
        <artifactId>spring-cloud-starter-netflix-zuul</artifactId>
    </dependency>
    <!--fastjson start-->
    <dependency>
        <groupId>com.alibaba</groupId>
        <artifactId>fastjson</artifactId>
        <version>1.2.28</version>
    </dependency>
    <!--fastjson end-->
    <!--lombok start-->
    <dependency>
        <groupId>org.projectlombok</groupId>
        <artifactId>lombok</artifactId>
        <scope>provided</scope>
    </dependency>
    <!--lombok end-->
    <!--jwt start-->
    <dependency>
        <groupId>com.auth0</groupId>
        <artifactId>java-jwt</artifactId>
        <version>3.3.0</version>
    </dependency>
    <!--jwt end-->
</dependencies>

<dependencyManagement>
    <dependencies>
        <dependency>
            <groupId>org.springframework.cloud</groupId>
            <artifactId>spring-cloud-dependencies</artifactId>
            <version>${spring-cloud.version}</version>
            <type>pom</type>
            <scope>import</scope>
        </dependency>
    </dependencies>
</dependencyManagement>

<build>
    <plugins>
        <plugin>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-maven-plugin</artifactId>
        </plugin>
    </plugins>
</build>

二、配置文件设置

##端口号
server.port=8201

spring.application.name=zuul

eureka.client.service-url.defaultZone=http://localhost:8200/eureka/

ribbon.ReadTimeout= 60000
ribbon.ConnectTimeout= 60000

##测试项目
zuul.routes.apply1.path=/apply1/**
zuul.routes.apply1.service-id=apply1

##测试项目

三、token实现

3.1、token过滤器实现

package com.java.zuul.filters;

import com.alibaba.fastjson.JSONObject;
import com.java.zuul.jwt.JWTUtil;
import com.java.zuul.util.Cache;
import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.exception.ZuulException;
import lombok.extern.java.Log;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

import java.io.IOException;

import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.PRE_TYPE;

@Component
@Log
public class TokenFilter extends ZuulFilter {
    @Override
    public String filterType() {
        return PRE_TYPE;
    }

    @Override
    public int filterOrder() {
        return 1;
    }

    @Override
    public boolean shouldFilter() {
        RequestContext requestContext = RequestContext.getCurrentContext();
        HttpServletRequest request = requestContext.getRequest();
        //完整路径接口
        String url = request.getRequestURI();
        String local_url = "/login";
        /**
         * 如果是登录接口不进行token验证
         */
        if (local_url.equals(url)){
            return false;
        }else{
            return true;
        }

    }

    @Override
    public Object run() throws ZuulException {
        RequestContext ctx = RequestContext.getCurrentContext();
        HttpServletRequest request = ctx.getRequest();
        String token_header = request.getHeader("token") == null ? "" : request.getHeader("token");
        if (token_header.equals("")){
            try {
                ctx.setSendZuulResponse(false);
                ctx.getResponse().setContentType("text/html;charset=utf-8");
                ctx.getResponse().getWriter().write("{\"code\": -1,\"message\": \"必须添加token验证\"}");
            } catch (IOException e) {
                e.printStackTrace();
              }
            return null;
        }

        JSONObject jsonObject = JSONObject.parseObject(Cache.getCache(token_header).toString());
    if (jsonObject.getInteger("state") == 0){
        try {
            ctx.setSendZuulResponse(false);
            ctx.getResponse().setContentType("text/html;charset=utf-8");
            ctx.getResponse().getWriter().write("{\"code\": -1,\"message\": \"token值过期或不存在\"}");
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
        }
        JSONObject accout = jsonObject.getJSONObject("account");
        String username = accout.getString("username");
        String secret = accout.getString("secret");
        if(!JWTUtil.verify(token_header,username,secret)){
            try {
                ctx.setSendZuulResponse(false);
                ctx.getResponse().setContentType("text/html;charset=utf-8");
                ctx.getResponse().getWriter().write("{\"code\": -1,\"message\": \"token验证失败\"}");
            } catch (IOException e) {
                e.printStackTrace();
            }
            return null;
        }

        /**
         * 通过登录名称获取权限接口list,在跟当前的url路径对比,
         * 如果相等说明有这个接口的权限
         */
        //完整路径接口
        String url = request.getRequestURI();
        log.info("url1"+url);


        return null;
    }
}

3.2、hashmap缓存实现

package com.java.zuul.util;

import com.alibaba.fastjson.JSONObject;

import java.util.concurrent.ConcurrentHashMap;

public class Cache {

    private static ConcurrentHashMap<String, Object> cacheMap = new ConcurrentHashMap<>();

    /**
     * 获取缓存的对象
     *
     * @param account
     * @return
     */
    public static Object getCache(String account) {
        JSONObject jsonObject = new JSONObject();
        // 如果缓冲中有该账号,则返回value
        if (cacheMap.containsKey(account)) {
            jsonObject.put("state",1);
            jsonObject.put("account",cacheMap.get(account));
        } else {
            jsonObject.put("state",0);
        }
        return jsonObject;
    }

    /**
     * 设置缓存
     *
     * @param key
     */
    public static void setCache(String key,Object value) {
        // 一般是进行数据库查询,将查询的结果进行缓存
        cacheMap.put(key, value);
    }


    /**
     * 移除缓存信息
     *
     * @param account
     */
    public static void removeCache(String account) {
        cacheMap.remove(account);
    }
}

3.3、返回结果封装

package com.java.zuul.result;

public class Result {
    private String msg;
    private Integer status;
    private Object data;
    public static Result newAppResult(String msg, Integer status, Object data){
        return new Result(msg, status, data);
    }
    public Result(String msg, Integer status, Object data) {
        this.msg = msg;
        this.status = status;
        this.data = data;
    }
    public static Result OK(Object object){
        return  new Result("success",1,object);
    }
    public static Result OK(String message, Object object){
        return  new Result(message,1,object);
    }
    public static Result ERROR(String message){
        return  new Result(message,-1,null);
    }
    public static Result ERROR(String message, Integer status){
        return  new Result(message,status,null);
    }

    public String getMsg() {
        return msg;
    }

    public void setMsg(String msg) {
        this.msg = msg;
    }

    public Integer getStatus() {
        return status;
    }

    public void setStatus(Integer status) {
        this.status = status;
    }

    public Object getData() {
        return data;
    }

    public void setData(Object data) {
        this.data = data;
    }
}

3.4、登录功能实现

package com.java.zuul.controller;

import com.alibaba.fastjson.JSONObject;
import com.java.zuul.jwt.JWTUtil;
import com.java.zuul.result.Result;
import com.java.zuul.util.Cache;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;


@RestController
public class UserController {

    /**
     * 用户登录同时返回jwt校验码
     * @param username
     * @param secret
     * @return
     */
    @RequestMapping(value = "/login")
    public Object login(String username,String secret){
        if ("cdy".equals(username) && "111111".equals(secret)){
            String jwt =JWTUtil.sign(username,secret);
            JSONObject user = new JSONObject();
            user.put("username",username);
            user.put("secret",secret);
            Cache.setCache(jwt,user);
            return Result.OK(jwt);
        } else {
            return Result.ERROR("账号密码错误!");
        }
    }

    /**
     * 校验token是否失败
     * @param token
     * @param username
     * @param secret
     * @return
     */
    @RequestMapping(value = "verify")
    public Object verify(String token, String username, String secret){
        if (JWTUtil.verify(token,username,secret)){
            return Result.OK("校验成功!");
        } else {
            return Result.ERROR("校验失败");
        }
    }



}

四、限流实现

package com.java.zuul.filters;

import com.google.common.util.concurrent.RateLimiter;
import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.exception.ZuulException;
import lombok.extern.java.Log;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

import java.io.IOException;

import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.*;

@Component
@Log
public class RateLimiterFilter  extends ZuulFilter {
    //测试接口令牌数量
    private static RateLimiter RATE_LIMITER = RateLimiter.create(1);

    @Override
    public String filterType() {
        return PRE_TYPE;
    }

    @Override
    public int filterOrder() {
        // 执行顺序为1,值越小执行顺行越靠前
    //        return SEND_ERROR_FILTER_ORDER;
        return 3;
    }

    @Override
    public boolean shouldFilter() {
        RequestContext requestContext = RequestContext.getCurrentContext();
        HttpServletRequest request = requestContext.getRequest();
        //完整路径接口
        String url = request.getRequestURI();
        log.info("url3:"+url);
        //去掉最后一层路径接口(例如:/instance/{serviceId})后面的参数需要去掉才能判断,最好不要写这样的接口
        String lastUrl = url.substring(0,url.lastIndexOf("/"));
        //只对订单接口限流
        if ("/apply1/test".equalsIgnoreCase(url)) {
            return true;
        }
        return false;
    }

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

        //就相当于每调用一次tryAcquire()方法,令牌数量减1,当1000个用完后,那么后面进来的用户无法访问上面接口
        //当然这里只写类上面一个接口,可以这么写,实际可以在这里要加一层接口判断。
        if (!RATE_LIMITER.tryAcquire()) {
            HttpStatus httpStatus = HttpStatus.TOO_MANY_REQUESTS;
            requestContext.setSendZuulResponse(false);
            //HttpStatus.TOO_MANY_REQUESTS.value()里面有静态代码常量
                requestContext.setResponseStatusCode(HttpStatus.TOO_MANY_REQUESTS.value());
//            throw new ZuulException(
//                    httpStatus.getReasonPhrase(),
//                    httpStatus.value(),
//                    httpStatus.getReasonPhrase()
//            );

            try {
                requestContext.setSendZuulResponse(false);
                requestContext.getResponse().setContentType("text/html;charset=utf-8");
                requestContext.getResponse().getWriter().write("{\"code\": -1,\"message\": \"访问次数过多\"}");
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
      return null;
    }
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,928评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,192评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,468评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,186评论 1 286
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,295评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,374评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,403评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,186评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,610评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,906评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,075评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,755评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,393评论 3 320
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,079评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,313评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,934评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,963评论 2 351

推荐阅读更多精彩内容