Add select and disticnt builders to FetchSpecification

... useful.
ZeeQL · Jan 1, 2025 · 5cd4699 · 5cd4699
1 parent 8fba63b
commit 5cd4699
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/Sources/ZeeQL/Control/FetchSpecification+Builder.swift b/Sources/ZeeQL/Control/FetchSpecification+Builder.swift
@@ -83,6 +83,11 @@ public extension FetchSpecification {
   @inlinable
   func offset(_ value : Int) -> Self { transform { $0.fetchOffset = value } }
 
+  @inlinable
+  func distinct(_ usesDistinct: Bool = true) -> Self {
+    transform { $0.usesDistinct = usesDistinct }
+  }
+
 
   // MARK: - Prefetches
 
@@ -283,5 +288,19 @@ public extension DatabaseFetchSpecification
     }
   }
 
+  @inlinable
+  func select<each A: Attribute>(
+    _ attribute: repeat Swift.KeyPath<Object.FullEntity, each A>,
+    clear: Bool = false
+  ) -> Self
+  {
+    transform {
+      if clear { $0.fetchAttributeNames = [] }
+      for attributePath in repeat each attribute {
+        let attribute = Object.e[keyPath: attributePath]
+        $0.fetchAttributeNames.append(attribute.name)
+      }
+    }
+  }
   #endif // compiler(>=6)
 }