mirror of
https://github.com/lightninglabs/faraday.git
synced 2026-08-18 13:09:18 +02:00
6f4c01d0f6
To avoid the frdrpc package having dependencies to other libraries and making using the JSON/WASM stubs hard, we extract all the business logic and RPC server code into its own package called frdrpcserver.
101 lines
3.2 KiB
Go
101 lines
3.2 KiB
Go
package frdrpcserver
|
|
|
|
import (
|
|
"context"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/lightninglabs/faraday/frdrpc"
|
|
"github.com/lightninglabs/faraday/insights"
|
|
"github.com/lightninglabs/faraday/recommend"
|
|
)
|
|
|
|
// parseRecommendationRequest parses a close recommendation request and
|
|
// returns the config required to get recommendations.
|
|
func parseRecommendationRequest(ctx context.Context, cfg *Config,
|
|
req *frdrpc.CloseRecommendationRequest) *recommend.CloseRecommendationConfig {
|
|
|
|
// Create a close recommendations config with the minimum monitored
|
|
// value provided in the request and the default outlier multiplier.
|
|
recCfg := &recommend.CloseRecommendationConfig{
|
|
ChannelInsights: func() ([]*insights.ChannelInfo, error) {
|
|
return channelInsights(ctx, cfg)
|
|
},
|
|
MinimumMonitored: time.Second *
|
|
time.Duration(req.MinimumMonitored),
|
|
}
|
|
|
|
// Get the metric that the recommendations are being calculated based
|
|
// on.
|
|
switch req.Metric {
|
|
case frdrpc.CloseRecommendationRequest_UPTIME:
|
|
recCfg.Metric = recommend.UptimeMetric
|
|
|
|
case frdrpc.CloseRecommendationRequest_REVENUE:
|
|
recCfg.Metric = recommend.RevenueMetric
|
|
|
|
case frdrpc.CloseRecommendationRequest_INCOMING_VOLUME:
|
|
recCfg.Metric = recommend.IncomingVolume
|
|
|
|
case frdrpc.CloseRecommendationRequest_OUTGOING_VOLUME:
|
|
recCfg.Metric = recommend.OutgoingVolume
|
|
|
|
case frdrpc.CloseRecommendationRequest_TOTAL_VOLUME:
|
|
recCfg.Metric = recommend.Volume
|
|
}
|
|
|
|
return recCfg
|
|
}
|
|
|
|
// parseOutlierRequest parses a rpc outlier recommendation request and returns
|
|
// the close recommendation config and multiplier required.
|
|
func parseOutlierRequest(ctx context.Context, cfg *Config,
|
|
req *frdrpc.OutlierRecommendationsRequest) (
|
|
*recommend.CloseRecommendationConfig, float64) {
|
|
|
|
multiplier := recommend.DefaultOutlierMultiplier
|
|
if req.OutlierMultiplier != 0 {
|
|
multiplier = float64(req.OutlierMultiplier)
|
|
}
|
|
|
|
return parseRecommendationRequest(ctx, cfg, req.RecRequest), multiplier
|
|
}
|
|
|
|
// parseThresholdRequest parses a rpc threshold recommendation request and
|
|
// returns the close recommendation config and threshold required. The above
|
|
// threshold boolean is inverted to allow for
|
|
// a default that returns values below a threshold.
|
|
func parseThresholdRequest(ctx context.Context, cfg *Config,
|
|
req *frdrpc.ThresholdRecommendationsRequest) (
|
|
*recommend.CloseRecommendationConfig, float64) {
|
|
|
|
return parseRecommendationRequest(ctx, cfg, req.RecRequest),
|
|
float64(req.ThresholdValue)
|
|
}
|
|
|
|
// rpcResponse parses the response obtained getting a close recommendation
|
|
// and converts it to a close recommendation response.
|
|
func rpcResponse(report *recommend.Report) *frdrpc.CloseRecommendationsResponse {
|
|
resp := &frdrpc.CloseRecommendationsResponse{
|
|
TotalChannels: int32(report.TotalChannels),
|
|
ConsideredChannels: int32(report.ConsideredChannels),
|
|
}
|
|
|
|
for chanPoint, rec := range report.Recommendations {
|
|
resp.Recommendations = append(
|
|
resp.Recommendations, &frdrpc.Recommendation{
|
|
ChanPoint: chanPoint,
|
|
Value: float32(rec.Value),
|
|
RecommendClose: rec.RecommendClose,
|
|
},
|
|
)
|
|
}
|
|
|
|
// Sort the recommendations returned by value.
|
|
sort.SliceStable(resp.Recommendations, func(i, j int) bool {
|
|
return resp.Recommendations[i].Value <
|
|
resp.Recommendations[j].Value
|
|
})
|
|
|
|
return resp
|
|
}
|