mirror of
https://github.com/ipfs/kubo.git
synced 2026-03-09 18:28:08 +08:00
49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
package blockstore
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"gx/ipfs/QmRg1gKTHzc3CZXSKzem8aR4E3TubFhbgXwfVuWnSK5CC5/go-metrics-interface"
|
|
context "gx/ipfs/QmZy2y8t9zQH2a1b8q2ZSLKp17ATuJoCNxxyMFG5qFExpt/go-net/context"
|
|
)
|
|
|
|
// Next to each option is it aproximate memory usage per unit
|
|
type CacheOpts struct {
|
|
HasBloomFilterSize int // 1 bit
|
|
HasBloomFilterHashes int // No size, 7 is usually best, consult bloom papers
|
|
HasARCCacheSize int // 32 bytes
|
|
}
|
|
|
|
func DefaultCacheOpts() CacheOpts {
|
|
return CacheOpts{
|
|
HasBloomFilterSize: 512 * 8 * 1024,
|
|
HasBloomFilterHashes: 7,
|
|
HasARCCacheSize: 64 * 1024,
|
|
}
|
|
}
|
|
|
|
func CachedBlockstore(bs GCBlockstore,
|
|
ctx context.Context, opts CacheOpts) (cbs GCBlockstore, err error) {
|
|
cbs = bs
|
|
|
|
if opts.HasBloomFilterSize < 0 || opts.HasBloomFilterHashes < 0 ||
|
|
opts.HasARCCacheSize < 0 {
|
|
return nil, errors.New("all options for cache need to be greater than zero")
|
|
}
|
|
|
|
if opts.HasBloomFilterSize != 0 && opts.HasBloomFilterHashes == 0 {
|
|
return nil, errors.New("bloom filter hash count can't be 0 when there is size set")
|
|
}
|
|
|
|
ctx = metrics.CtxSubScope(ctx, "bs.cache")
|
|
|
|
if opts.HasARCCacheSize > 0 {
|
|
cbs, err = newARCCachedBS(ctx, cbs, opts.HasARCCacheSize)
|
|
}
|
|
if opts.HasBloomFilterSize != 0 {
|
|
cbs, err = bloomCached(cbs, ctx, opts.HasBloomFilterSize, opts.HasBloomFilterHashes)
|
|
}
|
|
|
|
return cbs, err
|
|
}
|