重新认识 Java 的 System.in

重新认识 Java 的 System.in

以前也写过不少命令行的程序，处理文件时总需要通过参数指定路径，直到今天看资料时发现了一种我自己从来没用过的方式。这种方式让我重新认识了 System.in。

下面是一个简单的 Cat 命令的例子，这里提供了 -n 参数用于配置是否显示行号。

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

/**
 * @author liuzh
 * @since 2017/4/15.
 */
public class Cat {
    public static void main(String[] args) throws IOException {
        //是否显示行号，使用参数 -n 启用
        boolean showNumber = args.length > 0 && Arrays.asList(args).contains("-n");
        int num = 0;
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        String line = reader.readLine();
        while (line != null) {
            if (showNumber) {
                num++;
                System.out.printf("%1$8s %2$s%n", num, line);
            } else {
                System.out.println(line);
            }
            line = reader.readLine();
        }
    }
}

这个方法中用到了参数，参数只用于判断是否存在-n这个参数，没有通过参数指定文件。

这里获取文件内容的方式就是 System.in，从输入流中读取。输入流中怎么提供文件内容呢？

就是通过输入重定向到命令。针对上面的 Cat.java 文件执行下面的命令：

javac Cat.java
java Cat -n < Cat.java

先使用 javac 编译，在通过 java 命令执行，通过输入重定向将 Cat.java 作为命令的输入流。

上面命令执行后，输出内容如下：

       1 import java.io.BufferedReader;
       2 import java.io.IOException;
       3 import java.io.InputStreamReader;
       4 import java.util.Arrays;
       5
       6 /**
       7  * @author liuzh
       8  * @since 2017/4/15.
       9  */
      10 public class Cat {
      11     public static void main(String[] args) throws IOException {
      12         boolean showNumber = args.length > 0 && Arrays.asList(args).contains("-n");
      13         int num = 0;
      14         BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
      15         String line = reader.readLine();
      16         while (line != null) {
      17             if (showNumber) {
      18                 num++;
      19                 System.out.printf("%1$8s %2$s%n", num, line);
      20             } else {
      21                 System.out.println(line);
      22             }
      23             line = reader.readLine();
      24         }
      25     }
      26 }

如果只是处理文件，和参数方式指定文件路径没太大的区别。但是如果通过管道方式，就可以很方便的将前面命令的输出流作为输入流继续进行处理。例如下面的命令：

java Cat -n < Cat.java | java Cat -n

前一个命令的输出会作为第二个命令的输入，这会在原有行号的基础上增加一个行号，结果如下：

       1        1 import java.io.BufferedReader;
       2        2 import java.io.IOException;
       3        3 import java.io.InputStreamReader;
       4        4 import java.util.Arrays;
       5        5
       6        6 /**
       7        7  * @author liuzh
       8        8  * @since 2017/4/15.
       9        9  */
      10       10 public class Cat {
      11       11     public static void main(String[] args) throws IOException {
      12       12         boolean showNumber = args.length > 0 && Arrays.asList(args).contains("-n");
      13       13         int num = 0;
      14       14         BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
      15       15         String line = reader.readLine();
      16       16         while (line != null) {
      17       17             if (showNumber) {
      18       18                 num++;
      19       19                 System.out.printf("%1$8s %2$s%n", num, line);
      20       20             } else {
      21       21                 System.out.println(line);
      22       22             }
      23       23             line = reader.readLine();
      24       24         }
      25       25     }
      26       26 }

合理使用这种方式可以在某些情况下起到良好的作用。