136 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			136 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2016 The go-ethereum Authors
 | |
| // This file is part of the go-ethereum library.
 | |
| //
 | |
| // The go-ethereum library is free software: you can redistribute it and/or modify
 | |
| // it under the terms of the GNU Lesser General Public License as published by
 | |
| // the Free Software Foundation, either version 3 of the License, or
 | |
| // (at your option) any later version.
 | |
| //
 | |
| // The go-ethereum library is distributed in the hope that it will be useful,
 | |
| // but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 | |
| // GNU Lesser General Public License for more details.
 | |
| //
 | |
| // You should have received a copy of the GNU Lesser General Public License
 | |
| // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
 | |
| 
 | |
| package storage
 | |
| 
 | |
| import (
 | |
| 	"fmt"
 | |
| 	"path/filepath"
 | |
| 	"time"
 | |
| 
 | |
| 	"github.com/ethereum/go-ethereum/log"
 | |
| )
 | |
| 
 | |
| /*
 | |
| NetStore is a cloud storage access abstaction layer for swarm
 | |
| it contains the shared logic of network served chunk store/retrieval requests
 | |
| both local (coming from DPA api) and remote (coming from peers via bzz protocol)
 | |
| it implements the ChunkStore interface and embeds LocalStore
 | |
| 
 | |
| It is called by the bzz protocol instances via Depo (the store/retrieve request handler)
 | |
| a protocol instance is running on each peer, so this is heavily parallelised.
 | |
| NetStore falls back to a backend (CloudStorage interface)
 | |
| implemented by bzz/network/forwarder. forwarder or IPFS or IPΞS
 | |
| */
 | |
| type NetStore struct {
 | |
| 	hashfunc   SwarmHasher
 | |
| 	localStore *LocalStore
 | |
| 	cloud      CloudStore
 | |
| }
 | |
| 
 | |
| // backend engine for cloud store
 | |
| // It can be aggregate dispatching to several parallel implementations:
 | |
| // bzz/network/forwarder. forwarder or IPFS or IPΞS
 | |
| type CloudStore interface {
 | |
| 	Store(*Chunk)
 | |
| 	Deliver(*Chunk)
 | |
| 	Retrieve(*Chunk)
 | |
| }
 | |
| 
 | |
| type StoreParams struct {
 | |
| 	ChunkDbPath   string
 | |
| 	DbCapacity    uint64
 | |
| 	CacheCapacity uint
 | |
| 	Radius        int
 | |
| }
 | |
| 
 | |
| func NewStoreParams(path string) (self *StoreParams) {
 | |
| 	return &StoreParams{
 | |
| 		ChunkDbPath:   filepath.Join(path, "chunks"),
 | |
| 		DbCapacity:    defaultDbCapacity,
 | |
| 		CacheCapacity: defaultCacheCapacity,
 | |
| 		Radius:        defaultRadius,
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // netstore contructor, takes path argument that is used to initialise dbStore,
 | |
| // the persistent (disk) storage component of LocalStore
 | |
| // the second argument is the hive, the connection/logistics manager for the node
 | |
| func NewNetStore(hash SwarmHasher, lstore *LocalStore, cloud CloudStore, params *StoreParams) *NetStore {
 | |
| 	return &NetStore{
 | |
| 		hashfunc:   hash,
 | |
| 		localStore: lstore,
 | |
| 		cloud:      cloud,
 | |
| 	}
 | |
| }
 | |
| 
 | |
| const (
 | |
| 	// maximum number of peers that a retrieved message is delivered to
 | |
| 	requesterCount = 3
 | |
| )
 | |
| 
 | |
| var (
 | |
| 	// timeout interval before retrieval is timed out
 | |
| 	searchTimeout = 3 * time.Second
 | |
| )
 | |
| 
 | |
| // store logic common to local and network chunk store requests
 | |
| // ~ unsafe put in localdb no check if exists no extra copy no hash validation
 | |
| // the chunk is forced to propagate (Cloud.Store) even if locally found!
 | |
| // caller needs to make sure if that is wanted
 | |
| func (self *NetStore) Put(entry *Chunk) {
 | |
| 	self.localStore.Put(entry)
 | |
| 
 | |
| 	// handle deliveries
 | |
| 	if entry.Req != nil {
 | |
| 		log.Trace(fmt.Sprintf("NetStore.Put: localStore.Put %v hit existing request...delivering", entry.Key.Log()))
 | |
| 		// closing C signals to other routines (local requests)
 | |
| 		// that the chunk is has been retrieved
 | |
| 		close(entry.Req.C)
 | |
| 		// deliver the chunk to requesters upstream
 | |
| 		go self.cloud.Deliver(entry)
 | |
| 	} else {
 | |
| 		log.Trace(fmt.Sprintf("NetStore.Put: localStore.Put %v stored locally", entry.Key.Log()))
 | |
| 		// handle propagating store requests
 | |
| 		// go self.cloud.Store(entry)
 | |
| 		go self.cloud.Store(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // retrieve logic common for local and network chunk retrieval requests
 | |
| func (self *NetStore) Get(key Key) (*Chunk, error) {
 | |
| 	var err error
 | |
| 	chunk, err := self.localStore.Get(key)
 | |
| 	if err == nil {
 | |
| 		if chunk.Req == nil {
 | |
| 			log.Trace(fmt.Sprintf("NetStore.Get: %v found locally", key))
 | |
| 		} else {
 | |
| 			log.Trace(fmt.Sprintf("NetStore.Get: %v hit on an existing request", key))
 | |
| 			// no need to launch again
 | |
| 		}
 | |
| 		return chunk, err
 | |
| 	}
 | |
| 	// no data and no request status
 | |
| 	log.Trace(fmt.Sprintf("NetStore.Get: %v not found locally. open new request", key))
 | |
| 	chunk = NewChunk(key, newRequestStatus(key))
 | |
| 	self.localStore.memStore.Put(chunk)
 | |
| 	go self.cloud.Retrieve(chunk)
 | |
| 	return chunk, nil
 | |
| }
 | |
| 
 | |
| // Close netstore
 | |
| func (self *NetStore) Close() {}
 |