added queryAllByKey (#22738)

Author: roger-bosman-da

sdk/compiler/damlc/daml-preprocessor/src/DA/Daml/Preprocessor.hs

@@ -78,6 +78,7 @@ preprocessorExceptions = Set.fromList $ map GHC.mkModuleName
     , "DA.Exception.AssertionFailed"
     , "DA.Exception.PreconditionFailed"
     , "DA.Fail"
+    , "DA.ContractKeys"
 
 
     -- These modules need to have the record preprocessor disabled.

sdk/compiler/damlc/daml-stdlib-src/DA/ContractKeys.daml

@@ -0,0 +1,17 @@
+-- Copyright (c) 2025 Digital Asset (Switzerland) GmbH and/or its affiliates. All rights reserved.
+-- SPDX-License-Identifier: Apache-2.0
+
+{-# LANGUAGE CPP #-}
+
+-- | Note: Docs TODO (https://github.com/digital-asset/daml/issues/22673)
+module DA.ContractKeys where
+
+#ifdef DAML_NUCK
+import DA.Internal.Template.Functions as DA
+
+queryNByKey : forall t k. HasQueryNByKey t k => Int -> k -> Update [(ContractId t, t)]
+queryNByKey = DA.queryNByKey
+
+queryAllByKey : forall t k. HasQueryNByKey t k => k -> Update [(ContractId t, t)]
+queryAllByKey = DA.queryNByKey (maxBound @Int)
+#endif

sdk/compiler/damlc/daml-stdlib-src/LibraryModules.daml

@@ -13,6 +13,7 @@ import DA.Action.State
 import DA.Assert
 import DA.BigNumeric
 import DA.Bifunctor
+import DA.ContractKeys
 import DA.Crypto.Text
 import DA.Date
 import DA.Either

sdk/compiler/damlc/daml-stdlib-src/Prelude.daml

@@ -18,7 +18,7 @@ import DA.Internal.LF as X hiding
 -- but we can hide them.
 import DA.Internal.Any as X
 import DA.Internal.Template as X
-import DA.Internal.Template.Functions as X
+import DA.Internal.Template.Functions as X hiding (queryNByKey)
 import DA.Internal.Compatible as X
 import DA.Internal.Assert as X
 import DA.Internal.Interface as X

sdk/compiler/damlc/tests/daml-test-files/QueryNByKey.daml

@@ -5,7 +5,7 @@
 
 module QueryNByKey where
 
--- import Daml.Script
+import DA.ContractKeys
 
 template WithKey
   with
@@ -20,17 +20,26 @@ template Helper
     p : Party
   where
     signatory p
-    choice PerformQuery : ()
+    choice PerformQueryNByKey : ()
       controller p
       do contracts <- queryNByKey @WithKey 1 p
          case contracts of
            (contract :: _) -> do
              pure ()
            [] ->
              abort "Assertion failed: queryNByKey returned an empty list."
+    choice PerformQueryAllByKey : ()
+      controller p
+      do contracts <- queryAllByKey @WithKey p
+         case contracts of
+           (contract :: _) -> do
+             pure ()
+           [] ->
+             abort "Assertion failed: queryAllByKey returned an empty list."
 
 -- Test just here to assert that queryNByKey compiles
 -- When we have an ledger implementation, assert things actually run
 -- test = script do
 --   p <- allocateParty "p"
---   submit p $ createAndExerciseCmd (Helper p) PerformQuery
+--   submit p $ createAndExerciseCmd (Helper p) PerformQueryNByKey
+--   submit p $ createAndExerciseCmd (Helper p) PerformQueryAllByKey