github.com/adnan-c/fabric_e2e_couchdb@v0.6.1-preview.0.20170228180935-21ce6b23cf91/events/producer/eventhelper.go (about)

     1  /*
     2  Copyright IBM Corp. 2016 All Rights Reserved.
     3  
     4  Licensed under the Apache License, Version 2.0 (the "License");
     5  you may not use this file except in compliance with the License.
     6  You may obtain a copy of the License at
     7  
     8  		 http://www.apache.org/licenses/LICENSE-2.0
     9  
    10  Unless required by applicable law or agreed to in writing, software
    11  distributed under the License is distributed on an "AS IS" BASIS,
    12  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13  See the License for the specific language governing permissions and
    14  limitations under the License.
    15  */
    16  
    17  package producer
    18  
    19  import (
    20  	"fmt"
    21  
    22  	"github.com/hyperledger/fabric/protos/common"
    23  	pb "github.com/hyperledger/fabric/protos/peer"
    24  	"github.com/hyperledger/fabric/protos/utils"
    25  	"github.com/op/go-logging"
    26  )
    27  
    28  var logger *logging.Logger // package-level logger
    29  
    30  func init() {
    31  	logger = logging.MustGetLogger("eventhub_producer")
    32  }
    33  
    34  // SendProducerBlockEvent sends block event to clients
    35  func SendProducerBlockEvent(block *common.Block) error {
    36  	bevent := &common.Block{}
    37  	bevent.Header = block.Header
    38  	bevent.Metadata = block.Metadata
    39  	bevent.Data = &common.BlockData{}
    40  	for _, d := range block.Data.Data {
    41  		ebytes := d
    42  		if ebytes != nil {
    43  			if env, err := utils.GetEnvelopeFromBlock(ebytes); err != nil {
    44  				logger.Errorf("Error getting tx from block(%s)\n", err)
    45  			} else if env != nil {
    46  				// get the payload from the envelope
    47  				payload, err := utils.GetPayload(env)
    48  				if err != nil {
    49  					return fmt.Errorf("Could not extract payload from envelope, err %s", err)
    50  				}
    51  
    52  				chdr, err := utils.UnmarshalChannelHeader(payload.Header.ChannelHeader)
    53  				if err != nil {
    54  					return err
    55  				}
    56  
    57  				if common.HeaderType(chdr.Type) == common.HeaderType_ENDORSER_TRANSACTION {
    58  					tx, err := utils.GetTransaction(payload.Data)
    59  					if err != nil {
    60  						return fmt.Errorf("Error unmarshalling transaction payload for block event: %s", err)
    61  					}
    62  					chaincodeActionPayload, err := utils.GetChaincodeActionPayload(tx.Actions[0].Payload)
    63  					if err != nil {
    64  						return fmt.Errorf("Error unmarshalling transaction action payload for block event: %s", err)
    65  					}
    66  					propRespPayload, err := utils.GetProposalResponsePayload(chaincodeActionPayload.Action.ProposalResponsePayload)
    67  					if err != nil {
    68  						return fmt.Errorf("Error unmarshalling proposal response payload for block event: %s", err)
    69  					}
    70  					//ENDORSER_ACTION, ProposalResponsePayload.Extension field contains ChaincodeAction
    71  					caPayload, err := utils.GetChaincodeAction(propRespPayload.Extension)
    72  					if err != nil {
    73  						return fmt.Errorf("Error unmarshalling chaincode action for block event: %s", err)
    74  					}
    75  					// Drop read write set from transaction before sending block event
    76  					// Performance issue with chaincode deploy txs and causes nodejs grpc
    77  					// to hit max message size bug
    78  					// Dropping the read write set may cause issues for security and
    79  					// we will need to revist when event security is addressed
    80  					caPayload.Results = nil
    81  					chaincodeActionPayload.Action.ProposalResponsePayload, err = utils.GetBytesProposalResponsePayload(propRespPayload.ProposalHash, caPayload.Response, caPayload.Results, caPayload.Events)
    82  					if err != nil {
    83  						return fmt.Errorf("Error marshalling tx proposal payload for block event: %s", err)
    84  					}
    85  					tx.Actions[0].Payload, err = utils.GetBytesChaincodeActionPayload(chaincodeActionPayload)
    86  					if err != nil {
    87  						return fmt.Errorf("Error marshalling tx action payload for block event: %s", err)
    88  					}
    89  					payload.Data, err = utils.GetBytesTransaction(tx)
    90  					if err != nil {
    91  						return fmt.Errorf("Error marshalling payload for block event: %s", err)
    92  					}
    93  					env.Payload, err = utils.GetBytesPayload(payload)
    94  					if err != nil {
    95  						return fmt.Errorf("Error marshalling tx envelope for block event: %s", err)
    96  					}
    97  					ebytes, err = utils.GetBytesEnvelope(env)
    98  					if err != nil {
    99  						return fmt.Errorf("Cannot marshal transaction %s", err)
   100  					}
   101  				}
   102  			}
   103  		}
   104  		bevent.Data.Data = append(bevent.Data.Data, ebytes)
   105  	}
   106  	return Send(CreateBlockEvent(bevent))
   107  }
   108  
   109  //CreateBlockEvent creates a Event from a Block
   110  func CreateBlockEvent(te *common.Block) *pb.Event {
   111  	return &pb.Event{Event: &pb.Event_Block{Block: te}}
   112  }
   113  
   114  //CreateChaincodeEvent creates a Event from a ChaincodeEvent
   115  func CreateChaincodeEvent(te *pb.ChaincodeEvent) *pb.Event {
   116  	return &pb.Event{Event: &pb.Event_ChaincodeEvent{ChaincodeEvent: te}}
   117  }
   118  
   119  //CreateRejectionEvent creates an Event from TxResults
   120  func CreateRejectionEvent(tx *pb.Transaction, errorMsg string) *pb.Event {
   121  	return &pb.Event{Event: &pb.Event_Rejection{Rejection: &pb.Rejection{Tx: tx, ErrorMsg: errorMsg}}}
   122  }