Add prometheus metrics

internal/service/transcribe.go

@@ -6,14 +6,21 @@ import (
 	"log"
 	"os"
 	"path/filepath"
+	"strconv"
+	"strings"
 	"time"
 
 	"git.vakhrushev.me/av/transcriber/internal/contract"
 	"git.vakhrushev.me/av/transcriber/internal/entity"
+	"git.vakhrushev.me/av/transcriber/internal/metrics"
 	"github.com/google/uuid"
 )
 
-const baseStorageDir = "data/files"
+const (
+	baseStorageDir = "data/files"
+
+	defaultAudioExt = "audio"
+)
 
 type TranscribeService struct {
 	jobRepo    contract.TranscriptJobRepository
@@ -43,7 +50,7 @@ func (s *TranscribeService) CreateTranscribeJob(file io.Reader, fileName string)
 	// Определяем расширение файла
 	ext := filepath.Ext(fileName)
 	if ext == "" {
-		ext = ".audio" // fallback если расширение не определено
+		ext = fmt.Sprintf(".%s", defaultAudioExt) // fallback если расширение не определено
 	}
 
 	// Создаем путь для сохранения файла
@@ -63,6 +70,12 @@ func (s *TranscribeService) CreateTranscribeJob(file io.Reader, fileName string)
 		return nil, err
 	}
 
+	if err := dst.Close(); err != nil {
+		return nil, err
+	}
+
+	metrics.InputFileSizeHistogram.WithLabelValues(ext).Observe(float64(size))
+
 	// Создаем запись в таблице files
 	fileRecord := &entity.File{
 		Id:        fileId,
@@ -121,7 +134,22 @@ func (s *TranscribeService) FindAndRunConversionJob() error {
 	destFileName := fmt.Sprintf("%s%s", destFileId, ".ogg")
 	destFilePath := filepath.Join(baseStorageDir, destFileName)
 
+	// Получаем расширение исходного файла для метрики
+	srcExt := strings.TrimPrefix(filepath.Ext(srcFile.FileName), ".")
+	if srcExt == "" {
+		srcExt = defaultAudioExt
+	}
+
+	// Измеряем время конвертации
+	startTime := time.Now()
 	err = s.converter.Convert(srcFilePath, destFilePath)
+	conversionDuration := time.Since(startTime)
+
+	// Записываем метрику времени конвертации
+	metrics.ConversionDurationHistogram.
+		WithLabelValues(srcExt, "ogg", strconv.FormatBool(err != nil)).
+		Observe(conversionDuration.Seconds())
+
 	if err != nil {
 		return err
 	}
@@ -131,6 +159,9 @@ func (s *TranscribeService) FindAndRunConversionJob() error {
 		return err
 	}
 
+	// Записываем метрику размера выходного файла
+	metrics.OutputFileSizeHistogram.WithLabelValues("ogg").Observe(float64(stat.Size()))
+
 	// Создаем запись в таблице files
 	destFileRecord := &entity.File{
 		Id:        destFileId,
@@ -218,7 +249,7 @@ func (s *TranscribeService) FindAndRunTranscribeCheckJob() error {
 		if _, ok := err.(*contract.JobNotFoundError); ok {
 			return nil
 		}
-		return err
+		return fmt.Errorf("failed find and acquire job: %s, %w", entity.StateTranscribe, err)
 	}
 
 	if job.RecognitionOpID == nil {