github.com/daeglee/go-ethereum@v0.0.0-20190504220456-cad3e8d18e9b/swarm/api/inspector.go (about)

     1  // Copyright 2019 The go-ethereum Authors
     2  // This file is part of the go-ethereum library.
     3  //
     4  // The go-ethereum 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 go-ethereum library is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY 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 go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package api
    18  
    19  import (
    20  	"context"
    21  	"fmt"
    22  
    23  	"github.com/ethereum/go-ethereum/swarm/network"
    24  	"github.com/ethereum/go-ethereum/swarm/storage"
    25  )
    26  
    27  type Inspector struct {
    28  	api      *API
    29  	hive     *network.Hive
    30  	netStore *storage.NetStore
    31  }
    32  
    33  func NewInspector(api *API, hive *network.Hive, netStore *storage.NetStore) *Inspector {
    34  	return &Inspector{api, hive, netStore}
    35  }
    36  
    37  // Hive prints the kademlia table
    38  func (inspector *Inspector) Hive() string {
    39  	return inspector.hive.String()
    40  }
    41  
    42  func (inspector *Inspector) ListKnown() []string {
    43  	res := []string{}
    44  	for _, v := range inspector.hive.Kademlia.ListKnown() {
    45  		res = append(res, fmt.Sprintf("%v", v))
    46  	}
    47  	return res
    48  }
    49  
    50  type HasInfo struct {
    51  	Addr string `json:"address"`
    52  	Has  bool   `json:"has"`
    53  }
    54  
    55  // Has checks whether each chunk address is present in the underlying datastore,
    56  // the bool in the returned structs indicates if the underlying datastore has
    57  // the chunk stored with the given address (true), or not (false)
    58  func (inspector *Inspector) Has(chunkAddresses []storage.Address) []HasInfo {
    59  	results := make([]HasInfo, 0)
    60  	for _, addr := range chunkAddresses {
    61  		res := HasInfo{}
    62  		res.Addr = addr.String()
    63  		res.Has = inspector.netStore.Has(context.Background(), addr)
    64  		results = append(results, res)
    65  	}
    66  	return results
    67  }