c++ - 重定向两个单独的标准输入作为程序的输入

Question

以下示例输出来自 stdin 的输入:

#include <iostream>

int main(int argc, char **argv)
{
    std::string s1, s2;
    while (getline(std::cin, s1))
    {
        std::cout << s1 << " 1\n";
    }

    while (getline(std::cin, s2))
    {
        std::cout << s2 << "2\n";
    }
    return 0;
}

我像这样运行程序:cat main.cpp | ./main 输出

#include <iostream> 1
 1
int main(int argc, char **argv) 1
{ 1
 1
    std::string s1, s2; 1
    while (getline(std::cin, s1)) 1
    { 1
        std::cout << s1 << " 1\n"; 1
    } 1
         1
    while (getline(std::cin, s2)) 1
    { 1
        std::cout << s2 << "2\n"; 1
    } 1
} 1

有没有办法通过 stdin 将两个单独的输入传送到程序？也就是说，如果我写类似 cat main.cpp cat main.cpp | 的内容./main，那么第一个输入被分配给 s1，第二个输入被分配给 s2？

Answer 1

据我所知这是不可能的，但您可以简单地使用两个包含您的输入的文件，将它们作为命令行参数传递并从中读取。

#include <iostream>
#include <fstream>

int main(int argc, char **argv)
{
    if (argc != 3) 
      return -1;

    std::ifstream ifs1(argv[1]);
    std::ifstream ifs2(argv[2]);

    if (!ifs1.is_open() || !ifs2.is_open())
      return -2;

    std::string s1, s2;
    while (getline(ifs1, s1))
    {
      std::cout << s1 << " 1\n";
    }

    while (getline(ifs2, s2))
    {
      std::cout << s2 << "2\n";
    }
    return 0;
}

并这样调用它:

./main main.cpp main.cpp

您甚至可以使其更加灵活，允许您读取任意数量的文件:

#include <iostream>
#include <fstream>

int main(int argc, char **argv)
{
    if (argc == 1) 
      return -1; // no input file given

    for (int i = 1; i < argc; ++i) { 
      std::ifstream ifs(argv[i]);

      if (!ifs.is_open()) {
        std::cout << argv[i] << ":file not found\n";
        continue;
      }

      std::string line;
      while (getline(ifs, line))
      {
        std::cout << line << " " << i << "\n";
      }
    }
}