user_source_click_flow.go 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. package user
  2. import (
  3. "eta/eta_mini_ht_api/models"
  4. "gorm.io/gorm"
  5. "time"
  6. )
  7. const (
  8. userSourceClickFlows = "user_source_click_flows"
  9. )
  10. // UserReportClickFlow 用户点击研报流水记录
  11. type UserSourceClickFlow struct {
  12. ID int `gorm:"column:id;primaryKey;autoIncrement:'id'"`
  13. TraceId string `gorm:"column:trace_id"`
  14. UserID int `gorm:"column:user_id"` // 用户ID\
  15. Mobile string `gorm:"column:mobile"`
  16. SourceId int `gorm:"column:source_id"` // 研报ID
  17. SourceType SourceType `gorm:"column:source_type;type:enum('REPORT','VIDEO','AUDIO')"`
  18. ClickTime time.Time `gorm:"column:click_time"` // 点击时间
  19. ReadDurationSeconds int64 `gorm:"column:read_duration_seconds"`
  20. IPAddress string `gorm:"column:ip_address"` // IP地址
  21. Location string `gorm:"column:location"` // 地理位置
  22. Hidden bool `gorm:"column:hidden;type:tinyint(1);default:0"`
  23. Referer string `gorm:"column:referer"` // 来源页面
  24. AdditionalData string `gorm:"column:additional_data"` // 额外数据
  25. }
  26. func (v *UserSourceClickFlow) BeforeCreate(tx *gorm.DB) (err error) {
  27. v.ClickTime = time.Now()
  28. return
  29. }
  30. func UpdateSourceClicks(count UserSourceClickFlow) (err error) {
  31. db := models.Main()
  32. err = db.Table(userSourceClickFlows).Where("trace_id =?", count.TraceId).UpdateColumn("read_duration_seconds", count.ReadDurationSeconds).Error
  33. return
  34. }
  35. func CountSourceClicks(count UserSourceClickFlow) (err error) {
  36. db := models.Main()
  37. err = db.Create(&count).Error
  38. return
  39. }
  40. func GetClickRecordByTraceId(traceId string) (record UserSourceClickFlow, err error) {
  41. db := models.Main()
  42. err = db.Select("*").Where("trace_id=?", traceId).First(&record).Error
  43. return
  44. }
  45. type CountClickFlowById struct {
  46. SourceId int `gorm:"column:source_id"`
  47. Count int `gorm:"column:count"`
  48. }
  49. func GetTimeDurationReportCountsById(begin string, end string, limit int, sourceType SourceType) (ids []CountClickFlowById, err error) {
  50. db := models.Main()
  51. err = db.Table(userSourceClickFlows).Select("source_id,count(*) count").Where("source_id>0 and source_type=? and DATE(click_time) BETWEEN ? AND ? AND hidden =0", sourceType, begin, end).Group("source_id").Order("count desc").Limit(limit).Scan(&ids).Error
  52. return
  53. }
  54. func HiddenFlows(id int, sourceType SourceType) (err error) {
  55. db := models.Main()
  56. err = db.Table(userSourceClickFlows).Where("source_type=? and source_id =?", sourceType, id).Update("hidden", 1).Error
  57. return
  58. }
  59. func ShowFlows(id int, sourceType SourceType) (err error) {
  60. db := models.Main()
  61. err = db.Table(userSourceClickFlows).Where("source_type=? and source_id =?", sourceType, id).Update("hidden", 0).Error
  62. return
  63. }