接口调用超时的实现原理

　　平常开发过程中，如果涉及到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。

　　原文链接：https://www.cnblogs.com/-beyond/p/13204252.html

 

posted @ 2020-06-28 18:18  寻觅beyond  阅读(7957)  评论(2)  编辑  收藏  举报
返回顶部