Ori Newman 14fbe50636
[NOD-1493] Implement serialization in data stores (#983)
* [NOD-1493] Implement serialization in data stores

* [NOD-1493] Remove redundant functions

* [NOD-1493] Use bluesAnticoneSizesToDBBluesAnticoneSizes inside BlockGHOSTDAGDataToDBBlockGHOSTDAGData
2020-10-29 17:39:35 +02:00

52 lines
1.3 KiB
Go

package multiset
import (
"github.com/kaspanet/go-secp256k1"
"github.com/kaspanet/kaspad/domain/consensus/model"
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
"github.com/kaspanet/kaspad/domain/consensus/utils/hashes"
"github.com/pkg/errors"
)
type multiset struct {
ms *secp256k1.MultiSet
}
func (m multiset) Add(data []byte) {
m.ms.Add(data)
}
func (m multiset) Remove(data []byte) {
m.ms.Remove(data)
}
func (m multiset) Hash() *externalapi.DomainHash {
hash, err := hashes.FromBytes(m.ms.Finalize()[:])
if err != nil {
panic(errors.Errorf("this should never happen unless seckp hash size is different than %d",
externalapi.DomainHashSize))
}
return hash
}
func (m multiset) Serialize() []byte {
return m.Serialize()
}
// FromBytes deserializes the given bytes slice and returns a multiset.
func FromBytes(multisetBytes []byte) (model.Multiset, error) {
serialized := &secp256k1.SerializedMultiSet{}
if len(serialized) != len(multisetBytes) {
return nil, errors.Errorf("mutliset bytes expected to be in length of %d but got %d",
len(serialized), len(multisetBytes))
}
copy(serialized[:], multisetBytes)
ms, err := secp256k1.DeserializeMultiSet(serialized)
if err != nil {
return nil, err
}
return &multiset{ms: ms}, nil
}