transcriber/internal/repo/sqlite/transcript_job_sqlite.go

package sqlite

import (
	"database/sql"
	"fmt"
	"log"
	"time"

	"git.vakhrushev.me/av/transcriber/internal/entity"
	"github.com/doug-martin/goqu/v9"
)

type TranscriptJobRepository struct {
	db *sql.DB
	gq *goqu.Database
}

func NewTranscriptJobRepository(db *sql.DB, gq *goqu.Database) *TranscriptJobRepository {
	return &TranscriptJobRepository{db, gq}
}

func (repo *TranscriptJobRepository) Create(job *entity.TranscribeJob) error {
	record := goqu.Record{
		"id":          job.Id,
		"state":       job.State,
		"file_id":     job.FileID,
		"is_error":    job.IsError,
		"error_text":  job.ErrorText,
		"worker":      job.Worker,
		"acquired_at": job.AcquiredAt,
		"created_at":  job.CreatedAt,
	}
	query := repo.gq.Insert("transcribe_jobs").Rows(record)
	sql, args, err := query.ToSQL()
	if err != nil {
		return fmt.Errorf("failed to build query: %w", err)
	}

	_, err = repo.db.Exec(sql, args...)
	if err != nil {
		return fmt.Errorf("failed to insert transcribe job: %w", err)
	}

	return nil
}

func (repo *TranscriptJobRepository) Save(job *entity.TranscribeJob) error {
	record := goqu.Record{
		"state":       job.State,
		"file_id":     job.FileID,
		"is_error":    job.IsError,
		"error_text":  job.ErrorText,
		"worker":      job.Worker,
		"acquired_at": job.AcquiredAt,
	}
	query := repo.gq.Update("transcribe_jobs").Set(record).Where(goqu.C("id").Eq(job.Id))
	sql, args, err := query.ToSQL()
	if err != nil {
		return fmt.Errorf("failed to build query: %w", err)
	}

	_, err = repo.db.Exec(sql, args...)
	if err != nil {
		return fmt.Errorf("failed to update transcribe job: %w", err)
	}

	return nil
}

func (repo *TranscriptJobRepository) GetByID(id string) (*entity.TranscribeJob, error) {
	query := repo.gq.From("transcribe_jobs").Select(
		"id",
		"state",
		"file_id",
		"is_error",
		"error_text",
		"worker",
		"acquired_at",
		"created_at",
	).Where(goqu.C("id").Eq(id))
	sql, args, err := query.ToSQL()
	if err != nil {
		return nil, fmt.Errorf("failed to build query: %w", err)
	}

	var job entity.TranscribeJob
	err = repo.db.QueryRow(sql, args...).Scan(
		&job.Id,
		&job.State,
		&job.FileID,
		&job.IsError,
		&job.ErrorText,
		&job.Worker,
		&job.AcquiredAt,
		&job.CreatedAt,
	)
	if err != nil {
		return nil, fmt.Errorf("failed to get transcribe job: %w", err)
	}

	return &job, nil
}

func (repo *TranscriptJobRepository) FindAndAcquire(state, acquisitionId string, rottingTime time.Time) (*entity.TranscribeJob, error) {
	updateQuery := repo.gq.Update("transcribe_jobs").
		Set(
			goqu.Record{
				"worker":      acquisitionId,
				"acquired_at": time.Now(),
			},
		).
		Where(
			goqu.C("id").Eq(
				repo.gq.From("transcribe_jobs").Select("id").
					Where(
						goqu.And(
							goqu.C("state").Eq(state),
							goqu.C("is_error").Eq(0),
							goqu.Or(
								goqu.C("worker").IsNull(),
								goqu.C("acquired_at").Lt(rottingTime),
							),
						),
					).
					Limit(1),
			),
		)

	sql, args, err := updateQuery.ToSQL()
	if err != nil {
		return nil, fmt.Errorf("failed to build query: %w", err)
	}

	log.Printf("aquire sql: %s", sql)

	result, err := repo.db.Exec(sql, args...)
	if err != nil {
		return nil, fmt.Errorf("failed to aquire job with state %s: %w", state, err)
	}
	rowsAffected, err := result.RowsAffected()
	if err != nil {
		return nil, fmt.Errorf("failed check affected rows: %w", err)
	}
	if rowsAffected != 1 {
		return nil, fmt.Errorf("unexpected affected rows count: %d", rowsAffected)
	}

	selectQuery := repo.gq.From("transcribe_jobs").Select(
		"id",
		"state",
		"file_id",
		"is_error",
		"error_text",
		"worker",
		"acquired_at",
		"created_at",
	).Where(goqu.C("worker").Eq(acquisitionId))

	sql, args, err = selectQuery.ToSQL()
	if err != nil {
		return nil, fmt.Errorf("failed to build query: %w", err)
	}

	var job entity.TranscribeJob
	err = repo.db.QueryRow(sql, args...).Scan(
		&job.Id,
		&job.State,
		&job.FileID,
		&job.IsError,
		&job.ErrorText,
		&job.Worker,
		&job.AcquiredAt,
		&job.CreatedAt,
	)
	if err != nil {
		return nil, fmt.Errorf("failed to get transcribe job: %w", err)
	}

	return &job, nil
}