You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
regolancer/channels.go

66 lines
1.6 KiB
Go

2 years ago
package main
import (
"context"
"math"
"math/rand"
"time"
"github.com/lightningnetwork/lnd/lnrpc"
)
func (r *regolancer) getChannels() error {
ctx, cancel := context.WithTimeout(context.Background(), time.Second*30)
defer cancel()
channels, err := r.lnClient.ListChannels(ctx, &lnrpc.ListChannelsRequest{ActiveOnly: true, PublicOnly: true})
if err != nil {
return err
}
r.channels = channels.Channels
return nil
}
func (r *regolancer) getChannelCandidates(fromPerc, toPerc, amount int64, ignore []uint64) error {
ignoredSet := map[uint64]struct{}{}
for _, cid := range ignore {
ignoredSet[cid] = struct{}{}
}
2 years ago
for _, c := range r.channels {
if c.LocalBalance < c.Capacity*toPerc/100 && c.LocalBalance+amount < c.Capacity/2 {
r.toChannels = append(r.toChannels, c)
}
if c.RemoteBalance < c.Capacity*fromPerc/100 && c.RemoteBalance-amount < c.Capacity/2 {
if _, ok := ignoredSet[c.ChanId]; ok {
continue
}
2 years ago
r.fromChannels = append(r.fromChannels, c)
}
}
return nil
}
func min(args ...int64) (result int64) {
result = math.MaxInt64
for _, a := range args {
if a < result {
result = a
}
}
return
}
func (r *regolancer) pickChannelPair(amount int64) (from uint64, to uint64, maxAmount int64) {
fromIdx := rand.Int31n(int32(len(r.fromChannels)))
toIdx := rand.Int31n(int32(len(r.toChannels)))
fromChan := r.fromChannels[fromIdx]
toChan := r.toChannels[toIdx]
maxFrom := fromChan.Capacity/2 - fromChan.RemoteBalance
maxTo := toChan.Capacity/2 - toChan.LocalBalance
if amount == 0 {
maxAmount = min(maxFrom, maxTo)
} else {
maxAmount = min(maxFrom, maxTo, amount)
}
return fromChan.ChanId, toChan.ChanId, maxAmount
}