85 lines
1.8 KiB
Go
85 lines
1.8 KiB
Go
// Copyright (C) 2020 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package storagenodedb
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/common/storj"
|
|
"storj.io/storj/storagenode/pricing"
|
|
)
|
|
|
|
// ensures that pricingDB implements pricing.DB interface.
|
|
var _ pricing.DB = (*pricingDB)(nil)
|
|
|
|
// ErrPricing represents errors from the pricing database.
|
|
var ErrPricing = errs.Class("pricing error")
|
|
|
|
// PricingDBName represents the database name.
|
|
const PricingDBName = "pricing"
|
|
|
|
// pricing works with node pricing DB.
|
|
//
|
|
// architecture: Database
|
|
type pricingDB struct {
|
|
dbContainerImpl
|
|
}
|
|
|
|
// Store inserts or updates pricing model into the db.
|
|
func (db *pricingDB) Store(ctx context.Context, pricing pricing.Pricing) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
query := `INSERT OR REPLACE INTO pricing (
|
|
satellite_id,
|
|
egress_bandwidth_price,
|
|
repair_bandwidth_price,
|
|
audit_bandwidth_price,
|
|
disk_space_price
|
|
) VALUES(?,?,?,?,?)`
|
|
|
|
_, err = db.ExecContext(ctx, query,
|
|
pricing.SatelliteID,
|
|
pricing.EgressBandwidth,
|
|
pricing.RepairBandwidth,
|
|
pricing.AuditBandwidth,
|
|
pricing.DiskSpace,
|
|
)
|
|
|
|
return ErrPricing.Wrap(err)
|
|
}
|
|
|
|
// Get retrieves pricing model for specific satellite.
|
|
func (db *pricingDB) Get(ctx context.Context, satelliteID storj.NodeID) (_ *pricing.Pricing, err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
pricingModel := pricing.Pricing{
|
|
SatelliteID: satelliteID,
|
|
}
|
|
|
|
row := db.QueryRowContext(ctx,
|
|
`SELECT egress_bandwidth_price,
|
|
repair_bandwidth_price,
|
|
audit_bandwidth_price,
|
|
disk_space_price
|
|
FROM pricing WHERE satellite_id = ?`,
|
|
satelliteID,
|
|
)
|
|
|
|
err = row.Scan(
|
|
&pricingModel.EgressBandwidth,
|
|
&pricingModel.RepairBandwidth,
|
|
&pricingModel.AuditBandwidth,
|
|
&pricingModel.DiskSpace,
|
|
)
|
|
|
|
if err == sql.ErrNoRows {
|
|
err = nil
|
|
}
|
|
|
|
return &pricingModel, ErrPricing.Wrap(err)
|
|
}
|