86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
package market
|
|
|
|
import (
|
|
"context"
|
|
"github.com/filecoin-project/specs-actors/actors/abi/big"
|
|
"sync"
|
|
|
|
"github.com/filecoin-project/specs-actors/actors/builtin"
|
|
"github.com/ipfs/go-cid"
|
|
logging "github.com/ipfs/go-log"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
"github.com/filecoin-project/lotus/chain/actors"
|
|
"github.com/filecoin-project/lotus/chain/stmgr"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/node/impl/full"
|
|
)
|
|
|
|
var log = logging.Logger("market_adapter")
|
|
|
|
type FundMgr struct {
|
|
sm *stmgr.StateManager
|
|
mpool full.MpoolAPI
|
|
|
|
lk sync.Mutex
|
|
available map[address.Address]types.BigInt
|
|
}
|
|
|
|
func NewFundMgr(sm *stmgr.StateManager, mpool full.MpoolAPI) *FundMgr {
|
|
return &FundMgr{
|
|
sm: sm,
|
|
mpool: mpool,
|
|
|
|
available: map[address.Address]types.BigInt{},
|
|
}
|
|
}
|
|
|
|
func (fm *FundMgr) EnsureAvailable(ctx context.Context, addr, wallet address.Address, amt types.BigInt) (cid.Cid, error) {
|
|
fm.lk.Lock()
|
|
avail, ok := fm.available[addr]
|
|
if !ok {
|
|
bal, err := fm.sm.MarketBalance(ctx, addr, nil)
|
|
if err != nil {
|
|
fm.lk.Unlock()
|
|
return cid.Undef, err
|
|
}
|
|
|
|
avail = types.BigSub(bal.Escrow, bal.Locked)
|
|
}
|
|
|
|
toAdd := types.NewInt(0)
|
|
avail = types.BigSub(avail, amt)
|
|
if avail.LessThan(types.NewInt(0)) {
|
|
// TODO: some rules around adding more to avoid doing stuff on-chain
|
|
// all the time
|
|
toAdd = avail.Neg()
|
|
avail = types.NewInt(0)
|
|
}
|
|
fm.available[addr] = avail
|
|
|
|
fm.lk.Unlock()
|
|
|
|
if toAdd.LessThanEqual(big.Zero()) {
|
|
return cid.Undef, nil
|
|
} else {
|
|
var err error
|
|
params, err := actors.SerializeParams(&addr)
|
|
if err != nil {
|
|
return cid.Undef, err
|
|
}
|
|
|
|
smsg, err := fm.mpool.MpoolPushMessage(ctx, &types.Message{
|
|
To: builtin.StorageMarketActorAddr,
|
|
From: wallet,
|
|
Value: toAdd,
|
|
Method: builtin.MethodsMarket.AddBalance,
|
|
Params: params,
|
|
})
|
|
if err != nil {
|
|
return cid.Undef, err
|
|
}
|
|
|
|
return smsg.Cid(), nil
|
|
}
|
|
}
|