blockservice.go 1.84 KB
Newer Older
1 2 3 4
package blockservice

import (
	"fmt"
Jeromy's avatar
Jeromy committed
5
	"time"
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

	ds "github.com/jbenet/datastore.go"
	bitswap "github.com/jbenet/go-ipfs/bitswap"
	blocks "github.com/jbenet/go-ipfs/blocks"
	u "github.com/jbenet/go-ipfs/util"

	mh "github.com/jbenet/go-multihash"
)

// BlockService is a block datastore.
// It uses an internal `datastore.Datastore` instance to store values.
type BlockService struct {
	Datastore ds.Datastore
	Remote    *bitswap.BitSwap
}

// NewBlockService creates a BlockService with given datastore instance.
Jeromy's avatar
Jeromy committed
23
func NewBlockService(d ds.Datastore, rem *bitswap.BitSwap) (*BlockService, error) {
24 25 26
	if d == nil {
		return nil, fmt.Errorf("BlockService requires valid datastore")
	}
Jeromy's avatar
Jeromy committed
27
	if rem == nil {
28
		u.PErr("Caution: blockservice running in local (offline) mode.\n")
Jeromy's avatar
Jeromy committed
29 30
	}
	return &BlockService{Datastore: d, Remote: rem}, nil
31 32 33 34 35 36
}

// AddBlock adds a particular block to the service, Putting it into the datastore.
func (s *BlockService) AddBlock(b *blocks.Block) (u.Key, error) {
	k := b.Key()
	dsk := ds.NewKey(string(k))
Jeromy's avatar
Jeromy committed
37 38 39 40 41
	u.DOut("storing [%s] in datastore\n", k.Pretty())
	err := s.Datastore.Put(dsk, b.Data)
	if err != nil {
		return k, err
	}
42
	if s.Remote != nil {
43
		err = s.Remote.HaveBlock(b)
44
	}
Jeromy's avatar
Jeromy committed
45
	return k, err
46 47 48 49 50 51 52
}

// GetBlock retrieves a particular block from the service,
// Getting it from the datastore using the key (hash).
func (s *BlockService) GetBlock(k u.Key) (*blocks.Block, error) {
	dsk := ds.NewKey(string(k))
	datai, err := s.Datastore.Get(dsk)
Jeromy's avatar
Jeromy committed
53 54 55 56 57 58 59 60 61
	if err == nil {
		bdata, ok := datai.([]byte)
		if !ok {
			return nil, fmt.Errorf("data associated with %s is not a []byte", k)
		}
		return &blocks.Block{
			Multihash: mh.Multihash(k),
			Data:      bdata,
		}, nil
62
	} else if err == ds.ErrNotFound && s.Remote != nil {
Jeromy's avatar
Jeromy committed
63 64 65 66 67 68 69
		blk, err := s.Remote.GetBlock(k, time.Second*5)
		if err != nil {
			return nil, err
		}
		return blk, nil
	} else {
		return nil, u.ErrNotFound
70 71
	}
}