Java 中限制方法的返回时间最新方法

沃梦达教程 Java开发

2023-07-14

最近在研究ChatGPT的API调用，因为ChatGPT的API调用时间通常超过30秒，所以我们希望在程序中限制这个方法的执行时间，不要让方法花太长时间去执行了，今天通过本文给大家分享Java中如何限制方法的返回时间，感兴趣的朋友跟随小编一起看看吧

最近在研究 ChatGPT 的 API 调用。

因为 ChatGPT 的 API 调用时间通常超过 30 秒。

所以我们希望在程序中限制这个方法的执行时间，不要让方法花太长时间去执行了。

JDK 方法

可以使用 JDK 中的 ExecutorService 方法来对调用的方法进行处理。

代码如下：

        ExecutorService executor = newCachedThreadPool();
        Callable<Object> task = new Callable<Object>() {
            public String call() {
                return callChatGPT(content);
            }
        };
        Future<Object> future = executor.submit(task);
        try {
            Object result = future.get(15, TimeUnit.SECONDS);
            contentGPT = (String) result;
        } catch (TimeoutException ex) {
            contentGPT = "ChatGPT API Time out, Please Re-try it";
        } catch (InterruptedException e) {
            contentGPT = "ChatGPT API Time out, Please Re-try it";
        } catch (ExecutionException e) {
            contentGPT = "ChatGPT API Time out, Please Re-try it";
        } finally {
            future.cancel(true); // may or may not desire this
        }

在我们的调用方法 callChatGPT 中，我们配置了一个 ExecutorService 执行器。

在这个执行器中，我们配置一个任务。

然后这个任务我们指定了执行时间为 15 秒。

如果这个方法的执行时间超过了 15 秒，程序将会抛出一个异常。