cache key with context

Author: qzchenwl

src/main/java/org/dataloader/CacheKey.java

@@ -35,4 +35,8 @@ public interface CacheKey<K> {
      * @return the cache key
      */
     Object getKey(K input);
+
+    default Object getKeyWithContext(K input, Object context) {
+        return getKey(input);
+    }
 }

src/main/java/org/dataloader/DataLoaderHelper.java

@@ -104,7 +104,14 @@ CompletableFuture<V> load(K key, Object loadContext) {
             boolean batchingEnabled = loaderOptions.batchingEnabled();
             boolean cachingEnabled = loaderOptions.cachingEnabled();
 
-            Object cacheKey = cachingEnabled ? getCacheKey(nonNull(key)) : null;
+            Object cacheKey = null;
+            if (cachingEnabled) {
+                if (loadContext == null) {
+                    cacheKey = getCacheKey(key);
+                } else {
+                    cacheKey = getCacheKeyWithContext(key, loadContext);
+                }
+            }
             stats.incrementLoadCount();
 
             if (cachingEnabled) {
@@ -135,6 +142,12 @@ Object getCacheKey(K key) {
                 loaderOptions.cacheKeyFunction().get().getKey(key) : key;
     }
 
+    @SuppressWarnings("unchecked")
+    Object getCacheKeyWithContext(K key, Object context) {
+        return loaderOptions.cacheKeyFunction().isPresent() ?
+                loaderOptions.cacheKeyFunction().get().getKeyWithContext(key, context): key;
+    }
+
     DispatchResult<V> dispatch() {
         boolean batchingEnabled = loaderOptions.batchingEnabled();
         //