tercul-backend/internal/data/sql/monetization_repository.go
google-labs-jules[bot] 781b313bf1 feat: Complete all pending tasks from TASKS.md
This commit addresses all the high-priority tasks outlined in the TASKS.md file, significantly improving the application's observability, completing key features, and refactoring critical parts of the codebase.

### Observability

- **Centralized Logging:** Implemented a new structured, context-aware logging system using `zerolog`. A new logging middleware injects request-specific information (request ID, user ID, trace ID) into the logger, and all application logging has been refactored to use this new system.
- **Prometheus Metrics:** Added Prometheus metrics for database query performance by creating a GORM plugin that automatically records query latency and totals.
- **OpenTelemetry Tracing:** Fully instrumented all application services in `internal/app` and data repositories in `internal/data/sql` with OpenTelemetry tracing, providing deep visibility into application performance.

### Features

- **Analytics:** Implemented like, comment, and bookmark counting. The respective command handlers now call the analytics service to increment counters when these actions are performed.
- **Enrichment Tool:** Built a new, extensible `enrich` command-line tool to fetch data from external sources. The initial implementation enriches author data using the Open Library API.

### Refactoring & Fixes

- **Decoupled Testing:** Refactored the testing utilities in `internal/testutil` to be database-agnostic, promoting the use of mock-based unit tests and improving test speed and reliability.
- **Build Fixes:** Resolved numerous build errors, including a critical import cycle between the logging, observability, and authentication packages.
- **Search Service:** Fixed the search service integration by implementing the `GetWorkContent` method in the localization service, allowing the search indexer to correctly fetch and index work content.
2025-10-05 05:26:27 +00:00

107 lines
5.5 KiB
Go

package sql
import (
"context"
"tercul/internal/domain"
"tercul/internal/domain/work"
"go.opentelemetry.io/otel"
"go.opentelemetry.io/otel/trace"
"gorm.io/gorm"
)
type monetizationRepository struct {
domain.BaseRepository[domain.Monetization]
db *gorm.DB
tracer trace.Tracer
}
// NewMonetizationRepository creates a new MonetizationRepository.
func NewMonetizationRepository(db *gorm.DB) domain.MonetizationRepository {
return &monetizationRepository{
BaseRepository: NewBaseRepositoryImpl[domain.Monetization](db),
db: db,
tracer: otel.Tracer("monetization.repository"),
}
}
func (r *monetizationRepository) AddMonetizationToWork(ctx context.Context, workID uint, monetizationID uint) error {
ctx, span := r.tracer.Start(ctx, "AddMonetizationToWork")
defer span.End()
workRecord := &work.Work{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: workID}}}
monetization := &domain.Monetization{BaseModel: domain.BaseModel{ID: monetizationID}}
return r.db.WithContext(ctx).Model(workRecord).Association("Monetizations").Append(monetization)
}
func (r *monetizationRepository) RemoveMonetizationFromWork(ctx context.Context, workID uint, monetizationID uint) error {
ctx, span := r.tracer.Start(ctx, "RemoveMonetizationFromWork")
defer span.End()
workRecord := &work.Work{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: workID}}}
monetization := &domain.Monetization{BaseModel: domain.BaseModel{ID: monetizationID}}
return r.db.WithContext(ctx).Model(workRecord).Association("Monetizations").Delete(monetization)
}
func (r *monetizationRepository) AddMonetizationToAuthor(ctx context.Context, authorID uint, monetizationID uint) error {
ctx, span := r.tracer.Start(ctx, "AddMonetizationToAuthor")
defer span.End()
author := &domain.Author{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: authorID}}}
monetization := &domain.Monetization{BaseModel: domain.BaseModel{ID: monetizationID}}
return r.db.WithContext(ctx).Model(author).Association("Monetizations").Append(monetization)
}
func (r *monetizationRepository) RemoveMonetizationFromAuthor(ctx context.Context, authorID uint, monetizationID uint) error {
ctx, span := r.tracer.Start(ctx, "RemoveMonetizationFromAuthor")
defer span.End()
author := &domain.Author{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: authorID}}}
monetization := &domain.Monetization{BaseModel: domain.BaseModel{ID: monetizationID}}
return r.db.WithContext(ctx).Model(author).Association("Monetizations").Delete(monetization)
}
func (r *monetizationRepository) AddMonetizationToBook(ctx context.Context, bookID uint, monetizationID uint) error {
ctx, span := r.tracer.Start(ctx, "AddMonetizationToBook")
defer span.End()
book := &domain.Book{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: bookID}}}
monetization := &domain.Monetization{BaseModel: domain.BaseModel{ID: monetizationID}}
return r.db.WithContext(ctx).Model(book).Association("Monetizations").Append(monetization)
}
func (r *monetizationRepository) RemoveMonetizationFromBook(ctx context.Context, bookID uint, monetizationID uint) error {
ctx, span := r.tracer.Start(ctx, "RemoveMonetizationFromBook")
defer span.End()
book := &domain.Book{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: bookID}}}
monetization := &domain.Monetization{BaseModel: domain.BaseModel{ID: monetizationID}}
return r.db.WithContext(ctx).Model(book).Association("Monetizations").Delete(monetization)
}
func (r *monetizationRepository) AddMonetizationToPublisher(ctx context.Context, publisherID uint, monetizationID uint) error {
ctx, span := r.tracer.Start(ctx, "AddMonetizationToPublisher")
defer span.End()
publisher := &domain.Publisher{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: publisherID}}}
monetization := &domain.Monetization{BaseModel: domain.BaseModel{ID: monetizationID}}
return r.db.WithContext(ctx).Model(publisher).Association("Monetizations").Append(monetization)
}
func (r *monetizationRepository) RemoveMonetizationFromPublisher(ctx context.Context, publisherID uint, monetizationID uint) error {
ctx, span := r.tracer.Start(ctx, "RemoveMonetizationFromPublisher")
defer span.End()
publisher := &domain.Publisher{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: publisherID}}}
monetization := &domain.Monetization{BaseModel: domain.BaseModel{ID: monetizationID}}
return r.db.WithContext(ctx).Model(publisher).Association("Monetizations").Delete(monetization)
}
func (r *monetizationRepository) AddMonetizationToSource(ctx context.Context, sourceID uint, monetizationID uint) error {
ctx, span := r.tracer.Start(ctx, "AddMonetizationToSource")
defer span.End()
source := &domain.Source{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: sourceID}}}
monetization := &domain.Monetization{BaseModel: domain.BaseModel{ID: monetizationID}}
return r.db.WithContext(ctx).Model(source).Association("Monetizations").Append(monetization)
}
func (r *monetizationRepository) RemoveMonetizationFromSource(ctx context.Context, sourceID uint, monetizationID uint) error {
ctx, span := r.tracer.Start(ctx, "RemoveMonetizationFromSource")
defer span.End()
source := &domain.Source{TranslatableModel: domain.TranslatableModel{BaseModel: domain.BaseModel{ID: sourceID}}}
monetization := &domain.Monetization{BaseModel: domain.BaseModel{ID: monetizationID}}
return r.db.WithContext(ctx).Model(source).Association("Monetizations").Delete(monetization)
}