-
Notifications
You must be signed in to change notification settings - Fork 59
/
unixfs_testutil.go
69 lines (56 loc) · 1.67 KB
/
unixfs_testutil.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package unixfs
import (
"context"
"os"
"testing"
chunk "github.com/ipfs/boxo/chunker"
files "github.com/ipfs/boxo/files"
"github.com/ipfs/boxo/ipld/merkledag"
"github.com/ipfs/boxo/ipld/unixfs/importer/balanced"
ihelper "github.com/ipfs/boxo/ipld/unixfs/importer/helpers"
"github.com/ipfs/go-cid"
"github.com/ipfs/go-cidutil"
ipldformat "github.com/ipfs/go-ipld-format"
mh "github.com/multiformats/go-multihash"
"github.com/stretchr/testify/require"
)
const (
defaultHashFunction = uint64(mh.BLAKE2B_MIN + 31)
unixfsChunkSize = uint64(1 << 10)
unixfsLinksPerLevel = 1024
)
func WriteUnixfsDAGTo(t *testing.T, path string, into ipldformat.DAGService) cid.Cid {
file, err := os.Open(path)
require.NoError(t, err)
defer file.Close()
stat, err := file.Stat()
require.NoError(t, err)
// get a IPLD reader path file
// required to write the Unixfs DAG blocks to a filestore
rpf, err := files.NewReaderPathFile(file.Name(), file, stat)
require.NoError(t, err)
// generate the dag and get the root
// import to UnixFS
prefix, err := merkledag.PrefixForCidVersion(1)
require.NoError(t, err)
prefix.MhType = defaultHashFunction
bufferedDS := ipldformat.NewBufferedDAG(context.Background(), into)
params := ihelper.DagBuilderParams{
Maxlinks: unixfsLinksPerLevel,
RawLeaves: true,
CidBuilder: cidutil.InlineBuilder{
Builder: prefix,
Limit: 126,
},
Dagserv: bufferedDS,
NoCopy: true,
}
db, err := params.New(chunk.NewSizeSplitter(rpf, int64(unixfsChunkSize)))
require.NoError(t, err)
nd, err := balanced.Layout(db)
require.NoError(t, err)
err = bufferedDS.Commit()
require.NoError(t, err)
require.NoError(t, rpf.Close())
return nd.Cid()
}