pkg/levin: local_peerlist_new -> node

Signed-off-by: Ciro S. Costa <utxobr@protonmail.com>
This commit is contained in:
Ciro S. Costa 2021-04-25 16:41:24 -04:00
parent 8fdd71fd91
commit 5e09fd4047
3 changed files with 11 additions and 13 deletions

View file

@ -1,6 +1,7 @@
package main
import (
"encoding/json"
"flag"
"fmt"
"io/ioutil"
@ -33,11 +34,11 @@ func run() error {
return fmt.Errorf("portable storage from bytes: %w", err)
}
// if err := json.NewEncoder(os.Stdout).Encode(&ps); err != nil {
// return fmt.Errorf("encode: %w", err)
// }
if err := json.NewEncoder(os.Stdout).Encode(&ps); err != nil {
return fmt.Errorf("encode: %w", err)
}
levin.NewLocalPeerListFromEntries(ps.Entries)
// levin.NewNodeFromEntries(ps.Entries)
// for addr := range pl.Peers {
// fmt.Println(addr)

View file

@ -61,7 +61,7 @@ func (c *Client) Close() error {
return nil
}
func (c *Client) Handshake(ctx context.Context) (*LocalPeerList, error) {
func (c *Client) Handshake(ctx context.Context) (*Node, error) {
payload := (&PortableStorage{
Entries: []Entry{
{
@ -117,7 +117,7 @@ again:
return nil, fmt.Errorf("new portable storage from bytes: %w", err)
}
peerList := NewLocalPeerListFromEntries(ps.Entries)
peerList := NewNodeFromEntries(ps.Entries)
return &peerList, nil
}

View file

@ -3,10 +3,9 @@ package levin
import (
"fmt"
"net"
"runtime"
)
type LocalPeerList struct {
type Node struct {
Peers map[string]*Peer
Id uint64
@ -16,7 +15,7 @@ type LocalPeerList struct {
TopVersion uint8
}
func (l *LocalPeerList) GetPeers() map[string]*Peer {
func (l *Node) GetPeers() map[string]*Peer {
return l.Peers
}
@ -90,8 +89,8 @@ func ParsePeerList(entry Entry) map[string]*Peer {
}
// TODO less panic'ing
func NewLocalPeerListFromEntries(entries Entries) LocalPeerList {
lpl := LocalPeerList{}
func NewNodeFromEntries(entries Entries) Node {
lpl := Node{}
for _, entry := range entries {
@ -122,8 +121,6 @@ func NewLocalPeerListFromEntries(entries Entries) LocalPeerList {
}
}
runtime.Breakpoint()
return lpl
}