How do I save an entire string as a txt file in Go?

huangapple
go评论109阅读模式
英文:

How do I save an entire string as a txt file in Go?

问题

我正在创建一个简单的Go文字处理程序。从命令行开始，我有两个提示：

$输入标题：

$输入正文：

该程序应该将文档保存为txt文件并打印到命令行。如果用户输入一个单词的标题和一个单词的正文，程序可以正常工作。但是，如果用户输入一个由多个单词组成的标题，会出现以下情况：

  1. $输入标题：这是一个标题
  3. $输入正文：s
  5. $  标题
  7. -bash: 标题: 找不到命令

以下是我目前的代码：

  1. package main
  3. import (
  4.     "fmt"
  5.     "io/ioutil"
  6. )
  8. //创建文档的结构体
  9. type Document struct {
  10.     Title string
  11.     Body []byte
  12. }
  14. //将文档保存为txt文件
  15. func (p *Document) save() error {
  16.     filename := p.Title + ".txt"
  17.     return ioutil.WriteFile(filename, p.Body, 0600)
  18. }
  20. //加载文档
  21. func loadPage(title string) (*Document, error) {
  22.     filename := title + ".txt"
  23.     body, err := ioutil.ReadFile(filename)
  24.     if err != nil {
  25.         return nil, err
  26.     }
  27.     return &Document{Title: title, Body: body}, nil
  28. }
  30. //输入文档的标题和正文
  31. func main() {
  32.     fmt.Print("输入标题：")
  33.     var Title string
  34.     fmt.Scanln(&Title)
  36.     fmt.Print("输入正文：")
  37.     var Body []byte
  38.     fmt.Scanln(&Body)
  40. //保存文档并在命令行上显示
  41.     p1 := &Document{Title: Title, Body: []byte(Body)}
  42.     p1.save()
  43.     p2, _ := loadPage(Title)
  44.     fmt.Println(string(p2.Body))
  45. }
英文:

I'm creating a simple word processing program in Go. From the command line, I have two prompts:

$Enter Title:

$Enter Body:

The program is supposed to save the document as a txt file and print it to the command line. The program works if the user user types in a one-word Title and a one-word Body. But if the user types in a several-word title, this will happen:

  1. $Enter Title: Here is a title
  3. $Enter Body: s
  5. $  title
  7. -bash: title: command not found

Here is the code I have so far:

  1. package main
  3. import (
  4.     "fmt"
  5.     "io/ioutil"
  6. )
  8. //Create struct for a document
  9. type Document struct {
  10. 	Title string
  11. 	Body []byte
  12. }
  14. //Save document as txt file
  15. func (p *Document) save() error {
  16. 	filename := p.Title + ".txt"
  17. 	return ioutil.WriteFile(filename, p.Body, 0600)
  18. }
  20. //Load document
  21. func loadPage(title string) (*Document, error) {
  22. 	filename := title + ".txt"
  23. 	body, err := ioutil.ReadFile(filename)
  24. 	if err != nil {
  25. 		return nil, err
  26. 	}
  27. 	return &Document{Title: title, Body: body}, nil
  28. }
  30. //Input document title and body. 
  31. func main() {
  32.     fmt.Print("Enter Title: ")
  33.     var Title string
  34.     fmt.Scanln(&Title)
  36.     fmt.Print("Enter Body: ")
  37.     var Body []byte
  38.     fmt.Scanln(&Body)
  40. //Save document and display on command line
  41. 	p1 := &Document{Title: Title, Body: []byte(Body)}
  42. 	p1.save()
  43. 	p2, _ := loadPage(Title)
  44. 	fmt.Println(string(p2.Body))
  45. }

答案1

得分: 0

使用bufio.ReadString代替fmt.Scanln怎么样？

不完全了解Scanln的工作原理，但我相当确定问题出在对该函数的误用上。
使用bufio的示例代码：

  1. package main
  3. import (
  4.         "bufio"
  5.         "fmt"
  6.         "io/ioutil"
  7.         "log"
  8.         "os"
  9.         "strings"
  10. )
  12. // Document代表文档的数据。
  13. type Document struct {
  14.         Title string
  15.         Body  []byte
  16. }
  18. // save将文档以txt文件的形式保存到磁盘上。
  19. func (p *Document) save() error {
  20.         filename := p.Title + ".txt"
  21.         return ioutil.WriteFile(filename, p.Body, 0600)
  22. }
  24. // loadPage从磁盘加载文档。
  25. func loadPage(title string) (*Document, error) {
  26.         filename := title + ".txt"
  27.         body, err := ioutil.ReadFile(filename)
  28.         if err != nil {
  29.                 return nil, err
  30.         }
  31.         return &Document{Title: title, Body: body}, nil
  32. }
  34. // 输入文档的标题和内容。
  35. func main() {
  36.         reader := bufio.NewReader(os.Stdin)
  37.         fmt.Print("输入标题：")
  38.         title, err := reader.ReadString('\n')
  39.         if err != nil {
  40.                 log.Fatal(err)
  41.         }
  42.         title = strings.TrimSpace(title)
  44.         fmt.Print("输入内容：")
  45.         body, err := reader.ReadString('\n')
  46.         if err != nil {
  47.                 log.Fatal(err)
  48.         }
  49.         body = strings.TrimSpace(body)
  51.         // 保存文档并在命令行上显示
  52.         p1 := &Document{Title: title, Body: []byte(body)}
  53.         if err := p1.save(); err != nil {
  54.                 log.Fatal(err)
  55.         }
  56.         p2, err := loadPage(title)
  57.         if err != nil {
  58.                 log.Fatal(err)
  59.         }
  60.         fmt.Println(string(p2.Body))
  61. }
英文:

What about using bufio.ReadString instead of fmt.Scanln?
Not 100% how Scanln works, but I am pretty sure the issue comes from a misuse of that function.
Example with bufio:

  1. package main
  3. import (
  4.         "bufio"
  5.         "fmt"
  6.         "io/ioutil"
  7.         "log"
  8.         "os"
  9.         "strings"
  10. )
  12. // Document represent the document's data.
  13. type Document struct {
  14.         Title string
  15.         Body  []byte
  16. }
  18. // Save dumps document as txt file on disc.
  19. func (p *Document) save() error {
  20.         filename := p.Title + ".txt"
  21.         return ioutil.WriteFile(filename, p.Body, 0600)
  22. }
  24. // loadPage loads a document from disc.
  25. func loadPage(title string) (*Document, error) {
  26.         filename := title + ".txt"
  27.         body, err := ioutil.ReadFile(filename)
  28.         if err != nil {
  29.                 return nil, err
  30.         }
  31.         return &Document{Title: title, Body: body}, nil
  32. }
  34. // Input document title and body.
  35. func main() {
  36.         reader := bufio.NewReader(os.Stdin)
  37.         fmt.Print("Enter Title: ")
  38.         title, err := reader.ReadString('\n')
  39.         if err != nil {
  40.                 log.Fatal(err)
  41.         }
  42.         title = strings.TrimSpace(title)
  44.         fmt.Print("Enter Body: ")
  45.         body, err := reader.ReadString('\n')
  46.         if err != nil {
  47.                 log.Fatal(err)
  48.         }
  49.         body = strings.TrimSpace(body)
  51.         //Save document and display on command line
  52.         p1 := &Document{Title: title, Body: []byte(body)}
  53.         if err := p1.save(); err != nil {
  54.                 log.Fatal(err)
  55.         }
  56.         p2, err := loadPage(title)
  57.         if err != nil {
  58.                 log.Fatal(err)
  59.         }
  60.         fmt.Println(string(p2.Body))
  61. }

huangapple
  • 本文由 发表于 2015年6月20日 04:24:17
  • 转载请务必保留本文链接：https://go.coder-hub.com/30946686.html
  • go
  • string
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定