54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
|
package repository
|
||
|
|
|
||
|
|
import (
|
||
|
|
"gorm.io/gorm"
|
||
|
|
|
||
|
|
"dating-app/internal/models"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ReportRepository struct {
|
||
|
|
db *gorm.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewReportRepository(db *gorm.DB) *ReportRepository {
|
||
|
|
return &ReportRepository{db: db}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *ReportRepository) Create(report *models.Report) error {
|
||
|
|
return r.db.Create(report).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *ReportRepository) ListByUser(userID uint) ([]models.Report, error) {
|
||
|
|
var reports []models.Report
|
||
|
|
err := r.db.Where("user_id = ?", userID).Order("created_at desc").Find(&reports).Error
|
||
|
|
return reports, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// CountsByUserIDs تعداد گزارشهای ثبتشده روی چند کاربر رو با هم، در یک
|
||
|
|
// کوئری برمیگردونه (برای نشون دادن نشان تعداد گزارش روی لیست کاربران، بدون
|
||
|
|
// یک کوئری جدا بهازای هر کاربر).
|
||
|
|
func (r *ReportRepository) CountsByUserIDs(userIDs []uint) (map[uint]int64, error) {
|
||
|
|
result := make(map[uint]int64, len(userIDs))
|
||
|
|
if len(userIDs) == 0 {
|
||
|
|
return result, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type row struct {
|
||
|
|
UserID uint
|
||
|
|
Count int64
|
||
|
|
}
|
||
|
|
var rows []row
|
||
|
|
err := r.db.Model(&models.Report{}).
|
||
|
|
Select("user_id, count(*) as count").
|
||
|
|
Where("user_id IN ?", userIDs).
|
||
|
|
Group("user_id").
|
||
|
|
Find(&rows).Error
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
for _, rr := range rows {
|
||
|
|
result[rr.UserID] = rr.Count
|
||
|
|
}
|
||
|
|
return result, nil
|
||
|
|
}
|