2823fb45f5
Phone+OTP auth, staged onboarding, admin approval workflow, and a full discovery/swipe/match/chat system with photo upload and approximate distance (haversine) based on user location. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package repository
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
|
|
"dating-app/internal/models"
|
|
)
|
|
|
|
type PhotoRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewPhotoRepository(db *gorm.DB) *PhotoRepository {
|
|
return &PhotoRepository{db: db}
|
|
}
|
|
|
|
func (r *PhotoRepository) ListByUser(userID uint) ([]models.Photo, error) {
|
|
var photos []models.Photo
|
|
err := r.db.Where("user_id = ?", userID).Order("sort_order asc, id asc").Find(&photos).Error
|
|
return photos, err
|
|
}
|
|
|
|
func (r *PhotoRepository) ListByUserIDs(userIDs []uint) ([]models.Photo, error) {
|
|
var photos []models.Photo
|
|
if len(userIDs) == 0 {
|
|
return photos, nil
|
|
}
|
|
err := r.db.Where("user_id IN ?", userIDs).Order("sort_order asc, id asc").Find(&photos).Error
|
|
return photos, err
|
|
}
|
|
|
|
func (r *PhotoRepository) CountByUser(userID uint) (int64, error) {
|
|
var count int64
|
|
err := r.db.Model(&models.Photo{}).Where("user_id = ?", userID).Count(&count).Error
|
|
return count, err
|
|
}
|
|
|
|
func (r *PhotoRepository) Create(photo *models.Photo) error {
|
|
return r.db.Create(photo).Error
|
|
}
|
|
|
|
func (r *PhotoRepository) FindByID(id uint) (*models.Photo, error) {
|
|
var photo models.Photo
|
|
if err := r.db.First(&photo, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &photo, nil
|
|
}
|
|
|
|
func (r *PhotoRepository) Delete(id uint) error {
|
|
return r.db.Delete(&models.Photo{}, id).Error
|
|
}
|