如何触发关闭 Guava AbstractScheduledService?

How to triggerShutdown a Guava AbstractScheduledService?

我正在使用一些从 AbstractScheduledService 继承的服务,这些服务由 ServiceManager 管理。一切正常,但现在,有一项服务 runOneIteration 需要相当长的时间,结果,我的进程终止时间太长(超过五秒)。

还有其他服务继承自 AbstractExecutionThreadService,它们有类似的问题,我可以通过

解决
@Override
protected final void triggerShutdown() {
    if (thread != null) thread.interrupt();
}

并在 run 方法中存储 private volatile thread。但是,如 this issue.

中所述,AbstractScheduledService 没有 triggerShutdown

我已经考虑过让 runOneIteration 做更少工作的替代方案,但它既丑陋又低效。

我无法覆盖 stopAsync,因为它是最终版本,我看不到任何其他内容。 有这样的钩子吗?

你能用这个吗?您有什么理由不能自己添加 triggerShutdown 吗?

class GuavaServer {
    public static void main(String[] args) throws InterruptedException {
        GuavaServer gs = new GuavaServer();
        Set<ForceStoppableScheduledService> services = new HashSet<>();
        ForceStoppableScheduledService ts = gs.new ForceStoppableScheduledService();
        services.add(ts);
        ServiceManager manager = new ServiceManager(services);
        manager.addListener(new Listener() {
            public void stopped() {
                System.out.println("Stopped");
            }

            public void healthy() {
                System.out.println("Health");
            }

            public void failure(Service service) {
                System.out.println("Failure");
                System.exit(1);
            }
        }, MoreExecutors.directExecutor());

        manager.startAsync(); // start all the services asynchronously
        Thread.sleep(3000);
        manager.stopAsync();
        //maybe make a manager.StopNOW()?
        for (ForceStoppableScheduledService service : services) {
            service.triggerShutdown();
        }
    }

    public class ForceStoppableScheduledService extends AbstractScheduledService {

        Thread thread;

        @Override
        protected void runOneIteration() throws Exception {
            thread = Thread.currentThread();
            try {
                System.out.println("Working");
                Thread.sleep(10000);
            } catch (InterruptedException e) {// can your long process throw InterruptedException?
                System.out.println("Thread was interrupted, Failed to complete operation");
            } finally {
                thread = null;
            }
            System.out.println("Done");
        }

        @Override
        protected Scheduler scheduler() {
            return Scheduler.newFixedRateSchedule(0, 1, TimeUnit.SECONDS);
        }

        protected void triggerShutdown() {
            if (thread != null) thread.interrupt();
        }
    }
}