Golang 基础语法的使用

package customer

import (
    "encoding/json"
    "fmt"
    "os"
)

type customer struct {
    Name    string
    Sex     string
    Age     int
    Phone   string
    Email   string
    Address string
}

type CusSlice []customer

type MainDisplay struct {
    CusList CusSlice
}

func (self *MainDisplay) SaveJsonToFile() {
    dir, err := os.Getwd()
    fmt.Println("当前执行路径:", dir, err)
    file := dir + "/data.json"
    var fh *os.File

    fi, _ := os.Stat(file)
    if fi == nil {
        fh, _ = os.Create(file)
    } else {
        fh, _ = os.OpenFile(file, os.O_RDWR, 0666)
    }
    defer fh.Close()

    bytes, _ := json.Marshal(*self)
    fh.Write(bytes)
}

func (self *MainDisplay) LoadJsonByFile() {
    dir, err := os.Getwd()
    fmt.Println("当前执行路径:", dir, err)
    file := dir + "/data.json"
    var fh *os.File

    fi, _ := os.Stat(file)
    if fi == nil {
        fh, _ = os.Create(file)
    } else {
        fh, _ = os.OpenFile(file, os.O_RDWR, 0666)
    }
    defer fh.Close()

    bytes := make([]byte, fi.Size())
    fh.Seek(0, 0)
    fh.Read(bytes)
    json.Unmarshal(bytes, self)

}

func (self *MainDisplay) AddCustomer() {
    var Name string
    var Sex string
    var Age int
    var Phone string
    var Email string
    var Address string

    fmt.Print("请输入姓名:")
    fmt.Scanln(&Name)
    fmt.Print("请输入性别:")
    fmt.Scanln(&Sex)
    fmt.Print("请输入年龄:")
    fmt.Scanln(&Age)
    fmt.Print("请输入电话:")
    fmt.Scanln(&Phone)
    fmt.Print("请输入邮箱:")
    fmt.Scanln(&Email)
    fmt.Print("请输入住址:")
    fmt.Scanln(&Address)

    self.CusList = append(self.CusList, customer{Name, Sex, Age, Phone, Email, Address})
    fmt.Println("添加成功!")
    fmt.Println()
    fmt.Println()
    fmt.Println()
}

func (self *MainDisplay) ModifyCustomer() {
    var Name string
    var Sex string
    var Age int
    var Phone string
    var Email string
    var Address string
    var id int

    fmt.Print("请输入要修改的序号:")
    fmt.Scanln(&id)
    if len(self.CusList) < id {
        fmt.Println("修改失败!序列号不存在!")
        fmt.Println()
        fmt.Println()
        fmt.Println()
        return
    }
    fmt.Print("请输入修改后的姓名:")
    fmt.Scanln(&Name)
    fmt.Print("请输入修改后的性别:")
    fmt.Scanln(&Sex)
    fmt.Print("请输入修改后的年龄:")
    fmt.Scanln(&Age)
    fmt.Print("请输入修改后的电话:")
    fmt.Scanln(&Phone)
    fmt.Print("请输入修改后的邮箱:")
    fmt.Scanln(&Email)
    fmt.Print("请输入修改后的住址:")
    fmt.Scanln(&Address)

    if Name == "" {
        Name = self.CusList[id-1].Name
    }
    if Sex == "" {
        Sex = self.CusList[id-1].Sex
    }
    if Age == 0 {
        Age = self.CusList[id-1].Age
    }
    if Phone == "" {
        Phone = self.CusList[id-1].Phone
    }
    if Email == "" {
        Email = self.CusList[id-1].Email
    }
    if Address == "" {
        Address = self.CusList[id-1].Address
    }
    self.CusList[id-1].Name = Name
    self.CusList[id-1].Sex = Sex
    self.CusList[id-1].Age = Age
    self.CusList[id-1].Phone = Phone
    self.CusList[id-1].Email = Email
    self.CusList[id-1].Address = Address

    fmt.Println("修改成功!")
    fmt.Println()
    fmt.Println()
    fmt.Println()
}

func (self *MainDisplay) ShowCustomer() {

    fmt.Printf("%20v%20v%20v%20v%20v%20v%20v\n\n", "Sort", "Name", "Sex", "Age", "Phone", "Email", "Address")
    for index, val := range self.CusList {
        fmt.Printf("%20v%20v%20v%20v%20v%20v%20v\n", index+1, val.Name, val.Sex, val.Age, val.Phone, val.Email, val.Address)
    }
    fmt.Println()
    fmt.Println()
    fmt.Println()
}

func (self *MainDisplay) DeleteCustomer() {
    var id int

    fmt.Print("请输入要修改的序号:")
    fmt.Scanln(&id)
    if len(self.CusList) < id {
        fmt.Println("修改失败!序列号不存在!")
        fmt.Println()
        fmt.Println()
        fmt.Println()
        return
    }
    choise1 := ""
    fmt.Printf("是否确认删除'%v'退出?(y/n)", self.CusList[id-1].Name)
    fmt.Scanln(&choise1)
    if choise1 == "y" {
        for i := id; i < len(self.CusList); i++ {
            self.CusList[i-1] = self.CusList[i]
        }
        self.CusList = self.CusList[:len(self.CusList)-2]
    }
    fmt.Println()
    fmt.Println()
    fmt.Println()

}

func (self *MainDisplay) ViewMain() {

    self.LoadJsonByFile()

mainfor:
    for {
        var choise int
        fmt.Println("-----------------客户信息管理系统-----------------")
        fmt.Println("                 1. 客 户 列 表")
        fmt.Println("                 2. 添 加 客 户")
        fmt.Println("                 3. 修 改 客 户")
        fmt.Println("                 4. 删 除 客 户")
        fmt.Println("                 5. 退      出")
        fmt.Print("清选择(1~5)操作:")
        fmt.Scanln(&choise)
        switch choise {
        case 1:
            self.ShowCustomer()
        case 2:
            self.AddCustomer()
        case 3:
            self.ModifyCustomer()
        case 4:
            self.DeleteCustomer()
        case 5:
            choise1 := ""
            fmt.Print("是否确认退出?(y/n)")
            fmt.Scanln(&choise1)
            if choise1 == "y" {
                break mainfor
            }
            fmt.Println()
            fmt.Println()
            fmt.Println()

        default:
            fmt.Println("输入错误,请重新选择!")
            fmt.Println()
            fmt.Println()
            fmt.Println()
        }
    }
    self.SaveJsonToFile()
}

func CustomerFactory() *MainDisplay {
    return &MainDisplay{}
}

func TestCustomer() {
    m := CustomerFactory()
    m.ViewMain()
}