github.com/braveheart12/just@v0.8.7/core/message/message.go

/*
 *    Copyright 2019 Insolar Technologies
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

// Package message represents message that messagebus can route
package message

import (
	"bytes"
	"encoding/gob"
	"io"
	"io/ioutil"

	"github.com/ugorji/go/codec"

	"github.com/pkg/errors"

	"github.com/insolar/insolar/core"
)

// GetEmptyMessage constructs specified message
func getEmptyMessage(mt core.MessageType) (core.Message, error) {
	switch mt {

	// Logicrunner
	case core.TypeCallMethod:
		return &CallMethod{}, nil
	case core.TypeCallConstructor:
		return &CallConstructor{}, nil
	case core.TypeReturnResults:
		return &ReturnResults{}, nil
	case core.TypeExecutorResults:
		return &ExecutorResults{}, nil
	case core.TypeValidateCaseBind:
		return &ValidateCaseBind{}, nil
	case core.TypeValidationResults:
		return &ValidationResults{}, nil
	case core.TypePendingFinished:
		return &PendingFinished{}, nil
	case core.TypeStillExecuting:
		return &StillExecuting{}, nil

	// Ledger
	case core.TypeGetCode:
		return &GetCode{}, nil
	case core.TypeGetObject:
		return &GetObject{}, nil
	case core.TypeGetDelegate:
		return &GetDelegate{}, nil
	case core.TypeGetChildren:
		return &GetChildren{}, nil
	case core.TypeUpdateObject:
		return &UpdateObject{}, nil
	case core.TypeRegisterChild:
		return &RegisterChild{}, nil
	case core.TypeJetDrop:
		return &JetDrop{}, nil
	case core.TypeSetRecord:
		return &SetRecord{}, nil
	case core.TypeGetObjectIndex:
		return &GetObjectIndex{}, nil
	case core.TypeValidationCheck:
		return &ValidationCheck{}, nil
	case core.TypeGetPendingRequests:
		return &GetPendingRequests{}, nil
	case core.TypeGetJet:
		return &GetJet{}, nil
	case core.TypeAbandonedRequestsNotification:
		return &AbandonedRequestsNotification{}, nil
	case core.TypeGetPendingRequestID:
		return &GetPendingRequestID{}, nil
	case core.TypeGetRequest:
		return &GetRequest{}, nil

	// heavy sync
	case core.TypeHeavyStartStop:
		return &HeavyStartStop{}, nil
	case core.TypeHeavyPayload:
		return &HeavyPayload{}, nil
	case core.TypeHeavyReset:
		return &HeavyReset{}, nil
	// Bootstrap
	case core.TypeBootstrapRequest:
		return &GenesisRequest{}, nil

	// NodeCert
	case core.TypeNodeSignRequest:
		return &NodeSignPayload{}, nil
	default:
		return nil, errors.Errorf("unimplemented message type %d", mt)
	}
}

// MustSerializeBytes returns encoded core.Message, panics on error.
func MustSerializeBytes(msg core.Message) []byte {
	r, err := Serialize(msg)
	if err != nil {
		panic(err)
	}
	b, err := ioutil.ReadAll(r)
	if err != nil {
		panic(err)
	}
	return b
}

// Serialize returns io.Reader on buffer with encoded core.Message.
func Serialize(msg core.Message) (io.Reader, error) {
	buff := &bytes.Buffer{}
	_, err := buff.Write([]byte{byte(msg.Type())})
	if err != nil {
		return nil, err
	}

	enc := codec.NewEncoder(buff, &codec.CborHandle{})
	enc.MustEncode(msg)
	return buff, err
}

// Deserialize returns decoded message.
func Deserialize(buff io.Reader) (core.Message, error) {
	b := make([]byte, 1)
	_, err := buff.Read(b)
	if err != nil {
		return nil, errors.New("too short slice for deserialize message")
	}

	msg, err := getEmptyMessage(core.MessageType(b[0]))
	if err != nil {
		return nil, err
	}
	enc := codec.NewDecoder(buff, &codec.CborHandle{})
	if err = enc.Decode(msg); err != nil {
		return nil, err
	}
	return msg, nil
}

// ToBytes serialize a core.Message to bytes.
func ToBytes(msg core.Message) []byte {
	reqBuff, err := Serialize(msg)
	if err != nil {
		panic(errors.Wrap(err, "failed to serialize message"))
	}
	buff, err := ioutil.ReadAll(reqBuff)
	if err != nil {
		panic(errors.Wrap(err, "failed to serialize message"))
	}
	return buff
}

// SerializeParcel returns io.Reader on buffer with encoded core.Parcel.
func SerializeParcel(parcel core.Parcel) (io.Reader, error) {
	buff := &bytes.Buffer{}
	enc := gob.NewEncoder(buff)
	err := enc.Encode(parcel)
	return buff, err
}

// DeserializeParcel returns decoded signed message.
func DeserializeParcel(buff io.Reader) (core.Parcel, error) {
	var signed Parcel
	enc := gob.NewDecoder(buff)
	err := enc.Decode(&signed)
	return &signed, err
}

// ParcelToBytes deserialize a core.Parcel to bytes.
func ParcelToBytes(msg core.Parcel) []byte {
	reqBuff, err := SerializeParcel(msg)
	if err != nil {
		panic("failed to serialize message: " + err.Error())
	}
	buf, err := ioutil.ReadAll(reqBuff)
	if err != nil {
		panic("failed to serialize message: " + err.Error())
	}
	return buf
}

func init() {
	// Bootstrap
	gob.Register(&NodeSignPayload{})

	// Logicrunner
	gob.Register(&CallConstructor{})
	gob.Register(&CallMethod{})
	gob.Register(&ReturnResults{})
	gob.Register(&ExecutorResults{})
	gob.Register(&ValidateCaseBind{})
	gob.Register(&ValidationResults{})
	gob.Register(&PendingFinished{})
	gob.Register(&StillExecuting{})

	// Ledger
	gob.Register(&GetCode{})
	gob.Register(&GetObject{})
	gob.Register(&GetDelegate{})
	gob.Register(&UpdateObject{})
	gob.Register(&RegisterChild{})
	gob.Register(&JetDrop{})
	gob.Register(&SetRecord{})
	gob.Register(&GetObjectIndex{})
	gob.Register(&SetBlob{})
	gob.Register(&ValidateRecord{})
	gob.Register(&ValidationCheck{})
	gob.Register(&GetPendingRequests{})
	gob.Register(&GetJet{})
	gob.Register(&AbandonedRequestsNotification{})
	gob.Register(&HotData{})
	gob.Register(&GetPendingRequestID{})
	gob.Register(&GetRequest{})

	// heavy
	gob.Register(&HeavyStartStop{})
	gob.Register(&HeavyPayload{})
	gob.Register(&HeavyReset{})

	// Bootstrap
	gob.Register(&GenesisRequest{})
	gob.Register(&Parcel{})
	gob.Register(core.RecordRef{})
	gob.Register(&GetChildren{})

	// NodeCert
	gob.Register(&NodeSignPayload{})
}