Spring Cloud Feign统一设置验证token实现方法解析

Spring Cloud Feign统一设置验证token实现方法解析

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

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

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

static class DynamicAuthTokenTarget implements Target {

public DynamicAuthTokenTarget(Class clazz,

UrlAndTokenProvider provider,

ThreadLocal requestIdProvider);

...

@Override

public Request apply(RequestTemplate input) {

TokenIdAndPublicURL urlAndToken = provider.get();

if (input.url().indexOf("http") != 0) {

input.insert(0, urlAndToken.publicURL);

}

input.header("http://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的日志。

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

@Configuration

public 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

**/

@Component

public 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.isBlhwAOBrfWyank(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

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。