mirror of https://github.com/rocboss/paopao-ce
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.0 KiB
53 lines
1.0 KiB
2 years ago
|
// Copyright 2023 ROC. All rights reserved.
|
||
|
// Use of this source code is governed by a MIT style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
// Core service implement base sqlx+mysql. All sub-service
|
||
|
// will declare here and provide initial function.
|
||
|
|
||
|
package sakila
|
||
|
|
||
|
import (
|
||
|
"strings"
|
||
|
"sync"
|
||
|
|
||
|
"github.com/jmoiron/sqlx"
|
||
|
"github.com/rocboss/paopao-ce/internal/conf"
|
||
|
"github.com/sirupsen/logrus"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
_db *sqlx.DB
|
||
|
_once sync.Once
|
||
|
)
|
||
|
|
||
|
func sqlxDB() *sqlx.DB {
|
||
|
_once.Do(func() {
|
||
|
_db = conf.MustSqlxDB()
|
||
|
})
|
||
|
return _db
|
||
|
}
|
||
|
|
||
|
func c(query string) *sqlx.Stmt {
|
||
|
db := sqlxDB()
|
||
|
stmt, err := db.Preparex(db.Rebind(t(query)))
|
||
|
if err != nil {
|
||
|
logrus.Fatalf("prepare query(%s) error: %s", query, err)
|
||
|
}
|
||
|
return stmt
|
||
|
}
|
||
|
|
||
|
func n(query string) *sqlx.NamedStmt {
|
||
|
db := sqlxDB()
|
||
|
stmt, err := db.PrepareNamed(t(query))
|
||
|
if err != nil {
|
||
|
logrus.Fatalf("prepare named query(%s) error: %s", query, err)
|
||
|
}
|
||
|
return stmt
|
||
|
}
|
||
|
|
||
|
// t repace table prefix for query
|
||
|
func t(query string) string {
|
||
|
return strings.Replace(query, "@", conf.DatabaseSetting.TablePrefix, -1)
|
||
|
}
|