Как принять ввод в одну строку, используя библиотеку кобры в Go

Question

Я пишу код на языке go с использованием кобры, в настоящее время я даю следующие данные:

 Calc add 
           Enter the Number of inputs
           2
           Enter the Numbers
           2
           4
 Output: Sum is : 6

В этом случае те, кто знаком с коброй, Calc - это мой проект, а add - команда, которую я использую. Я хочу, чтобы ввод был Calc add N2 2 4(в единственной строке), и вывод должен отображаться, где N - это переменная, которая определяет количество входов, а 2 4 - это числа, которые будут добавлены.

КОД ДЛЯ ДОБАВЛЕНИЯ КОМАНДЫ:

package cmd

import (
    "fmt"

    "github.com/spf13/cobra"
)

// addCmd represents the add command
var addCmd = &cobra.Command{
    Use:   "add",
    Short: "Addition value of given Numbers",

    Run: func(cmd *cobra.Command, args []string) {
        length := 0
    fmt.Println("Enter the number of inputs")
    fmt.Scanln(&length)
    fmt.Println("Enter the inputs")
    numbers := make([]int, length)
    for i := 0; i < length; i++ {
        fmt.Scanln(&numbers[i])
    }
      fmt.Println(numbers)

      sum:=0

for _, numbers := range numbers {

sum += numbers

}

fmt.Println("The Sum :",sum)


 },
}

func init() {
    RootCmd.AddCommand(addCmd)


}

п

Answer 1

Это выполнит вашу цель. Возьми свой номер во флаг --input, Дайте другие числа, чтобы добавить в качестве аргументов.

func NewCmd() *cobra.Command {
    var input int
    c := &cobra.Command{
        Use:   "add",
        Short: "Addition value of given Numbers",

        Run: func(cmd *cobra.Command, args []string) {
            if len(args) != input {
                fmt.Println(fmt.Sprintf("You need to provide %v number to sum up", input))
                os.Exit(1)
            }
            numbers := make([]int, input)
            for i := 0; i < input; i++ {
                num, _ := strconv.Atoi(args[i])
                numbers[i] = num
            }
            sum := 0
            for _, numbers := range numbers {
                sum += numbers
            }
            fmt.Println("The Sum :", sum)
        },
    }
    c.Flags().IntVar(&input, "input", 0, "Number of input")
    return c
}

func init() {
    cmd := NewCmd()
    RootCmd.AddCommand(cmd)
}

Входные данные:

Calc add --input=3 6 3 6

Выход: сумма: 15