...

2025-05-23 09:33:05 +04:00
parent a16ac8f627
commit 79d66e4b6b
34 changed files with 603 additions and 608 deletions
--- a/pkg/herojobs/processjob.go
+++ b/pkg/herojobs/processjob.go
@@ -0,0 +1,148 @@
+package herojobs
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"time"
+)
+
+// processJob processes a job
+func (d *WatchDog) processJob(job *Job) {
+	// First, initialize OurDB connection and save the job
+	if err := job.initOurDbClient(); err != nil {
+		log.Printf("Error initializing OurDB client: %v", err)
+	} else {
+		// Set job status to active and save it to OurDB
+		job.Status = JobStatusActive
+		job.TimeStart = time.Now().Unix()
+
+		if err := job.Save(); err != nil {
+			log.Printf("Error saving job to OurDB: %v", err)
+		}
+
+		// Make sure to close the DB connection when we're done
+		defer job.Close()
+	}
+
+	// Also update Redis for backward compatibility
+	err := d.redisClient.UpdateJobStatus(job.JobID, JobStatusActive)
+	if err != nil {
+		log.Printf("Error updating job status in Redis: %v", err)
+	}
+
+	// Create a context with timeout
+	timeout := time.Duration(job.Timeout) * time.Second
+	if timeout == 0 {
+		timeout = 60 * time.Second // Default timeout: 60 seconds
+	}
+	ctx, cancel := context.WithTimeout(d.ctx, timeout)
+	defer cancel()
+
+	// Process job based on parameters type
+	switch job.ParamsType {
+	case ParamsTypeHeroScript:
+		d.processHeroScript(ctx, job)
+	case ParamsTypeRhaiScript:
+		d.processRhaiScript(ctx, job)
+	case ParamsTypeOpenRPC:
+		d.processOpenRPC(ctx, job)
+	case ParamsTypeAI:
+		d.processAI(ctx, job)
+	default:
+		// Unknown parameters type
+		errMsg := fmt.Sprintf("Unknown parameters type: %s", job.ParamsType)
+		log.Printf("Error processing job %s: %s", job.JobID, errMsg)
+
+		// Update job status to error in both OurDB and Redis
+		job.Finish(JobStatusError, "", fmt.Errorf(errMsg))
+
+		if err := d.redisClient.UpdateJobError(job.JobID, errMsg); err != nil {
+			log.Printf("Error updating job error in Redis: %v", err)
+		}
+	}
+}
+
+// processHeroScript processes a job with HeroScript parameters
+func (d *WatchDog) processHeroScript(ctx context.Context, job *Job) {
+	// Start a goroutine to process the job
+	d.wg.Add(1)
+	go func() {
+		defer d.wg.Done()
+
+		// TODO: Implement HeroScript processing
+		errMsg := "HeroScript processing not implemented yet"
+		log.Printf("Error processing job %s: %s", job.JobID, errMsg)
+
+		// Update job status in OurDB
+		job.Finish(JobStatusError, "", fmt.Errorf(errMsg))
+
+		// Also update Redis for backward compatibility
+		if err := d.redisClient.UpdateJobError(job.JobID, errMsg); err != nil {
+			log.Printf("Error updating job error in Redis: %v", err)
+		}
+	}()
+}
+
+// processRhaiScript processes a job with RhaiScript parameters
+func (d *WatchDog) processRhaiScript(ctx context.Context, job *Job) {
+	// Start a goroutine to process the job
+	d.wg.Add(1)
+	go func() {
+		defer d.wg.Done()
+
+		// TODO: Implement RhaiScript processing
+		errMsg := "RhaiScript processing not implemented yet"
+		log.Printf("Error processing job %s: %s", job.JobID, errMsg)
+
+		// Update job status in OurDB
+		job.Finish(JobStatusError, "", fmt.Errorf(errMsg))
+
+		// Also update Redis for backward compatibility
+		if err := d.redisClient.UpdateJobError(job.JobID, errMsg); err != nil {
+			log.Printf("Error updating job error in Redis: %v", err)
+		}
+	}()
+}
+
+// processOpenRPC processes a job with OpenRPC parameters
+func (d *WatchDog) processOpenRPC(ctx context.Context, job *Job) {
+	// Start a goroutine to process the job
+	d.wg.Add(1)
+	go func() {
+		defer d.wg.Done()
+
+		// TODO: Implement OpenRPC processing
+		errMsg := "OpenRPC processing not implemented yet"
+		log.Printf("Error processing job %s: %s", job.JobID, errMsg)
+
+		// Update job status in OurDB
+		job.Finish(JobStatusError, "", fmt.Errorf(errMsg))
+
+		// Also update Redis for backward compatibility
+		if err := d.redisClient.UpdateJobError(job.JobID, errMsg); err != nil {
+			log.Printf("Error updating job error in Redis: %v", err)
+		}
+	}()
+}
+
+// processAI processes a job with AI parameters
+func (d *WatchDog) processAI(ctx context.Context, job *Job) {
+	// Start a goroutine to process the job
+	d.wg.Add(1)
+	go func() {
+		defer d.wg.Done()
+
+		// TODO: Implement AI processing
+		errMsg := "AI processing not implemented yet"
+		log.Printf("Error processing job %s: %s", job.JobID, errMsg)
+
+		// Update job status in OurDB
+		job.Finish(JobStatusError, "", fmt.Errorf(errMsg))
+
+		// Also update Redis for backward compatibility
+		if err := d.redisClient.UpdateJobError(job.JobID, errMsg); err != nil {
+			log.Printf("Error updating job error in Redis: %v", err)
+		}
+	}()
+}