add context propagation
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package meteo
|
||||
|
||||
import (
|
||||
"context"
|
||||
"io"
|
||||
"log/slog"
|
||||
)
|
||||
@@ -17,22 +18,22 @@ func NewService(repo Repository) *Service {
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Service) IngestCSV(r io.Reader, fs *FileStats) error {
|
||||
func (s *Service) IngestCSV(ctx context.Context, r io.Reader, fs *FileStats) error {
|
||||
accepted, rejected, err := s.csv.Parse(r)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// TODO call insertToDB
|
||||
s.insertAcceptedToDB(accepted, fs)
|
||||
s.insertRejectedToDB(rejected, fs)
|
||||
s.insertAcceptedToDB(ctx, accepted, fs)
|
||||
s.insertRejectedToDB(ctx, rejected, fs)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Service) insertAcceptedToDB(data []MeteoData, fs *FileStats) error {
|
||||
func (s *Service) insertAcceptedToDB(ctx context.Context, data []MeteoData, fs *FileStats) error {
|
||||
var err error
|
||||
fs.RowsInserted, err = s.repo.InsertAcceptedMeteoData(data)
|
||||
fs.RowsInserted, err = s.repo.InsertAcceptedMeteoData(ctx, data)
|
||||
if err != nil {
|
||||
slog.Error("error", "err", err)
|
||||
return err
|
||||
@@ -41,9 +42,9 @@ func (s *Service) insertAcceptedToDB(data []MeteoData, fs *FileStats) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Service) insertRejectedToDB(data []RejectedMeteoData, fs *FileStats) error {
|
||||
func (s *Service) insertRejectedToDB(ctx context.Context, data []RejectedMeteoData, fs *FileStats) error {
|
||||
var err error
|
||||
fs.RowsRejected, err = s.repo.InsertRejectedMeteoData(data)
|
||||
fs.RowsRejected, err = s.repo.InsertRejectedMeteoData(ctx, data)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user