storage.go 427 B

12345678910111213141516171819202122232425262728293031
  1. package storage
  2. /**
  3. TODO use a CLI flag to clear the database on init
  4. */
  5. import (
  6. "database/sql"
  7. _ "github.com/go-sql-driver/mysql"
  8. "log"
  9. )
  10. type Storage struct {
  11. Name string
  12. DB *sql.DB
  13. }
  14. var Service Storage = Storage{}
  15. func (s *Storage) Open(dbName string) error {
  16. var err error
  17. s.DB, err = sql.Open("mysql", dbName)
  18. if err != nil {
  19. log.Fatal(err)
  20. }
  21. return err
  22. }
  23. func (s *Storage) Close() {
  24. s.DB.Close()
  25. }