mirror of
https://github.com/genxium/DelayNoMore
synced 2024-12-27 12:08:21 +00:00
93 lines
2.5 KiB
Go
93 lines
2.5 KiB
Go
package models
|
|
|
|
import (
|
|
"battle_srv/storage"
|
|
"database/sql"
|
|
. "dnmshared"
|
|
|
|
sq "github.com/Masterminds/squirrel"
|
|
"github.com/jmoiron/sqlx"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func rowValues(rows *sqlx.Rows, cols []string) []interface{} {
|
|
results := make([]interface{}, len(cols))
|
|
for i := range results {
|
|
results[i] = new(interface{})
|
|
}
|
|
if err := rows.Scan(results[:]...); err != nil {
|
|
panic(err)
|
|
}
|
|
return results
|
|
}
|
|
|
|
func exist(t string, cond sq.Eq) (bool, error) {
|
|
c, err := getCount(t, cond)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return c >= 1, nil
|
|
}
|
|
|
|
func getCount(t string, cond sq.Eq) (int, error) {
|
|
query, args, err := sq.Select("count(1)").From(t).Where(cond).ToSql()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
//Logger.Debug("getCount", zap.String("sql", query), zap.Any("args", args))
|
|
var c int
|
|
err = storage.MySQLManagerIns.Get(&c, query, args...)
|
|
return c, err
|
|
}
|
|
|
|
func insert(t string, cols []string, vs []interface{}) (sql.Result, error) {
|
|
query, args, err := sq.Insert(t).Columns(cols...).Values(vs...).ToSql()
|
|
Logger.Debug("txInsert", zap.String("sql", query))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result, err := storage.MySQLManagerIns.Exec(query, args...)
|
|
return result, err
|
|
}
|
|
|
|
func txInsert(tx *sqlx.Tx, t string, cols []string, vs []interface{}) (sql.Result, error) {
|
|
query, args, err := sq.Insert(t).Columns(cols...).Values(vs...).ToSql()
|
|
//Logger.Debug("txInsert", zap.String("sql", query))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result, err := tx.Exec(query, args...)
|
|
return result, err
|
|
}
|
|
|
|
func getFields(t string, fields []string, cond sq.Eq, dest interface{}) error {
|
|
query, args, err := sq.Select(fields...).From(t).Where(cond).Limit(1).ToSql()
|
|
Logger.Debug("getFields", zap.String("sql", query), zap.Any("args", args))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = storage.MySQLManagerIns.Get(dest, query, args...)
|
|
return err
|
|
}
|
|
|
|
func getObj(t string, cond sq.Eq, dest interface{}) error {
|
|
query, args, err := sq.Select("*").From(t).Where(cond).Limit(1).ToSql()
|
|
Logger.Debug("getObj", zap.String("sql", query), zap.Any("args", args))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = storage.MySQLManagerIns.Get(dest, query, args...)
|
|
return err
|
|
}
|
|
|
|
func getList(t string, cond sq.Eq, dest interface{}) error {
|
|
query, args, err := sq.Select("*").From(t).Where(cond).ToSql()
|
|
Logger.Debug("getList", zap.String("sql", query), zap.Any("args", args))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = storage.MySQLManagerIns.Select(dest, query, args...)
|
|
//Logger.Debug("getList", zap.Error(err))
|
|
return err
|
|
}
|