go入门项目：(1) 基于命令行的图书的增删查改

Go 语言入门练手项目系列

• 01 基于命令行的图书的增删查改
  
• 02 文件管理
  持续更新中...
  

> 本文来自博客园，作者：Arway，转载请注明原文链接：https://www.cnblogs.com/cenjw/p/gobeginner-proj-bookstore-cli.html介绍

这是一个基于go应用程序命令行的bookstore小项目，具体实现了图书的增删改查，适合刚入门go的朋友熟悉go的语法。
知识点

• go flag，提供了命令行参数解析的功能
  
• go json的序列化与反序列化
  
• go ioutil 读写文件
  

功能演示

• 列出全部图书
  

1. $ .\bookstore.exe get -all
2. Id      Title                   Author
3. 1       Go语言圣经              Ailen A
4. 2       Java编程思想            Bruce Eckel
5. 3       流畅的Python            Luciano Ramalho
6. 4       PHP权威指南             CENJW

复制代码

• 通过id获取相应图书
  

1. $ .\bookstore.exe get -id 1
2. Id      Title           Author
3. 1       Go语言圣经      Ailen A

复制代码

• 添加图书
  

1. $ .\bookstore.exe add -id 123 -title 'Head First Go' -author unknown
2. add successfully!

复制代码

• 添加图书，如果已经存在相同id的图书，提示已存在
  

1. $ .\bookstore.exe add -id 1 -title test -author test
2. 2022/06/24 18:05:44 Book already exists.

复制代码

• 更新图书
  

1. $ .\bookstore.exe update -id 123 -title 'head first go' -author UNKNOWN
2. update successfully!

复制代码

• 删除图书
  

1. $ .\bookstore.exe delete -id 123
2. Delete successfully!

复制代码

本文来自博客园，作者：Arway，转载请注明原文链接：https://www.cnblogs.com/cenjw/p/gobeginner-proj-bookstore-cli.html

功能实现

main.go

1. func main() {
2.         // get book/books
3.         getCmd := flag.NewFlagSet("get", flag.ExitOnError)
4.         getAll := getCmd.Bool("all", false, "List all books")
5.         getId := getCmd.String("id", "", "Get a book by id")
7.         // add/update book
8.         addCmd := flag.NewFlagSet("add", flag.ExitOnError)
9.         addId := addCmd.String("id", "", "Book id")
10.         addTitle := addCmd.String("title", "", "Book title")
11.         addAuthor := addCmd.String("author", "", "Book author")
13.         // delete book
14.         deleteCmd := flag.NewFlagSet("delete", flag.ExitOnError)
15.         deleteId := deleteCmd.String("id", "", "Book id")
16.        
17.         // check command
18.         if len(os.Args) < 2 {
19.                 log.Fatal("Expected get, add, update, delete commands.")
20.         }
22.         switch os.Args[1] {
23.         case "get":
24.                 handleGetBooks(getCmd, getAll, getId)
25.         case "add":
26.                 handleAddOrUpdateBook(addCmd, addId, addTitle, addAuthor, true)
27.         case "update":
28.                 handleAddOrUpdateBook(addCmd, addId, addTitle, addAuthor, false)
29.         case "delete":
30.                 handleDeleteBook(deleteCmd, deleteId)
31.         default:
32.                 fmt.Println("Pls provide get, add, update, delete commands.")
33.         }
35. }

复制代码

books.go

1. type Book struct {
2.         Id     string `json:"id"`
3.         Title  string `json:"title"`
4.         Author string `json:"author"`
5. }
7. // get all the books logic
8. func handleGetBooks(getCmd *flag.FlagSet, all *bool, id *string) {
9.         getCmd.Parse(os.Args[2:])
11.         if !*all && *id == "" {
12.                 fmt.Println("subcommand --all or --id needed") // PrintDefault打印集合中所有注册好的flag的默认值。除非另外配置，默认输出到标准错误输出中。
13.                 getCmd.PrintDefaults()
14.                 os.Exit(1)
15.         }
17.         if *all {
18.                 books := getBooks()
19.                 fmt.Println("Id\tTitle\t\t\tAuthor\t")
20.                 for _, book := range books {
21.                         fmt.Printf("%v\t%v\t\t%v\t\n", book.Id, book.Title, book.Author)
22.                 }
23.         }
25.         if *id != "" {
26.                 var isExist bool
27.                 books := getBooks()
28.                 fmt.Println("Id\tTitle\t\tAuthor\t")
29.                 for _, book := range books {
30.                         if book.Id == *id {
31.                                 fmt.Printf("%v\t%v\t%v\t\n", book.Id, book.Title, book.Author)
32.                                 isExist = true
33.                         }
34.                 }
36.                 if !isExist {
37.                         log.Fatal("Book not found!")
38.                 }
39.         }
40. }
42. // add or update a book logic
43. // addBook: true denote add a new book to file, false denote update a book
44. func handleAddOrUpdateBook(addCmd *flag.FlagSet, id, title, author *string, addBook bool) {
45.         addCmd.Parse(os.Args[2:])
47.         if *id == "" || *title == "" || *author == "" {
48.                 addCmd.PrintDefaults()
49.                 log.Fatal("Pls provide id, title, author commands.")
50.         }
51.         books := getBooks()
52.         var newBook Book
53.         if addBook {
54.                 for _, book := range books {
55.                         if book.Id == *id {
56.                                 log.Fatal("Book already exists.")       
57.                         } else {
58.                                 newBook = Book{*id, *title, *author}
59.                                 books = append(books, newBook)
60.                                 break
61.                         }
62.                 }
63.                
64.         } else {
65.                 var isExist bool
66.                 for i, book := range books {
67.                         if book.Id == *id {
68.                                 books[i] = Book{Id: *id, Title: *title, Author: *author}
69.                                 isExist = true
70.                         }
71.                 }
73.                 err := saveBooks(books)
74.                 checkError(err)
76.                 if !isExist {
77.                         log.Fatal("Book not found")
78.                 }
80.         }
82.         saveBooks(books)
84.         fmt.Printf("%s successfully!\n", os.Args[1])
85. }
87. func handleDeleteBook(deleteCmd *flag.FlagSet, id *string) {
88.         deleteCmd.Parse(os.Args[2:])
90.         if *id == "" {
91.                 deleteCmd.PrintDefaults()
92.                 log.Fatal("Pls provide id command.")
93.         }
95.         if *id != "" {
96.                 var isExist bool
97.                 books := getBooks()
98.                 for i, book := range books {
99.                         if book.Id == *id {
100.                                 books = append(books[:i], books[i+1:]...)
101.                                 isExist = true
102.                         }
103.                 }
105.                 if !isExist {
106.                         log.Fatal("Book not found!")
107.                 }
109.                 err := saveBooks(books)
110.                 checkError(err)
112.                 fmt.Println("Delete successfully!")
113.         }
114. }
116. func saveBooks(books []Book) error {
117.         bookBytes, err := json.Marshal(books)
118.         checkError(err)
119.         err = ioutil.WriteFile("./books.json", bookBytes, 0644)
120.         return err
121. }
123. func getBooks() (books []Book) {
124.         bookBytes, err := ioutil.ReadFile("./books.json")
125.         checkError(err)
126.         err = json.Unmarshal(bookBytes, &books)
127.         checkError(err)
128.         return books
129. }
131. func checkError(err error) {
132.         if err != nil {
133.                 log.Fatal("Error happended ", err)
134.         }
135. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！