forked from free-lancers/electricity_bill_calc_service
feat(topup):基本完成商户充值记录的数据库操作部分。
This commit is contained in:
徐涛
22
model/top_up.go
Normal file
22
model/top_up.go
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"electricity_bill_calc/types"
|
||||||
|
|
||||||
|
"github.com/shopspring/decimal"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TopUp struct {
|
||||||
|
TopUpCode string `json:"topUpCode" db:"top_up_code"`
|
||||||
|
Park string `json:"parkId" db:"park_id"`
|
||||||
|
Tenement string `json:"tenementId" db:"tenement_id"`
|
||||||
|
TenementName string `json:"tenementName" db:"tenement_name"`
|
||||||
|
Meter string `json:"meterId" db:"meter_id"`
|
||||||
|
MeterAddress *string `json:"meterAddress" db:"meter_address"`
|
||||||
|
ToppedUpAt types.DateTime `json:"toppedUpAt" db:"topped_up_at"`
|
||||||
|
Amount decimal.Decimal `json:"amount" db:"amount"`
|
||||||
|
PaymentType int16 `json:"paymentType" db:"payment_type"`
|
||||||
|
SuccessfulSynchronized bool `json:"successfulSynchronized" db:"successful_synchronized"`
|
||||||
|
SynchronizedAt *types.DateTime `json:"synchronizedAt" db:"synchronized_at"`
|
||||||
|
CancelledAt *types.DateTime `json:"cancelledAt" db:"cancelled_at"`
|
||||||
|
}
|
||||||
186
repository/top_up.go
Normal file
186
repository/top_up.go
Normal file
@@ -0,0 +1,186 @@
|
|||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"electricity_bill_calc/cache"
|
||||||
|
"electricity_bill_calc/config"
|
||||||
|
"electricity_bill_calc/global"
|
||||||
|
"electricity_bill_calc/logger"
|
||||||
|
"electricity_bill_calc/model"
|
||||||
|
"electricity_bill_calc/tools/serial"
|
||||||
|
"electricity_bill_calc/types"
|
||||||
|
"electricity_bill_calc/vo"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/doug-martin/goqu/v9"
|
||||||
|
_ "github.com/doug-martin/goqu/v9/dialect/postgres"
|
||||||
|
"github.com/georgysavva/scany/v2/pgxscan"
|
||||||
|
"go.uber.org/zap"
|
||||||
|
)
|
||||||
|
|
||||||
|
type _TopUpRepository struct {
|
||||||
|
log *zap.Logger
|
||||||
|
ds goqu.DialectWrapper
|
||||||
|
}
|
||||||
|
|
||||||
|
var TopUpRepository = _TopUpRepository{
|
||||||
|
log: logger.Named("Repository", "TopUp"),
|
||||||
|
ds: goqu.Dialect("postgres"),
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检索符合条件的商户充值记录
|
||||||
|
func (tur _TopUpRepository) ListTopUps(pid string, startDate, endDate *types.Date, keyword *string, page uint) ([]*model.TopUp, int64, error) {
|
||||||
|
tur.log.Info("查询符合条件的商户充值记录", zap.String("Park", pid), logger.DateFieldp("StartDate", startDate), logger.DateFieldp("EndDate", endDate), zap.Stringp("keyword", keyword), zap.Uint("page", page))
|
||||||
|
cacheConditions := []string{
|
||||||
|
pid,
|
||||||
|
cache.NullableStringKey(keyword),
|
||||||
|
cache.NullableConditionKey(startDate),
|
||||||
|
cache.NullableConditionKey(endDate),
|
||||||
|
fmt.Sprintf("%d", page),
|
||||||
|
}
|
||||||
|
if topUps, total, err := cache.RetrievePagedSearch[[]*model.TopUp]("top_ups", cacheConditions...); err == nil && topUps != nil && len(*topUps) > 0 {
|
||||||
|
tur.log.Info("从缓存中获取到商户充值记录", zap.Int("Count", len(*topUps)), zap.Int64("Total", total))
|
||||||
|
return *topUps, total, nil
|
||||||
|
}
|
||||||
|
ctx, cancel := global.TimeoutContext()
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
topUpQuery := tur.ds.
|
||||||
|
From(goqu.T("tenement_top_ups").As("t")).
|
||||||
|
LeftJoin(goqu.T("tenement").As("te"), goqu.On(goqu.I("te.id").Eq(goqu.I("t.tenement_id")), goqu.I("te.park_id").Eq(goqu.I("t.park_id")))).
|
||||||
|
LeftJoin(goqu.T("meter").As("m"), goqu.On(goqu.I("m.code").Eq(goqu.I("t.meter_id")), goqu.I("m.park_id").Eq(goqu.I("t.park_id")))).
|
||||||
|
Select("t.*", goqu.I("te.full_name").As("tenement_name"), goqu.I("m.address").As("meter_address")).
|
||||||
|
Where(goqu.I("t.park_id").Eq(pid))
|
||||||
|
countQuery := tur.ds.
|
||||||
|
From(goqu.T("tenement_top_ups").As("t")).
|
||||||
|
LeftJoin(goqu.T("tenement").As("te"), goqu.On(goqu.I("te.id").Eq(goqu.I("t.tenement_id")), goqu.I("te.park_id").Eq(goqu.I("t.park_id")))).
|
||||||
|
LeftJoin(goqu.T("meter").As("m"), goqu.On(goqu.I("m.code").Eq(goqu.I("t.meter_id")), goqu.I("m.park_id").Eq(goqu.I("t.park_id")))).
|
||||||
|
Select(goqu.COUNT("t.*")).
|
||||||
|
Where(goqu.I("t.park_id").Eq(pid))
|
||||||
|
|
||||||
|
if keyword != nil && len(*keyword) > 0 {
|
||||||
|
pattern := fmt.Sprintf("%%%s%%", *keyword)
|
||||||
|
topUpQuery = topUpQuery.Where(goqu.Or(
|
||||||
|
goqu.I("te.full_name").ILike(pattern),
|
||||||
|
goqu.I("te.short_name").ILike(pattern),
|
||||||
|
goqu.I("te.abbr").ILike(pattern),
|
||||||
|
goqu.I("m.code").ILike(pattern),
|
||||||
|
goqu.I("m.address").ILike(pattern),
|
||||||
|
))
|
||||||
|
countQuery = countQuery.Where(goqu.Or(
|
||||||
|
goqu.I("te.full_name").ILike(pattern),
|
||||||
|
goqu.I("te.short_name").ILike(pattern),
|
||||||
|
goqu.I("te.abbr").ILike(pattern),
|
||||||
|
goqu.I("m.code").ILike(pattern),
|
||||||
|
goqu.I("m.address").ILike(pattern),
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
if startDate != nil {
|
||||||
|
topUpQuery = topUpQuery.Where(goqu.I("t.topped_up_at").Gte(startDate.ToBeginningOfDate()))
|
||||||
|
countQuery = countQuery.Where(goqu.I("t.topped_up_at").Gte(startDate.ToBeginningOfDate()))
|
||||||
|
}
|
||||||
|
|
||||||
|
if endDate != nil {
|
||||||
|
topUpQuery = topUpQuery.Where(goqu.I("t.topped_up_at").Lte(endDate.ToEndingOfDate()))
|
||||||
|
countQuery = countQuery.Where(goqu.I("t.topped_up_at").Lte(endDate.ToEndingOfDate()))
|
||||||
|
}
|
||||||
|
|
||||||
|
startRow := (page - 1) * config.ServiceSettings.ItemsPageSize
|
||||||
|
topUpQuery = topUpQuery.Order(goqu.I("t.topped_up_at").Desc()).Offset(startRow).Limit(config.ServiceSettings.ItemsPageSize)
|
||||||
|
|
||||||
|
topUpSql, topUpArgs, _ := topUpQuery.Prepared(true).ToSQL()
|
||||||
|
countSql, countArgs, _ := countQuery.Prepared(true).ToSQL()
|
||||||
|
|
||||||
|
var (
|
||||||
|
topUps []*model.TopUp = make([]*model.TopUp, 0)
|
||||||
|
total int64 = 0
|
||||||
|
)
|
||||||
|
if err := pgxscan.Select(ctx, global.DB, &topUps, topUpSql, topUpArgs...); err != nil {
|
||||||
|
tur.log.Error("查询商户充值记录失败", zap.Error(err))
|
||||||
|
return topUps, 0, err
|
||||||
|
}
|
||||||
|
if err := pgxscan.Get(ctx, global.DB, &total, countSql, countArgs...); err != nil {
|
||||||
|
tur.log.Error("查询商户充值记录总数失败", zap.Error(err))
|
||||||
|
return topUps, 0, err
|
||||||
|
}
|
||||||
|
cache.CachePagedSearch(topUps, total, []string{fmt.Sprintf("top_up:%s", pid), fmt.Sprintf("tenement:%s", pid), fmt.Sprintf("meter:%s", pid)}, "top_ups", cacheConditions...)
|
||||||
|
return topUps, total, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 取得一个充值记录的详细信息
|
||||||
|
func (tur _TopUpRepository) GetTopUp(pid, topUpCode string) (*model.TopUp, error) {
|
||||||
|
tur.log.Info("查询充值记录", zap.String("Park", pid), zap.String("TopUpCode", topUpCode))
|
||||||
|
if topUp, err := cache.RetrieveEntity[model.TopUp]("top_up", topUpCode); err == nil && topUp != nil {
|
||||||
|
tur.log.Info("从缓存中获取到充值记录")
|
||||||
|
return topUp, err
|
||||||
|
}
|
||||||
|
ctx, cancel := global.TimeoutContext()
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
topUpSql, topUpArgs, _ := tur.ds.
|
||||||
|
From(goqu.T("tenement_top_ups").As("t")).
|
||||||
|
LeftJoin(goqu.T("tenement").As("te"), goqu.On(goqu.I("te.id").Eq(goqu.I("t.tenement_id")), goqu.I("te.park_id").Eq(goqu.I("t.park_id")))).
|
||||||
|
LeftJoin(goqu.T("meter").As("m"), goqu.On(goqu.I("m.code").Eq(goqu.I("t.meter_id")), goqu.I("m.park_id").Eq(goqu.I("t.park_id")))).
|
||||||
|
Select("t.*", goqu.I("te.full_name").As("tenement_name"), goqu.I("m.address").As("meter_address")).
|
||||||
|
Where(goqu.I("t.park_id").Eq(pid), goqu.I("t.top_up_code").Eq(topUpCode)).
|
||||||
|
Prepared(true).ToSQL()
|
||||||
|
|
||||||
|
var topUp model.TopUp
|
||||||
|
if err := pgxscan.Get(ctx, global.DB, &topUp, topUpSql, topUpArgs...); err != nil {
|
||||||
|
tur.log.Error("查询充值记录失败", zap.Error(err))
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
cache.CacheEntity(&topUp, []string{fmt.Sprintf("top_up:%s", pid), fmt.Sprintf("tenement:%s", pid), fmt.Sprintf("meter:%s", pid)}, "top_up", topUpCode)
|
||||||
|
return &topUp, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建一条新的充值记录
|
||||||
|
func (tur _TopUpRepository) CreateTopUp(pid string, form *vo.TopUpCreationForm) error {
|
||||||
|
tur.log.Info("创建一条新的充值记录", zap.String("Park", pid), zap.String("Tenement", form.Tenement), zap.String("Meter", form.Meter))
|
||||||
|
ctx, cancel := global.TimeoutContext()
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
serial.StringSerialRequestChan <- 1
|
||||||
|
topUpCode := serial.Prefix("U", <-serial.StringSerialResponseChan)
|
||||||
|
topUpTime := types.Now()
|
||||||
|
topUpSql, topUpArgs, _ := tur.ds.
|
||||||
|
Insert("tenement_top_ups").
|
||||||
|
Cols("top_up_code", "park_id", "tenement_id", "meter_id", "topped_up_at", "amount", "payment_type").
|
||||||
|
Vals(goqu.Vals{
|
||||||
|
topUpCode,
|
||||||
|
pid,
|
||||||
|
form.Tenement,
|
||||||
|
form.Meter,
|
||||||
|
topUpTime,
|
||||||
|
form.Amount,
|
||||||
|
model.PAYMENT_CASH,
|
||||||
|
}).
|
||||||
|
Prepared(true).ToSQL()
|
||||||
|
|
||||||
|
if _, err := global.DB.Exec(ctx, topUpSql, topUpArgs...); err != nil {
|
||||||
|
tur.log.Error("创建充值记录失败", zap.Error(err))
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
cache.AbolishRelation(fmt.Sprintf("top_ups:%s", pid))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 删除一条充值记录
|
||||||
|
func (tur _TopUpRepository) DeleteTopUp(pid, topUpCode string) error {
|
||||||
|
tur.log.Info("删除一条充值记录", zap.String("Park", pid), zap.String("TopUpCode", topUpCode))
|
||||||
|
ctx, cancel := global.TimeoutContext()
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
topUpSql, topUpArgs, _ := tur.ds.
|
||||||
|
Update("tenement_top_ups").
|
||||||
|
Set(goqu.Record{"cancelled_at": types.Now()}).
|
||||||
|
Where(goqu.I("park_id").Eq(pid), goqu.I("top_up_code").Eq(topUpCode)).
|
||||||
|
Prepared(true).ToSQL()
|
||||||
|
|
||||||
|
if _, err := global.DB.Exec(ctx, topUpSql, topUpArgs...); err != nil {
|
||||||
|
tur.log.Error("删除充值记录失败", zap.Error(err))
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
cache.AbolishRelation(fmt.Sprintf("top_ups:%s", pid))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
9
vo/top_up.go
Normal file
9
vo/top_up.go
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
package vo
|
||||||
|
|
||||||
|
import "github.com/shopspring/decimal"
|
||||||
|
|
||||||
|
type TopUpCreationForm struct {
|
||||||
|
Tenement string `json:"tenement"`
|
||||||
|
Meter string `json:"meter"`
|
||||||
|
Amount decimal.Decimal `json:"amount"`
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user