mirror of
https://github.com/lightningdevkit/rust-lightning.git
synced 2025-02-27 00:06:34 +01:00
67736b7480
P2PGossipSync logs before delegating to NetworkGraph in its EventHandler. In order to share this handling with RapidGossipSync, NetworkGraph needs to take a logger so that it can implement EventHandler instead.
88 lines
4.6 KiB
Rust
88 lines
4.6 KiB
Rust
// This file is licensed under the Apache License, Version 2.0 <LICENSE-APACHE
|
|
// or http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your option.
|
|
// You may not use this file except in accordance with one or both of these
|
|
// licenses.
|
|
|
|
//! This module contains a simple key-value store trait KVStorePersister that
|
|
//! allows one to implement the persistence for [`ChannelManager`], [`NetworkGraph`],
|
|
//! and [`ChannelMonitor`] all in one place.
|
|
|
|
use core::ops::Deref;
|
|
use bitcoin::hashes::hex::ToHex;
|
|
use io::{self};
|
|
use routing::scoring::WriteableScore;
|
|
|
|
use crate::{chain::{keysinterface::{Sign, KeysInterface}, self, transaction::{OutPoint}, chaininterface::{BroadcasterInterface, FeeEstimator}, chainmonitor::{Persist, MonitorUpdateId}, channelmonitor::{ChannelMonitor, ChannelMonitorUpdate}}, ln::channelmanager::ChannelManager, routing::gossip::NetworkGraph};
|
|
use super::{logger::Logger, ser::Writeable};
|
|
|
|
/// Trait for a key-value store for persisting some writeable object at some key
|
|
/// Implementing `KVStorePersister` provides auto-implementations for [`Persister`]
|
|
/// and [`Persist`] traits. It uses "manager", "network_graph",
|
|
/// and "monitors/{funding_txo_id}_{funding_txo_index}" for keys.
|
|
pub trait KVStorePersister {
|
|
/// Persist the given writeable using the provided key
|
|
fn persist<W: Writeable>(&self, key: &str, object: &W) -> io::Result<()>;
|
|
}
|
|
|
|
/// Trait that handles persisting a [`ChannelManager`], [`NetworkGraph`], and [`WriteableScore`] to disk.
|
|
pub trait Persister<'a, Signer: Sign, M: Deref, T: Deref, K: Deref, F: Deref, L: Deref, S>
|
|
where M::Target: 'static + chain::Watch<Signer>,
|
|
T::Target: 'static + BroadcasterInterface,
|
|
K::Target: 'static + KeysInterface<Signer = Signer>,
|
|
F::Target: 'static + FeeEstimator,
|
|
L::Target: 'static + Logger,
|
|
S: WriteableScore<'a>,
|
|
{
|
|
/// Persist the given ['ChannelManager'] to disk, returning an error if persistence failed.
|
|
fn persist_manager(&self, channel_manager: &ChannelManager<Signer, M, T, K, F, L>) -> Result<(), io::Error>;
|
|
|
|
/// Persist the given [`NetworkGraph`] to disk, returning an error if persistence failed.
|
|
fn persist_graph(&self, network_graph: &NetworkGraph<L>) -> Result<(), io::Error>;
|
|
|
|
/// Persist the given [`WriteableScore`] to disk, returning an error if persistence failed.
|
|
fn persist_scorer(&self, scorer: &S) -> Result<(), io::Error>;
|
|
}
|
|
|
|
impl<'a, A: KVStorePersister, Signer: Sign, M: Deref, T: Deref, K: Deref, F: Deref, L: Deref, S> Persister<'a, Signer, M, T, K, F, L, S> for A
|
|
where M::Target: 'static + chain::Watch<Signer>,
|
|
T::Target: 'static + BroadcasterInterface,
|
|
K::Target: 'static + KeysInterface<Signer = Signer>,
|
|
F::Target: 'static + FeeEstimator,
|
|
L::Target: 'static + Logger,
|
|
S: WriteableScore<'a>,
|
|
{
|
|
/// Persist the given ['ChannelManager'] to disk with the name "manager", returning an error if persistence failed.
|
|
fn persist_manager(&self, channel_manager: &ChannelManager<Signer, M, T, K, F, L>) -> Result<(), io::Error> {
|
|
self.persist("manager", channel_manager)
|
|
}
|
|
|
|
/// Persist the given [`NetworkGraph`] to disk with the name "network_graph", returning an error if persistence failed.
|
|
fn persist_graph(&self, network_graph: &NetworkGraph<L>) -> Result<(), io::Error> {
|
|
self.persist("network_graph", network_graph)
|
|
}
|
|
|
|
/// Persist the given [`WriteableScore`] to disk with name "scorer", returning an error if persistence failed.
|
|
fn persist_scorer(&self, scorer: &S) -> Result<(), io::Error> {
|
|
self.persist("scorer", &scorer)
|
|
}
|
|
}
|
|
|
|
impl<ChannelSigner: Sign, K: KVStorePersister> Persist<ChannelSigner> for K {
|
|
// TODO: We really need a way for the persister to inform the user that its time to crash/shut
|
|
// down once these start returning failure.
|
|
// A PermanentFailure implies we need to shut down since we're force-closing channels without
|
|
// even broadcasting!
|
|
|
|
fn persist_new_channel(&self, funding_txo: OutPoint, monitor: &ChannelMonitor<ChannelSigner>, _update_id: MonitorUpdateId) -> Result<(), chain::ChannelMonitorUpdateErr> {
|
|
let key = format!("monitors/{}_{}", funding_txo.txid.to_hex(), funding_txo.index);
|
|
self.persist(&key, monitor)
|
|
.map_err(|_| chain::ChannelMonitorUpdateErr::PermanentFailure)
|
|
}
|
|
|
|
fn update_persisted_channel(&self, funding_txo: OutPoint, _update: &Option<ChannelMonitorUpdate>, monitor: &ChannelMonitor<ChannelSigner>, _update_id: MonitorUpdateId) -> Result<(), chain::ChannelMonitorUpdateErr> {
|
|
let key = format!("monitors/{}_{}", funding_txo.txid.to_hex(), funding_txo.index);
|
|
self.persist(&key, monitor)
|
|
.map_err(|_| chain::ChannelMonitorUpdateErr::PermanentFailure)
|
|
}
|
|
}
|