Implemented invalidateRecords() + Unit tests (#143)

* Implemented invalidateRecords() + tests

* Changed the way how entries are removed from cache

* Changed the way how sobject type is deferred to assume there is collection of single type passed in. Refactoring.

* Changed key in unit tests + removed static string key

* Changed unit test object to profile

Author: patrykacc

force-app/main/default/classes/main/cached-soql/SOQLCache.cls

@@ -14,6 +14,7 @@
 **/
 @SuppressWarnings('PMD.ExcessivePublicCount, PMD.ExcessiveClassLength, PMD.CyclomaticComplexity, PMD.CognitiveComplexity, PMD.PropertyNamingConventions, PMD.FieldDeclarationsShouldBeAtStart, PMD.ApexDoc, PMD.ExcessiveParameterList')
 public virtual inherited sharing class SOQLCache implements Cacheable {
+
     public interface Selector {
         Cacheable query();
     }
@@ -61,6 +62,37 @@ public virtual inherited sharing class SOQLCache implements Cacheable {
         mock.setMock(mockId, new List<SObject>{ record });
     }
 
+    public static void invalidateRecords(List<SObject> records) {
+        if (records.isEmpty()) {
+            return;
+        }
+
+        List<CacheManager.Cacheable> caches = new List<CacheManager.Cacheable>{
+                CacheManager.SOQLOrgCache,
+                CacheManager.SOQLSessionCache,
+                CacheManager.ApexTransaction
+        };
+
+        Set<Id> recordIdsToBeRemovedFromCache = new Map<Id, SObject>(records).keySet();
+        String key = records[0].getSObjectType().toString();
+
+        for (CacheManager.Cacheable cache : caches) {
+            if (!cache.contains(key)) {
+                break;
+            }
+
+            List<CacheItem> clearedCacheItems = new List<CacheItem>();
+
+            for (CacheItem cacheItem :  (List<CacheItem>) cache.get(key)) {
+                if (!recordIdsToBeRemovedFromCache.contains(cacheItem.id)) {
+                    clearedCacheItems.add(cacheItem);
+                }
+            }
+
+            cache.put(key, clearedCacheItems);
+        }
+    }
+
     public static Cacheable of(SObjectType ofObject) {
         return new SOQLCache(ofObject);
     }

force-app/main/default/classes/main/cached-soql/SOQLCache_Test.cls

@@ -559,6 +559,31 @@ private class SOQLCache_Test {
         Assert.isTrue(profile2.isSet('UserLicenseId'), 'The profile UserLicenseId should not be set.');
     }
 
+    @IsTest
+    static void recordsClearedFromCache() {
+        // Setup
+        Profile profile = (Profile) SOQLCache.of(Profile.SObjectType)
+                .with(Profile.Id, Profile.Name)
+                .whereEqual(Profile.Name, 'System Administrator')
+                .cacheInOrgCache()
+                .maxHoursWithoutRefresh(5)
+                .toObject();
+
+        List<SOQLCache.CacheItem> cacheItems = (List<SOQLCache.CacheItem>) CacheManager.SOQLOrgCache.get('Profile');
+
+        //Verify initial setup
+        Assert.isTrue(CacheManager.SOQLOrgCache.contains('Profile'), 'Key should exist.');
+        Assert.isFalse(cacheItems.isEmpty(), 'Cache item should be present');
+
+        //Test
+        SOQLCache.invalidateRecords(new List<Profile>{profile});
+
+        // Verify
+        Assert.isTrue(CacheManager.SOQLOrgCache.contains('Profile'), 'Key should still exist.');
+        cacheItems = (List<SOQLCache.CacheItem>) CacheManager.SOQLOrgCache.get('Profile');
+        Assert.isTrue(cacheItems.isEmpty(), 'Cache items should be empty');
+    }
+
     static User minimumAccessUser() {
         return new User(
             Alias = 'newUser',