From 3190e0e6567dfc5f9e9e49bcc6e31f597d0c95e6 Mon Sep 17 00:00:00 2001
From: Oliver Fesseler <ofesseler@gmail.com>
Date: Fri, 18 Nov 2016 16:13:46 +0100
Subject: [PATCH] initial commit of glusterfs_exporter

---
 main.go | 192 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 192 insertions(+)
 create mode 100644 main.go

diff --git a/main.go b/main.go
new file mode 100644
index 0000000..665a01c
--- /dev/null
+++ b/main.go
@@ -0,0 +1,192 @@
+// Copyright 2015 The Prometheus Authors
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// A minimal example of how to include Prometheus instrumentation.
+package main
+
+import (
+	"flag"
+	"net/http"
+	"os/exec"
+
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/promhttp"
+	"bytes"
+	"encoding/xml"
+	"io/ioutil"
+	"time"
+	"github.com/prometheus/common/log"
+	"strconv"
+)
+
+type CliOutput struct {
+	XMLName xml.Name  `xml:"cliOutput"`
+	OpRet int         `xml:"opRet"`
+	OpErrno int       `xml:"opErrno"`
+	OpErrstr string   `xml:"opErrstr"`
+	VolInfo VolInfo   `xml:"volInfo"`
+}
+
+type VolInfo struct {
+	XMLName xml.Name  `xml:"volInfo"`
+	Volumes Volumes   `xml:"volumes"`
+}
+
+type Volumes struct {
+	XMLName xml.Name  `xml:"volumes"`
+	Volume []Volume   `xml:"volume"`
+	Count int         `xml:"count"`
+}
+
+type Volume struct {
+	XMLName xml.Name  `xml:"volume"`
+	Name string       `xml:"name"`
+	Id string         `xml:"id"`
+	Status int        `xml:"status"`
+	StatusStr string  `xml:"statusStr"`
+	BrickCount int    `xml:"brickCount"`
+	Bricks []Brick    `xml:"bricks"`
+	DistCount int     `xml:"distCount"`
+}
+
+type Brick struct {
+	Uuid string       `xml:"brick>uuid"`
+	Name string       `xml:"brick>name"`
+	HostUuid string   `xml:"brick>hostUuid"`
+	IsArbiter int     `xml:"brick>isArbiter"`
+}
+
+var (
+	// Error number from GlusterFS
+	errno = prometheus.NewGaugeVec(
+		prometheus.GaugeOpts{
+			Name:"glusterfs_errno",
+			Help:"Error Number Glusterfs",
+		},
+		[]string{},
+	)
+
+	// creates a gauge of active nodes in glusterfs
+	volume_count = prometheus.NewGaugeVec(
+		prometheus.GaugeOpts{
+			Name:"glusterfs_volume_count",
+			Help:"Number of active glusterfs nodes",
+		},
+		[]string{},
+	)
+
+	// Count of bricks for gluster volume
+	brick_count = prometheus.NewGaugeVec(
+		prometheus.GaugeOpts{
+			Name:"glusterfs_brick_count",
+			Help:"Count of bricks for gluster volume",
+		},
+		[]string{"volume"},
+	)
+
+	// distribution count of bricks
+	distribution_count = prometheus.NewGaugeVec(
+		prometheus.GaugeOpts{
+			Name:"glusterfs_nodes_active",
+			Help:"distribution count of bricks",
+		},
+		[]string{"volume"},
+	)
+)
+
+func init() {
+	// register metric to prometheus's default registry
+	prometheus.MustRegister(errno)
+	prometheus.MustRegister(volume_count)
+	prometheus.MustRegister(brick_count)
+	prometheus.MustRegister(distribution_count)
+}
+
+func glusterVolumeInfo(sec_int int) {
+	for {
+		// Gluster Info
+		cmd_profile := exec.Command("/usr/sbin/gluster", "volume", "info", "--xml")
+		//cmd_profile := exec.Command("/home/oli/dev/glusterfs_exporter_go/gluster_info")
+
+		stdOutbuff := &bytes.Buffer{}
+
+		cmd_profile.Stdout = stdOutbuff
+
+		err := cmd_profile.Run()
+
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		var vol CliOutput
+		b, err := ioutil.ReadAll(stdOutbuff)
+		if err != nil {
+			log.Fatal(err)
+		}
+		xml.Unmarshal(b, &vol)
+
+		// set opErrno
+		errno.WithLabelValues().Set(float64(vol.OpErrno))
+		log.Debug("opErrno: %v", vol.OpErrno)
+
+		// set volume count
+		volume_count.WithLabelValues().Set(float64(vol.VolInfo.Volumes.Count))
+		log.Debug("volume_count: %v", vol.VolInfo.Volumes.Count)
+
+		// Volume based values
+		for _, v := range vol.VolInfo.Volumes.Volume {
+			// brick count with volume label
+			brick_count.WithLabelValues(v.Name).Set(float64(v.BrickCount))
+			log.Debug("opErrno: %v", vol.OpErrno)
+
+			// distribution count with volume label
+			distribution_count.WithLabelValues(v.Name).Set(float64(v.DistCount))
+			log.Debug("opErrno: %v", vol.OpErrno)
+
+		}
+		time.Sleep(time.Duration(sec_int) * time.Second)
+	}
+}
+
+func glusterProfile(sec_int int) {
+	// Gluster Profile
+
+
+	// Get gluster volumes, then call gluster profile on every volume
+
+	//  gluster volume profile gv_leoticket info cumulative --xml
+	//cmd_profile := exec.Command("/usr/sbin/gluster", "volume", "profile", "gv_leoticket", "info", "cumulative", "--xml")
+}
+
+func main() {
+
+	// commandline arguments
+	var (
+		addr = flag.String("listen-address", ":9189",  "The address to listen on for HTTP requests.")
+		sec  = flag.String("scrape-seconds", "2",      "Frequency of scraping glusterfs in seconds")
+	)
+
+	flag.Parse()
+
+	// ensure that sec is int
+	sec_int, err := strconv.Atoi(*sec)
+	if err != nil {
+		log.Fatal("Parameter -scrape-seconds is not an int value")
+	}
+
+	// gluster volume info
+	go glusterVolumeInfo(sec_int)
+
+	http.Handle("/metrics", promhttp.Handler())
+	log.Fatal(http.ListenAndServe(*addr, nil))
+}