Implement FileFetchScheduler and refactor Main class

This commit is contained in:
Guntis Smaukstelis
2023-05-20 21:53:03 +03:00
parent bbdfaa06d8
commit b4955e2619
3 changed files with 36 additions and 16 deletions
@@ -0,0 +1,29 @@
package fetch
import cats.effect.IO
import db.DBService
import fs2.Stream
import org.typelevel.log4cats.Logger
import org.typelevel.log4cats.slf4j.Slf4jLogger
object FileFetchScheduler {
def of(dbService: DBService): IO[FileFetchScheduler] = {
Slf4jLogger.create[IO].map {
new FileFetchScheduler(dbService, _)
}
}
}
class FileFetchScheduler(dbService: DBService, log: Logger[IO]) {
def run: Stream[IO, Unit] = {
val fetchTask = FileNameService.generateCurrentHour.flatMap(FetchService.fetchSingleFile)
Scheduler.scheduleTask(fetchTask)
.evalMap { case (name, content) =>
log.info(s"fetched: $name") *>
dbService.save(name, content).attempt.flatMap {
case Right(savedName) => log.info(s"saved: $savedName")
case Left(err) => log.error(s"error: $err")
}
}
}
}