add http endpoint to get the size of a collection (#5910)

1 files changed, 55 insertions, 0 deletions

diff --git a/weed/server/master_server_handlers_admin.go b/weed/server/master_server_handlers_admin.go
index 7479b5535..e003d7d9f 100644
--- a/weed/server/master_server_handlers_admin.go
+++ b/weed/server/master_server_handlers_admin.go
@@ -178,3 +178,58 @@ func (ms *MasterServer) getVolumeGrowOption(r *http.Request) (*topology.VolumeGr
 	}
 	return volumeGrowOption, nil
 }
+
+func (ms *MasterServer) collectionInfoHandler(w http.ResponseWriter, r *http.Request) {
+	//get collection from request
+	collectionName := r.FormValue("collection")
+	if collectionName == "" {
+		writeJsonError(w, r, http.StatusBadRequest, fmt.Errorf("collection is required"))
+		return
+	}
+	//output details of the volumes?
+	detail := r.FormValue("detail") == "true"
+	//collect collection info
+	collection, ok := ms.Topo.FindCollection(collectionName)
+	if !ok {
+		writeJsonError(w, r, http.StatusBadRequest, fmt.Errorf("collection %s does not exist", collectionName))
+		return
+	}
+
+	volumeLayouts := collection.GetAllVolumeLayouts()
+
+	if detail {
+		//prepare the json response
+		all_stats := make([]map[string]interface{}, len(volumeLayouts))
+		for i, volumeLayout := range volumeLayouts {
+			volumeLayoutStats := volumeLayout.Stats()
+			m := make(map[string]interface{})
+			m["Version"] = util.Version()
+			m["Collection"] = collectionName
+			m["TotalSize"] = volumeLayoutStats.TotalSize
+			m["FileCount"] = volumeLayoutStats.FileCount
+			m["UsedSize"] = volumeLayoutStats.UsedSize
+			all_stats[i] = m
+		}
+		//write it
+		writeJsonQuiet(w, r, http.StatusOK, all_stats)
+	} else {
+		//prepare the json response
+		collectionStats := map[string]interface{}{
+			"Version":     util.Version(),
+			"Collection":  collectionName,
+			"TotalSize":   uint64(0),
+			"FileCount":   uint64(0),
+			"UsedSize":    uint64(0),
+			"VolumeCount": uint64(0),
+		}
+		for _, volumeLayout := range volumeLayouts {
+			volumeLayoutStats := volumeLayout.Stats()
+			collectionStats["TotalSize"] = collectionStats["TotalSize"].(uint64) + volumeLayoutStats.TotalSize
+			collectionStats["FileCount"] = collectionStats["FileCount"].(uint64) + volumeLayoutStats.FileCount
+			collectionStats["UsedSize"] = collectionStats["UsedSize"].(uint64) + volumeLayoutStats.UsedSize
+			collectionStats["VolumeCount"] = collectionStats["VolumeCount"].(uint64) + 1
+		}
+		//write it
+		writeJsonQuiet(w, r, http.StatusOK, collectionStats)
+	}
+}