平常开发过程中,如果涉及到RPC调用,对于服务调用方和服务提供方,都是可以设置接口超时时间的。以调用方为例,调用方需要调用远程的一个接口,为了保证服务的质量,一般会设置调用接口的超时时间,比如将调用接口的超时时间设置为1秒,当调用远程接口后,经过1秒还未拿到结果,那么就认为是超时了,调用方就不会继续等待服务提供方返回结果,而是直接抛出一个SocketTimeOutException。
其实不仅仅是RPC接口调用需要设置超时时间,数据库、缓存这些一样的,一般都会设置超时时间,不能让程序无休止的等待下去。
那么问题来了!!应用或者说框架,是如何设置超时时间的呢?我们设置超时时间为1秒,那么程序是怎么保证1秒后就停止调用了呢?这也就是本文要介绍的,接口调用设置超时时间的原理。
本文将以设置调用数据库超时为例,介绍设置超时时间是怎么实现的,明白这个原理后,其他比如RPC调用、缓存调用的超时原理也就明白。
CallUtils-带有超时的执行工具类
下面是一个CallUtils,包含一个线程池和execute方法,execute方法中默认超时时间为1秒。
package cn.ganlixin.util; import java.util.concurrent.*; /** * 带有超时的任务执行器 */ public class CallUtils { /** * 用于执行任务的线程池 */ private static ExecutorService executorService = Executors.newFixedThreadPool(5); /** * 执行任务,并且超时时间为1秒 * * @param callable 需要执行的任务 * @param <T> 任务执行完毕后返回值类型 * @return 任务结果 */ public static <T> T execute(Callable<T> callable) { // 提交任务 Future<T> future = executorService.submit(callable); try { return future.get(1, TimeUnit.SECONDS); } catch (InterruptedException | ExecutionException | TimeoutException e) { e.printStackTrace(); } return null; } }
UserService-将调用数据库
UserService使用CallUtils来执行数据库操作,因为CallUtils设置有调用超时:
package cn.ganlixin.service; import cn.ganlixin.dao.UserDAO; import cn.ganlixin.model.User; import cn.ganlixin.util.CallUtils; import java.util.List; public class UserService { private UserDAO userDAO = new UserDAO(); public List<User> getAllUser() { return CallUtils.execute(() -> userDAO.getAllUserFromDB()); } }
UserDAO-耗时的数据库操作
userDAO中,作为测试,只提供了一个getAllUserFromDB方法,休眠2秒来模拟数据库操作,因为UserService使用CallUtils来调用该方法,CallUtils的超时时间为1秒,所以该数据库操作必定会超时:
package cn.ganlixin.dao; import cn.ganlixin.model.User; import java.util.List; import java.util.concurrent.TimeUnit; public class UserDAO { /** * 从DB获取全量user列表 */ public List<User> getAllUserFromDB() { try { // 模拟数据库操作,耗时2秒 TimeUnit.SECONDS.sleep(2); } catch (InterruptedException e) { e.printStackTrace(); } return null; } }
进行测试
创建测试类,测试UserService的getAllUser方法,因为getAllUser方法中使用CallUtils来调用数据库操作,CallUtils的超时时间为1秒,而数据库操作需要2秒,所以getAllUser方法必然会超时。
package cn.ganlixin.test; import cn.ganlixin.service.UserService; import org.junit.Test; public class TestUserService { public UserService userService = new UserService(); @Test public void testGetAllUser() { userService.getAllUser(); } }
执行测试,输出如下:
总结
本文演示了接口超时调用的原理,实现接口调用超时,无非是通过将任务提交到线程池后,使用future.get,设置超时时间即可。
上面的代码很多细节都不太规范,比如涉及到数据库的超时,应该是数据库连接池的超时配置,而我在演示时是直接使用CallUtils来替代了,但是明白这个原理就OK。