71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"electricity_bill_calc/config"
|
|
"electricity_bill_calc/global"
|
|
"electricity_bill_calc/model"
|
|
"electricity_bill_calc/utils"
|
|
"time"
|
|
|
|
"xorm.io/builder"
|
|
)
|
|
|
|
type _ChargeRepository struct{}
|
|
|
|
var ChargeRepo _ChargeRepository
|
|
|
|
func (_ChargeRepository) ListPagedChargeRecord(keyword, beginDate, endDate string, page int) ([]model.ChargeWithName, int64, error) {
|
|
var cond = builder.NewCond()
|
|
if len(keyword) != 0 {
|
|
cond.And(builder.Like{"u.name", keyword})
|
|
}
|
|
if len(beginDate) != 0 {
|
|
beginTime, err := time.Parse("2006-01-02", beginDate)
|
|
if err != nil {
|
|
return make([]model.ChargeWithName, 0), 0, err
|
|
}
|
|
cond.And(builder.Gte{"c.created_at": beginTime})
|
|
}
|
|
if len(endDate) != 0 {
|
|
endTime, err := time.Parse("2006-01-02", endDate)
|
|
if err != nil {
|
|
return make([]model.ChargeWithName, 0), 0, err
|
|
}
|
|
cond.And(builder.Lte{"c.created_at": endTime})
|
|
}
|
|
startItem := (page - 1) * config.ServiceSettings.ItemsPageSize
|
|
total, err := global.DBConn.
|
|
Table("user_charge").Alias("c").
|
|
Join("INNER", []string{"user_detail", "d"}, "c.user_id=d.id").
|
|
Where(cond).
|
|
Count(&model.ChargeWithName{})
|
|
if err != nil {
|
|
return nil, -1, err
|
|
}
|
|
charges := make([]model.ChargeWithName, 0)
|
|
err = global.DBConn.
|
|
Table("user_charge").Alias("c").
|
|
Join("INNER", []string{"user_detail", "d"}, "c.user_id=d.id").
|
|
Where(cond).
|
|
Limit(config.ServiceSettings.ItemsPageSize, startItem).
|
|
Find(&charges)
|
|
return charges, total, err
|
|
}
|
|
|
|
func (_ChargeRepository) LastValidChargeTo(uid string) (time.Time, error) {
|
|
veryBlankTime, _ := time.Parse("2006-01-02 15:04:05", "0001-01-01 00:00:00")
|
|
var records = make([]time.Time, 0)
|
|
err := global.DBConn.Where(builder.Eq{"settled": true, "cancelled": false, "refunded": false}).Cols("charged_to").Find(&records)
|
|
if err != nil {
|
|
return veryBlankTime, nil
|
|
}
|
|
lastValid := utils.Reduce(records, veryBlankTime, func(acc, elem time.Time) time.Time {
|
|
if elem.After(acc) {
|
|
return elem
|
|
} else {
|
|
return acc
|
|
}
|
|
})
|
|
return lastValid, nil
|
|
}
|