From e30e98a4960762ee4d1c65d593e1e9196b648dca Mon Sep 17 00:00:00 2001
From: Cameron Sparr <cameronsparr@gmail.com>
Date: Mon, 14 Dec 2015 17:11:54 -0600
Subject: [PATCH] 0.3.0: leofs & lustre2

---
 plugins/leofs/leofs.go     |  5 ++++-
 plugins/lustre2/lustre2.go | 12 ++++++------
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/plugins/leofs/leofs.go b/plugins/leofs/leofs.go
index 32e5ee99a..5a1df7c56 100644
--- a/plugins/leofs/leofs.go
+++ b/plugins/leofs/leofs.go
@@ -197,6 +197,8 @@ func (l *LeoFS) gatherServer(endpoint string, serverType ServerType, acc plugins
 		"node": nodeNameTrimmed,
 	}
 	i := 0
+
+	fields := make(map[string]interface{})
 	for scanner.Scan() {
 		key := KeyMapping[serverType][i]
 		val, err := retrieveTokenAfterColon(scanner.Text())
@@ -207,9 +209,10 @@ func (l *LeoFS) gatherServer(endpoint string, serverType ServerType, acc plugins
 		if err != nil {
 			return fmt.Errorf("Unable to parse the value:%s, err:%s", val, err)
 		}
-		acc.Add(key, fVal, tags)
+		fields[key] = fVal
 		i++
 	}
+	acc.AddFields("leofs", fields, tags)
 	return nil
 }
 
diff --git a/plugins/lustre2/lustre2.go b/plugins/lustre2/lustre2.go
index 57217ec06..64af3050b 100644
--- a/plugins/lustre2/lustre2.go
+++ b/plugins/lustre2/lustre2.go
@@ -149,19 +149,19 @@ func (l *Lustre2) GetLustreProcStats(fileglob string, wanted_fields []*mapping,
 			return err
 		}
 
+		fields := make(map[string]interface{})
 		for _, line := range lines {
-			fields := strings.Fields(line)
-
+			parts := strings.Fields(line)
 			for _, wanted := range wanted_fields {
 				var data uint64
-				if fields[0] == wanted.inProc {
+				if parts[0] == wanted.inProc {
 					wanted_field := wanted.field
 					// if not set, assume field[1]. Shouldn't be field[0], as
 					// that's a string
 					if wanted_field == 0 {
 						wanted_field = 1
 					}
-					data, err = strconv.ParseUint((fields[wanted_field]), 10, 64)
+					data, err = strconv.ParseUint((parts[wanted_field]), 10, 64)
 					if err != nil {
 						return err
 					}
@@ -169,11 +169,11 @@ func (l *Lustre2) GetLustreProcStats(fileglob string, wanted_fields []*mapping,
 					if wanted.reportAs != "" {
 						report_name = wanted.reportAs
 					}
-					acc.Add(report_name, data, tags)
-
+					fields[report_name] = data
 				}
 			}
 		}
+		acc.AddFields("lustre2", fields)
 	}
 	return nil
 }