Merge branch 'main' into felipecsl--cloud-eppo

Author: felipecsl

build.gradle

@@ -38,8 +38,9 @@ dependencies {
   implementation 'org.slf4j:slf4j-api:2.0.13'
   // Logback classic 1.3.x is compatible with java 8
   implementation 'ch.qos.logback:logback-classic:1.3.14'
+  implementation 'org.jetbrains:annotations:13.0'
 
-  testImplementation 'cloud.eppo:sdk-common-jvm:3.5.0-SNAPSHOT:tests'
+  testImplementation 'cloud.eppo:sdk-common-jvm:3.5.0:tests'
   testImplementation platform('org.junit:junit-bom:5.10.2')
   testImplementation 'org.junit.jupiter:junit-jupiter'
   testImplementation 'com.github.tomakehurst:wiremock-jre8:2.35.2'

src/main/java/cloud/eppo/EppoClient.java

@@ -1,8 +1,11 @@
 package cloud.eppo;
 
+import cloud.eppo.api.IAssignmentCache;
 import cloud.eppo.logging.AssignmentLogger;
 import cloud.eppo.logging.BanditLogger;
 import java.util.Timer;
+import java.util.concurrent.TimeUnit;
+import org.jetbrains.annotations.Nullable;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -36,9 +39,11 @@ private EppoClient(
       String host,
       String sdkName,
       String sdkVersion,
-      AssignmentLogger assignmentLogger,
-      BanditLogger banditLogger,
-      boolean isGracefulModel) {
+      @Nullable AssignmentLogger assignmentLogger,
+      @Nullable BanditLogger banditLogger,
+      boolean isGracefulMode,
+      @Nullable IAssignmentCache assignmentCache,
+      @Nullable IAssignmentCache banditAssignmentCache) {
     super(
         apiKey,
         host,
@@ -47,13 +52,13 @@ private EppoClient(
         assignmentLogger,
         banditLogger,
         null,
-        isGracefulModel,
+        isGracefulMode,
         false,
         true,
         null,
-        null, // TODO assignmentCache,
-        null // TODO banditCache
-        );
+        assignmentCache,
+        banditAssignmentCache
+    );
   }
 
   /** Stops the client from polling Eppo for updated flag and bandit configurations */
@@ -73,6 +78,12 @@ public static class Builder {
     private long pollingIntervalMs = DEFAULT_POLLING_INTERVAL_MS;
     private String host = DEFAULT_HOST;
 
+    // Assignment and bandit caching on by default. To disable, call
+    // `builder.assignmentCache(null).banditAssignmentCache(null);`
+    private IAssignmentCache assignmentCache = new LRUInMemoryAssignmentCache(100);
+    private IAssignmentCache banditAssignmentCache =
+        new ExpiringInMemoryAssignmentCache(10, TimeUnit.MINUTES);
+
     /** Sets the API Key--created within the eppo application--to use. This is required. */
     public Builder apiKey(String apiKey) {
       this.apiKey = apiKey;
@@ -135,6 +146,16 @@ public Builder host(String host) {
       return this;
     }
 
+    public Builder assignmentCache(IAssignmentCache assignmentCache) {
+      this.assignmentCache = assignmentCache;
+      return this;
+    }
+
+    public Builder banditAssignmentCache(IAssignmentCache banditAssignmentCache) {
+      this.banditAssignmentCache = banditAssignmentCache;
+      return this;
+    }
+
     public EppoClient buildAndInit() {
       AppDetails appDetails = AppDetails.getInstance();
       String sdkName = appDetails.getName();
@@ -153,7 +174,15 @@ public EppoClient buildAndInit() {
 
       instance =
           new EppoClient(
-              apiKey, sdkName, sdkVersion, host, assignmentLogger, banditLogger, isGracefulMode);
+              apiKey,
+              sdkName,
+              sdkVersion,
+              host,
+              assignmentLogger,
+              banditLogger,
+              isGracefulMode,
+              assignmentCache,
+              banditAssignmentCache);
 
       // Stop any active polling
       stopPolling();