59 lines
1.7 KiB
Go
59 lines
1.7 KiB
Go
|
// Copyright (C) 2020 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package compensation
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/shopspring/decimal"
|
||
|
|
"github.com/spf13/pflag"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Rates configures the payment rates for network operations.
|
||
|
|
type Rates struct {
|
||
|
|
AtRestGBHours Rate // For data at rest in dollars per gigabyte-hour.
|
||
|
|
GetTB Rate // For data the node has sent for reads in dollars per terabyte.
|
||
|
|
PutTB Rate // For data the node has received for writes in dollars per terabyte.
|
||
|
|
GetRepairTB Rate // For data the node has sent for repairs in dollars per terabyte.
|
||
|
|
PutRepairTB Rate // For data the node has received for repairs in dollars per terabyte.
|
||
|
|
GetAuditTB Rate // For data the node has sent for audits in dollars per terabyte.
|
||
|
|
}
|
||
|
|
|
||
|
|
// Rate is a wrapper type around a decimal.Decimal.
|
||
|
|
type Rate decimal.Decimal
|
||
|
|
|
||
|
|
var _ pflag.Value = (*Rate)(nil)
|
||
|
|
|
||
|
|
// RateFromString parses the string form of the rate into a Rate.
|
||
|
|
func RateFromString(value string) (Rate, error) {
|
||
|
|
r, err := decimal.NewFromString(value)
|
||
|
|
if err != nil {
|
||
|
|
return Rate{}, err
|
||
|
|
}
|
||
|
|
return Rate(r), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// String returns the string form of the Rate.
|
||
|
|
func (rate Rate) String() string {
|
||
|
|
return decimal.Decimal(rate).String()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set updates the Rate to be equal to the parsed string.
|
||
|
|
func (rate *Rate) Set(s string) error {
|
||
|
|
r, err := decimal.NewFromString(s)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
*rate = Rate(r)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Type returns a unique string representing the type of the Rate.
|
||
|
|
func (rate Rate) Type() string {
|
||
|
|
return "rate"
|
||
|
|
}
|
||
|
|
|
||
|
|
// RequireRateFromString parses the Rate from the string or panics.
|
||
|
|
func RequireRateFromString(s string) Rate {
|
||
|
|
return Rate(decimal.RequireFromString(s))
|
||
|
|
}
|