github.com/igggame/nebulas-go@v2.1.0+incompatible/net/pb/types.go (about)

     1  // Copyright (C) 2018 go-nebulas authors
     2  //
     3  // This file is part of the go-nebulas library.
     4  //
     5  // the go-nebulas library is free software: you can redistribute it and/or modify
     6  // it under the terms of the GNU General Public License as published by
     7  // the Free Software Foundation, either version 3 of the License, or
     8  // (at your option) any later version.
     9  //
    10  // the go-nebulas library is distributed in the hope that it will be useful,
    11  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13  // GNU General Public License for more details.
    14  //
    15  // You should have received a copy of the GNU General Public License
    16  // along with the go-nebulas library.  If not, see <http://www.gnu.org/licenses/>.
    17  //
    18  
    19  package netpb
    20  
    21  import (
    22  	"github.com/gogo/protobuf/proto"
    23  	"github.com/nebulasio/go-nebulas/util/logging"
    24  	"github.com/sirupsen/logrus"
    25  )
    26  
    27  // HelloMessageFromProto parse the data into Hello message
    28  func HelloMessageFromProto(data []byte) (*Hello, error) {
    29  	pb := new(Hello)
    30  
    31  	if err := proto.Unmarshal(data, pb); err != nil {
    32  		logging.VLog().WithFields(logrus.Fields{
    33  			"err": err,
    34  		}).Debug("Failed to unmarshal Hello message.")
    35  		return nil, err
    36  	}
    37  
    38  	return pb, nil
    39  }
    40  
    41  // OKMessageFromProto parse the data into OK message
    42  func OKMessageFromProto(data []byte) (*OK, error) {
    43  	pb := new(OK)
    44  
    45  	if err := proto.Unmarshal(data, pb); err != nil {
    46  		logging.VLog().WithFields(logrus.Fields{
    47  			"err": err,
    48  		}).Debug("Failed to unmarshal OK message.")
    49  		return nil, err
    50  	}
    51  
    52  	return pb, nil
    53  }