mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-21 18:37:45 +08:00
54 lines
1.0 KiB
Go
54 lines
1.0 KiB
Go
package coreapi
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/ipfs/go-ipfs/pin"
|
|
|
|
cid "gx/ipfs/QmR8BauakNcBa3RbE4nbQu76PDiJgoQgz8AJdhJuiU4TAw/go-cid"
|
|
ipld "gx/ipfs/QmRL22E4paat7ky7vx9MLpR97JHHbFPrg3ytFQw6qp1y1s/go-ipld-format"
|
|
)
|
|
|
|
type dagAPI struct {
|
|
ipld.DAGService
|
|
|
|
core *CoreAPI
|
|
}
|
|
|
|
type pinningAdder CoreAPI
|
|
|
|
func (adder *pinningAdder) Add(ctx context.Context, nd ipld.Node) error {
|
|
defer adder.blockstore.PinLock().Unlock()
|
|
|
|
if err := adder.dag.Add(ctx, nd); err != nil {
|
|
return err
|
|
}
|
|
|
|
adder.pinning.PinWithMode(nd.Cid(), pin.Recursive)
|
|
|
|
return adder.pinning.Flush()
|
|
}
|
|
|
|
func (adder *pinningAdder) AddMany(ctx context.Context, nds []ipld.Node) error {
|
|
defer adder.blockstore.PinLock().Unlock()
|
|
|
|
if err := adder.dag.AddMany(ctx, nds); err != nil {
|
|
return err
|
|
}
|
|
|
|
cids := cid.NewSet()
|
|
|
|
for _, nd := range nds {
|
|
c := nd.Cid()
|
|
if cids.Visit(c) {
|
|
adder.pinning.PinWithMode(c, pin.Recursive)
|
|
}
|
|
}
|
|
|
|
return adder.pinning.Flush()
|
|
}
|
|
|
|
func (api *dagAPI) Pinning() ipld.NodeAdder {
|
|
return (*pinningAdder)(api.core)
|
|
}
|