go 语言命令行程序

标签：project   handler   tmp   handle   please   menu   cmd   string   number   

// lab3 project main.go

// minglinghang project main.go

package main

import (

    "fmt"

    "math"

    "os"

    "strings"

    "unsafe"

)

func add() {

    var firstNum, secondNum int

    //fmt.Println("Please input two integers: ")

    fmt.Scanln(&firstNum, &secondNum)

    fmt.Printf("%d + %d = %d\n", firstNum, secondNum, firstNum+secondNum)

}

func help() {

    ShowAllCommand(list)

}

func sub() {

    var firstNum, secondNum int

    //fmt.Println("Please input two integers: ")

    fmt.Scanln(&firstNum, &secondNum)

    fmt.Printf("%d - %d = %d\n", firstNum, secondNum, firstNum-secondNum)

}

func mul() {

    var firstNum, secondNum int

    //fmt.Println("Please input two integers: ")

    fmt.Scanln(&firstNum, &secondNum)

    fmt.Printf("%d * %d = %d\n", firstNum, secondNum, firstNum*secondNum)

}

func div() {

    var firstNum, secondNum int

    //fmt.Println("Please input two integers: ")

    fmt.Scanln(&firstNum, &secondNum)

    fmt.Printf("%d / %d = %.2f\n", firstNum, secondNum, float64(firstNum)/float64(secondNum))

}

func pow() {

    var firstNum, secondNum int

    //fmt.Println("Please input two integers: ")

    fmt.Scanln(&firstNum, &secondNum)

    fmt.Printf("%d ^ %d = %.0f\n", firstNum, secondNum, math.Pow(float64(firstNum), float64(secondNum)))

}

func mod() {

    var firstNum, secondNum int

    //fmt.Println("Please input two integers: ")

    fmt.Scanln(&firstNum, &secondNum)

    fmt.Printf("%d mod %d = %d\n", firstNum, secondNum, firstNum%secondNum)

}

func quit() {

    os.Exit(1)

}

type DataNode struct {

    next *DataNode

    cmd string

    desc string

    handler func()

}

type linknode struct {

    next *linknode

}

type linklist struct {

    head *linknode

    tail *linknode

}

func CreateLinklist() *linklist {

    return &linklist{}

}

func AddNode(list *linklist, newnode *linknode) {

    if list.head == nil {

        list.head = newnode

        list.tail = list.head

    } else {

        list.tail.next = newnode

        list.tail = newnode

    }

}

func ShowAllCommand(list *linklist) {

    for node := list.head; node != nil; node = node.next {

        tmp := (*DataNode)(unsafe.Pointer(node))

        fmt.Printf("%s - %s\n", tmp.cmd, tmp.desc)

    }

}

func searchcondition(node *DataNode, cmd interface{}) bool {

    str := cmd.(string)

    //if strings.Compare(str, node.cmd) {

    if node.cmd == str {

        return true

    } else {

        return false

    }

}

func search(list *linklist, cmd string) *DataNode {

    for node := list.head; node != nil; node = node.next {

        tmp := (*DataNode)(unsafe.Pointer(node))

        if searchcondition(tmp, cmd) {

            return tmp

        }

    }

    return (*DataNode)(unsafe.Pointer(list.head))

}

func Init(list *linklist) {

    menuconfig(list, "help", "this is help cmd", help)

    menuconfig(list, "add", "this is add cmd", add)

    menuconfig(list, "sub", "this is sub cmd", sub)

    menuconfig(list, "mul", "this is mul cmd", mul)

    menuconfig(list, "div", "this is div cmd", div)

    menuconfig(list, "mod", "this is mod cmd", mod)

    menuconfig(list, "pow", "this is pow cmd", pow)

    menuconfig(list, "quit", "this is quit cmd", quit)

    /*node := &DataNode{nil, "help", "this is help cmd", help}

    AddNode(list, (*linknode)(unsafe.Pointer(node)))

    node := &DataNode{nil, "add", "this is add cmd", add}

    AddNode(list, (*linknode)(unsafe.Pointer(node)))

    node = &DataNode{nil, "sub", "this is sub cmd", sub}

    AddNode(list, (*linknode)(unsafe.Pointer(node)))

    node = &DataNode{nil, "mul", "this is mul cmd", mul}

    AddNode(list, (*linknode)(unsafe.Pointer(node)))

    node = &DataNode{nil, "div", "this is div cmd", div}

    AddNode(list, (*linknode)(unsafe.Pointer(node)))

    node = &DataNode{nil, "mod", "this is mod cmd", mod}

    AddNode(list, (*linknode)(unsafe.Pointer(node)))

    node = &DataNode{nil, "pow", "this is pow cmd", pow}

    AddNode(list, (*linknode)(unsafe.Pointer(node)))

    node = &DataNode{nil, "quit", "this is squit cmd", quit}

    AddNode(list, (*linknode)(unsafe.Pointer(node)))

    */

}

func menuconfig(list *linklist, cmd string, desc string, handler func()) {

    node := &DataNode{nil, cmd, desc, handler}

    AddNode(list, (*linknode)(unsafe.Pointer(node)))

}

const CMD_LENTH = 10

func Excutemenu(list *linklist) {

    for {

        var argc int = 0

        var cmdi string

        var arg [5]string

        fmt.Println("Input a cmd number >")

        fmt.Scanf("%s", &cmdi)

        var cmdis []string = strings.Split(cmdi, " ")

        for (argc < len(cmdis)) && argc < CMD_LENTH {

            arg[argc] = cmdis[argc]

            argc = argc + 1

        }

        node := search(list, arg[0])

        if node != nil {

            node.handler()

        } else {

            fmt.Printf("this is a wrong cmd ")

        }

    }

}

var list *linklist

func main() {

    list = CreateLinklist()

    Init(list)

    Excutemenu(list)

}

go 语言命令行程序

标签：project   handler   tmp   handle   please   menu   cmd   string   number   

原文地址：http://www.cnblogs.com/zeroArn/p/7029455.html