56 lines
1.9 KiB
Go
56 lines
1.9 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
xsql "database/sql"
|
|
|
|
"go-common/app/service/main/vip/model"
|
|
"go-common/library/database/sql"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const (
|
|
_activityOrderSQL = "SELECT id,mid,order_no,product_id,months,panel_type,associate_state,ctime,mtime FROM vip_order_activity_record WHERE order_no = ?;"
|
|
_updateActivityStateSQL = "UPDATE vip_order_activity_record SET associate_state =? WHERE order_no = ?;"
|
|
_productBuyCountSQL = "SELECT current_count as count FROM vip_product_pay_record WHERE mid = ? AND months=? AND panel_type = ?;"
|
|
)
|
|
|
|
//ActivityOrder get activity order by order_no.
|
|
func (d *Dao) ActivityOrder(c context.Context, orderNO string) (res *model.VipOrderActivityRecord, err error) {
|
|
res = new(model.VipOrderActivityRecord)
|
|
if err = d.olddb.QueryRow(c, _activityOrderSQL, orderNO).
|
|
Scan(&res.ID, &res.Mid, &res.OrderNO, &res.ProductID, &res.Months, &res.PanelType, &res.AssociateState, &res.Ctime, &res.Mtime); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
res = nil
|
|
err = nil
|
|
return
|
|
}
|
|
err = errors.Wrapf(err, "dao activity order(%s)", orderNO)
|
|
}
|
|
return
|
|
}
|
|
|
|
// UpdateActivityState update act vip grant state.
|
|
func (d *Dao) UpdateActivityState(c context.Context, state int8, orderNO string) (aff int64, err error) {
|
|
var res xsql.Result
|
|
if res, err = d.olddb.Exec(c, _updateActivityStateSQL, state, orderNO); err != nil {
|
|
err = errors.Wrapf(err, "dao update associate state(%d,%s)", state, orderNO)
|
|
return
|
|
}
|
|
return res.RowsAffected()
|
|
}
|
|
|
|
// CountProductBuy get user by product count.
|
|
func (d *Dao) CountProductBuy(c context.Context, mid int64, months int32, panelType string) (count int64, err error) {
|
|
row := d.olddb.QueryRow(c, _productBuyCountSQL, mid, months, panelType)
|
|
if err = row.Scan(&count); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
err = nil
|
|
return
|
|
}
|
|
err = errors.Wrapf(err, "dao update associate state(%d,%d,%s)", mid, months, panelType)
|
|
}
|
|
return
|
|
}
|