@@ -30,31 +30,32 @@ type LedgerBackendKey struct {
3030
3131// Funder is a multi-ledger funder.
3232// funders is a map of LedgerIDs corresponding to a funder on some chain.
33- // egoisticChains is a map of LedgerIDs corresponding to a boolean indicating whether the chain should be funded last.
33+ // egoistic controls whether the funder uses the egoisticIndex to control the funding order.
34+ // egoisticIndex controls which participant index will fund last.
3435type Funder struct {
35- funders map [LedgerBackendKey ]channel.Funder
36- egoisticChains map [LedgerBackendKey ]bool
36+ funders map [LedgerBackendKey ]channel.Funder
37+ egoistic bool
38+ egoisticIndex int
3739}
3840
3941// NewFunder creates a new funder.
4042func NewFunder () * Funder {
4143 return & Funder {
42- funders : make (map [LedgerBackendKey ]channel.Funder ),
43- egoisticChains : make ( map [ LedgerBackendKey ] bool ) ,
44+ funders : make (map [LedgerBackendKey ]channel.Funder ),
45+ egoistic : false ,
4446 }
4547}
4648
4749// RegisterFunder registers a funder for a given ledger.
4850func (f * Funder ) RegisterFunder (l LedgerBackendID , lf channel.Funder ) {
4951 key := LedgerBackendKey {BackendID : l .BackendID (), LedgerID : string (l .LedgerID ().MapKey ())}
5052 f .funders [key ] = lf
51- f .egoisticChains [key ] = false
5253}
5354
54- // SetEgoisticChain sets the egoistic chain flag for a given ledger.
55- func (f * Funder ) SetEgoisticChain ( l LedgerBackendID , id int , egoistic bool ) {
56- key := LedgerBackendKey { BackendID : l . BackendID (), LedgerID : string ( l . LedgerID (). MapKey ())}
57- f .egoisticChains [ key ] = egoistic
55+ // SetEgoisticPart sets the egoistic chain flag for a given ledger.
56+ func (f * Funder ) SetEgoisticPart ( index int ) {
57+ f . egoisticIndex = index
58+ f .egoistic = true
5859}
5960
6061// Fund funds a multi-ledger channel. It dispatches funding calls to all
@@ -74,9 +75,8 @@ func (f *Funder) Fund(ctx context.Context, request channel.FundingReq) error {
7475 var egoisticLedgers []LedgerBackendID
7576 var nonEgoisticLedgers []LedgerBackendID
7677
77- for _ , l := range ledgerIDs {
78- key := LedgerBackendKey {BackendID : l .BackendID (), LedgerID : string (l .LedgerID ().MapKey ())}
79- if f .egoisticChains [key ] {
78+ for i , l := range ledgerIDs {
79+ if f .egoistic && f .egoisticIndex == i {
8080 egoisticLedgers = append (egoisticLedgers , l )
8181 } else {
8282 nonEgoisticLedgers = append (nonEgoisticLedgers , l )
@@ -128,6 +128,5 @@ func fundLedgers(ctx context.Context, request channel.FundingReq, assetIDs []Led
128128 return err
129129 }
130130 }
131-
132131 return nil
133132}
0 commit comments