highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-11-10 14:17:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

watchtower: add PostEvaluateFilterFn call-back option

Browse Source 
This commit adds a new PostEvaluateFilterFn call-back option to
`wtdb.ClientSessionListCfg`. This call-back is run _after_ all the other
evaluation call-backs in `wtdb.ClientSessionListCfg` and is only used to
determine if the session should be included in the returned list or not.
This commit is contained in:
Elle Mouton
2023-03-20 17:28:53 +02:00
parent 7bc86ca42e
commit 49cd23725a
2 changed files with 33 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
watchtower/wtdb/client_db.go
View File

@@ -2145,6 +2145,12 @@ type ClientSessionListCfg struct {
// function, then it will not be passed to any of the other call backs // function, then it will not be passed to any of the other call backs
// and won't be included in the return list. // and won't be included in the return list.
PreEvaluateFilterFn ClientSessionFilterFn PreEvaluateFilterFn ClientSessionFilterFn
// PostEvaluateFilterFn will be run _after_ all the other call-back
// functions in ClientSessionListCfg. If a session fails this filter
// function then all it means is that it won't be included in the list
// of sessions to return.
PostEvaluateFilterFn ClientSessionFilterFn
} }
// NewClientSessionCfg constructs a new ClientSessionListCfg. // NewClientSessionCfg constructs a new ClientSessionListCfg.
@@ -2189,6 +2195,19 @@ func WithPreEvalFilterFn(fn ClientSessionFilterFn) ClientSessionListOption {
} }
} }
// WithPostEvalFilterFn constructs a functional option that will set a call-back
// function that will be used to determine if a session should be included in
// the returned list. This differs from WithPreEvalFilterFn since that call-back
// is used to determine if the session should be evaluated at all (and thus
// run against the other ClientSessionListCfg call-backs) whereas the session
// will only reach the PostEvalFilterFn call-back once it has already been
// evaluated by all the other call-backs.
func WithPostEvalFilterFn(fn ClientSessionFilterFn) ClientSessionListOption {
return func(cfg *ClientSessionListCfg) {
cfg.PostEvaluateFilterFn = fn
}
}
// getClientSession loads the full ClientSession associated with the serialized // getClientSession loads the full ClientSession associated with the serialized
// session id. This method populates the CommittedUpdates, AckUpdates and Tower // session id. This method populates the CommittedUpdates, AckUpdates and Tower
// in addition to the ClientSession's body. // in addition to the ClientSession's body.
@@ -2232,6 +2251,12 @@ func (c *ClientDB) getClientSession(sessionsBkt, chanIDIndexBkt kvdb.RBucket,
return nil, err return nil, err
} }
if cfg.PostEvaluateFilterFn != nil &&
!cfg.PostEvaluateFilterFn(session) {
return nil, ErrSessionFailedFilterFn
}
return session, nil return session, nil
} }

10
watchtower/wtmock/client_db.go
View File

@@ -260,8 +260,6 @@ func (m *ClientDB) listClientSessions(tower *wtdb.TowerID,
continue continue
} }
sessions[session.ID] = &session
if cfg.PerMaxHeight != nil { if cfg.PerMaxHeight != nil {
for chanID, index := range m.ackedUpdates[session.ID] { for chanID, index := range m.ackedUpdates[session.ID] {
cfg.PerMaxHeight( cfg.PerMaxHeight(
@@ -285,6 +283,14 @@ func (m *ClientDB) listClientSessions(tower *wtdb.TowerID,
cfg.PerCommittedUpdate(&session, &update) cfg.PerCommittedUpdate(&session, &update)
} }
} }
if cfg.PostEvaluateFilterFn != nil &&
!cfg.PostEvaluateFilterFn(&session) {
continue
}
sessions[session.ID] = &session
} }
return sessions, nil return sessions, nil