修改为sql数据库

This commit is contained in:
bjdgyc
2021-07-16 11:25:06 +08:00
parent 884f41d2f8
commit e7ef29c4ad
9 changed files with 36 additions and 27 deletions

View File

@@ -20,7 +20,7 @@ func GetXdb() *xorm.Engine {
func initDb() {
var err error
xdb, err = xorm.NewEngine(base.Cfg.DbType, base.Cfg.DbDsn)
xdb, err = xorm.NewEngine(base.Cfg.DbType, base.Cfg.DbSource)
// xdb.ShowSQL(true)
if err != nil {
base.Fatal(err)

View File

@@ -69,12 +69,12 @@ func Find(data interface{}, limit, page int) error {
}
func CountPrefix(fieldName string, prefix string, data interface{}) int {
n, _ := xdb.Where(fieldName + " like '" + prefix + "%' ").Count(data)
n, _ := xdb.Where(fieldName+" like ?", prefix+"%").Count(data)
return int(n)
}
func Prefix(fieldName string, prefix string, data interface{}, limit, page int) error {
where := xdb.Where(fieldName + " like '" + prefix + "%' ")
where := xdb.Where(fieldName+" like ?", prefix+"%")
if limit == 0 {
return where.Find(data)
}

View File

@@ -12,7 +12,7 @@ import (
func preIpData() {
tmpDb := path.Join(os.TempDir(), "anylink_test.db")
base.Cfg.DbType = "sqlite3"
base.Cfg.DbDsn = tmpDb
base.Cfg.DbSource = tmpDb
initDb()
}