bot/WeatherTool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor FileFetchScheduler

Browse Source
This commit is contained in:
Guntis Smaukstelis
2023-05-23 19:16:57 +03:00
parent 6f005c56d7
commit bd302618d3
1 changed files with 17 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/scala/fetch/FileFetchScheduler.scala
+5 -4
View File
@@ -9,16 +9,18 @@ import org.typelevel.log4cats.slf4j.Slf4jLogger
object FileFetchScheduler { object FileFetchScheduler {
def of(dbService: DBService, fetch: FetchService): IO[FileFetchScheduler] = { def of(dbService: DBService, fetch: FetchService): IO[FileFetchScheduler] = {
Scheduler.of.flatMap { scheduler =>
Slf4jLogger.create[IO].map { Slf4jLogger.create[IO].map {
new FileFetchScheduler(dbService, fetch, _) new FileFetchScheduler(dbService, fetch, scheduler, _)
}
} }
} }
} }
class FileFetchScheduler(dbService: DBService, fetch: FetchService, log: Logger[IO]) {
class FileFetchScheduler(dbService: DBService, fetch: FetchService, scheduler: Scheduler, log: Logger[IO]) {
def run: Stream[IO, Unit] = { def run: Stream[IO, Unit] = {
val fetchTask = FileNameService.generateCurrentHour.flatMap(fetch.fetchSingleFile) val fetchTask = FileNameService.generateCurrentHour.flatMap(fetch.fetchSingleFile)
Stream.eval(Scheduler.of).flatMap { scheduler =>
scheduler.scheduleTask(fetchTask) scheduler.scheduleTask(fetchTask)
.evalMap { .evalMap {
case Left(fetchErr) => case Left(fetchErr) =>
@@ -33,5 +35,4 @@ class FileFetchScheduler(dbService: DBService, fetch: FetchService, log: Logger[
} }
} }
} }
}
} }