mirror of
https://github.com/SamyRai/tercul-backend.git
synced 2025-12-27 04:01:34 +00:00
d50722dad5
Some checks failed
- Updated database models and repositories to replace uint IDs with UUIDs. - Modified test fixtures to generate and use UUIDs for authors, translations, users, and works. - Adjusted mock implementations to align with the new UUID structure. - Ensured all relevant functions and methods are updated to handle UUIDs correctly. - Added necessary imports for UUID handling in various files.
78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"tercul/internal/data/sql"
|
|
"tercul/internal/enrichment"
|
|
"tercul/internal/platform/config"
|
|
"tercul/internal/platform/db"
|
|
"tercul/internal/platform/log"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func main() {
|
|
// 1. Parse command-line arguments
|
|
entityType := flag.String("type", "", "The type of entity to enrich (e.g., 'author')")
|
|
entityIDStr := flag.String("id", "", "The ID of the entity to enrich")
|
|
flag.Parse()
|
|
|
|
if *entityType == "" || *entityIDStr == "" {
|
|
fmt.Println("Usage: go run cmd/tools/enrich/main.go --type <entity_type> --id <entity_id>")
|
|
os.Exit(1)
|
|
}
|
|
|
|
entityID, err := uuid.Parse(*entityIDStr)
|
|
if err != nil {
|
|
fmt.Printf("Invalid entity ID: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// 2. Initialize dependencies
|
|
cfg, err := config.LoadConfig()
|
|
if err != nil {
|
|
log.Fatal(err, "Failed to load config")
|
|
}
|
|
log.Init("enrich-tool", "development")
|
|
database, err := db.InitDB(cfg, nil) // No metrics needed for this tool
|
|
if err != nil {
|
|
log.Fatal(err, "Failed to initialize database")
|
|
}
|
|
defer func() {
|
|
if err := db.Close(database); err != nil {
|
|
log.Error(err, "Error closing database")
|
|
}
|
|
}()
|
|
|
|
repos := sql.NewRepositories(database, cfg)
|
|
enrichmentSvc := enrichment.NewService()
|
|
|
|
// 3. Fetch, enrich, and save the entity
|
|
ctx := context.Background()
|
|
log.Info(fmt.Sprintf("Enriching %s with ID %d", *entityType, entityID))
|
|
|
|
switch *entityType {
|
|
case "author":
|
|
author, err := repos.Author.GetByID(ctx, entityID)
|
|
if err != nil {
|
|
log.Fatal(err, "Failed to get author")
|
|
}
|
|
|
|
if err := enrichmentSvc.EnrichAuthor(ctx, author); err != nil {
|
|
log.Fatal(err, "Failed to enrich author")
|
|
}
|
|
|
|
if err := repos.Author.Update(ctx, author); err != nil {
|
|
log.Fatal(err, "Failed to save enriched author")
|
|
}
|
|
|
|
log.Info("Successfully enriched and saved author")
|
|
|
|
default:
|
|
log.Fatal(fmt.Errorf("unknown entity type: %s", *entityType), "Enrichment failed")
|
|
}
|
|
}
|