()
| 174 | } |
| 175 | |
| 176 | func InitDB() (err error) { |
| 177 | db, err := chooseDB("SQL_DSN", false) |
| 178 | if err == nil { |
| 179 | if common.DebugEnabled { |
| 180 | db = db.Debug() |
| 181 | } |
| 182 | DB = db |
| 183 | sqlDB, err := DB.DB() |
| 184 | if err != nil { |
| 185 | return err |
| 186 | } |
| 187 | sqlDB.SetMaxIdleConns(common.GetEnvOrDefault("SQL_MAX_IDLE_CONNS", 100)) |
| 188 | sqlDB.SetMaxOpenConns(common.GetEnvOrDefault("SQL_MAX_OPEN_CONNS", 1000)) |
| 189 | sqlDB.SetConnMaxLifetime(time.Second * time.Duration(common.GetEnvOrDefault("SQL_MAX_LIFETIME", 60))) |
| 190 | |
| 191 | if !common.IsMasterNode { |
| 192 | return nil |
| 193 | } |
| 194 | if common.UsingMySQL { |
| 195 | //_, _ = sqlDB.Exec("ALTER TABLE channels MODIFY model_mapping TEXT;") // TODO: delete this line when most users have upgraded |
| 196 | } |
| 197 | common.SysLog("database migration started") |
| 198 | err = migrateDB() |
| 199 | return err |
| 200 | } else { |
| 201 | common.FatalLog(err) |
| 202 | } |
| 203 | return err |
| 204 | } |
| 205 | |
| 206 | func InitLogDB() (err error) { |
| 207 | if os.Getenv("LOG_SQL_DSN") == "" { |
no test coverage detected