时间戳

首页 > 解决方案 > 读取任意数量的输入标准输入

go - 读取任意数量的输入标准输入

问题描述

我需要从中读取任意数量的输入stdin。我有以下代码可以正常工作并从命令行读取任意数量的输入：

// Reads from command-line arguments
package main

import(
        "fmt"
        "os"
        "strconv"
)

func main(){
        for _,arg := range os.Args[1:] {
                n, err := strconv.Atoi(arg)
                if err != nil {
                        fmt.Fprintf(os.Stderr, "Error: %v\n", err)
                        os.Exit(1)
                }
                fmt.Printf("%d\n", n)
        }
}

但是，我想将其更改为从 读取stdin，我所做的如下：

// Reads input from stdin
package main

import "fmt"

func main(){
        var a, b, c, d int
        count, err := fmt.Scan(&a, &b, &c, &d)
        fmt.Printf("Arguments Read: %d\n", count)
        fmt.Printf("%d, %d, %d, %d\n", a, b, c, d)
        if err != nil {
                fmt.Printf("%v\n", err)
        }
}

但在代码的第二个版本中，我必须从stdin. 如何从中读取任意数量的参数stdin？

标签： go

解决方案

您可以循环条件并继续从标准输入读取输入，直到满足该条件。在下面的示例中，如果输入是“退出”程序完成读取并输出所有提供的参数。

inputOpen := true
var args []string

fmt.Println("Provide argument, type exit to finish")
for inputOpen {
    var arg string
    var err error

    if _, err = fmt.Scan(&arg); err != nil {
        fmt.Printf("%v\n", err)
    }


    if arg == "exit" {
        inputOpen = false
    } else {
        args = append(args, arg)
    }
}

fmt.Println(fmt.Sprintf("Arguments: %v", args))

推荐阅读