Dependency inject logger and replace println
This commit is contained in:
@@ -20,11 +20,20 @@ import io.circe.syntax._
|
||||
import parse.Aggregate.AggregateValueImplicits.aggregateValueEncoder
|
||||
import parse.Aggregate.{AggregateKey, UserQuery}
|
||||
import org.http4s.circe.jsonEncoder
|
||||
|
||||
import org.typelevel.log4cats.Logger
|
||||
import org.typelevel.log4cats.slf4j.Slf4jLogger
|
||||
import scala.concurrent.duration.DurationInt
|
||||
|
||||
|
||||
object Server {
|
||||
def of(dbService: DBService, fetch: FetchService): IO[Server] = {
|
||||
Slf4jLogger.create[IO].map {
|
||||
new Server(dbService, fetch, _)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class Server(dbService: DBService, fetch: FetchService, log: Logger[IO]) {
|
||||
|
||||
// Define the extension method `pretty` for Json
|
||||
implicit class JsonPrettyPrinter(json: Json) {
|
||||
@@ -38,25 +47,24 @@ object Server {
|
||||
|
||||
// http://0.0.0.0:8080/api/query/20230414_2200-20230501_1230/Liepāja,Rēzekne/tempMax/max
|
||||
case GET -> Root / "query" / DateTimeRange(from, to) / CityList(cities) / field / AggKey(key) =>
|
||||
DBService.of.flatMap(_.getInRange(from, to)
|
||||
dbService.getInRange(from, to)
|
||||
.map(Parser.queryData(UserQuery(cities, field, key), _))
|
||||
.flatMap(result => Ok(result.asJson.pretty)))
|
||||
.flatMap(result => Ok(result.asJson.pretty))
|
||||
|
||||
// http://0.0.0.0:8080/api/fetch/date/20230514
|
||||
case GET -> Root / "fetch" / "date" / ValidDate(date) =>
|
||||
val result = for {
|
||||
fetchResultEither <- FetchService.fetchFromDate(date).attempt
|
||||
fetchResultEither <- fetch.fetchFromDate(date).attempt
|
||||
fetchServiceError = fetchResultEither.left.toOption.map(e => s"FetchServiceError: ${e.getMessage}").toList
|
||||
fetchResult = fetchResultEither.getOrElse(List.empty)
|
||||
(fetchErrors, successDownloads) = fetchResult.partitionMap(identity)
|
||||
dbService <- DBService.of
|
||||
saveResults <- successDownloads.traverse { case (name, content) => dbService.save(name, content) }
|
||||
(saveErrors, successSaves) = saveResults.partitionMap(identity)
|
||||
// successes = successDownloads.map(s => s"fetched: ${s._1}") ++ successSaves.map(s => s"saved: $s")
|
||||
successes = successSaves
|
||||
errors = fetchServiceError ++ fetchErrors.map(e => s"FetchError: ${e.getMessage}") ++ saveErrors.map(e => s"SaveError: ${e.getMessage}")
|
||||
_ <- IO.println(s"errors: $errors")
|
||||
_ <- IO.println(s"successes: $successes")
|
||||
_ <- log.error(s"errors: $errors")
|
||||
_ <- log.info(s"successes: $successes")
|
||||
} yield (successes, errors)
|
||||
|
||||
result.flatMap { case (successes, errors) =>
|
||||
@@ -68,19 +76,19 @@ object Server {
|
||||
|
||||
// http://0.0.0.0:8080/api/show/all_dates
|
||||
case GET -> Root / "show" / "all_dates" =>
|
||||
DBService.of.flatMap(_.getDates().flatMap(dates =>
|
||||
dbService.getDates().flatMap(dates =>
|
||||
Ok(dates.asJson.pretty)
|
||||
))
|
||||
)
|
||||
|
||||
// http://0.0.0.0:8080/api/show/date/20230423
|
||||
case GET -> Root / "show" / "date" / ValidDate(date) =>
|
||||
DBService.of.flatMap(_.getDateFileNames(date).flatMap(fileNames =>
|
||||
dbService.getDateFileNames(date).flatMap(fileNames =>
|
||||
Ok(fileNames.asJson.pretty)
|
||||
))
|
||||
)
|
||||
|
||||
// http://0.0.0.0:8080/api/show/file/20230423_12:30.csv
|
||||
case GET -> Root / "show" / "file" / (fileName: String) =>
|
||||
DBService.of.flatMap(_.readFile(fileName).flatMap(content => Ok(content.asJson)))
|
||||
dbService.readFile(fileName).flatMap(content => Ok(content.asJson))
|
||||
|
||||
// http://0.0.0.0:8080/api/help
|
||||
case GET -> Root / "help" => {
|
||||
@@ -98,14 +106,14 @@ object Server {
|
||||
}
|
||||
}
|
||||
|
||||
val corsConfig = CORSConfig.default
|
||||
private val corsConfig = CORSConfig.default
|
||||
.withAnyOrigin(true)
|
||||
.withAnyMethod(true)
|
||||
.withAllowedMethods(Some(Set(Method.GET, Method.POST)))
|
||||
.withAllowCredentials(false)
|
||||
.withMaxAge(1.day)
|
||||
|
||||
val apiRoutesCors = CORS(apiRoutes, corsConfig)
|
||||
private val apiRoutesCors = CORS(apiRoutes, corsConfig)
|
||||
|
||||
private val httpApp = Router(
|
||||
"/" -> staticcontent.fileService[IO](FileService.Config("./web/dist")),
|
||||
|
||||
Reference in New Issue
Block a user