change StartWriteBatch

Signed-off-by: Fedor Partanskiy <[email protected]>

Author: pfi79

shim/handler.go

@@ -544,17 +544,17 @@ func (h *Handler) sendBatch(channelID string, txid string) error {
 	return nil
 }
 
-func (h *Handler) handleStartWriteBatch(channelID string, txID string) error {
+func (h *Handler) handleStartWriteBatch(channelID string, txID string) (isSupportWriteBatch bool) {
 	if !h.usePeerWriteBatch {
-		return errors.New("peer does not support write batch")
+		return false
 	}
 
 	txCtxID := transactionContextID(channelID, txID)
 	h.startWriteBatchMutex.Lock()
 	defer h.startWriteBatchMutex.Unlock()
 
 	h.startWriteBatch[txCtxID] = true
-	return nil
+	return true
 }
 
 func (h *Handler) handleFinishWriteBatch(channelID string, txID string) error {

shim/interfaces.go

@@ -374,8 +374,9 @@ type ChaincodeStubInterface interface {
 	// StartWriteBatch enables a mode where all changes are not immediately forwarded to the feast,
 	// but accumulate in the cache. The cache is sent in large batches either at the end of transaction
 	// execution or after the FinishWriteBatch call.
+	// StartWriteBatch returns a bool indication that the peer supports or does not support writing by batches.
 	// IMPORTANT: in this mode, the expected order of transaction execution and expected errors can be changed.
-	StartWriteBatch() error
+	StartWriteBatch() (isSupportWriteBatch bool)
 
 	// FinishWriteBatch sends accumulated changes in large batches to the peer
 	// if StartWriteBatch has been called before it.

shim/stub.go

@@ -582,7 +582,7 @@ func (s *ChaincodeStub) GetQueryResultWithPagination(query string, pageSize int3
 // --------- Batch State functions ----------
 
 // StartWriteBatch documentation can be found in interfaces.go
-func (s *ChaincodeStub) StartWriteBatch() error {
+func (s *ChaincodeStub) StartWriteBatch() (isSupportWriteBatch bool) {
 	return s.handler.handleStartWriteBatch(s.ChannelID, s.TxID)
 }
 

shim/stub_test.go

@@ -321,7 +321,7 @@ func TestChaincodeStubHandlers(t *testing.T) {
 			resType: peer.ChaincodeMessage_RESPONSE,
 			payload: []byte("myvalue"),
 			testFunc: func(s *ChaincodeStub, h *Handler, t *testing.T, payload []byte) {
-				s.StartWriteBatch() //nolint:errcheck
+				s.StartWriteBatch()
 				err := s.PutState("key", payload)
 				assert.NoError(t, err)
 				err = s.PutPrivateData("col", "key", payload)
@@ -339,8 +339,8 @@ func TestChaincodeStubHandlers(t *testing.T) {
 				err = s.FinishWriteBatch()
 				assert.NoError(t, err)
 
-				s.StartWriteBatch() //nolint:errcheck
-				s.StartWriteBatch() //nolint:errcheck
+				s.StartWriteBatch()
+				s.StartWriteBatch()
 				err = s.PutState("key", payload)
 				assert.NoError(t, err)
 				err = s.PutPrivateData("col", "key", payload)
@@ -609,8 +609,8 @@ func TestChaincodeStubHandlers(t *testing.T) {
 				resp := s.InvokeChaincode("cc", [][]byte{}, "channel")
 				assert.Equal(t, payload, resp.GetPayload())
 
-				s.StartWriteBatch() //nolint:errcheck
-				s.StartWriteBatch() //nolint:errcheck
+				s.StartWriteBatch()
+				s.StartWriteBatch()
 				err = s.PutState("key", payload)
 				assert.NoError(t, err)
 				err = s.FinishWriteBatch()