1 package customer
  2 
  3 import (
  4     "encoding/json"
  5     "fmt"
  6     "os"
  7 )
  8 
  9 type customer struct {
 10     Name    string
 11     Sex     string
 12     Age     int
 13     Phone   string
 14     Email   string
 15     Address string
 16 }
 17 
 18 type CusSlice []customer
 19 
 20 type MainDisplay struct {
 21     CusList CusSlice
 22 }
 23 
 24 func (self *MainDisplay) SaveJsonToFile() {
 25     dir, err := os.Getwd()
 26     fmt.Println("當(dāng)前執(zhí)行路徑:", dir, err)
 27     file := dir + "/data.json"
 28     var fh *os.File
 29 
 30     fi, _ := os.Stat(file)
 31     if fi == nil {
 32         fh, _ = os.Create(file)
 33     } else {
 34         fh, _ = os.OpenFile(file, os.O_RDWR, 0666)
 35     }
 36     defer fh.Close()
 37 
 38     bytes, _ := json.Marshal(*self)
 39     fh.Write(bytes)
 40 }
 41 
 42 func (self *MainDisplay) LoadJsonByFile() {
 43     dir, err := os.Getwd()
 44     fmt.Println("當(dāng)前執(zhí)行路徑:", dir, err)
 45     file := dir + "/data.json"
 46     var fh *os.File
 47 
 48     fi, _ := os.Stat(file)
 49     if fi == nil {
 50         fh, _ = os.Create(file)
 51     } else {
 52         fh, _ = os.OpenFile(file, os.O_RDWR, 0666)
 53     }
 54     defer fh.Close()
 55 
 56     bytes := make([]byte, fi.Size())
 57     fh.Seek(0, 0)
 58     fh.Read(bytes)
 59     json.Unmarshal(bytes, self)
 60 
 61 }
 62 
 63 func (self *MainDisplay) AddCustomer() {
 64     var Name string
 65     var Sex string
 66     var Age int
 67     var Phone string
 68     var Email string
 69     var Address string
 70 
 71     fmt.Print("請(qǐng)輸入姓名:")
 72     fmt.Scanln(&Name)
 73     fmt.Print("請(qǐng)輸入性別:")
 74     fmt.Scanln(&Sex)
 75     fmt.Print("請(qǐng)輸入年齡:")
 76     fmt.Scanln(&Age)
 77     fmt.Print("請(qǐng)輸入電話:")
 78     fmt.Scanln(&Phone)
 79     fmt.Print("請(qǐng)輸入郵箱:")
 80     fmt.Scanln(&Email)
 81     fmt.Print("請(qǐng)輸入住址:")
 82     fmt.Scanln(&Address)
 83 
 84     self.CusList = append(self.CusList, customer{Name, Sex, Age, Phone, Email, Address})
 85     fmt.Println("添加成功!")
 86     fmt.Println()
 87     fmt.Println()
 88     fmt.Println()
 89 }
 90 
 91 func (self *MainDisplay) ModifyCustomer() {
 92     var Name string
 93     var Sex string
 94     var Age int
 95     var Phone string
 96     var Email string
 97     var Address string
 98     var id int
 99 
100     fmt.Print("請(qǐng)輸入要修改的序號(hào):")
101     fmt.Scanln(&id)
102     if len(self.CusList) < id {
103         fmt.Println("修改失敗!序列號(hào)不存在!")
104         fmt.Println()
105         fmt.Println()
106         fmt.Println()
107         return
108     }
109     fmt.Print("請(qǐng)輸入修改后的姓名:")
110     fmt.Scanln(&Name)
111     fmt.Print("請(qǐng)輸入修改后的性別:")
112     fmt.Scanln(&Sex)
113     fmt.Print("請(qǐng)輸入修改后的年齡:")
114     fmt.Scanln(&Age)
115     fmt.Print("請(qǐng)輸入修改后的電話:")
116     fmt.Scanln(&Phone)
117     fmt.Print("請(qǐng)輸入修改后的郵箱:")
118     fmt.Scanln(&Email)
119     fmt.Print("請(qǐng)輸入修改后的住址:")
120     fmt.Scanln(&Address)
121 
122     if Name == "" {
123         Name = self.CusList[id-1].Name
124     }
125     if Sex == "" {
126         Sex = self.CusList[id-1].Sex
127     }
128     if Age == 0 {
129         Age = self.CusList[id-1].Age
130     }
131     if Phone == "" {
132         Phone = self.CusList[id-1].Phone
133     }
134     if Email == "" {
135         Email = self.CusList[id-1].Email
136     }
137     if Address == "" {
138         Address = self.CusList[id-1].Address
139     }
140     self.CusList[id-1].Name = Name
141     self.CusList[id-1].Sex = Sex
142     self.CusList[id-1].Age = Age
143     self.CusList[id-1].Phone = Phone
144     self.CusList[id-1].Email = Email
145     self.CusList[id-1].Address = Address
146 
147     fmt.Println("修改成功!")
148     fmt.Println()
149     fmt.Println()
150     fmt.Println()
151 }
152 
153 func (self *MainDisplay) ShowCustomer() {
154 
155     fmt.Printf("%20v%20v%20v%20v%20v%20v%20v\n\n", "Sort", "Name", "Sex", "Age", "Phone", "Email", "Address")
156     for index, val := range self.CusList {
157         fmt.Printf("%20v%20v%20v%20v%20v%20v%20v\n", index+1, val.Name, val.Sex, val.Age, val.Phone, val.Email, val.Address)
158     }
159     fmt.Println()
160     fmt.Println()
161     fmt.Println()
162 }
163 
164 func (self *MainDisplay) DeleteCustomer() {
165     var id int
166 
167     fmt.Print("請(qǐng)輸入要修改的序號(hào):")
168     fmt.Scanln(&id)
169     if len(self.CusList) < id {
170         fmt.Println("修改失敗!序列號(hào)不存在!")
171         fmt.Println()
172         fmt.Println()
173         fmt.Println()
174         return
175     }
176     choise1 := ""
177     fmt.Printf("是否確認(rèn)刪除'%v'退出?(y/n)", self.CusList[id-1].Name)
178     fmt.Scanln(&choise1)
179     if choise1 == "y" {
180         for i := id; i < len(self.CusList); i++ {
181             self.CusList[i-1] = self.CusList[i]
182         }
183         self.CusList = self.CusList[:len(self.CusList)-2]
184     }
185     fmt.Println()
186     fmt.Println()
187     fmt.Println()
188 
189 }
190 
191 func (self *MainDisplay) ViewMain() {
192 
193     self.LoadJsonByFile()
194 
195 mainfor:
196     for {
197         var choise int
198         fmt.Println("-----------------客戶信息管理系統(tǒng)-----------------")
199         fmt.Println("                 1. 客 戶 列 表")
200         fmt.Println("                 2. 添 加 客 戶")
201         fmt.Println("                 3. 修 改 客 戶")
202         fmt.Println("                 4. 刪 除 客 戶")
203         fmt.Println("                 5. 退      出")
204         fmt.Print("清選擇(1~5)操作:")
205         fmt.Scanln(&choise)
206         switch choise {
207         case 1:
208             self.ShowCustomer()
209         case 2:
210             self.AddCustomer()
211         case 3:
212             self.ModifyCustomer()
213         case 4:
214             self.DeleteCustomer()
215         case 5:
216             choise1 := ""
217             fmt.Print("是否確認(rèn)退出?(y/n)")
218             fmt.Scanln(&choise1)
219             if choise1 == "y" {
220                 break mainfor
221             }
222             fmt.Println()
223             fmt.Println()
224             fmt.Println()
225 
226         default:
227             fmt.Println("輸入錯(cuò)誤,請(qǐng)重新選擇!")
228             fmt.Println()
229             fmt.Println()
230             fmt.Println()
231         }
232     }
233     self.SaveJsonToFile()
234 }
235 
236 func CustomerFactory() *MainDisplay {
237     return &MainDisplay{}
238 }
239 
240 func TestCustomer() {
241     m := CustomerFactory()
242     m.ViewMain()
243 }