mirror of
https://github.com/SamyRai/tercul-backend.git
synced 2025-12-27 05:11:34 +00:00
8797cec718
This commit includes the following changes: - Refactored all data repositories in `internal/data/sql/` to use a consistent `sql` package and to align with the new `domain` models. - Fixed the GraphQL structure by moving the server creation logic from `internal/app` to `cmd/api`, which resolved an import cycle. - Corrected numerous incorrect import paths for packages like `graph`, `linguistics`, `syncjob`, and the legacy `models` package. - Resolved several package and function redeclaration errors. - Removed legacy migration code.
113 lines
3.2 KiB
Go
113 lines
3.2 KiB
Go
package linguistics
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"tercul/internal/domain"
|
|
"time"
|
|
|
|
"github.com/hibiken/asynq"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
const (
|
|
TaskLinguisticAnalysis = "analysis:linguistic"
|
|
)
|
|
|
|
// LinguisticSyncJob manages the linguistic analysis sync process
|
|
type LinguisticSyncJob struct {
|
|
DB *gorm.DB
|
|
Analyzer Analyzer
|
|
Client *asynq.Client
|
|
}
|
|
|
|
// NewLinguisticSyncJob creates a new LinguisticSyncJob
|
|
func NewLinguisticSyncJob(db *gorm.DB, analyzer Analyzer, client *asynq.Client) *LinguisticSyncJob {
|
|
return &LinguisticSyncJob{
|
|
DB: db,
|
|
Analyzer: analyzer,
|
|
Client: client,
|
|
}
|
|
}
|
|
|
|
// AnalysisPayload contains data for the linguistic analysis task
|
|
type AnalysisPayload struct {
|
|
WorkID uint `json:"work_id"`
|
|
}
|
|
|
|
// EnqueueAnalysisForWork enqueues a linguistic analysis task for a specific work
|
|
func EnqueueAnalysisForWork(client *asynq.Client, workID uint) error {
|
|
payload := AnalysisPayload{WorkID: workID}
|
|
data, err := json.Marshal(payload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
task := asynq.NewTask(TaskLinguisticAnalysis, data)
|
|
_, err = client.Enqueue(task, asynq.ProcessIn(5*time.Second))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Printf("Enqueued linguistic analysis task for work ID %d", workID)
|
|
return nil
|
|
}
|
|
|
|
// EnqueueAnalysisForAllWorks enqueues linguistic analysis tasks for all works
|
|
func (j *LinguisticSyncJob) EnqueueAnalysisForAllWorks() error {
|
|
log.Println("Enqueueing linguistic analysis jobs for all works...")
|
|
|
|
var workIDs []uint
|
|
if err := j.DB.Model(&domain.Work{}).Pluck("id", &workIDs).Error; err != nil {
|
|
return fmt.Errorf("error fetching work IDs: %w", err)
|
|
}
|
|
|
|
for _, workID := range workIDs {
|
|
if err := EnqueueAnalysisForWork(j.Client, workID); err != nil {
|
|
log.Printf("Error enqueueing linguistic analysis for work ID %d: %v", workID, err)
|
|
} else {
|
|
log.Printf("Enqueued linguistic analysis for work ID %d", workID)
|
|
}
|
|
}
|
|
|
|
log.Println("Linguistic analysis jobs enqueued successfully.")
|
|
return nil
|
|
}
|
|
|
|
// HandleLinguisticAnalysis handles the linguistic analysis task
|
|
func (j *LinguisticSyncJob) HandleLinguisticAnalysis(ctx context.Context, t *asynq.Task) error {
|
|
var payload AnalysisPayload
|
|
if err := json.Unmarshal(t.Payload(), &payload); err != nil {
|
|
return fmt.Errorf("failed to unmarshal linguistic analysis payload: %v", err)
|
|
}
|
|
|
|
log.Printf("Processing linguistic analysis for work ID %d", payload.WorkID)
|
|
|
|
// Check if analysis already exists
|
|
var count int64
|
|
if err := j.DB.Model(&domain.LanguageAnalysis{}).Where("work_id = ?", payload.WorkID).Count(&count).Error; err != nil {
|
|
return fmt.Errorf("error checking existing analysis: %w", err)
|
|
}
|
|
|
|
// Skip if analysis already exists
|
|
if count > 0 {
|
|
log.Printf("Linguistic analysis already exists for work ID %d, skipping", payload.WorkID)
|
|
return nil
|
|
}
|
|
|
|
// Perform the analysis
|
|
if err := j.Analyzer.AnalyzeWork(ctx, payload.WorkID); err != nil {
|
|
return fmt.Errorf("error analyzing work ID %d: %w", payload.WorkID, err)
|
|
}
|
|
|
|
log.Printf("Completed linguistic analysis for work ID %d", payload.WorkID)
|
|
return nil
|
|
}
|
|
|
|
// RegisterLinguisticHandlers registers the linguistic analysis task handlers
|
|
func RegisterLinguisticHandlers(mux *asynq.ServeMux, job *LinguisticSyncJob) {
|
|
mux.HandleFunc(TaskLinguisticAnalysis, job.HandleLinguisticAnalysis)
|
|
}
|