swarm: linting of simulation package

This commit is contained in:
Fabio Barone 2018-07-20 11:49:52 -05:00
parent 965014df84
commit bf87fdc3d7
6 changed files with 24 additions and 9 deletions

View file

@ -43,7 +43,7 @@ func (s *Simulation) SetNodeItem(id discover.NodeID, key interface{}, value inte
s.buckets[id].Store(key, value) s.buckets[id].Store(key, value)
} }
// NodeItems returns a map of items from all nodes that are all set under the // NodesItems returns a map of items from all nodes that are all set under the
// same BucketKey. // same BucketKey.
func (s *Simulation) NodesItems(key interface{}) (values map[discover.NodeID]interface{}) { func (s *Simulation) NodesItems(key interface{}) (values map[discover.NodeID]interface{}) {
s.mu.RLock() s.mu.RLock()

View file

@ -135,7 +135,7 @@ func (s *Simulation) ConnectNodesStar(id discover.NodeID, ids []discover.NodeID)
return nil return nil
} }
// ConnectNodesStar connects all nodes in a star topology // ConnectNodesStarPivot connects all nodes in a star topology
// with the center at already set pivot node. // with the center at already set pivot node.
// If ids argument is nil, all nodes that are up will be connected. // If ids argument is nil, all nodes that are up will be connected.
func (s *Simulation) ConnectNodesStarPivot(ids []discover.NodeID) (err error) { func (s *Simulation) ConnectNodesStarPivot(ids []discover.NodeID) (err error) {

View file

@ -72,7 +72,7 @@ func (f *PeerEventsFilter) MsgCode(c uint64) *PeerEventsFilter {
func (s *Simulation) PeerEvents(ctx context.Context, ids []discover.NodeID, filters ...*PeerEventsFilter) <-chan PeerEvent { func (s *Simulation) PeerEvents(ctx context.Context, ids []discover.NodeID, filters ...*PeerEventsFilter) <-chan PeerEvent {
eventC := make(chan PeerEvent) eventC := make(chan PeerEvent)
// wait group to make sure all subscriptons to admin peerEvents are established // wait group to make sure all subscriptions to admin peerEvents are established
// before this function returns. // before this function returns.
var subsWG sync.WaitGroup var subsWG sync.WaitGroup
for _, id := range ids { for _, id := range ids {

View file

@ -29,7 +29,7 @@ var (
DefaultHTTPSimAddr = ":8888" DefaultHTTPSimAddr = ":8888"
) )
//`With`(builder) pattern constructor for Simulation to //WithServer implements the builder pattern constructor for Simulation to
//start with a HTTP server //start with a HTTP server
func (s *Simulation) WithServer(addr string) *Simulation { func (s *Simulation) WithServer(addr string) *Simulation {
//assign default addr if nothing provided //assign default addr if nothing provided
@ -46,7 +46,12 @@ func (s *Simulation) WithServer(addr string) *Simulation {
Addr: addr, Addr: addr,
Handler: s.handler, Handler: s.handler,
} }
go s.httpSrv.ListenAndServe() go func() {
err := s.httpSrv.ListenAndServe()
if err != nil {
log.Error("Error starting the HTTP server", "error", err)
}
}()
return s return s
} }
@ -55,7 +60,7 @@ func (s *Simulation) addSimulationRoutes() {
s.handler.POST("/runsim", s.RunSimulation) s.handler.POST("/runsim", s.RunSimulation)
} }
// StartNetwork starts all nodes in the network // RunSimulation is the actual POST endpoint runner
func (s *Simulation) RunSimulation(w http.ResponseWriter, req *http.Request) { func (s *Simulation) RunSimulation(w http.ResponseWriter, req *http.Request) {
log.Debug("RunSimulation endpoint running") log.Debug("RunSimulation endpoint running")
s.runC <- struct{}{} s.runC <- struct{}{}

View file

@ -96,7 +96,12 @@ func sendRunSignal(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("Request failed: %v", err) t.Fatalf("Request failed: %v", err)
} }
defer resp.Body.Close() defer func() {
err := resp.Body.Close()
if err != nil {
log.Error("Error closing response body", "err", err)
}
}()
log.Debug("Signal sent") log.Debug("Signal sent")
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {
t.Fatalf("err %s", resp.Status) t.Fatalf("err %s", resp.Status)

View file

@ -195,7 +195,7 @@ func (s *Simulation) AddNodesAndConnectStar(count int, opts ...AddNodeOption) (i
return ids, nil return ids, nil
} }
//Upload a snapshot //UploadSnapshot uploads a snapshot to the simulation
//This method tries to open the json file provided, applies the config to all nodes //This method tries to open the json file provided, applies the config to all nodes
//and then loads the snapshot into the Simulation network //and then loads the snapshot into the Simulation network
func (s *Simulation) UploadSnapshot(snapshotFile string, opts ...AddNodeOption) error { func (s *Simulation) UploadSnapshot(snapshotFile string, opts ...AddNodeOption) error {
@ -203,7 +203,12 @@ func (s *Simulation) UploadSnapshot(snapshotFile string, opts ...AddNodeOption)
if err != nil { if err != nil {
return err return err
} }
defer f.Close() defer func() {
err := f.Close()
if err != nil {
log.Error("Error closing snapshot file", "err", err)
}
}()
jsonbyte, err := ioutil.ReadAll(f) jsonbyte, err := ioutil.ReadAll(f)
if err != nil { if err != nil {
return err return err