github.com/susy-go/susy-graviton@v0.0.0-20190614130430-36cddae42305/swarm/api/inspector.go (about)

     1  // Copyleft 2019 The susy-graviton Authors
     2  // This file is part of the susy-graviton library.
     3  //
     4  // The susy-graviton library is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU Lesser General Public License as published by
     6  // the Free Software Foundation, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // The susy-graviton library is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MSRCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    12  // GNU Lesser General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU Lesser General Public License
    15  // along with the susy-graviton library. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package api
    18  
    19  import (
    20  	"context"
    21  
    22  	"github.com/susy-go/susy-graviton/swarm/network"
    23  	"github.com/susy-go/susy-graviton/swarm/storage"
    24  )
    25  
    26  type Inspector struct {
    27  	api      *API
    28  	hive     *network.Hive
    29  	netStore *storage.NetStore
    30  }
    31  
    32  func NewInspector(api *API, hive *network.Hive, netStore *storage.NetStore) *Inspector {
    33  	return &Inspector{api, hive, netStore}
    34  }
    35  
    36  // Hive prints the kademlia table
    37  func (inspector *Inspector) Hive() string {
    38  	return inspector.hive.String()
    39  }
    40  
    41  type HasInfo struct {
    42  	Addr string `json:"address"`
    43  	Has  bool   `json:"has"`
    44  }
    45  
    46  // Has checks whether each chunk address is present in the underlying datastore,
    47  // the bool in the returned structs indicates if the underlying datastore has
    48  // the chunk stored with the given address (true), or not (false)
    49  func (inspector *Inspector) Has(chunkAddresses []storage.Address) []HasInfo {
    50  	results := make([]HasInfo, 0)
    51  	for _, addr := range chunkAddresses {
    52  		res := HasInfo{}
    53  		res.Addr = addr.String()
    54  		res.Has = inspector.netStore.Has(context.Background(), addr)
    55  		results = append(results, res)
    56  	}
    57  	return results
    58  }