-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add ETag-caching (and AWS SDK v2) support
This change adds these improvements: * Facia data is only re-downloaded & re-parsed if the S3 content has _changed_, thanks to ETag-caching - see https://github.com/guardian/etag-caching . This library has already been used in DotCom PROD with guardian/frontend#26338 * AWS SDK v2: the FAPI client itself now has a `fapi-s3-sdk-v2` artifact. An example PR consuming this updated version of the FAPI client is at: guardian/ophan#5506 Updated FAPI artifact layout ---------------------------- To use FAPI with the new AWS SDK v2 support, users must now have a dependency on *two* FAPI artifacts: * `fapi-s3-sdk-v2` * `fapi-client-playXX` Due to needing to support the matrix of: * AWS SDK v1 & v2 * Play-JSON 2.8, 2.9, 3.0 ...it's best not to try to produce an artifact that corresponds to every single combination of those! Consequently, we provide an artifacts that are specific to the different versions of AWS SDK (or at least, could do - if AWS SDK v1 was moved out of common code), and artifacts that are specific to the different versions of Play-JSON, and allow the user to combine them as needed. A similar approach was used with `guardian/play-secret-rotation`: guardian/play-secret-rotation#8 In order for the different artifacts to have interfaces they can use to join together and become a single useful Facia client, we have a `fapi-client-core` artifact. Any code that doesn't depend on the JSON classes, or the actual AWS SDK version (which isn't much!), can live in there. In particular, we have: * `com.gu.facia.client.ApiClient`, an existing type that is now a trait, with 2 implementations - one that uses the existing `com.gu.facia.client.S3Client` abstraction on S3 behaviour * `com.gu.facia.client.etagcaching.fetching.S3FetchBehaviour`, a new trait that exposes just enough interface to allow the conditional fetching used for ETag-based caching, but doesn't tie you to any specific version of the AWS SDK.
- Loading branch information
Showing
13 changed files
with
204 additions
and
76 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
11 changes: 3 additions & 8 deletions
11
.../scala/com/gu/facia/client/S3Client.scala → ...m/gu/facia/client/AmazonSdkS3Client.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
91 changes: 68 additions & 23 deletions
91
facia-json/src/main/scala/com/gu/facia/client/ApiClient.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,37 +1,82 @@ | ||
package com.gu.facia.client | ||
|
||
import com.gu.facia.client.models.{ConfigJson, CollectionJson} | ||
import com.gu.etagcaching.FreshnessPolicy.AlwaysWaitForRefreshedValue | ||
import com.gu.etagcaching.aws.s3.{ObjectId, S3ByteArrayFetching} | ||
import com.gu.etagcaching.fetching.Fetching | ||
import com.gu.etagcaching.{ConfigCache, ETagCache} | ||
import com.gu.facia.client.models.{CollectionJson, ConfigJson} | ||
import play.api.libs.json.{Format, Json} | ||
|
||
import scala.concurrent.{ExecutionContext, Future} | ||
|
||
object ApiClient { | ||
val Encoding = "utf-8" | ||
trait ApiClient { | ||
def config: Future[ConfigJson] | ||
|
||
def collection(id: String): Future[Option[CollectionJson]] | ||
} | ||
|
||
case class ApiClient( | ||
object ApiClient { | ||
private val Encoding = "utf-8" | ||
|
||
/** | ||
* Legacy constructor for creating a client that does not support caching. Use `ApiClient.withCaching()` instead. | ||
*/ | ||
def apply( | ||
bucket: String, | ||
/** e.g., CODE, PROD, DEV ... */ | ||
environment: String, | ||
environment: String, // e.g., CODE, PROD, DEV ... | ||
s3Client: S3Client | ||
)(implicit executionContext: ExecutionContext) { | ||
import com.gu.facia.client.ApiClient._ | ||
|
||
private def retrieve[A: Format](key: String): Future[Option[A]] = s3Client.get(bucket, key) map { | ||
case FaciaSuccess(bytes) => | ||
Some(Json.fromJson[A](Json.parse(new String(bytes, Encoding))) getOrElse { | ||
throw new JsonDeserialisationError(s"Could not deserialize JSON in $bucket, $key") | ||
}) | ||
case FaciaNotAuthorized(message) => throw new BackendError(message) | ||
case FaciaNotFound(_) => None | ||
case FaciaUnknownError(message) => throw new BackendError(message) | ||
)(implicit executionContext: ExecutionContext): ApiClient = new ApiClient { | ||
val env: Environment = Environment(environment) | ||
|
||
private def retrieve[A: Format](key: String): Future[Option[A]] = s3Client.get(bucket, key).map(translateFaciaResult[A](_)) | ||
|
||
def config: Future[ConfigJson] = | ||
retrieve[ConfigJson](env.configS3Path).map(getOrWarnAboutMissingConfig) | ||
|
||
def collection(id: String): Future[Option[CollectionJson]] = | ||
retrieve[CollectionJson](env.collectionS3Path(id)) | ||
} | ||
|
||
private def getOrWarnAboutMissingConfig(configOpt: Option[ConfigJson]): ConfigJson = | ||
configOpt.getOrElse(throw BackendError("Config was missing!! NOT GOOD!")) | ||
|
||
private def translateFaciaResult[B: Format](faciaResult: FaciaResult): Option[B] = faciaResult match { | ||
case FaciaSuccess(bytes) => Some(parseBytes(bytes)) | ||
case FaciaNotAuthorized(message) => throw BackendError(message) | ||
case FaciaNotFound(_) => None | ||
case FaciaUnknownError(message) => throw BackendError(message) | ||
} | ||
|
||
def config: Future[ConfigJson] = | ||
retrieve[ConfigJson](s"$environment/frontsapi/config/config.json").map(_ getOrElse { | ||
throw new BackendError("Config was missing!! OH MY GOD") | ||
}) | ||
private def parseBytes[B: Format](bytes: Array[Byte]): B = | ||
Json.fromJson[B](Json.parse(new String(bytes, Encoding))) getOrElse { | ||
throw JsonDeserialisationError(s"Could not deserialize JSON") | ||
} | ||
|
||
/** | ||
* @param s3Fetching see scaladoc on `S3ByteArrayFetching` i.e. use `S3ObjectFetching.byteArraysWith(s3AsyncClient)` | ||
*/ | ||
def withCaching( | ||
bucket: String, | ||
environment: Environment, | ||
s3Fetching: S3ByteArrayFetching, | ||
configureCollectionCache: ConfigCache = _.maximumSize(10000) // at most 1GB RAM worst case | ||
)(implicit ec: ExecutionContext): ApiClient = new ApiClient { | ||
private val fetching = | ||
s3Fetching.keyOn[String](path => ObjectId(bucket, path)) | ||
|
||
def eTagCache[B: Format](configureCache: ConfigCache) = new ETagCache( | ||
fetching.thenParsing(parseBytes[B]), | ||
AlwaysWaitForRefreshedValue, | ||
configureCache | ||
) | ||
|
||
private val configCache = eTagCache[ConfigJson](_.maximumSize(1)) | ||
private val collectionCache = eTagCache[CollectionJson](configureCollectionCache) | ||
|
||
def collection(id: String): Future[Option[CollectionJson]] = | ||
retrieve[CollectionJson](s"$environment/frontsapi/collection/$id/collection.json") | ||
override def config: Future[ConfigJson] = | ||
configCache.get(environment.configS3Path).map(getOrWarnAboutMissingConfig) | ||
|
||
override def collection(id: String): Future[Option[CollectionJson]] = | ||
collectionCache.get(environment.collectionS3Path(id)) | ||
} | ||
} |
47 changes: 36 additions & 11 deletions
47
facia-json/src/test/scala/com/gu/facia/client/ApiClientSpec.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,33 +1,58 @@ | ||
package com.gu.facia.client | ||
|
||
import com.gu.etagcaching.aws.s3.ObjectId | ||
import com.gu.etagcaching.fetching.{ETaggedData, Fetching, Missing, MissingOrETagged} | ||
import com.gu.facia.client.lib.ResourcesHelper | ||
import org.scalatest.OptionValues | ||
import org.scalatest.concurrent.{IntegrationPatience, ScalaFutures} | ||
import org.scalatest.flatspec.AnyFlatSpec | ||
import org.scalatest.matchers.should.Matchers | ||
|
||
import scala.concurrent.ExecutionContext.Implicits.global | ||
import scala.concurrent.Future | ||
import scala.concurrent.{ExecutionContext, Future} | ||
import scala.util.hashing.MurmurHash3 | ||
|
||
object FakeS3Fetching extends Fetching[ObjectId, Array[Byte]] with ResourcesHelper { | ||
private def pretendETagFor(bytes: Array[Byte]): String = MurmurHash3.bytesHash(bytes).toHexString | ||
|
||
override def fetch(objectId: ObjectId)(implicit ec: ExecutionContext): Future[MissingOrETagged[Array[Byte]]] = Future { | ||
slurpBytes(objectId.key).fold(Missing: MissingOrETagged[Array[Byte]]) { bytes => | ||
ETaggedData(pretendETagFor(bytes), bytes) | ||
} | ||
} | ||
|
||
override def fetchOnlyIfETagChanged(objectId: ObjectId, oldETag: String)(implicit ec: ExecutionContext): Future[Option[MissingOrETagged[Array[Byte]]]] = { | ||
fetch(objectId).map { | ||
case taggedData: ETaggedData[_] => | ||
Option.unless(oldETag == taggedData.eTag)(taggedData) // simulate a Not-Modified response, if there's no change in ETag | ||
case x => Some(x) | ||
} | ||
} | ||
} | ||
|
||
class ApiClientSpec extends AnyFlatSpec with Matchers with OptionValues with ScalaFutures with IntegrationPatience { | ||
import scala.concurrent.ExecutionContext.Implicits.global | ||
|
||
object FakeS3Client extends S3Client with ResourcesHelper { | ||
override def get(bucket: String, path: String): Future[FaciaResult] = Future { | ||
slurpOrDie(path) | ||
} | ||
} | ||
|
||
val client: ApiClient = ApiClient("not used", "DEV", FakeS3Client) | ||
val legacyClient: ApiClient = ApiClient("not used", "DEV", FakeS3Client) | ||
val cachingClient: ApiClient = ApiClient.withCaching("not used", Environment.Dev, FakeS3Fetching) | ||
|
||
"ApiClient" should "fetch the config" in { | ||
val config = client.config.futureValue | ||
for ((name, client) <- Map("legacy" -> legacyClient, "caching" -> cachingClient)) { | ||
s"$name ApiClient" should "fetch the config" in { | ||
val config = client.config.futureValue | ||
|
||
config.collections should have size 334 | ||
config.fronts should have size 79 | ||
} | ||
config.collections should have size 334 | ||
config.fronts should have size 79 | ||
} | ||
|
||
it should "fetch a collection" in { | ||
val collectionOpt = client.collection("2409-31b3-83df0-de5a").futureValue | ||
it should "fetch a collection" in { | ||
val collectionOpt = cachingClient.collection("2409-31b3-83df0-de5a").futureValue | ||
|
||
collectionOpt.value.live should have size 8 | ||
collectionOpt.value.live should have size 8 | ||
} | ||
} | ||
} |
16 changes: 5 additions & 11 deletions
16
facia-json/src/test/scala/com/gu/facia/client/lib/ApiTestClient.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,27 +1,21 @@ | ||
package com.gu.facia.client.lib | ||
|
||
import com.amazonaws.auth.{ AWSStaticCredentialsProvider, BasicAWSCredentials} | ||
import com.amazonaws.regions.Regions | ||
import com.amazonaws.services.s3.{AmazonS3, AmazonS3ClientBuilder} | ||
import com.gu.facia.client.models.{CollectionJson, ConfigJson} | ||
import com.gu.facia.client.{AmazonSdkS3Client, ApiClient} | ||
import com.gu.facia.client.{ApiClient, Environment} | ||
import play.api.libs.json.{Format, Json} | ||
|
||
import scala.concurrent.ExecutionContext.Implicits.global | ||
import scala.concurrent.Future | ||
|
||
object Amazon { | ||
val amazonS3Client = AmazonS3ClientBuilder.standard().withRegion(Regions.EU_WEST_1).withCredentials(new AWSStaticCredentialsProvider(new BasicAWSCredentials("key", "pass"))).build() | ||
} | ||
|
||
class ApiTestClient extends ApiClient("bucket", "DEV", AmazonSdkS3Client(Amazon.amazonS3Client)) with ResourcesHelper { | ||
class ApiTestClient extends ApiClient with ResourcesHelper { | ||
private val environment = Environment.Dev | ||
|
||
private def retrieve[A: Format](key: String): Future[Option[A]] = | ||
Future.successful(slurp(key).map(Json.parse).flatMap(_.asOpt[A])) | ||
|
||
override def config: Future[ConfigJson] = | ||
retrieve[ConfigJson](s"$environment/frontsapi/config/config.json").map(_.get) | ||
retrieve[ConfigJson](environment.configS3Path).map(_.get) | ||
|
||
override def collection(id: String): Future[Option[CollectionJson]] = | ||
retrieve[CollectionJson](s"$environment/frontsapi/collection/$id/collection.json") | ||
retrieve[CollectionJson](environment.collectionS3Path(id)) | ||
} |
10 changes: 8 additions & 2 deletions
10
facia-json/src/test/scala/com/gu/facia/client/lib/ResourcesHelper.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,11 +1,17 @@ | ||
package com.gu.facia.client.lib | ||
|
||
import com.gu.facia.client.FaciaSuccess | ||
import org.apache.commons.io.IOUtils | ||
|
||
import java.nio.charset.StandardCharsets.UTF_8 | ||
|
||
trait ResourcesHelper { | ||
def slurpBytes(path: String): Option[Array[Byte]] = | ||
Option(getClass.getClassLoader.getResource(path)).map(url => IOUtils.toByteArray(url.openStream())) | ||
|
||
def slurp(path: String): Option[String] = | ||
Option(getClass.getClassLoader.getResource(path)).map(scala.io.Source.fromURL(_).mkString) | ||
slurpBytes(path).map(bytes => new String(bytes, UTF_8)) | ||
|
||
def slurpOrDie(path: String) = slurp(path).map(_.getBytes).map(FaciaSuccess.apply).getOrElse { | ||
def slurpOrDie(path: String) = slurpBytes(path).map(FaciaSuccess.apply).getOrElse { | ||
throw new RuntimeException(s"Required resource $path not on class path") } | ||
} |
15 changes: 15 additions & 0 deletions
15
fapi-client-core/src/main/scala/com/gu/facia/client/Environment.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,15 @@ | ||
package com.gu.facia.client | ||
|
||
case class Environment(name: String) { | ||
private val s3PathPrefix: String = s"$name/frontsapi" | ||
|
||
val configS3Path: String = s"$s3PathPrefix/config/config.json" | ||
def collectionS3Path(id: String): String = s"$s3PathPrefix/collection/$id/collection.json" | ||
} | ||
|
||
object Environment { | ||
val Prod = Environment("PROD") | ||
val Code = Environment("CODE") | ||
val Dev = Environment("DEV") | ||
val Test = Environment("TEST") | ||
} |
File renamed without changes.
File renamed without changes.
16 changes: 16 additions & 0 deletions
16
fapi-client-core/src/main/scala/com/gu/facia/client/S3Client.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,16 @@ | ||
package com.gu.facia.client | ||
|
||
import scala.concurrent.Future | ||
|
||
/** | ||
* Legacy class for mocking in tests, but also previously used to allow library users to define a properly | ||
* asynchronous S3 client (back when the one in the AWS SDK was synchronous only). | ||
* | ||
* Note that use of `S3Client` is now discouraged, as `facia-scala-client` now supports caching using the | ||
* `etag-caching` library, which provides its own more powerful abstraction for fetching & parsing data, and | ||
* `com.gu.etagcaching.aws.sdkv2.s3.S3ObjectFetching`, a ready-made implementation of fetching that includes | ||
* support for ETags & Conditional Requests, decoding S3 exceptions, etc. | ||
*/ | ||
trait S3Client { | ||
def get(bucket: String, path: String): Future[FaciaResult] | ||
} |
16 changes: 16 additions & 0 deletions
16
fapi-client-core/src/main/scala/com/gu/facia/client/package.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,16 @@ | ||
package com.gu.facia | ||
|
||
import com.gu.etagcaching.aws.s3.ObjectId | ||
import com.gu.etagcaching.fetching.Fetching | ||
|
||
package object client { | ||
/** | ||
* AWS SDK v2 | ||
* | ||
* Add "com.gu.etag-caching" %% "aws-s3-sdk-v2" as dependency, | ||
* import com.gu.etagcaching.aws.sdkv2.s3.S3ObjectFetching | ||
* | ||
* S3ObjectFetching.byteArrayWith(s3AsyncClient) | ||
*/ | ||
type S3FetchBehaviour = Fetching[ObjectId, Array[Byte]] | ||
} |
Oops, something went wrong.