delete from cache by fields

Author: bitops

src/__tests__/cache.test.js

@@ -111,8 +111,10 @@ describe('createCachingMethods', () => {
   it('adds the right methods', () => {
     expect(api.findOneById).toBeDefined()
     expect(api.findManyByIds).toBeDefined()
-    expect(api.findByFields).toBeDefined()
     expect(api.deleteFromCacheById).toBeDefined()
+
+    expect(api.findByFields).toBeDefined()
+    expect(api.deleteFromCacheByFields).toBeDefined()
   })
 
   it('finds one with ObjectId', async () => {
@@ -249,7 +251,7 @@ describe('createCachingMethods', () => {
     expect(value).toBeUndefined()
   })
 
-  it(`deletes from cache`, async () => {
+  it(`deletes from cache by ID`, async () => {
     for (const doc of [docs.one, docs.two, stringDoc]) {
       await api.findOneById(doc._id, { ttl: 1 })
 
@@ -263,7 +265,7 @@ describe('createCachingMethods', () => {
     }
   })
 
-  it('deletes from DataLoader cache', async () => {
+  it('deletes from DataLoader cache by ID', async () => {
     for (const id of [docs.one._id, docs.two._id, stringDoc._id]) {
       await api.findOneById(id)
       expect(collection.find).toHaveBeenCalled()
@@ -277,6 +279,19 @@ describe('createCachingMethods', () => {
       expect(collection.find).toHaveBeenCalled()
     }
   })
+
+  it(`deletes from cache by fields`, async () => {
+    const fields = { foo: 'bar' }
+    await api.findByFields(fields, { ttl: 1 })
+
+    const valueBefore = await cache.get(cacheKeyByFields(fields))
+    expect(valueBefore).toEqual(EJSON.stringify([docs.one, docs.two]))
+
+    await api.deleteFromCacheByFields(fields)
+
+    const valueAfter = await cache.get(cacheKeyByFields(fields))
+    expect(valueAfter).toBeUndefined()
+  })
 })
 
 describe('isValidObjectIdString', () => {

src/__tests__/datasource.test.js

@@ -18,6 +18,9 @@ describe('MongoDataSource', () => {
     const source = new Users(users)
     source.initialize()
     expect(source.findOneById).toBeDefined()
+    expect(source.findByFields).toBeDefined()
+    expect(source.deleteFromCacheById).toBeDefined()
+    expect(source.deleteFromCacheByFields).toBeDefined()
     expect(source.collection).toEqual(users)
   })
 })

src/cache.js

@@ -163,6 +163,25 @@ export const createCachingMethods = ({ collection, model, cache }) => {
     deleteFromCacheById: async id => {
       loader.clear(JSON.stringify({ id }))
       await cache.delete(cachePrefix + idToString(id))
+    },
+    deleteFromCacheByFields: async fields => {
+      const cleanedFields = {}
+
+      Object.keys(fields)
+        .sort()
+        .forEach(key => {
+          if (typeof key !== 'undefined') {
+            cleanedFields[key] = Array.isArray(fields[key])
+              ? fields[key]
+              : [fields[key]]
+          }
+        })
+
+      const loaderJSON = JSON.stringify(cleanedFields)
+
+      const key = cachePrefix + loaderJSON
+      loader.clear(loaderJSON)
+      await cache.delete(key)
     }
   }