Sqlite.go 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. package db
  2. import (
  3. "file-manage-ui/data"
  4. _ "github.com/glebarez/sqlite"
  5. "log"
  6. "xorm.io/xorm"
  7. )
  8. var DB *xorm.Engine
  9. func init() {
  10. Open()
  11. }
  12. func Open() {
  13. db, err := xorm.NewEngine("sqlite", "./file/email.db")
  14. if err != nil {
  15. log.Fatal(err)
  16. }
  17. //创建表
  18. err = db.Sync2(new(data.KVStore)) //new(entity.Email), new(entity.User), new(entity.FromEmail), new(entity.Code)
  19. if err != nil {
  20. log.Fatalf("同步表结构时出错: %s", err)
  21. }
  22. DB = db
  23. DB.ShowSQL(true)
  24. DB.Logger().ShowSQL(true)
  25. }
  26. func Close() {
  27. defer DB.Close()
  28. }
  29. //func InsertEmail(email entity.Email) {
  30. // _, err := DB.Insert(&email)
  31. // if err != nil {
  32. // log.Fatalf("插入数据时出错: %s", err)
  33. // }
  34. // fmt.Println("数据插入成功")
  35. //}
  36. //
  37. //func GetEmails() {
  38. // var emails []entity.Email
  39. // err := DB.Find(&emails)
  40. // if err != nil {
  41. // log.Fatal(err)
  42. // }
  43. //
  44. // for _, email := range emails {
  45. // fmt.Printf("%+v\n", email)
  46. // }
  47. //}
  48. //func GetEmailById() {
  49. // var emails []entity.Email
  50. // err := DB.ID(&emails)
  51. // if err != nil {
  52. // log.Fatal(err)
  53. // }
  54. //
  55. // for _, email := range emails {
  56. // fmt.Printf("%+v\n", email)
  57. // }
  58. //}
  59. //
  60. //func UpdateEmail(id int64, newEmail string) {
  61. // email := new(entity.Email)
  62. // has, err := DB.ID(id).Get(email)
  63. // if err != nil {
  64. // log.Fatal(err)
  65. // }
  66. // if !has {
  67. // fmt.Println("数据未找到")
  68. // return
  69. // }
  70. //
  71. // email.Email = newEmail
  72. // _, err = DB.ID(id).Update(email)
  73. // if err != nil {
  74. // log.Fatalf("更新数据时出错: %s", err)
  75. // }
  76. // fmt.Println("数据更新成功")
  77. //}
  78. //
  79. //func DeleteEmail(id int64) {
  80. // _, err := DB.ID(id).Delete(&entity.Email{})
  81. // if err != nil {
  82. // log.Fatalf("删除数据时出错: %s", err)
  83. // }
  84. // fmt.Println("数据删除成功")
  85. //}