[FAB-3649] add getTransient() API

Change-Id: Iae5d98b9c4d9eb43d22e43164740cee09d50b355
Signed-off-by: Luis Sanchez <sanchezl@us.ibm.com>

Author: Luis Sanchez

shim/src/main/java/org/hyperledger/fabric/shim/ChaincodeStub.java

@@ -17,6 +17,7 @@
 import java.time.Instant;
 import java.util.Arrays;
 import java.util.List;
+import java.util.Map;
 
 import static java.nio.charset.StandardCharsets.UTF_8;
 import static java.util.stream.Collectors.toList;
@@ -309,4 +310,10 @@ default void putStringState(String key, String value) {
 	 */
 	byte[] getCreator();
 
+	/**
+	 * Returns the transient map associated with the current transaction.
+	 *
+	 * @return
+	 */
+	Map<String, byte[]> getTransient();
 }

shim/src/main/java/org/hyperledger/fabric/shim/impl/ChaincodeStubImpl.java

@@ -18,6 +18,7 @@
 import org.hyperledger.fabric.protos.ledger.queryresult.KvQueryResult.KV;
 import org.hyperledger.fabric.protos.peer.ChaincodeEventPackage.ChaincodeEvent;
 import org.hyperledger.fabric.protos.peer.ChaincodeShim.QueryResultBytes;
+import org.hyperledger.fabric.protos.peer.ProposalPackage.ChaincodeProposalPayload;
 import org.hyperledger.fabric.protos.peer.ProposalPackage.Proposal;
 import org.hyperledger.fabric.protos.peer.ProposalPackage.SignedProposal;
 import org.hyperledger.fabric.shim.Chaincode.Response;
@@ -30,6 +31,7 @@
 import java.time.Instant;
 import java.util.Collections;
 import java.util.List;
+import java.util.Map;
 import java.util.function.Function;
 import java.util.stream.Collectors;
 
@@ -44,6 +46,7 @@ class ChaincodeStubImpl implements ChaincodeStub {
 	private final SignedProposal signedProposal;
 	private final Instant txTimestamp;
 	private final ByteString creator;
+	private final Map<String, ByteString> transientMap;
 	private ChaincodeEvent event;
 
 	ChaincodeStubImpl(String txId, Handler handler, List<ByteString> args, SignedProposal signedProposal) {
@@ -54,17 +57,20 @@ class ChaincodeStubImpl implements ChaincodeStub {
 		if(this.signedProposal == null) {
 			this.creator = null;
 			this.txTimestamp = null;
+			this.transientMap = Collections.emptyMap();
 		} else {
 			try {
 				final Proposal proposal = Proposal.parseFrom(signedProposal.getProposalBytes());
 				final Header header = Header.parseFrom(proposal.getHeader());
 				final ChannelHeader channelHeader = ChannelHeader.parseFrom(header.getChannelHeader());
 				validateProposalType(channelHeader);
 				final SignatureHeader signatureHeader = SignatureHeader.parseFrom(header.getSignatureHeader());
-
+				final ChaincodeProposalPayload chaincodeProposalPayload = ChaincodeProposalPayload.parseFrom(proposal.getPayload());
 				final Timestamp timestamp = channelHeader.getTimestamp();
+
 				this.txTimestamp = Instant.ofEpochSecond(timestamp.getSeconds(), timestamp.getNanos());
 				this.creator = signatureHeader.getCreator();
+				this.transientMap = chaincodeProposalPayload.getTransientMapMap();
 			} catch (InvalidProtocolBufferException e) {
 				throw new RuntimeException(e);
 			}
@@ -236,4 +242,9 @@ public byte[] getCreator() {
 		if(creator == null) return null;
 		return creator.toByteArray();
 	}
+
+	@Override
+	public Map<String, byte[]> getTransient() {
+		return transientMap.entrySet().stream().collect(Collectors.toMap(x -> x.getKey(), x -> x.getValue().toByteArray()));
+	}
 }

shim/src/test/java/org/hyperledger/fabric/shim/impl/ChaincodeStubImplTest.java

@@ -17,6 +17,7 @@
 import org.hyperledger.fabric.protos.peer.ChaincodeEventPackage.ChaincodeEvent;
 import org.hyperledger.fabric.protos.peer.ChaincodeShim.QueryResponse;
 import org.hyperledger.fabric.protos.peer.ChaincodeShim.QueryResultBytes;
+import org.hyperledger.fabric.protos.peer.ProposalPackage.ChaincodeProposalPayload;
 import org.hyperledger.fabric.protos.peer.ProposalPackage.Proposal;
 import org.hyperledger.fabric.protos.peer.ProposalPackage.SignedProposal;
 import org.hyperledger.fabric.shim.Chaincode;
@@ -38,8 +39,8 @@
 
 import static java.nio.charset.StandardCharsets.UTF_8;
 import static org.hamcrest.MatcherAssert.assertThat;
-import static org.hamcrest.Matchers.contains;
 import static org.hamcrest.Matchers.*;
+import static org.hamcrest.Matchers.contains;
 import static org.hyperledger.fabric.protos.common.Common.HeaderType.ENDORSER_TRANSACTION_VALUE;
 import static org.junit.Assert.assertNotNull;
 import static org.mockito.ArgumentMatchers.*;
@@ -400,4 +401,30 @@ public void testGetCreator() {
 		assertThat(stub.getCreator(), is(creator));
 	}
 
+	@Test
+	public void testGetTransient() {
+		final SignedProposal signedProposal = SignedProposal.newBuilder()
+				.setProposalBytes(Proposal.newBuilder()
+						.setHeader(Header.newBuilder()
+								.setChannelHeader(ChannelHeader.newBuilder()
+										.setType(ENDORSER_TRANSACTION_VALUE)
+										.setTimestamp(Timestamp.getDefaultInstance())
+										.build().toByteString()
+								)
+								.build().toByteString()
+						)
+						.setPayload(ChaincodeProposalPayload.newBuilder()
+								.putTransientMap("key0", ByteString.copyFromUtf8("value0"))
+								.putTransientMap("key1", ByteString.copyFromUtf8("value1"))
+								.build().toByteString()
+						)
+						.build().toByteString()
+				).build();
+		final ChaincodeStubImpl stub = new ChaincodeStubImpl("txid", handler, new ArrayList<>(), signedProposal);
+		assertThat(stub.getTransient(), allOf(
+				hasEntry("key0", "value0".getBytes(UTF_8)),
+				hasEntry("key1", "value1".getBytes(UTF_8))
+				));
+	}
+
 }