clearLocal, clearAllLocal

Author: Brian Detering

README.md

@@ -50,6 +50,7 @@ with a few differences. Read through the [Facebook Dataloader documentation](htt
 - `prime` will always overwrite the cache. Facebook Dataloader will only write to
 its cache if a value is not already present. Prime is asyncronous and returns a Promise.
 - dataloader results must be either `null` or a JSON object.
+- two functions: `clearLocal(key)` and `clearAllLocal()` allow you to clear the local cache only.
 
 ### Instantiation
 

index.js

@@ -137,6 +137,14 @@ module.exports = fig => {
                 rDel(this.keySpace, key).then(() => this.loader.clear(key)) :
                 Q.reject(new TypeError('key parameter is required'));
         }
+
+        clearAllLocal() {
+            return Q(this.loader.clearAll());
+        }
+
+        clearLocal(key) {
+            return Q(this.loader.clear(key));
+        }
     };
 };
 

test.js

@@ -14,6 +14,10 @@ describe('redis-dataloader', () => {
             key, (err, resp) => err ? reject(err) : resolve(resp)
         ));
 
+        this.rSet = (k, v) => Q.Promise((resolve, reject) => redis.set(
+            k, v, (err, resp) => err ? reject(err) : resolve(resp)
+        ));
+
         this.keySpace = 'key-space';
         this.data = {
             json: { foo: 'bar' },
@@ -253,4 +257,40 @@ describe('redis-dataloader', () => {
             }).done();
         });
     });
+
+    describe('clearAllLocal', () => {
+        it('should clear all local in-memory cache', done => {
+            this.loader.loadMany(['json', 'null'])
+            .then(() => this.loader.clearAllLocal())
+            .then(() => this.rSet(
+                `${this.keySpace}:json`, JSON.stringify({ new: 'valeo' })
+            ))
+            .then(() => this.rSet(
+                `${this.keySpace}:null`, JSON.stringify({ foo: 'bar' })
+            ))
+            .then(() => this.loader.loadMany(['null', 'json']))
+            .then(data => {
+                expect(data).to.deep.equal([{ foo: 'bar' }, { new: 'valeo' }]);
+                done();
+            }).done();
+        });
+    });
+
+    describe('clearLocal', () => {
+        it('should clear local cache for a specific key', done => {
+            this.loader.loadMany(['json', 'null'])
+            .then(() => this.loader.clearLocal('json'))
+            .then(() => this.rSet(
+                `${this.keySpace}:json`, JSON.stringify({ new: 'valeo' })
+            ))
+            .then(() => this.rSet(
+                `${this.keySpace}:null`, JSON.stringify({ foo: 'bar' })
+            ))
+            .then(() => this.loader.loadMany(['null', 'json']))
+            .then(data => {
+                expect(data).to.deep.equal([null, { new: 'valeo' }]);
+                done();
+            }).done();
+        });
+    });
 });