153 lines
4.2 KiB
Go
153 lines
4.2 KiB
Go
package rest
|
|
|
|
import (
|
|
"github.com/eyebluecn/tank/code/core"
|
|
"github.com/eyebluecn/tank/code/tool/builder"
|
|
"github.com/eyebluecn/tank/code/tool/result"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/eyebluecn/tank/code/tool/uuid"
|
|
"time"
|
|
)
|
|
|
|
type FootprintDao struct {
|
|
BaseDao
|
|
}
|
|
|
|
//find by uuid. if not found return nil.
|
|
func (this *FootprintDao) FindByUuid(uuid string) *Footprint {
|
|
var entity = &Footprint{}
|
|
db := core.CONTEXT.GetDB().Where("uuid = ?", uuid).First(entity)
|
|
if db.Error != nil {
|
|
if db.Error.Error() == result.DB_ERROR_NOT_FOUND {
|
|
return nil
|
|
} else {
|
|
panic(db.Error)
|
|
}
|
|
}
|
|
return entity
|
|
}
|
|
|
|
//find by uuid. if not found panic NotFound error
|
|
func (this *FootprintDao) CheckByUuid(uuid string) *Footprint {
|
|
entity := this.FindByUuid(uuid)
|
|
if entity == nil {
|
|
panic(result.NotFound("not found record with uuid = %s", uuid))
|
|
}
|
|
return entity
|
|
}
|
|
|
|
func (this *FootprintDao) Page(page int, pageSize int, userUuid string, sortArray []builder.OrderPair) *Pager {
|
|
|
|
var wp = &builder.WherePair{}
|
|
|
|
if userUuid != "" {
|
|
wp = wp.And(&builder.WherePair{Query: "user_uuid = ?", Args: []interface{}{userUuid}})
|
|
}
|
|
|
|
var conditionDB *gorm.DB
|
|
conditionDB = core.CONTEXT.GetDB().Model(&Footprint{}).Where(wp.Query, wp.Args...)
|
|
|
|
var count int64 = 0
|
|
db := conditionDB.Count(&count)
|
|
this.PanicError(db.Error)
|
|
|
|
var footprints []*Footprint
|
|
db = conditionDB.Order(this.GetSortString(sortArray)).Offset(page * pageSize).Limit(pageSize).Find(&footprints)
|
|
this.PanicError(db.Error)
|
|
pager := NewPager(page, pageSize, int(count), footprints)
|
|
|
|
return pager
|
|
}
|
|
|
|
func (this *FootprintDao) Create(footprint *Footprint) *Footprint {
|
|
|
|
timeUUID, _ := uuid.NewV4()
|
|
footprint.Uuid = string(timeUUID.String())
|
|
footprint.CreateTime = time.Now()
|
|
footprint.UpdateTime = time.Now()
|
|
footprint.Sort = time.Now().UnixNano() / 1e6
|
|
db := core.CONTEXT.GetDB().Create(footprint)
|
|
this.PanicError(db.Error)
|
|
|
|
return footprint
|
|
}
|
|
|
|
func (this *FootprintDao) Save(footprint *Footprint) *Footprint {
|
|
|
|
footprint.UpdateTime = time.Now()
|
|
db := core.CONTEXT.GetDB().Save(footprint)
|
|
this.PanicError(db.Error)
|
|
|
|
return footprint
|
|
}
|
|
|
|
func (this *FootprintDao) Delete(footprint *Footprint) {
|
|
|
|
db := core.CONTEXT.GetDB().Delete(&footprint)
|
|
this.PanicError(db.Error)
|
|
}
|
|
|
|
func (this *FootprintDao) CountBetweenTime(startTime time.Time, endTime time.Time) int64 {
|
|
var count int64
|
|
db := core.CONTEXT.GetDB().Model(&Footprint{}).Where("create_time >= ? AND create_time <= ?", startTime, endTime).Count(&count)
|
|
this.PanicError(db.Error)
|
|
return count
|
|
}
|
|
|
|
func (this *FootprintDao) UvBetweenTime(startTime time.Time, endTime time.Time) int64 {
|
|
|
|
var wp = &builder.WherePair{Query: "create_time >= ? AND create_time <= ?", Args: []interface{}{startTime, endTime}}
|
|
|
|
var count int64
|
|
db := core.CONTEXT.GetDB().Model(&Footprint{}).Where(wp.Query, wp.Args...).Count(&count)
|
|
if count == 0 {
|
|
return 0
|
|
}
|
|
|
|
db = core.CONTEXT.GetDB().Model(&Footprint{}).Where("create_time >= ? AND create_time <= ?", startTime, endTime).Select("COUNT(DISTINCT(ip))")
|
|
this.PanicError(db.Error)
|
|
row := db.Row()
|
|
err := row.Scan(&count)
|
|
this.PanicError(err)
|
|
return count
|
|
}
|
|
|
|
func (this *FootprintDao) AvgCostBetweenTime(startTime time.Time, endTime time.Time) int64 {
|
|
|
|
var wp = &builder.WherePair{Query: "create_time >= ? AND create_time <= ?", Args: []interface{}{startTime, endTime}}
|
|
|
|
var count int64
|
|
db := core.CONTEXT.GetDB().Model(&Footprint{}).Where(wp.Query, wp.Args...).Count(&count)
|
|
if count == 0 {
|
|
return 0
|
|
}
|
|
|
|
var cost float64
|
|
db = core.CONTEXT.GetDB().Model(&Footprint{}).Where(wp.Query, wp.Args...).Select("AVG(cost)")
|
|
this.PanicError(db.Error)
|
|
row := db.Row()
|
|
err := row.Scan(&cost)
|
|
this.PanicError(err)
|
|
return int64(cost)
|
|
}
|
|
|
|
func (this *FootprintDao) DeleteByCreateTimeBefore(createTime time.Time) {
|
|
db := core.CONTEXT.GetDB().Where("create_time < ?", createTime).Delete(Footprint{})
|
|
this.PanicError(db.Error)
|
|
}
|
|
|
|
func (this *FootprintDao) DeleteByUserUuid(userUuid string) {
|
|
|
|
db := core.CONTEXT.GetDB().Where("user_uuid = ?", userUuid).Delete(Footprint{})
|
|
this.PanicError(db.Error)
|
|
|
|
}
|
|
|
|
//System cleanup.
|
|
func (this *FootprintDao) Cleanup() {
|
|
this.logger.Info("[FootprintDao][DownloadTokenDao] clean up. Delete all Footprint")
|
|
db := core.CONTEXT.GetDB().Where("uuid is not null").Delete(Footprint{})
|
|
this.PanicError(db.Error)
|
|
}
|