/internal/database/database.go

https://github.com/cnbattle/douyin · Go · 26 lines · 23 code · 3 blank · 0 comment · 2 complexity · 0ab4cec3f3f267be4e08cdb8471c3b02 MD5 · raw file

  1. package database
  2. import (
  3. "github.com/cnbattle/douyin/internal/database/model"
  4. "github.com/jinzhu/gorm"
  5. _ "github.com/jinzhu/gorm/dialects/sqlite"
  6. "log"
  7. )
  8. var (
  9. Local *gorm.DB
  10. localDialect = "sqlite3"
  11. localArgs = "./database.db"
  12. )
  13. func init() {
  14. var err error
  15. Local, err = gorm.Open(localDialect, localArgs)
  16. if err != nil {
  17. log.Panic(err)
  18. }
  19. Local.LogMode(false)
  20. Local.DB().SetMaxOpenConns(10)
  21. Local.DB().SetMaxIdleConns(20)
  22. Local.AutoMigrate(&model.Video{})
  23. }