45 lines
1.7 KiB
Go
45 lines
1.7 KiB
Go
package retrievaladapter
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
"github.com/filecoin-project/go-fil-markets/retrievalmarket"
|
|
retrievaltoken "github.com/filecoin-project/go-fil-markets/shared/tokenamount"
|
|
retrievaltypes "github.com/filecoin-project/go-fil-markets/shared/types"
|
|
"github.com/filecoin-project/go-sectorbuilder"
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/markets/utils"
|
|
"github.com/filecoin-project/lotus/storage"
|
|
)
|
|
|
|
type retrievalProviderNode struct {
|
|
miner *storage.Miner
|
|
sb sectorbuilder.Interface
|
|
full api.FullNode
|
|
}
|
|
|
|
// NewRetrievalProviderNode returns a new node adapter for a retrieval provider that talks to the
|
|
// Lotus Node
|
|
func NewRetrievalProviderNode(miner *storage.Miner, sb sectorbuilder.Interface, full api.FullNode) retrievalmarket.RetrievalProviderNode {
|
|
return &retrievalProviderNode{miner, sb, full}
|
|
}
|
|
|
|
func (rpn *retrievalProviderNode) UnsealSector(ctx context.Context, sectorID uint64, offset uint64, length uint64) (io.ReadCloser, error) {
|
|
si, err := rpn.miner.GetSectorInfo(sectorID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return rpn.sb.ReadPieceFromSealedSector(ctx, sectorID, offset, length, si.Ticket.TicketBytes, si.CommD)
|
|
}
|
|
|
|
func (rpn *retrievalProviderNode) SavePaymentVoucher(ctx context.Context, paymentChannel address.Address, voucher *retrievaltypes.SignedVoucher, proof []byte, expectedAmount retrievaltoken.TokenAmount) (retrievaltoken.TokenAmount, error) {
|
|
localVoucher, err := utils.FromSharedSignedVoucher(voucher)
|
|
if err != nil {
|
|
return retrievaltoken.FromInt(0), err
|
|
}
|
|
added, err := rpn.full.PaychVoucherAdd(ctx, paymentChannel, localVoucher, proof, utils.FromSharedTokenAmount(expectedAmount))
|
|
return utils.ToSharedTokenAmount(added), err
|
|
}
|