无需调用Shell命令，如何更新Spring Shell提示符？（Java11/2.1.11）

要实现后台任务或外部事件触发后立即更新Shell提示符，核心是让动态提示符的数据源实时更新，并触发LineReader重绘当前终端行，而不是仅在用户回车后刷新。以下是具体实现步骤：

1. 实现线程安全的状态持有类 #

用一个Spring Bean维护服务状态，确保多线程下的安全访问：

import org.springframework.stereotype.Component;
import java.util.concurrent.atomic.AtomicReference;

@Component
public class ServiceStatusHolder {
    private final AtomicReference<String> currentStatus = new AtomicReference<>("IDLE");

    public void updateStatus(String newStatus) {
        currentStatus.set(newStatus);
    }

    public String getCurrentStatus() {
        return currentStatus.get();
    }
}

2. 自定义动态PromptProvider #

让Spring Shell的提示符从状态持有类获取最新状态：

import org.jline.utils.AttributedString;
import org.jline.utils.AttributedStyle;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.shell.jline.PromptProvider;
import org.springframework.stereotype.Component;

@Component
public class DynamicStatusPrompt implements PromptProvider {

    private final ServiceStatusHolder statusHolder;

    @Autowired
    public DynamicStatusPrompt(ServiceStatusHolder statusHolder) {
        this.statusHolder = statusHolder;
    }

    @Override
    public AttributedString getPrompt() {
        String status = statusHolder.getCurrentStatus();
        // 可以根据状态设置不同颜色，比如ERROR用红色
        AttributedStyle style = status.equals("ERROR") 
            ? AttributedStyle.DEFAULT.foreground(AttributedStyle.RED)
            : AttributedStyle.DEFAULT.foreground(AttributedStyle.GREEN);
        return new AttributedString("service-manager [" + status + "]> ", style);
    }
}

3. 后台任务/外部事件中触发提示符更新 #

在状态变更后，调用LineReader的redrawLine()和flush()方法强制重绘终端行：

import org.jline.reader.LineReader;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

@Component
public class ServiceStatusMonitor {

    private final ServiceStatusHolder statusHolder;
    private final LineReader lineReader;

    @Autowired
    public ServiceStatusMonitor(ServiceStatusHolder statusHolder, LineReader lineReader) {
        this.statusHolder = statusHolder;
        this.lineReader = lineReader;
    }

    @Scheduled(fixedRate = 5000) // 模拟每5秒检查一次服务状态
    public void checkAndUpdateStatus() {
        // 替换为实际的服务状态检查逻辑
        String newStatus = checkExternalServiceStatus();
        statusHolder.updateStatus(newStatus);

        // 关键：触发终端重绘，更新提示符
        if (lineReader != null) {
            lineReader.getTerminal().writer().flush();
            lineReader.redrawLine();
            lineReader.flush();
        }
    }

    private String checkExternalServiceStatus() {
        // 模拟状态变更
        return Math.random() > 0.5 ? "RUNNING" : "ERROR";
    }
}

关键注意点 #

• 线程安全：必须用线程安全的方式维护状态（比如AtomicReference），避免后台线程和Shell主线程的并发访问问题。
• 重绘逻辑：redrawLine()会重新调用PromptProvider.getPrompt()获取最新状态，并刷新当前终端行，即使用户正在输入命令，也会保留已输入内容并更新提示符。
• LineReader注入：Spring Shell 2.x中LineReader是可直接注入的Bean，无需手动创建。

内容的提问来源于stack exchange，提问作者Phil B.