mirror of
https://git.sr.ht/~phw/scotty
synced 2025-06-01 19:38:34 +02:00
3b545a0fd6
This will allow cancelling the export if the import fails before the export finished. For now the context isn't passed on to the actual export functions, hence there is not yet any cancellation happening.
59 lines
1.9 KiB
Go
59 lines
1.9 KiB
Go
/*
|
|
Copyright © 2023-2025 Philipp Wolfer <phw@uploadedlobster.com>
|
|
|
|
Scotty is free software: you can redistribute it and/or modify it under the
|
|
terms of the GNU General Public License as published by the Free Software
|
|
Foundation, either version 3 of the License, or (at your option) any later version.
|
|
|
|
Scotty is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
|
A PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along with
|
|
Scotty. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package backends
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"time"
|
|
|
|
"go.uploadedlobster.com/scotty/internal/models"
|
|
)
|
|
|
|
type ExportProcessor[T models.ListensResult | models.LovesResult] interface {
|
|
ExportBackend() models.Backend
|
|
Process(ctx context.Context, wg *sync.WaitGroup, oldestTimestamp time.Time, results chan T, progress chan models.TransferProgress)
|
|
}
|
|
|
|
type ListensExportProcessor struct {
|
|
Backend models.ListensExport
|
|
}
|
|
|
|
func (p ListensExportProcessor) ExportBackend() models.Backend {
|
|
return p.Backend
|
|
}
|
|
|
|
func (p ListensExportProcessor) Process(ctx context.Context, wg *sync.WaitGroup, oldestTimestamp time.Time, results chan models.ListensResult, progress chan models.TransferProgress) {
|
|
wg.Add(1)
|
|
defer wg.Done()
|
|
defer close(results)
|
|
p.Backend.ExportListens(oldestTimestamp, results, progress)
|
|
}
|
|
|
|
type LovesExportProcessor struct {
|
|
Backend models.LovesExport
|
|
}
|
|
|
|
func (p LovesExportProcessor) ExportBackend() models.Backend {
|
|
return p.Backend
|
|
}
|
|
|
|
func (p LovesExportProcessor) Process(ctx context.Context, wg *sync.WaitGroup, oldestTimestamp time.Time, results chan models.LovesResult, progress chan models.TransferProgress) {
|
|
wg.Add(1)
|
|
defer wg.Done()
|
|
defer close(results)
|
|
p.Backend.ExportLoves(oldestTimestamp, results, progress)
|
|
}
|