go-libp2p-kad-dht/DHTMessage.go

package dht

// A helper struct to make working with protbuf types easier
type DHTMessage struct {
	Type     PBDHTMessage_MessageType
	Key      string
	Value    []byte
	Response bool
	Id       uint64
	Success  bool
}

func (m *DHTMessage) ToProtobuf() *PBDHTMessage {
	pmes := new(PBDHTMessage)
	if m.Value != nil {
		pmes.Value = m.Value
	}

	pmes.Type = &m.Type
	pmes.Key = &m.Key
	pmes.Response = &m.Response
	pmes.Id = &m.Id
	pmes.Success = &m.Success

	return pmes
}