操做Unix系统下的pipe管道
发布者:admin 发表于:439天前 阅读数:562 评论:0

将一个程序的输出传递给另一个程序的输入时,Unix管道很有用。 例如:

echo "test case" | wc -l
1

在Go应用程序中,可以使用os.Stdin读入管道的左侧,并像文件描述符一样工作。为了证明这一点,本节将在管道的左侧进行输入并返回单词列表及其出现次数。这些单词将以空白为分隔标记。

实践

1.建立 pipes.go:

package main

import (
    "bufio"
    "fmt"
    "io"
    "os"
)

// WordCount 获取一个io.Reader并返回一个map,每个单词作为一个键,它的出现次数为对应的值
func WordCount(f io.Reader) map[string]int {
    result := make(map[string]int)

    // 建立scanner处理文件io.Reader接口
    scanner := bufio.NewScanner(f)
    scanner.Split(bufio.ScanWords)

    for scanner.Scan() {
        result[scanner.Text()]++
    }

    if err := scanner.Err(); err != nil {
        fmt.Fprintln(os.Stderr, "reading input:", err)
    }

    return result
}

func main() {
    fmt.Printf("string: number_of_occurrences\n\n")
    for key, value := range WordCount(os.Stdin) {
        fmt.Printf("%s: %d\n", key, value)
    }
}

2.运行:

echo "some string" | go run pipes.go

3.多执行一些输入:

$ echo "test case" | go run pipes.go
string: number_of_occurrences
test: 1
case: 1
$ echo "test case test" | go run pipes.go
string: number_of_occurrences
test: 2
case: 1

说明

在go中使用管道非常简单,特别是如果你熟悉使用I/O接口。例如第1章 I/O 和文件系统中的管道配方创建应用程序(https//en.wikipedia.org/wiki/Tee_(command)),其中所有管道输入都会立即写入到标准输出和文件。

本节使用scanner来标记os.Stdin对象的io.Reader接口。请注意在完成所有读取后必须检查错误。