221229命令行参数编程

2022-12-28  本文已影响0人  里欧老师
  1. 关于golang的文件组织
    <projName>
    |-<bin>
    |-<pkg>
    |-<src>
    |-<calu>
    main.go
    |-<simpleMath>
    add.go
    sqrt.go
    <>表示文件夹,src中每个文件夹内的所有文件,开头都应该是package 文件名,这样的话同一个文件夹下所有文件可以视为是同一个文件

  2. os.Args
    可执行文件本身是第一个args[0]

// calc.go
package main

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

var Usage = func() {
    fmt.Println("USAGE: calcu command [arguments] ...")
    fmt.Println("\nThe commands are:\n\tadd\tAddition of two values.\n\tsqrt\tSquare root of a non-negative value.")
}

func main() {
    args := os.Args
    if args == nil || len(args) < 3 {
        Usage()
        return
    }

    switch args[1] {
    case "add":
        if len(args) != 4 {
            fmt.Println("USAGE: calcu add <integer#1> <integer#2>")
            return
        }

        v1, err1 := strconv.Atoi(args[2])
        v2, err2 := strconv.Atoi(args[3])
        if err1 != nil || err2 != nil {
            fmt.Println("USAGE: calcu add <integer#1> <integer#2>")
            return
        }
        ret := simpleMath.Add(v1, v2)
        fmt.Println("Result: ", ret)

    case "sqrt":
        if len(args) != 3 {
            fmt.Println("USAGE: calcu sqrt <integer>")
            return
        }
        v, err := strconv.Atoi(args[2])
        if err != nil {
            fmt.Println("USAGE: calcu sqrt <integer>")
            return
        }
        ret := simpleMath.Sqrt(v)
        fmt.Println("Result: ", ret)

    default:
        Usage()
    }
}

// add.go
package simpleMath

func Add(a int, b int) int {
    return a + b
}
// sqrt.go
package simpleMath

import (
    "math"
)

func Sqrt(i int) int {
    v := math.Sqrt(float64(i))
    return int(v)
}

上一篇 下一篇

猜你喜欢

热点阅读