/
block_getter.go
65 lines (57 loc) · 2.33 KB
/
block_getter.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
// Copyright 2016 Keybase Inc. All rights reserved.
// Use of this source code is governed by a BSD
// license that can be found in the LICENSE file.
package libkbfs
import (
"github.com/keybase/client/go/kbfs/data"
"github.com/keybase/client/go/kbfs/kbfscrypto"
"github.com/keybase/client/go/kbfs/libkey"
"golang.org/x/net/context"
)
// blockGetter provides the API for the block retrieval worker to obtain blocks.
type blockGetter interface {
getBlock(
context.Context, libkey.KeyMetadata, data.BlockPointer, data.Block,
DiskBlockCacheType) error
assembleBlock(
context.Context, libkey.KeyMetadata, data.BlockPointer, data.Block,
[]byte, kbfscrypto.BlockCryptKeyServerHalf) error
assembleBlockLocal(
context.Context, libkey.KeyMetadata, data.BlockPointer, data.Block,
[]byte, kbfscrypto.BlockCryptKeyServerHalf) error
}
// realBlockGetter obtains real blocks using the APIs available in Config.
type realBlockGetter struct {
config blockOpsConfig
}
// getBlock implements the interface for realBlockGetter.
func (bg *realBlockGetter) getBlock(
ctx context.Context, kmd libkey.KeyMetadata, blockPtr data.BlockPointer,
block data.Block, cacheType DiskBlockCacheType) error {
bserv := bg.config.BlockServer()
buf, blockServerHalf, err := bserv.Get(
ctx, kmd.TlfID(), blockPtr.ID, blockPtr.Context, cacheType)
if err != nil {
return err
}
if _, isLocal := bserv.(blockServerLocal); isLocal {
return assembleBlockLocal(
ctx, bg.config.keyGetter(), bg.config.Codec(),
bg.config.cryptoPure(), kmd, blockPtr, block, buf, blockServerHalf)
}
return assembleBlock(
ctx, bg.config.keyGetter(), bg.config.Codec(), bg.config.cryptoPure(),
kmd, blockPtr, block, buf, blockServerHalf)
}
func (bg *realBlockGetter) assembleBlock(ctx context.Context,
kmd libkey.KeyMetadata, ptr data.BlockPointer, block data.Block, buf []byte,
serverHalf kbfscrypto.BlockCryptKeyServerHalf) error {
return assembleBlock(ctx, bg.config.keyGetter(), bg.config.Codec(),
bg.config.cryptoPure(), kmd, ptr, block, buf, serverHalf)
}
func (bg *realBlockGetter) assembleBlockLocal(ctx context.Context,
kmd libkey.KeyMetadata, ptr data.BlockPointer, block data.Block, buf []byte,
serverHalf kbfscrypto.BlockCryptKeyServerHalf) error {
return assembleBlockLocal(ctx, bg.config.keyGetter(), bg.config.Codec(),
bg.config.cryptoPure(), kmd, ptr, block, buf, serverHalf)
}