Added support for SOSL's OFFSET, FOR VIEW & FOR REFERENCE

ISObjectQueryBuilder.setAsUpdate() renamed to forUpdate() for consistency

Author: jongpie

src/classes/ISObjectQueryBuilder.cls

@@ -37,7 +37,10 @@ public interface ISObjectQueryBuilder {
 
     // Additional query option methods
     ISObjectQueryBuilder limitCount(Integer limitCount);
-    ISObjectQueryBuilder setAsUpdate();
+    ISObjectQueryBuilder offset(Integer numberOfRowsToSkip);
+    ISObjectQueryBuilder forReference();
+    ISObjectQueryBuilder forUpdate();
+    ISObjectQueryBuilder forView();
     ISObjectQueryBuilder usingScope(QueryFilterScope filterScope);
 
     // Query string methods

src/classes/SObjectQueryBuilder.cls

@@ -15,7 +15,10 @@ public class SObjectQueryBuilder extends QueryBuilder implements ISObjectQueryBu
     private Set<String> queryFields;
     private List<String> childRelationshipQueries;
     private QueryFilterScope filterScope;
+    private Integer offset;
+    private Boolean forReference;
     private Boolean forUpdate;
+    private Boolean forView;
     private SObjectType sobjectType;
     private Map<String, Schema.SObjectField> sobjectTypeFieldMap;
 
@@ -25,7 +28,9 @@ public class SObjectQueryBuilder extends QueryBuilder implements ISObjectQueryBu
 
         this.queryFields              = new Set<String>();
         this.childRelationshipQueries = new List<String>();
+        this.forReference             = false;
         this.forUpdate                = false;
+        this.forView                = false;
 
         this.addCommonQueryFields();
     }
@@ -123,11 +128,26 @@ public class SObjectQueryBuilder extends QueryBuilder implements ISObjectQueryBu
         return this;
     }
 
-    public ISObjectQueryBuilder setAsUpdate() {
+    public ISObjectQueryBuilder offset(Integer numberOfRowsToSkip) {
+        this.offset = numberOfRowsToSkip;
+        return this;
+    }
+
+    public ISObjectQueryBuilder forReference() {
+        this.forReference = true;
+        return this;
+    }
+
+    public ISObjectQueryBuilder forUpdate() {
         this.forUpdate = true;
         return this;
     }
 
+    public ISObjectQueryBuilder forView() {
+        this.forView = true;
+        return this;
+    }
+
     public ISObjectQueryBuilder usingScope(QueryFilterScope filterScope) {
         this.filterScope = filterScope;
         return this;

src/classes/SObjectQueryBuilder_Tests.cls

@@ -407,7 +407,7 @@ private class SObjectQueryBuilder_Tests {
 
         Test.startTest();
         SObjectQueryBuilder query = (SObjectQueryBuilder)new SObjectQueryBuilder(sobjectType).addFields(convertToQueryFields(fields));
-        query.setAsUpdate();
+        query.forUpdate();
         Test.stopTest();
 
         System.assert(query.getQuery().contains('FOR UPDATE'));