Refactor project structure

internal/adapter/sqlite/file_repo.go

@@ -0,0 +1,56 @@
+package sqlite
+
+import (
+	"database/sql"
+	"fmt"
+
+	"git.vakhrushev.me/av/transcriber/internal/entity"
+	"github.com/doug-martin/goqu/v9"
+)
+
+type FileRepository struct {
+	db *sql.DB
+	gq *goqu.Database
+}
+
+func NewFileRepository(conn *sql.DB, gq *goqu.Database) *FileRepository {
+	return &FileRepository{conn, gq}
+}
+
+func (repo *FileRepository) Create(file *entity.File) error {
+	record := goqu.Record{
+		"id":         file.Id,
+		"storage":    file.Storage,
+		"file_name":  file.FileName,
+		"size":       file.Size,
+		"created_at": file.CreatedAt,
+	}
+	query := repo.gq.Insert("files").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 file: %w", err)
+	}
+
+	return nil
+}
+
+func (repo *FileRepository) GetByID(id string) (*entity.File, error) {
+	query := repo.gq.From("files").Select("id", "storage", "file_name", "size", "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 file entity.File
+	err = repo.db.QueryRow(sql, args...).Scan(&file.Id, &file.Storage, &file.FileName, &file.Size, &file.CreatedAt)
+	if err != nil {
+		return nil, fmt.Errorf("failed to get file: %w", err)
+	}
+
+	return &file, nil
+}

internal/adapter/sqlite/transcript_job_repo.go

@@ -0,0 +1,212 @@
+package sqlite
+
+import (
+	"database/sql"
+	"fmt"
+	"time"
+
+	"git.vakhrushev.me/av/transcriber/internal/contract"
+	"git.vakhrushev.me/av/transcriber/internal/entity"
+	goqu "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,
+		"acquisition_id":     job.AcquisitionID,
+		"acquire_time":       job.AcquireTime,
+		"delay_time":         job.DelayTime,
+		"recognition_op_id":  job.RecognitionOpID,
+		"transcription_text": job.TranscriptionText,
+		"created_at":         job.CreatedAt,
+		"updated_at":         job.UpdatedAt,
+	}
+	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,
+		"acquisition_id":     job.AcquisitionID,
+		"acquire_time":       job.AcquireTime,
+		"delay_time":         job.DelayTime,
+		"recognition_op_id":  job.RecognitionOpID,
+		"transcription_text": job.TranscriptionText,
+		"updated_at":         job.UpdatedAt,
+	}
+	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",
+		"acquisition_id",
+		"acquire_time",
+		"delay_time",
+		"recognition_op_id",
+		"transcription_text",
+		"created_at",
+		"updated_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.AcquisitionID,
+		&job.AcquireTime,
+		&job.DelayTime,
+		&job.RecognitionOpID,
+		&job.TranscriptionText,
+		&job.CreatedAt,
+		&job.UpdatedAt,
+	)
+	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{
+				"acquisition_id": acquisitionId,
+				"acquire_time":   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("delay_time").IsNull(),
+								goqu.C("delay_time").Lt(time.Now()),
+							),
+							goqu.Or(
+								goqu.C("acquisition_id").IsNull(),
+								goqu.C("acquire_time").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 == 0 {
+		e := contract.JobNotFoundError{State: state, Message: "appropriate job not found"}
+		return nil, &e
+	}
+	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",
+		"acquisition_id",
+		"acquire_time",
+		"delay_time",
+		"recognition_op_id",
+		"transcription_text",
+		"created_at",
+		"updated_at",
+	).Where(goqu.C("acquisition_id").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.AcquisitionID,
+		&job.AcquireTime,
+		&job.DelayTime,
+		&job.RecognitionOpID,
+		&job.TranscriptionText,
+		&job.CreatedAt,
+		&job.UpdatedAt,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("failed to get transcribe job: %w", err)
+	}
+
+	return &job, nil
+}