/internal/database/database.go
https://github.com/cnbattle/douyin · Go · 26 lines · 23 code · 3 blank · 0 comment · 2 complexity · 0ab4cec3f3f267be4e08cdb8471c3b02 MD5 · raw file
- package database
- import (
- "github.com/cnbattle/douyin/internal/database/model"
- "github.com/jinzhu/gorm"
- _ "github.com/jinzhu/gorm/dialects/sqlite"
- "log"
- )
- var (
- Local *gorm.DB
- localDialect = "sqlite3"
- localArgs = "./database.db"
- )
- func init() {
- var err error
- Local, err = gorm.Open(localDialect, localArgs)
- if err != nil {
- log.Panic(err)
- }
- Local.LogMode(false)
- Local.DB().SetMaxOpenConns(10)
- Local.DB().SetMaxIdleConns(20)
- Local.AutoMigrate(&model.Video{})
- }