first fairshare collector prototype

Author: vpenso

sshare.go

@@ -19,10 +19,13 @@ import (
         "io/ioutil"
         "os/exec"
         "log"
+        "strings"
+        "strconv"
+        "github.com/prometheus/client_golang/prometheus"
 )
 
 func FairShareData() []byte {
-        cmd := exec.Command("sshare", "-o account,fairshare")
+        cmd := exec.Command("sshare", "-nPo account,fairshare", "|", "grep '^ [a-z]'", "|", "tr -d ' '" )
         stdout, err := cmd.StdoutPipe()
         if err != nil {
                 log.Fatal(err)
@@ -36,3 +39,46 @@ func FairShareData() []byte {
         }
         return out
 }
+
+type FairShareMetrics struct {
+        fairshare float64
+}
+
+func ParseFairShareMetrics() map[string]*FairShareMetrics {
+        accounts := make(map[string]*FairShareMetrics)
+        lines := strings.Split(string(FairShareData()), "\n")
+        for _, line := range lines {
+                if strings.Contains(line,"|") {
+                        account := strings.Split(line,"|")[0]
+                        _,key := accounts[account]
+                        if !key {
+                                accounts[account] = &FairShareMetrics{0}
+                        }
+                        fairshare,_ := strconv.ParseFloat(strings.Split(line,"|")[1],64)
+                        accounts[account].fairshare = fairshare
+                }
+        }
+        return accounts
+}
+
+type FairShareCollector struct {
+        fairshare *prometheus.Desc
+}
+
+func NewFairShareCollector() *FairShareCollector {
+        labels := []string{"account"}
+        return &FairShareCollector{
+                fairshare: prometheus.NewDesc("slurm_account_fairshare","FairShare for account" , labels,nil),
+        }
+}
+
+func (fsc *FairShareCollector) Describe(ch chan<- *prometheus.Desc) {
+        ch <- fsc.fairshare
+}
+
+func (fsc *FairShareCollector) Collect(ch chan<- prometheus.Metric) {
+        fsm := ParseFairShareMetrics()
+        for f := range fsm {
+                ch <- prometheus.MustNewConstMetric(fsc.fairshare, prometheus.GaugeValue, fsm[f].fairshare, f)
+        }
+}