alphago9
682aa3907a
- refine config loading with --config and current-directory .env - improve sqlite busy handling and runtime DB tuning - restore telegram payment notification and wallet input flow - add behavior-based config and callback recovery tests
52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package dao
|
|
|
|
import (
|
|
"github.com/assimon/luuu/config"
|
|
"github.com/assimon/luuu/model/mdb"
|
|
"github.com/assimon/luuu/util/log"
|
|
"github.com/gookit/color"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
var RuntimeDB *gorm.DB
|
|
|
|
func RuntimeInit() error {
|
|
var err error
|
|
runtimePath := config.GetRuntimeSqlitePath()
|
|
color.Green.Printf("[runtime_db] sqlite filename: %s\n", runtimePath)
|
|
RuntimeDB, err = openDB(runtimePath, &gorm.Config{
|
|
NamingStrategy: schema.NamingStrategy{
|
|
SingularTable: true,
|
|
},
|
|
Logger: logger.Default.LogMode(logger.Error),
|
|
})
|
|
if err != nil {
|
|
color.Red.Printf("[runtime_db] sqlite open DB,err=%s\n", err)
|
|
return err
|
|
}
|
|
|
|
concurrency := config.GetQueueConcurrency()
|
|
if concurrency < 2 {
|
|
concurrency = 2
|
|
}
|
|
if concurrency > 16 {
|
|
concurrency = 16
|
|
}
|
|
if _, err = configureSQLite(RuntimeDB, concurrency); err != nil {
|
|
color.Red.Printf("[runtime_db] sqlite connDB err:%s", err.Error())
|
|
return err
|
|
}
|
|
if err = RuntimeDB.Exec("DROP INDEX IF EXISTS transaction_lock_token_amount_uindex").Error; err != nil {
|
|
return err
|
|
}
|
|
if err = RuntimeDB.AutoMigrate(&mdb.TransactionLock{}); err != nil {
|
|
color.Red.Printf("[runtime_db] sqlite migrate DB(TransactionLock),err=%s\n", err)
|
|
return err
|
|
}
|
|
|
|
log.Sugar.Debug("[runtime_db] sqlite connDB success")
|
|
return nil
|
|
}
|