Spring Cloud中Feign如何统一设置验证token

来源:互联网 发布:怎样解除淘宝15天到账 编辑:程序博客网 时间:2024/05/29 17:48

前面我们大致的聊了下如何保证各个微服务之前调用的认证问题
- Spring Cloud中如何保证各个微服务之间调用的安全性
- Spring Cloud中如何保证各个微服务之间调用的安全性(下篇)

原理是通过每个微服务请求之前都从认证服务获取认证之后的token,然后将token放入到请求头中带过去,这样被调用方通过验证token来判断是否合法请求

我们也在zuul中通过前置过滤器来统一设置token, 其实还漏掉了一种,那就是业务服务调用业务服务的时候,是没有zuul这种前置过滤器的,那么我们该如何设置呢?

其实也挺简单的,因为我们服务之前的调用是依赖于Feign的,我们可以从Feign上来做文章

如果你仔细看过Feign的文档的话,肯定会注意到下面一段代码:

tatic class DynamicAuthTokenTarget<T> implements Target<T> {    public DynamicAuthTokenTarget(Class<T> clazz,                                  UrlAndTokenProvider provider,                                  ThreadLocal<String> requestIdProvider);    ...    @Override    public Request apply(RequestTemplate input) {      TokenIdAndPublicURL urlAndToken = provider.get();      if (input.url().indexOf("http") != 0) {        input.insert(0, urlAndToken.publicURL);      }      input.header("X-Auth-Token", urlAndToken.tokenId);      input.header("X-Request-ID", requestIdProvider.get());      return input.request();    }  }  ...  Bank bank = Feign.builder()          .target(new DynamicAuthTokenTarget(Bank.class, provider, requestIdProvider));

我们可以为Feign设置一个请求拦截器,在调用之前做一些事情,添加请求头信息

原生的Feign都能添加拦截器,Spring Cloud中那肯定也是可以的

前面的文章中我们讲过通过自定义配置来覆盖默认配置,我们创建了一个FeignConfiguration配置类来配置Feign的日志

今天终于又派上用场了, 可以在这边自定义一个拦截器

@Configurationpublic class FeignConfiguration {    /**     * 日志级别     * @return     */    @Bean      Logger.Level feignLoggerLevel() {          return Logger.Level.FULL;      }    /**     * 创建Feign请求拦截器,在发送请求前设置认证的token,各个微服务将token设置到环境变量中来达到通用     * @return     */    @Bean    public FeignBasicAuthRequestInterceptor basicAuthRequestInterceptor() {        return new FeignBasicAuthRequestInterceptor();    }}/** * Feign请求拦截器 * @author yinjihuan * @create 2017-11-10 17:25 **/public class FeignBasicAuthRequestInterceptor  implements RequestInterceptor {    public FeignBasicAuthRequestInterceptor() {    }    @Override    public void apply(RequestTemplate template) {        template.header("Authorization", System.getProperty("fangjia.auth.token"));    }}

大致的步骤呢就是通过设置Feign的拦截器来设置token, 因为这边是通用的,所有呢token的值通过环境变量来传递

每个微服务只需要将获取的token信息设置到环境变量中即可

 System.setProperty("fangjia.auth.token", token);

上次我们也讲过如果来自动获取token,我们是通过定时任务来定时刷新的,当时还创建了AuthService来获取token

今天在优化下,因为这个也需要通用,所以直接将Service去掉了

/** * 定时刷新token * * @author yinjihuan * @create 2017-11-09 15:39 **/@Componentpublic class TokenScheduledTask {    private static Logger logger = LoggerFactory.getLogger(TokenScheduledTask.class);    public final static long ONE_Minute = 60 * 1000 * 60 * 20;    @Autowired    private AuthRemoteClient authRemoteClient;    /**     * 刷新Token     */    @Scheduled(fixedDelay = ONE_Minute)    public void reloadApiToken() {        String token = this.getToken();        while (StringUtils.isBlank(token)) {            try {                Thread.sleep(1000);                token = this.getToken();            } catch (InterruptedException e) {                logger.error("", e);            }        }        System.setProperty("fangjia.auth.token", token);    }    public String getToken() {        AuthQuery query = new AuthQuery();        query.setAccessKey("1");        query.setSecretKey("1");        ResponseData response = authRemoteClient.auth(query);        return response.getData() == null ? "" : response.getData().toString();    }}

到现在,各个微服务直接的安全认证就结束啦,用起来不麻烦,下面总结下

  • 定义一个用于认证的微服务,可以有用户信息,和业务无关,提供认证接口
  • 定义JWT工具类,提供生成token和检查token的方法,公用
  • 定义认证token的过滤器,公用
  • 在需要认证的微服务中注册过滤器即可实现拦截操作, 也可做成公用的,就是所有都需要认证
  • 在调用方配置TokenScheduledTask,实现定时刷新token, 也可做成公用的

如果不需要认证,那么不注册拦截过滤器即可,也不用 TokenScheduledTask,对业务代码无任何倾入性

具体代码可以参考我的github:

https://github.com/yinjihuan/spring-cloud

原创粉丝点击