highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-08-03 10:12:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

channeldb: export route serialization

Browse Source
This commit is contained in:
Joost Jager
2019-06-14 15:01:48 +02:00
parent 67ddb72eab
commit eb4e65e54f
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
channeldb/payments.go
View File

@@ -484,7 +484,7 @@ func serializePaymentAttemptInfo(w io.Writer, a *PaymentAttemptInfo) error {
return err return err
} }
if err := serializeRoute(w, a.Route); err != nil { if err := SerializeRoute(w, a.Route); err != nil {
return err return err
} }
@@ -497,7 +497,7 @@ func deserializePaymentAttemptInfo(r io.Reader) (*PaymentAttemptInfo, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
a.Route, err = deserializeRoute(r) a.Route, err = DeserializeRoute(r)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -533,7 +533,8 @@ func deserializeHop(r io.Reader) (*route.Hop, error) {
return h, nil return h, nil
} }
func serializeRoute(w io.Writer, r route.Route) error { // SerializeRoute serializes a route.
func SerializeRoute(w io.Writer, r route.Route) error {
if err := WriteElements(w, if err := WriteElements(w,
r.TotalTimeLock, r.TotalAmount, r.SourcePubKey[:], r.TotalTimeLock, r.TotalAmount, r.SourcePubKey[:],
); err != nil { ); err != nil {
@@ -553,7 +554,8 @@ func serializeRoute(w io.Writer, r route.Route) error {
return nil return nil
} }
func deserializeRoute(r io.Reader) (route.Route, error) { // DeserializeRoute deserializes a route.
func DeserializeRoute(r io.Reader) (route.Route, error) {
rt := route.Route{} rt := route.Route{}
if err := ReadElements(r, if err := ReadElements(r,
&rt.TotalTimeLock, &rt.TotalAmount, &rt.TotalTimeLock, &rt.TotalAmount,

4
channeldb/payments_test.go
View File

@@ -203,12 +203,12 @@ func TestRouteSerialization(t *testing.T) {
t.Parallel() t.Parallel()
var b bytes.Buffer var b bytes.Buffer
if err := serializeRoute(&b, testRoute); err != nil { if err := SerializeRoute(&b, testRoute); err != nil {
t.Fatal(err) t.Fatal(err)
} }
r := bytes.NewReader(b.Bytes()) r := bytes.NewReader(b.Bytes())
route2, err := deserializeRoute(r) route2, err := DeserializeRoute(r)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }