standard commit

prober/result_handler.go

@@ -0,0 +1,31 @@
+package prober
+
+import "dtool/utils"
+
+func OutputHandler(data interface{}) (string, error) {
+	var output_str string
+	var err error
+	switch value := data.(type) {
+	case CacheStruct:
+		result := make(map[string]map[int][]string)
+		temp := make(map[int][]string)
+		if len(value.dict) > 0 {
+			for cache_id := range value.dict {
+				for rdns := range value.dict[cache_id] {
+					temp[cache_id] = append(temp[cache_id], rdns)
+				}
+			}
+		}
+		result[value.target] = temp
+		output_str, err = utils.ToJSON(result, "")
+	case RecursiveStruct:
+		result := make(map[string][]string)
+		temp := []string{}
+		for rdns := range value.dict {
+			temp = append(temp, rdns)
+		}
+		result[value.target] = temp
+		output_str, err = utils.ToJSON(result, "")
+	}
+	return output_str, err
+}