app/com/gu/aws/AwsS3Client.scala (33 lines of code) (raw):
package com.gu.aws
import com.amazonaws.regions.Regions
import com.amazonaws.services.s3.model.{GetObjectRequest, S3ObjectInputStream}
import com.amazonaws.services.s3.{AmazonS3, AmazonS3Client}
import com.gu.monitoring.SafeLogger
import com.gu.monitoring.SafeLogger._
import com.typesafe.scalalogging.StrictLogging
import play.api.libs.json.{JsValue, Json}
import scalaz.{-\/, \/, \/-}
import scala.io.Source
import scala.util.{Failure, Success, Try}
object AwsS3 extends StrictLogging {
lazy val client = AmazonS3Client.builder.withCredentials(CredentialsProvider).withRegion(Regions.EU_WEST_1).build()
def fetchObject(s3Client: AmazonS3, request: GetObjectRequest): Try[S3ObjectInputStream] = Try(
s3Client.getObject(request).getObjectContent,
)
def fetchJson(s3Client: AmazonS3, request: GetObjectRequest): String \/ JsValue = {
logger.info(s"Getting file from S3. Bucket: ${request.getBucketName} | Key: ${request.getKey}")
val attempt = for {
s3Stream <- fetchObject(s3Client, request)
json <- Try(Json.parse(Source.fromInputStream(s3Stream).mkString))
_ <- Try(s3Stream.close())
} yield json
attempt match {
case Success(json) =>
logger.info(s"Successfully loaded ${request.getKey} from ${request.getBucketName}")
\/-(json)
case Failure(ex) =>
SafeLogger.error(scrub"Failed to load JSON from S3 bucket ${request.getBucketName}", ex)
-\/(s"Failed to load JSON due to $ex")
}
}
}