Feature/extra summary

pkg/visor/hypervisor.go

@@ -228,6 +228,7 @@ func (hv *Hypervisor) makeMux() chi.Router {
 				r.Get("/dmsg", hv.getDmsg())
 
 				r.Get("/visors", hv.getVisors())
+				r.Get("/visors-summary", hv.getVisorsExtraSummary())
 				r.Get("/visors/{pk}", hv.getVisor())
 				r.Get("/visors/{pk}/summary", hv.getVisorSummary())
 				r.Get("/visors/{pk}/health", hv.getHealth())
@@ -489,6 +490,99 @@ func (hv *Hypervisor) getVisorSummary() http.HandlerFunc {
 	})
 }
 
+type extraSummaryWithDmsgResp struct {
+	extraSummaryResp
+	IsHypervisor bool                           `json:"is_hypervisor"`
+	DmsgStats    *dmsgtracker.DmsgClientSummary `json:"dmsg_stats"`
+}
+
+func (hv *Hypervisor) getVisorsExtraSummary() http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		hv.mu.RLock()
+		wg := new(sync.WaitGroup)
+		wg.Add(len(hv.visors))
+
+		i := 0
+		if hv.visor != nil {
+			i++
+		}
+
+		dmsgStats := make(map[string]dmsgtracker.DmsgClientSummary)
+		wg.Add(1)
+		go func() {
+			summary := hv.getDmsgSummary()
+			for _, stat := range summary {
+				dmsgStats[stat.PK.String()] = stat
+			}
+			wg.Done()
+		}()
+
+		summaries := make([]extraSummaryWithDmsgResp, len(hv.visors)+i)
+
+		if hv.visor != nil {
+			summary, err := hv.visor.ExtraSummary()
+			if err != nil {
+				log.WithError(err).Warn("Failed to obtain summary of this visor.")
+				summary = &ExtraSummary{}
+				summary.Summary.PubKey = hv.visor.conf.PK
+			}
+
+			addr := dmsg.Addr{PK: hv.c.PK, Port: hv.c.DmsgPort}
+			summaries[0] = extraSummaryWithDmsgResp{
+				extraSummaryResp: extraSummaryResp{
+					TCPAddr:      addr.String(),
+					Online:       err == nil,
+					ExtraSummary: summary,
+				},
+				IsHypervisor: true,
+			}
+		}
+
+		for pk, c := range hv.visors {
+			go func(pk cipher.PubKey, c Conn, i int) {
+				log := hv.log(r).
+					WithField("visor_addr", c.Addr).
+					WithField("func", "getVisors")
+
+				log.Debug("Requesting summary via RPC.")
+
+				summary, err := c.API.ExtraSummary()
+				if err != nil {
+					log.WithError(err).
+						Warn("Failed to obtain summary via RPC.")
+					summary = &ExtraSummary{}
+					summary.Summary.PubKey = pk
+				} else {
+					log.Debug("Obtained summary via RPC.")
+				}
+				resp := extraSummaryWithDmsgResp{
+					extraSummaryResp: extraSummaryResp{
+						TCPAddr:      c.Addr.String(),
+						Online:       err == nil,
+						ExtraSummary: summary,
+					},
+					IsHypervisor: false,
+				}
+				resp.ExtraSummary = summary
+				summaries[i] = resp
+				wg.Done()
+			}(pk, c, i)
+			i++
+		}
+
+		wg.Wait()
+		for i := 0; i < len(summaries); i++ {
+			if stat, ok := dmsgStats[summaries[i].Summary.PubKey.String()]; ok {
+				summaries[i].DmsgStats = &stat
+			}
+		}
+
+		hv.mu.RUnlock()
+
+		httputil.WriteJSON(w, r, http.StatusOK, summaries)
+	}
+}
+
 // returns app summaries of a given node of pk
 func (hv *Hypervisor) getApps() http.HandlerFunc {
 	return hv.withCtx(hv.visorCtx, func(w http.ResponseWriter, r *http.Request, ctx *httpCtx) {