许多工作经验少的同学可能只是在背面试题时候了解过多线程,在工作中没有运用过线程池,我在这给大家提供一个容易理解的伪代码,来给大家提供运用线程池的思路和示例。
场景:现在有一个userIdList
,需要去调用外部系统的接口 来查询详情信息,外部接口没有提供批量查询的时候,我们为提高效率,采用多线程来进行查询
要使用线程池实现根据一个 userIdList
中的每个 ID 查询详情,可以遵循以下步骤:
-
准备查询逻辑: 首先,确保您有一个方法(例如
getUserDetailsById(String userId)
)能够接收一个用户 ID,并返回对应的用户详细信息。这个方法应该是线程安全的,或者其内部不涉及共享状态。 -
创建线程池: 使用
Javajava.util.concurrent.Executors
工厂类或直接实例化ThreadPoolExecutor
创建一个线程池。以下是一个使用固定线程池大小的例子:1int corePoolSize = 5; // 根据实际情况调整 2int maxPoolSize = 10; // 根据实际情况调整 3long keepAliveTime = 60L; // 闲置线程存活时间,单位为秒 4BlockingQueue<Runnable> workQueue = new LinkedBlockingQueue<>(); // 任务队列 5 6ThreadPoolExecutor threadPool = new ThreadPoolExecutor( 7 corePoolSize, 8 maxPoolSize, 9 keepAliveTime, 10 TimeUnit.SECONDS, 11 workQueue);
如果需要在应用关闭时优雅地关闭线程池,记得在适当的地方调用
threadPool.shutdown()
或threadPool.shutdownNow()
。 -
提交查询任务: 遍历
JavauserIdList
,为每个 ID 创建一个查询任务(实现Runnable
或Callable
接口),并将任务提交到线程池执行。这里假设您使用Callable<UserDetails>
返回查询结果:1List<Future<UserDetails>> futures = new ArrayList<>(); 2for (String userId : userIdList) { 3 Future<UserDetails> future = threadPool.submit(() -> getUserDetailsById(userId)); 4 futures.add(future); 5}
上述代码为每个用户 ID 创建一个查询任务,并将任务提交给线程池。
futures
列表将保存所有任务的Future
对象,用于后续获取查询结果。 -
获取查询结果: 由于查询任务是在多线程环境下异步执行的,您需要遍历
Javafutures
列表,通过调用Future.get()
方法等待每个任务完成并获取查询结果。这一步通常伴随超时机制,以防个别任务执行过久:1List<UserDetails> userDetailsList = new ArrayList<>(); 2for (Future<UserDetails> future : futures) { 3 try { 4 UserDetails userDetails = future.get(5, TimeUnit.SECONDS); // 设置超时时间 5 userDetailsList.add(userDetails); 6 } catch (InterruptedException | ExecutionException | TimeoutException e) { 7 // 处理异常,如记录日志、设置默认值等 8 System.err.println("Error fetching user details for a user: " + e.getMessage()); 9 } 10}
在上述代码中,
future.get(5, TimeUnit.SECONDS)
将阻塞等待指定的超时时间(这里是 5 秒),直到任务完成并返回结果。如果遇到异常(如线程被中断、任务执行异常或超时),则捕获并处理。
至此,已经使用线程池并发地查询了 userIdList
中每个用户的详情。请注意,实际应用中可能需要根据具体需求对线程池参数、异常处理、查询结果合并等方面进行更细致的调整。