53 lines
958 B
Go
53 lines
958 B
Go
package server
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/boltdb/bolt"
|
|
"github.com/gokins/core"
|
|
"github.com/gokins/gokins/comm"
|
|
"github.com/gokins/gokins/migrates"
|
|
"github.com/sirupsen/logrus"
|
|
"xorm.io/xorm"
|
|
)
|
|
|
|
func initDb() error {
|
|
var err error
|
|
dvs := "mysql"
|
|
ul := comm.Cfg.Datasource.Url
|
|
if comm.Cfg.Datasource.Driver != "" {
|
|
dvs = comm.Cfg.Datasource.Driver
|
|
}
|
|
comm.IsMySQL = dvs == "mysql"
|
|
if !comm.Installed {
|
|
if comm.IsMySQL {
|
|
err = migrates.UpMysqlMigrate(ul)
|
|
} else {
|
|
err = migrates.UpSqliteMigrate(ul)
|
|
}
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
db, err := xorm.NewEngine(dvs, comm.Cfg.Datasource.Url)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
db.ShowSQL(core.Debug)
|
|
comm.Db = db
|
|
return nil
|
|
}
|
|
|
|
func initCache() error {
|
|
pth := filepath.Join(comm.WorkPath, "cache.dat")
|
|
os.Remove(pth)
|
|
db, err := bolt.Open(pth, 0640, nil)
|
|
if err != nil {
|
|
logrus.Errorf("InitCache err:%v", err)
|
|
return err
|
|
}
|
|
comm.BCache = db
|
|
return nil
|
|
}
|