5e1b02ca8b
* intial changes to migrate statdb to masterdb framework * statdb refactor compiles * added TestCreateDoesNotExist testcase * Initial port of statdb to masterdb framework working * refactored statdb proto def to pkg/statdb * removed statdb/proto folder * moved pb.Node to storj.NodeID * CreateEntryIfNotExistsRequest moved pd.Node to storj.NodeID * moved the fields from pb.Node to statdb.UpdateRequest ported TestUpdateExists, TestUpdateUptimeExists, TestUpdateAuditSuccessExists TestUpdateBatchExists
69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package pb
|
|
|
|
import "storj.io/storj/pkg/storj"
|
|
|
|
// NodeIDsToLookupRequests converts NodeIDs to LookupRequests
|
|
func NodeIDsToLookupRequests(nodeIDs storj.NodeIDList) *LookupRequests {
|
|
var rq []*LookupRequest
|
|
for _, v := range nodeIDs {
|
|
r := &LookupRequest{NodeId: v}
|
|
rq = append(rq, r)
|
|
}
|
|
return &LookupRequests{LookupRequest: rq}
|
|
}
|
|
|
|
// LookupResponsesToNodes converts LookupResponses to Nodes
|
|
func LookupResponsesToNodes(responses *LookupResponses) []*Node {
|
|
var nodes []*Node
|
|
for _, v := range responses.LookupResponse {
|
|
n := v.GetNode()
|
|
nodes = append(nodes, n)
|
|
}
|
|
return nodes
|
|
}
|
|
|
|
// NodesToIDs extracts Node-s into a list of ids
|
|
func NodesToIDs(nodes []*Node) storj.NodeIDList {
|
|
ids := make(storj.NodeIDList, len(nodes))
|
|
for i, node := range nodes {
|
|
if node != nil {
|
|
ids[i] = node.Id
|
|
}
|
|
}
|
|
return ids
|
|
}
|
|
|
|
// CopyNode returns a deep copy of a node
|
|
// It would be better to use `proto.Clone` but it is curently incompatible
|
|
// with gogo's customtype extension.
|
|
// (see https://github.com/gogo/protobuf/issues/147)
|
|
func CopyNode(src *Node) (dst *Node) {
|
|
node := Node{Id: storj.NodeID{}}
|
|
copy(node.Id[:], src.Id[:])
|
|
if src.Address != nil {
|
|
node.Address = &NodeAddress{
|
|
Transport: src.Address.Transport,
|
|
Address: src.Address.Address,
|
|
}
|
|
}
|
|
if src.Metadata != nil {
|
|
node.Metadata = &NodeMetadata{
|
|
Email: src.Metadata.Email,
|
|
Wallet: src.Metadata.Wallet,
|
|
}
|
|
}
|
|
if src.Restrictions != nil {
|
|
node.Restrictions = &NodeRestrictions{
|
|
FreeBandwidth: src.Restrictions.FreeBandwidth,
|
|
FreeDisk: src.Restrictions.FreeDisk,
|
|
}
|
|
}
|
|
|
|
node.Type = src.Type
|
|
|
|
return &node
|
|
}
|