457f909a74
* fix - Satellite crashing on receiving a manipulated bandwidthagreement * provider.PeerIdentityFromContext called twice. Remove one * add storage node ID to serial number * remove serialNum query and transaction * add uuid to GeneratePayerBandwidthAllocation for testing * enable expected failure on duplicate serialnum cases * Revert "enable expected failure on duplicate serialnum cases" This reverts commit 5948f43ed1741c280f0bb34a86c1c490365417bc. * enable expected failure on duplicate serialnum cases
59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package satellitedb
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"storj.io/storj/pkg/bwagreement"
|
|
dbx "storj.io/storj/satellite/satellitedb/dbx"
|
|
)
|
|
|
|
type bandwidthagreement struct {
|
|
db *dbx.DB
|
|
}
|
|
|
|
func (b *bandwidthagreement) CreateAgreement(ctx context.Context, serialNum string, agreement bwagreement.Agreement) error {
|
|
_, err := b.db.Create_Bwagreement(
|
|
ctx,
|
|
dbx.Bwagreement_Signature(agreement.Signature),
|
|
dbx.Bwagreement_Serialnum(serialNum),
|
|
dbx.Bwagreement_Data(agreement.Agreement),
|
|
)
|
|
return err
|
|
}
|
|
|
|
func (b *bandwidthagreement) GetAgreements(ctx context.Context) ([]bwagreement.Agreement, error) {
|
|
rows, err := b.db.All_Bwagreement(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
agreements := make([]bwagreement.Agreement, len(rows))
|
|
for i, entry := range rows {
|
|
agreement := &agreements[i]
|
|
agreement.Signature = entry.Signature
|
|
agreement.Agreement = entry.Data
|
|
agreement.CreatedAt = entry.CreatedAt
|
|
}
|
|
return agreements, nil
|
|
}
|
|
|
|
func (b *bandwidthagreement) GetAgreementsSince(ctx context.Context, since time.Time) ([]bwagreement.Agreement, error) {
|
|
rows, err := b.db.All_Bwagreement_By_CreatedAt_Greater(ctx, dbx.Bwagreement_CreatedAt(since))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
agreements := make([]bwagreement.Agreement, len(rows))
|
|
for i, entry := range rows {
|
|
agreement := &agreements[i]
|
|
agreement.Signature = entry.Signature
|
|
agreement.Agreement = entry.Data
|
|
agreement.CreatedAt = entry.CreatedAt
|
|
}
|
|
return agreements, nil
|
|
}
|