av
f494dcb83e
- контекст проложен от воркера и обоих входов до внешних вызовов: ffmpeg и ffprobe заводятся через exec.CommandContext, SpeechKit и Object Storage принимают ctx вместо context.Background, скачивание записи идёт запросом с контекстом. Прежде остановка сервиса не доходила до чужой работы вовсе - прерванный шаг приговора не выносит: убитый по контексту ffmpeg отдаёт «signal: killed», от настоящего отказа неотличимо ни типом, ни errors.Is, и различает их только ctx.Err(). Задача остаётся на повтор, попытку не тратит и отправителю о несуществующем сбое не сообщает; воркер не считает остановку отказом, а задача не забирается вовсе, если нас уже остановили - клиента Bot API заводит единая точка internal/adapter/telegram: токен стоит в пути каждого обращения, а http.Client кладёт адрес в *url.Error целиком. Чистка на месте употребления закрывала один вызов из пяти — теперь свой Do чистит отказ, подменённый логгер вычищает токен из строк самой библиотеки, а транспорт бота токена не получает вовсе - принятие операции распознавания защищено от отмены своим пределом: SpeechKit мог её принять и начать считать деньги, а потерянный идентификатор заставил бы повтор оплатить ту же запись второй раз - приём по HTTP доводит запись до задачи независимо от отправителя: на контексте запроса один обрыв соединения терял полностью загруженную запись - ответ Telegram с не-2xx кодом больше не становится записью: прежде тело отказа доезжало до хранилища и умирало на ffprobe, уводя диагностику
96 lines
2.8 KiB
Go
96 lines
2.8 KiB
Go
package yandex
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/aws/aws-sdk-go-v2/config"
|
|
"github.com/aws/aws-sdk-go-v2/credentials"
|
|
"github.com/aws/aws-sdk-go-v2/feature/s3/manager"
|
|
"github.com/aws/aws-sdk-go-v2/service/s3"
|
|
"github.com/aws/smithy-go"
|
|
)
|
|
|
|
type s3Config struct {
|
|
Region string
|
|
AccessKey string
|
|
SecretKey string
|
|
BucketName string
|
|
Endpoint string
|
|
}
|
|
|
|
type yandexS3Service struct {
|
|
client *s3.Client
|
|
uploader *manager.Uploader
|
|
bucketName string
|
|
endpoint string
|
|
}
|
|
|
|
func newYandexS3Service(cfg s3Config) (*yandexS3Service, error) {
|
|
if cfg.Region == "" || cfg.AccessKey == "" || cfg.SecretKey == "" || cfg.BucketName == "" {
|
|
return nil, fmt.Errorf("missing required S3 configuration parameters")
|
|
}
|
|
|
|
// Создаем конфигурацию
|
|
awsCfg, err := config.LoadDefaultConfig(context.Background(),
|
|
config.WithRegion(cfg.Region),
|
|
config.WithCredentialsProvider(credentials.NewStaticCredentialsProvider(cfg.AccessKey, cfg.SecretKey, "")),
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to load AWS config: %w", err)
|
|
}
|
|
|
|
// Создаем клиент S3
|
|
var client *s3.Client
|
|
if cfg.Endpoint != "" {
|
|
// Кастомный endpoint (например, для MinIO)
|
|
client = s3.NewFromConfig(awsCfg, func(o *s3.Options) {
|
|
o.BaseEndpoint = aws.String(cfg.Endpoint)
|
|
o.UsePathStyle = true
|
|
})
|
|
} else {
|
|
// Стандартный AWS S3
|
|
client = s3.NewFromConfig(awsCfg)
|
|
}
|
|
|
|
uploader := manager.NewUploader(client)
|
|
|
|
return &yandexS3Service{
|
|
client: client,
|
|
uploader: uploader,
|
|
bucketName: cfg.BucketName,
|
|
endpoint: cfg.Endpoint,
|
|
}, nil
|
|
}
|
|
|
|
func (s *yandexS3Service) uploadFile(ctx context.Context, file io.Reader, fileName string) error {
|
|
_, err := s.uploader.Upload(ctx, &s3.PutObjectInput{
|
|
Bucket: aws.String(s.bucketName),
|
|
Key: aws.String(fileName),
|
|
Body: file,
|
|
})
|
|
if err != nil {
|
|
// Отказ SDK несёт полный URL объекта, то есть имя файла в хранилище, а
|
|
// оно — последняя часть ссылки на скачивание: цепочка `%w` уехала бы в
|
|
// журнал вместе с ключом. Наружу идёт класс отказа и только он — по
|
|
// нему «ключи отозваны» отличимо от «бакета нет» и от «сети нет», а
|
|
// адреса в коде отказа SDK не бывает.
|
|
var apiErr smithy.APIError
|
|
if errors.As(err, &apiErr) {
|
|
return fmt.Errorf("failed to upload file to S3: %s", apiErr.ErrorCode())
|
|
}
|
|
return errors.New("failed to upload file to S3")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *yandexS3Service) fileUrl(fileName string) string {
|
|
endpoint := strings.TrimRight(s.endpoint, "/")
|
|
return fmt.Sprintf("%s/%s/%s", endpoint, s.bucketName, fileName)
|
|
}
|