nidus-sync/platform/start.go

package platform

import (
	"context"
	"fmt"
	"strconv"
	"sync"
	"time"

	"github.com/Gleipnir-Technology/bob"
	bobpgx "github.com/Gleipnir-Technology/bob/drivers/pgx"
	//"github.com/Gleipnir-Technology/bob/dialect/psql"
	//"github.com/Gleipnir-Technology/bob/dialect/psql/sm"
	"github.com/Gleipnir-Technology/nidus-sync/config"
	"github.com/Gleipnir-Technology/nidus-sync/db"
	"github.com/Gleipnir-Technology/nidus-sync/db/enums"
	"github.com/Gleipnir-Technology/nidus-sync/db/models"
	//"github.com/Gleipnir-Technology/nidus-sync/platform/background"
	"github.com/Gleipnir-Technology/nidus-sync/platform/csv"
	"github.com/Gleipnir-Technology/nidus-sync/platform/email"
	"github.com/Gleipnir-Technology/nidus-sync/platform/file"
	"github.com/Gleipnir-Technology/nidus-sync/platform/geocode"
	"github.com/Gleipnir-Technology/nidus-sync/platform/text"
	"github.com/jackc/pgx/v5"
	//"github.com/Gleipnir-Technology/nidus-sync/userfile"
	//"github.com/google/uuid"
	"github.com/rs/zerolog/log"
)

var waitGroup sync.WaitGroup
var newOAuthTokenChannel chan struct{}

func StartAll(ctx context.Context) error {
	err := email.LoadTemplates()
	if err != nil {
		return fmt.Errorf("Failed to load email templates: %w", err)
	}

	err = text.StoreSources()
	if err != nil {
		return fmt.Errorf("Failed to store text source phone numbers: %w", err)
	}

	err = file.CreateDirectories()
	if err != nil {
		return fmt.Errorf("Failed to create file directories: %w", err)
	}

	err = initializeLabelStudio()
	if err != nil {
		return fmt.Errorf("init label studio: %w", err)
	}

	geocode.InitializeStadia(config.StadiaMapsAPIKey)

	newOAuthTokenChannel = make(chan struct{}, 10)

	waitGroup.Add(1)
	go func() {
		defer waitGroup.Done()
		refreshFieldseekerData(ctx, newOAuthTokenChannel)
	}()

	err = addWaitingJobs(ctx)
	if err != nil {
		log.Error().Err(err).Msg("Failed to add waiting background jobs")
	}
	return nil
}

func WaitForExit() {
	waitGroup.Wait()
}

func addWaitingJobs(ctx context.Context) error {
	jobs, err := models.Jobs.Query().All(ctx, db.PGInstance.BobDB)
	if err != nil {
		return fmt.Errorf("Failed to query waiting jobs: %w", err)
	}
	for _, job := range jobs {
		go func() {
			txn, err := db.PGInstance.BobDB.Begin(ctx)
			if err != nil {
				log.Error().Err(err).Msg("failed begin txn")
				return
			}
			err = handleJob(ctx, txn, job)
			if err != nil {
				log.Error().Err(err).Msg("failed begin txn")
				return
			}
			err = job.Delete(ctx, txn)
			if err != nil {
				log.Error().Err(err).Msg("failed delete job")
				return
			}
			txn.Commit(ctx)
		}()
	}
	return nil
}
func handleJob(ctx context.Context, txn bob.Executor, job *models.Job) error {
	switch job.Type {
	case enums.JobtypeCSVCommit:
		return csv.JobCommit(ctx, txn, job.RowID)
	case enums.JobtypeCSVImport:
		return csv.JobImport(ctx, txn, job.RowID)
	case enums.JobtypeLabelStudioAudioCreate:
		return handleJobLabelStudioAudioCreate(ctx, txn, job.RowID)
	case enums.JobtypeEmailSend:
		return email.Job(ctx, txn, job.RowID)
	case enums.JobtypeTextSend:
		return text.Job(ctx, txn, job.RowID)
	default:
		return fmt.Errorf("No handler for job type %s", string(job.Type))
	}
}
func handleJobLabelStudioAudioCreate(ctx context.Context, txn bob.Executor, row_id int32) error {
	return jobLabelStudioAudioCreate(ctx, txn, row_id)
}
func listenForJobs(ctx context.Context) {
	for {
		//es.SendQueuedEmails(ctx) // send any emails queued prior to listening for notificiations
		err := listenAndDoOneJob(ctx)
		if err != nil {
			log.Error().Err(err).Msg("Crashed listenAndDoOneJob")
		}

		select {
		case <-ctx.Done():
			return
		default:
			// If listenAndSendOneConn returned and ctx has not been cancelled that means there was a fatal database error.
			// Wait a while to avoid busy-looping while the database is unreachable.
			time.Sleep(time.Minute)
		}
	}
}
func listenAndDoOneJob(ctx context.Context) error {
	conn, err := db.PGInstance.PGXPool.Acquire(ctx)
	if err != nil {
		//if !pgconn.Timeout(err) {
		return fmt.Errorf("failed to acquire database connection to listen for queued emails: %w", err)
	}
	defer conn.Release()

	_, err = conn.Exec(ctx, "LISTEN new_job")
	if err != nil {
		//if !pgconn.Timeout(err) {
		return fmt.Errorf("failed to listen to outbound_email_queued: %w", err)
	}

	for {
		notification, err := conn.Conn().WaitForNotification(ctx)
		if err != nil {
			//if !pgconn.Timeout(err) {
			return fmt.Errorf("failed while waiting for notification of outbound_email_queued")
		}

		job_id, err := strconv.Atoi(notification.Payload)
		if err != nil {
			return fmt.Errorf("failed to parse int from payload '%s': %w", notification.Payload, err)
		}

		c := bobpgx.NewConn(conn.Conn())
		job, err := models.FindJob(ctx, c, int32(job_id))
		if err != nil {
			return fmt.Errorf("Failed to find job %d: %w", job_id, err)
		}

		//tx, err := c.BeginTx(ctx, pgx.TxOptions{})
		tx, err := conn.BeginTx(ctx, pgx.TxOptions{})
		if err != nil {
			return fmt.Errorf("Failed to start transaction: %w", err)
		}
		ctx, cancel := context.WithCancel(ctx)
		txn := bobpgx.NewTx(tx, cancel)

		sublog := log.With().Int32("job", job.ID).Int32("row_id", job.RowID).Logger()
		err = handleJob(ctx, txn, job)
		if err != nil {
			sublog.Error().Err(err).Msg("failed to handle job")
			txn.Rollback(ctx)
			return nil
		}
		err = job.Delete(ctx, txn)
		if err != nil {
			sublog.Error().Err(err).Msg("failed to delete job")
			txn.Rollback(ctx)
			return fmt.Errorf("delete job: %w", err)
		}
		txn.Commit(ctx)
	}
}