如何检查 bash 输入是否被重定向到 PHP 脚本？

Question

我正在编写一个 PHP 脚本，我希望能够选择性地使用文件作为脚本输入。这样：

$ php script.php < file.txt

实际上，我可以使用 file_get_contents 做到这一点：

$data = file_get_contents('php://stdin');

但是，如果我没有将文件传递给输入，脚本会无限期挂起，等待输入。

我尝试了以下方法，但没有成功：

$data = '';
$in = fopen('php://stdin', 'r');
do {
    $bytes = fread($in, 4096);
    // Maybe the input will be empty here?! But no, it's not :(
    if (empty($bytes)) {
        break;
    }
    $data .= $bytes;
} while (!feof($in));

脚本等待fread 返回一个值，但它永远不会返回。我猜它会像 file_get_contents 一样等待一些输入。

另一个尝试是用while { ... } 替换do { ... } while 循环，在尝试读取输入之前检查EOF。但这也没有用。

关于如何实现这一目标的任何想法？

Answer 1

您可以通过stream_set_blocking() 函数将 STDIN 设置为非阻塞。

function stdin()
{
    $stdin = '';
    $fh = fopen('php://stdin', 'r');
    stream_set_blocking($fh, false);
    while (($line = fgets($fh)) !== false) {
        $stdin .= $line;
    }
    return $stdin;
}

$stdin = stdin(); // returns the contents of STDIN or empty string if nothing is ready

显然，您可以根据需要将每次使用 line-at-a-time fgets() 更改为 hunk-at-a-time fread()。