From 1e81d83f782eb53186a74f5c0a3d8373c7003aec Mon Sep 17 00:00:00 2001 From: Elle Mouton Date: Tue, 22 Oct 2024 12:22:54 +0200 Subject: [PATCH] channeldb: export net.Addr encode/decode methods We'll move these to the new graphdb package later and import them from there. --- channeldb/addr.go | 8 ++++---- channeldb/addr_test.go | 4 ++-- channeldb/codec.go | 8 ++++---- channeldb/graph.go | 4 ++-- channeldb/nodes.go | 4 ++-- 5 files changed, 14 insertions(+), 14 deletions(-) diff --git a/channeldb/addr.go b/channeldb/addr.go index dd057265c..2064939c1 100644 --- a/channeldb/addr.go +++ b/channeldb/addr.go @@ -121,10 +121,10 @@ func encodeOnionAddr(w io.Writer, addr *tor.OnionAddr) error { return nil } -// deserializeAddr reads the serialized raw representation of an address and +// DeserializeAddr reads the serialized raw representation of an address and // deserializes it into the actual address. This allows us to avoid address // resolution within the channeldb package. -func deserializeAddr(r io.Reader) (net.Addr, error) { +func DeserializeAddr(r io.Reader) (net.Addr, error) { var addrType [1]byte if _, err := r.Read(addrType[:]); err != nil { return nil, err @@ -207,9 +207,9 @@ func deserializeAddr(r io.Reader) (net.Addr, error) { return address, nil } -// serializeAddr serializes an address into its raw bytes representation so that +// SerializeAddr serializes an address into its raw bytes representation so that // it can be deserialized without requiring address resolution. -func serializeAddr(w io.Writer, address net.Addr) error { +func SerializeAddr(w io.Writer, address net.Addr) error { switch addr := address.(type) { case *net.TCPAddr: return encodeTCPAddr(w, addr) diff --git a/channeldb/addr_test.go b/channeldb/addr_test.go index c761989c0..b8318b72f 100644 --- a/channeldb/addr_test.go +++ b/channeldb/addr_test.go @@ -117,7 +117,7 @@ func TestAddrSerialization(t *testing.T) { var b bytes.Buffer for _, test := range addrTests { - err := serializeAddr(&b, test.expAddr) + err := SerializeAddr(&b, test.expAddr) switch { case err == nil && test.serErr != "": t.Fatalf("expected serialization err for addr %v", @@ -136,7 +136,7 @@ func TestAddrSerialization(t *testing.T) { continue } - addr, err := deserializeAddr(&b) + addr, err := DeserializeAddr(&b) if err != nil { t.Fatalf("unable to deserialize address: %v", err) } diff --git a/channeldb/codec.go b/channeldb/codec.go index 07f125742..aaa7917ac 100644 --- a/channeldb/codec.go +++ b/channeldb/codec.go @@ -218,7 +218,7 @@ func WriteElement(w io.Writer, element interface{}) error { } case net.Addr: - if err := serializeAddr(w, e); err != nil { + if err := SerializeAddr(w, e); err != nil { return err } @@ -228,7 +228,7 @@ func WriteElement(w io.Writer, element interface{}) error { } for _, addr := range e { - if err := serializeAddr(w, addr); err != nil { + if err := SerializeAddr(w, addr); err != nil { return err } } @@ -451,7 +451,7 @@ func ReadElement(r io.Reader, element interface{}) error { } case *net.Addr: - addr, err := deserializeAddr(r) + addr, err := DeserializeAddr(r) if err != nil { return err } @@ -465,7 +465,7 @@ func ReadElement(r io.Reader, element interface{}) error { *e = make([]net.Addr, numAddrs) for i := uint32(0); i < numAddrs; i++ { - addr, err := deserializeAddr(r) + addr, err := DeserializeAddr(r) if err != nil { return err } diff --git a/channeldb/graph.go b/channeldb/graph.go index d7a4480d0..e51c68753 100644 --- a/channeldb/graph.go +++ b/channeldb/graph.go @@ -4016,7 +4016,7 @@ func putLightningNode(nodeBucket kvdb.RwBucket, aliasBucket kvdb.RwBucket, // no } for _, address := range node.Addresses { - if err := serializeAddr(&b, address); err != nil { + if err := SerializeAddr(&b, address); err != nil { return err } } @@ -4209,7 +4209,7 @@ func deserializeLightningNode(r io.Reader) (LightningNode, error) { var addresses []net.Addr for i := 0; i < numAddresses; i++ { - address, err := deserializeAddr(r) + address, err := DeserializeAddr(r) if err != nil { return LightningNode{}, err } diff --git a/channeldb/nodes.go b/channeldb/nodes.go index b6e1573cf..b5b1cfdec 100644 --- a/channeldb/nodes.go +++ b/channeldb/nodes.go @@ -273,7 +273,7 @@ func serializeLinkNode(w io.Writer, l *LinkNode) error { } for _, addr := range l.Addresses { - if err := serializeAddr(w, addr); err != nil { + if err := SerializeAddr(w, addr); err != nil { return err } } @@ -315,7 +315,7 @@ func deserializeLinkNode(r io.Reader) (*LinkNode, error) { node.Addresses = make([]net.Addr, numAddrs) for i := uint32(0); i < numAddrs; i++ { - addr, err := deserializeAddr(r) + addr, err := DeserializeAddr(r) if err != nil { return nil, err }