Merge pull request corda#2 from corda/oracle-primenumber_CCV

peterli-r3 · web-flow · commit 03fe800534ca · 2020-05-07T09:34:58.000-04:00
oracle-primenumber: added checkCommandVisibility to oracle service
diff --git a/Features/oracle-primenumber/service/src/main/java/net/corda/examples/oracle/service/service/Oracle.java b/Features/oracle-primenumber/service/src/main/java/net/corda/examples/oracle/service/service/Oracle.java
@@ -5,6 +5,7 @@
 import net.corda.core.node.ServiceHub;
 import net.corda.core.node.services.CordaService;
 import net.corda.core.serialization.SingletonSerializeAsToken;
+import net.corda.core.transactions.ComponentVisibilityException;
 import net.corda.core.transactions.FilteredTransaction;
 import net.corda.core.transactions.FilteredTransactionVerificationException;
 import net.corda.examples.oracle.base.contract.PrimeContract;
@@ -74,6 +75,15 @@ public TransactionSignature sign(FilteredTransaction ftx) throws FilteredTransac
 
         // Is it a Merkle tree we are willing to sign over?
         boolean isValidMerkleTree = ftx.checkWithFun(this::isCommandWithCorrectPrimeAndIAmSigner);
+        try {
+            /**
+             * Function that checks if all of the commands that should be signed by the input public key are visible.
+             * This functionality is required from Oracles to check that all of the commands they should sign are visible.
+             */
+            ftx.checkCommandVisibility(services.getMyInfo().getLegalIdentities().get(0).getOwningKey());
+        } catch (ComponentVisibilityException e) {
+            e.printStackTrace();
+        }
 
         if (isValidMerkleTree) {
             return services.createSignature(ftx, myKey);
