forked from TrueCloudLab/frostfs-api-go
137 lines
3.1 KiB
Go
137 lines
3.1 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"encoding/hex"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/nspcc-dev/neofs-api-go/object"
|
|
)
|
|
|
|
type (
|
|
// NodeStatus is a bitwise status field of the node.
|
|
NodeStatus uint64
|
|
)
|
|
|
|
const (
|
|
storageFullMask = 0x1
|
|
|
|
optionCapacity = "/Capacity:"
|
|
optionPrice = "/Price:"
|
|
)
|
|
|
|
var (
|
|
_ proto.Message = (*NodeInfo)(nil)
|
|
_ proto.Message = (*SpreadMap)(nil)
|
|
)
|
|
|
|
var requestEndianness = binary.BigEndian
|
|
|
|
// Equals checks whether two NodeInfo has same address.
|
|
func (m NodeInfo) Equals(n1 NodeInfo) bool {
|
|
return m.Address == n1.Address && bytes.Equal(m.PubKey, n1.PubKey)
|
|
}
|
|
|
|
// Full checks if node has enough space for storing users objects.
|
|
func (n NodeStatus) Full() bool {
|
|
return n&storageFullMask > 0
|
|
}
|
|
|
|
// SetFull changes state of node to indicate if node has enough space for storing users objects.
|
|
// If value is true - there's not enough space.
|
|
func (n *NodeStatus) SetFull(value bool) {
|
|
switch value {
|
|
case true:
|
|
*n |= NodeStatus(storageFullMask)
|
|
case false:
|
|
*n &= NodeStatus(^uint64(storageFullMask))
|
|
}
|
|
}
|
|
|
|
// Price returns price in 1e-8*GAS/Megabyte per month.
|
|
// User set price in GAS/Terabyte per month.
|
|
func (m NodeInfo) Price() uint64 {
|
|
for i := range m.Options {
|
|
if strings.HasPrefix(m.Options[i], optionPrice) {
|
|
n, err := strconv.ParseFloat(m.Options[i][len(optionPrice):], 64)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
return uint64(n*1e8) / uint64(object.UnitsMB) // UnitsMB == megabytes in 1 terabyte
|
|
}
|
|
}
|
|
return 0
|
|
}
|
|
|
|
// Capacity returns node's capacity as reported by user.
|
|
func (m NodeInfo) Capacity() uint64 {
|
|
for i := range m.Options {
|
|
if strings.HasPrefix(m.Options[i], optionCapacity) {
|
|
n, err := strconv.ParseUint(m.Options[i][len(optionCapacity):], 10, 64)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
return n
|
|
}
|
|
}
|
|
return 0
|
|
}
|
|
|
|
// String returns string representation of NodeInfo.
|
|
func (m NodeInfo) String() string {
|
|
return "(NodeInfo)<" +
|
|
"Address:" + m.Address +
|
|
", " +
|
|
"PublicKey:" + hex.EncodeToString(m.PubKey) +
|
|
", " +
|
|
"Options: [" + strings.Join(m.Options, ",") + "]>"
|
|
}
|
|
|
|
// String returns string representation of SpreadMap.
|
|
func (m SpreadMap) String() string {
|
|
result := make([]string, 0, len(m.NetMap))
|
|
for i := range m.NetMap {
|
|
result = append(result, m.NetMap[i].String())
|
|
}
|
|
return "(SpreadMap)<" +
|
|
"Epoch: " + strconv.FormatUint(m.Epoch, 10) +
|
|
", " +
|
|
"Netmap: [" + strings.Join(result, ",") + "]>"
|
|
}
|
|
|
|
// GetType is a Type field getter.
|
|
func (m Request) GetType() NodeType {
|
|
return m.Type
|
|
}
|
|
|
|
// SetType is a Type field setter.
|
|
func (m *Request) SetType(t NodeType) {
|
|
m.Type = t
|
|
}
|
|
|
|
// SetState is a State field setter.
|
|
func (m *Request) SetState(state Request_State) {
|
|
m.State = state
|
|
}
|
|
|
|
// SetInfo is an Info field getter.
|
|
func (m *Request) SetInfo(info NodeInfo) {
|
|
m.Info = info
|
|
}
|
|
|
|
// Size returns the size necessary for a binary representation of the state.
|
|
func (x Request_State) Size() int {
|
|
return 4
|
|
}
|
|
|
|
// Bytes returns a binary representation of the state.
|
|
func (x Request_State) Bytes() []byte {
|
|
data := make([]byte, x.Size())
|
|
|
|
requestEndianness.PutUint32(data, uint32(x))
|
|
|
|
return data
|
|
}
|