69 lines
2.4 KiB
Go
69 lines
2.4 KiB
Go
package income
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
model "go-common/app/admin/main/growup/model/income"
|
|
|
|
"go-common/library/database/sql"
|
|
"go-common/library/log"
|
|
)
|
|
|
|
const (
|
|
// count
|
|
_upAccountCountSQL = "SELECT count(*) FROM up_account WHERE %s is_deleted = ?"
|
|
// select
|
|
_upAccountSQL = "SELECT mid,total_income,total_unwithdraw_income,total_withdraw_income,withdraw_date_version,last_withdraw_time,mtime FROM up_account WHERE %s is_deleted = ? LIMIT ?,?"
|
|
_upAccountByMIDSQL = "SELECT mid,total_income,total_unwithdraw_income,withdraw_date_version,version FROM up_account WHERE mid = ? AND is_deleted = 0"
|
|
// update
|
|
_breachUpAccountSQL = "UPDATE up_account SET total_income = ?, total_unwithdraw_income = ?, version = ? WHERE mid = ? AND version = ? AND is_deleted = 0"
|
|
)
|
|
|
|
// UpAccountCount get up_account count
|
|
func (d *Dao) UpAccountCount(c context.Context, query string, isDeleted int) (total int64, err error) {
|
|
err = d.db.QueryRow(c, fmt.Sprintf(_upAccountCountSQL, query), isDeleted).Scan(&total)
|
|
if err == sql.ErrNoRows {
|
|
err = nil
|
|
}
|
|
return
|
|
}
|
|
|
|
// ListUpAccount list up account bu query
|
|
func (d *Dao) ListUpAccount(c context.Context, query string, isDeleted, from, limit int) (ups []*model.UpAccount, err error) {
|
|
ups = make([]*model.UpAccount, 0)
|
|
rows, err := d.db.Query(c, fmt.Sprintf(_upAccountSQL, query), isDeleted, from, limit)
|
|
if err != nil {
|
|
log.Error("ListUpAccount d.db.Query error(%v)", err)
|
|
return
|
|
}
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
up := &model.UpAccount{}
|
|
err = rows.Scan(&up.MID, &up.TotalIncome, &up.TotalUnwithdrawIncome, &up.TotalWithdrawIncome, &up.WithdrawDateVersion, &up.LastWithdrawTime, &up.MTime)
|
|
if err != nil {
|
|
log.Error("ListUpAccount rows scan error(%v)", err)
|
|
return
|
|
}
|
|
ups = append(ups, up)
|
|
}
|
|
err = rows.Err()
|
|
return
|
|
}
|
|
|
|
// GetUpAccount get up_account by mid
|
|
func (d *Dao) GetUpAccount(c context.Context, mid int64) (up *model.UpAccount, err error) {
|
|
up = &model.UpAccount{}
|
|
err = d.db.QueryRow(c, _upAccountByMIDSQL, mid).Scan(&up.MID, &up.TotalIncome, &up.TotalUnwithdrawIncome, &up.WithdrawDateVersion, &up.Version)
|
|
return
|
|
}
|
|
|
|
// TxBreachUpAccount breach up_account
|
|
func (d *Dao) TxBreachUpAccount(tx *sql.Tx, total, unwithdraw, mid, newVersion, oldVersion int64) (rows int64, err error) {
|
|
res, err := tx.Exec(_breachUpAccountSQL, total, unwithdraw, newVersion, mid, oldVersion)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return res.RowsAffected()
|
|
}
|