JavaFX中在Task内调用Robot截图时程序冻结问题求助

问题根源 #

JavaFX 中所有与UI交互的API（包括javafx.scene.robot.Robot的所有方法）都必须在JavaFX应用主线程上执行。而Service/Task的call()方法默认运行在独立的后台工作线程，你在后台线程直接调用robot.getScreenCapture()会触发线程冲突，导致永久阻塞。你把全部逻辑放到Platform.runLater()中可正常运行，就是因为此时所有代码都被调度到了主线程执行。

解决方案 #

把必须跑在主线程的截图操作和不需要占用主线程的像素统计逻辑拆分，截图阶段调度到主线程执行，完成后再在后台线程做计算即可。注意Platform.runLater()是异步调度，需要加同步机制等待截图完成再继续后台逻辑。

修改后的PixelCounterService核心代码如下：

package sample;

import javafx.concurrent.Service;
import javafx.concurrent.Task;
import javafx.application.Platform;
import javafx.scene.image.PixelReader;
import javafx.scene.image.WritableImage;
import javafx.scene.robot.Robot;
import javafx.stage.Screen;
import java.util.concurrent.CountDownLatch;

public class PixelCounterService extends Service<Integer> {
    private Robot robot;

    public PixelCounterService(Robot robot){
        this.robot = robot;
    }

    @Override
    protected Task<Integer> createTask() {
        return new Task<>() {
            @Override
            protected Integer call() throws Exception {
                Screen screen = Screen.getPrimary();
                // 同步工具，用于等待主线程截图完成
                CountDownLatch latch = new CountDownLatch(1);
                // 用数组存储截图结果，适配lambda变量必须为final的要求
                final WritableImage[] screenshotHolder = new WritableImage[1];

                // 截图操作调度到主线程执行
                Platform.runLater(() -> {
                    try {
                        screenshotHolder[0] = robot.getScreenCapture(null, screen.getBounds());
                    } finally {
                        latch.countDown();
                    }
                });

                // 阻塞等待截图完成，再执行后续后台计算
                latch.await();
                WritableImage screenshot = screenshotHolder[0];

                // 像素统计逻辑完全在后台线程执行，不会占用主线程资源
                PixelReader pixelReader  = screenshot.getPixelReader();
                int redPixels = 0;
                for(int i = 0; i < screenshot.getWidth(); i++){
                    for(int j = 0; j < screenshot.getHeight(); j++){
                        if(pixelReader.getColor(i, j).getRed() == 1){
                            redPixels++;
                        }
                    }
                }
                return redPixels;
            }
        };
    }
}

注意事项 #

• 已经生成的WritableImage和PixelReader可以安全在后台线程访问，不需要额外同步
• 如果需要支持任务中断，可给latch.await()添加超时时间，同时监听任务的取消状态
• 不要把像素统计逻辑放到Platform.runLater()中，否则大分辨率屏幕下计算会卡住UI交互

内容的提问来源于stack exchange，提问作者Czachodym