130 lines
3.3 KiB
Go
130 lines
3.3 KiB
Go
package storage
|
|
|
|
import (
|
|
"fmt"
|
|
"git.lumeweb.com/LumeWeb/libs5-go/encoding"
|
|
"github.com/vmihailenco/msgpack/v5"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
_ msgpack.CustomDecoder = (*StorageLocationMap)(nil)
|
|
_ msgpack.CustomEncoder = (*StorageLocationMap)(nil)
|
|
_ StorageLocation = (*StorageLocationImpl)(nil)
|
|
_ SignedStorageLocation = (*SignedStorageLocationImpl)(nil)
|
|
)
|
|
|
|
type StorageLocationMap map[int]NodeStorage
|
|
type NodeStorage map[string]NodeDetailsStorage
|
|
type NodeDetailsStorage map[int]interface{}
|
|
|
|
func (s *StorageLocationImpl) BytesURL() string {
|
|
return s.parts[0]
|
|
}
|
|
|
|
func (s *StorageLocationImpl) OutboardBytesURL() string {
|
|
if len(s.parts) == 1 {
|
|
return s.parts[0] + ".obao"
|
|
}
|
|
return s.parts[1]
|
|
}
|
|
|
|
func (s *StorageLocationImpl) String() string {
|
|
expiryDate := time.Unix(s.expiry, 0)
|
|
return "StorageLocationImpl(" + strconv.Itoa(s.Type()) + ", " + fmt.Sprint(s.parts) + ", expiry: " + expiryDate.Format(time.RFC3339) + ")"
|
|
}
|
|
|
|
type SignedStorageLocationImpl struct {
|
|
nodeID *encoding.NodeId
|
|
location StorageLocation
|
|
}
|
|
|
|
func NewSignedStorageLocation(NodeID *encoding.NodeId, Location StorageLocation) SignedStorageLocation {
|
|
return &SignedStorageLocationImpl{
|
|
nodeID: NodeID,
|
|
location: Location,
|
|
}
|
|
}
|
|
|
|
func (ssl *SignedStorageLocationImpl) String() string {
|
|
nodeString, _ := ssl.nodeID.ToString()
|
|
|
|
if nodeString == "" {
|
|
nodeString = "failed to decode node id"
|
|
}
|
|
|
|
return "SignedStorageLocationImpl(" + ssl.location.String() + ", " + nodeString + ")"
|
|
}
|
|
|
|
func (ssl *SignedStorageLocationImpl) NodeId() *encoding.NodeId {
|
|
return ssl.nodeID
|
|
}
|
|
func (ssl *SignedStorageLocationImpl) Location() StorageLocation {
|
|
return ssl.location
|
|
}
|
|
|
|
func (s *StorageLocationMap) DecodeMsgpack(dec *msgpack.Decoder) error {
|
|
if *s == nil {
|
|
*s = make(StorageLocationMap)
|
|
}
|
|
|
|
// Decode directly into a temp map
|
|
temp := make(map[int]map[string]map[int]interface{})
|
|
err := dec.Decode(&temp)
|
|
if err != nil {
|
|
return fmt.Errorf("error decoding msgpack: %w", err)
|
|
}
|
|
|
|
// Convert temp map to StorageLocationMap
|
|
for k, v := range temp {
|
|
nodeStorage, exists := (*s)[k]
|
|
if !exists {
|
|
nodeStorage = make(NodeStorage, len(v)) // preallocate if size is known
|
|
(*s)[k] = nodeStorage
|
|
}
|
|
|
|
for nk, nv := range v {
|
|
nodeDetailsStorage, exists := nodeStorage[nk]
|
|
if !exists {
|
|
nodeDetailsStorage = make(NodeDetailsStorage, len(nv)) // preallocate if size is known
|
|
nodeStorage[nk] = nodeDetailsStorage
|
|
}
|
|
|
|
for ndk, ndv := range nv {
|
|
nodeDetailsStorage[ndk] = ndv
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s StorageLocationMap) EncodeMsgpack(enc *msgpack.Encoder) error {
|
|
// Create a temporary map to hold the encoded data
|
|
tempMap := make(map[int]map[string]map[int]interface{})
|
|
|
|
// Populate the temporary map with data from storageLocationMap
|
|
for storageKey, nodeStorages := range s {
|
|
tempNodeStorages := make(map[string]map[int]interface{})
|
|
for nodeId, nodeDetails := range nodeStorages {
|
|
tempNodeStorages[nodeId] = nodeDetails
|
|
}
|
|
tempMap[storageKey] = tempNodeStorages
|
|
}
|
|
|
|
// Encode the temporary map using MessagePack
|
|
return enc.Encode(tempMap)
|
|
}
|
|
|
|
func NewStorageLocationMap() StorageLocationMap {
|
|
return StorageLocationMap{}
|
|
}
|
|
|
|
type StorageLocationProvider interface {
|
|
Start() error
|
|
Next() (SignedStorageLocation, error)
|
|
Upvote(uri SignedStorageLocation) error
|
|
Downvote(uri SignedStorageLocation) error
|
|
}
|