mirror of
https://github.com/SamyRai/tercul-backend.git
synced 2025-12-27 05:11:34 +00:00
32 lines
781 B
Go
32 lines
781 B
Go
package sql
|
|
|
|
import (
|
|
"context"
|
|
"tercul/internal/domain"
|
|
"tercul/internal/domain/edge"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type edgeRepository struct {
|
|
domain.BaseRepository[domain.Edge]
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewEdgeRepository creates a new EdgeRepository.
|
|
func NewEdgeRepository(db *gorm.DB) edge.EdgeRepository {
|
|
return &edgeRepository{
|
|
BaseRepository: NewBaseRepositoryImpl[domain.Edge](db),
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
// ListBySource finds edges by source table and ID
|
|
func (r *edgeRepository) ListBySource(ctx context.Context, sourceTable string, sourceID uint) ([]domain.Edge, error) {
|
|
var edges []domain.Edge
|
|
if err := r.db.WithContext(ctx).Where("source_table = ? AND source_id = ?", sourceTable, sourceID).Find(&edges).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return edges, nil
|
|
}
|