seperated responsibilities (part 1)
This commit is contained in:
Daniel Czerwonk
parent
a726f3f16e
commit
2f302f030f
116
main.go
Normal file
116
main.go
Normal file
@ -0,0 +1,116 @@
|
||||
/*
|
||||
Copyright 2016 Daniel Czerwonk (d.czerwonk@gmail.com)
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/exec"
|
||||
)
|
||||
|
||||
type session struct {
|
||||
name string
|
||||
ipVersion int
|
||||
established int
|
||||
imported int64
|
||||
exported int64
|
||||
uptime int
|
||||
}
|
||||
|
||||
const version string = "0.2.1"
|
||||
|
||||
var (
|
||||
showVersion = flag.Bool("version", false, "Print version information.")
|
||||
listenAddress = flag.String("web.listen-address", ":9200", "Address on which to expose metrics and web interface.")
|
||||
metricsPath = flag.String("web.telemetry-path", "/metrics", "Path under which to expose metrics.")
|
||||
birdClient = flag.String("bird.client", "birdc", "Binary to communicate with the bird routing daemon")
|
||||
)
|
||||
|
||||
func main() {
|
||||
initRegexes()
|
||||
flag.Parse()
|
||||
|
||||
if *showVersion {
|
||||
printVersion()
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
startServer()
|
||||
}
|
||||
|
||||
func printVersion() {
|
||||
fmt.Println("bird_bgp_exporter")
|
||||
fmt.Printf("Version: %s\n", version)
|
||||
fmt.Println("BGP metric exporter for bird routing daemon")
|
||||
}
|
||||
|
||||
func startServer() {
|
||||
fmt.Printf("Starting bird BGP exporter (Version: %s)\n", version)
|
||||
http.HandleFunc(*metricsPath, handleMetricsRequest)
|
||||
|
||||
fmt.Printf("Listening for %s on %s\n", *metricsPath, *listenAddress)
|
||||
log.Fatal(http.ListenAndServe(*listenAddress, nil))
|
||||
}
|
||||
|
||||
func handleMetricsRequest(w http.ResponseWriter, r *http.Request) {
|
||||
sessions := getSessions()
|
||||
|
||||
for _, s := range sessions {
|
||||
writeLineForSession(s, w)
|
||||
}
|
||||
}
|
||||
|
||||
func writeLineForSession(s *session, w io.Writer) {
|
||||
fmt.Fprintf(w, "bgp%d_session_up{name=\"%s\"} %d\n", s.ipVersion, s.name, s.established)
|
||||
fmt.Fprintf(w, "bgp%d_session_prefix_count_import{name=\"%s\"} %d\n", s.ipVersion, s.name, s.imported)
|
||||
fmt.Fprintf(w, "bgp%d_session_prefix_count_export{name=\"%s\"} %d\n", s.ipVersion, s.name, s.exported)
|
||||
fmt.Fprintf(w, "bgp%d_session_uptime{name=\"%s\"} %d\n", s.ipVersion, s.name, s.uptime)
|
||||
}
|
||||
|
||||
func getSessions() []*session {
|
||||
birdSessions := getSessionsFromBird(4)
|
||||
bird6Sessions := getSessionsFromBird(6)
|
||||
|
||||
return append(birdSessions, bird6Sessions...)
|
||||
}
|
||||
|
||||
func getSessionsFromBird(ipVersion int) []*session {
|
||||
client := *birdClient
|
||||
|
||||
if ipVersion == 6 {
|
||||
client += "6"
|
||||
}
|
||||
|
||||
output := getBirdOutput(client)
|
||||
return parseOutput(output, ipVersion)
|
||||
}
|
||||
|
||||
func getBirdOutput(birdClient string) []byte {
|
||||
b, err := exec.Command(birdClient, "show", "protocols", "all").Output()
|
||||
|
||||
if err != nil {
|
||||
b = make([]byte, 0)
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
return b
|
||||
}
|
||||
@ -20,115 +20,25 @@ package main
|
||||
import (
|
||||
"bufio"
|
||||
"bytes"
|
||||
"flag"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/exec"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
type session struct {
|
||||
name string
|
||||
ipVersion int
|
||||
established int
|
||||
imported int64
|
||||
exported int64
|
||||
uptime int
|
||||
}
|
||||
|
||||
const version string = "0.2"
|
||||
|
||||
var (
|
||||
protoRegex *regexp.Regexp
|
||||
routeRegex *regexp.Regexp
|
||||
uptimeRegex *regexp.Regexp
|
||||
showVersion = flag.Bool("version", false, "Print version information.")
|
||||
listenAddress = flag.String("web.listen-address", ":9200", "Address on which to expose metrics and web interface.")
|
||||
metricsPath = flag.String("web.telemetry-path", "/metrics", "Path under which to expose metrics.")
|
||||
birdClient = flag.String("bird.client", "birdc", "Binary to communicate with the bird routing daemon")
|
||||
protoRegex *regexp.Regexp
|
||||
routeRegex *regexp.Regexp
|
||||
uptimeRegex *regexp.Regexp
|
||||
)
|
||||
|
||||
func main() {
|
||||
initRegexes()
|
||||
flag.Parse()
|
||||
|
||||
if *showVersion {
|
||||
printVersion()
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
startServer()
|
||||
}
|
||||
|
||||
func printVersion() {
|
||||
fmt.Println("bird_bgp_exporter")
|
||||
fmt.Printf("Version: %s\n", version)
|
||||
fmt.Println("BGP exporter for bird routing daemon")
|
||||
}
|
||||
|
||||
func startServer() {
|
||||
fmt.Printf("Starting bird BGP exporter (Version: %s)\n", version)
|
||||
http.HandleFunc(*metricsPath, handleMetricsRequest)
|
||||
|
||||
fmt.Printf("Listening for %s on %s\n", *metricsPath, *listenAddress)
|
||||
log.Fatal(http.ListenAndServe(*listenAddress, nil))
|
||||
}
|
||||
|
||||
func initRegexes() {
|
||||
protoRegex, _ = regexp.Compile("^([^\\s]+)\\s+BGP\\s+([^\\s]+)\\s+([^\\s]+)\\s+([^\\s]+)\\s+(.*?)\\s*$")
|
||||
routeRegex, _ = regexp.Compile("^\\s+Routes:\\s+(\\d+) imported, \\d+ filtered, (\\d+) exported")
|
||||
uptimeRegex, _ = regexp.Compile("^(?:((\\d+):(\\d{2}):(\\d{2}))|\\d+)$")
|
||||
}
|
||||
|
||||
func handleMetricsRequest(w http.ResponseWriter, r *http.Request) {
|
||||
sessions := getSessions()
|
||||
|
||||
for _, s := range sessions {
|
||||
writeLineForSession(s, w)
|
||||
}
|
||||
}
|
||||
|
||||
func writeLineForSession(s *session, w io.Writer) {
|
||||
fmt.Fprintf(w, "bgp%d_session_up{name=\"%s\"} %d\n", s.ipVersion, s.name, s.established)
|
||||
fmt.Fprintf(w, "bgp%d_session_prefix_count_import{name=\"%s\"} %d\n", s.ipVersion, s.name, s.imported)
|
||||
fmt.Fprintf(w, "bgp%d_session_prefix_count_export{name=\"%s\"} %d\n", s.ipVersion, s.name, s.exported)
|
||||
fmt.Fprintf(w, "bgp%d_session_uptime{name=\"%s\"} %d\n", s.ipVersion, s.name, s.uptime)
|
||||
}
|
||||
|
||||
func getSessions() []*session {
|
||||
birdSessions := getSessionsFromBird(4)
|
||||
bird6Sessions := getSessionsFromBird(6)
|
||||
|
||||
return append(birdSessions, bird6Sessions...)
|
||||
}
|
||||
|
||||
func getSessionsFromBird(ipVersion int) []*session {
|
||||
client := *birdClient
|
||||
|
||||
if ipVersion == 6 {
|
||||
client += "6"
|
||||
}
|
||||
|
||||
output := getBirdOutput(client)
|
||||
return parseOutput(output, ipVersion)
|
||||
}
|
||||
|
||||
func getBirdOutput(birdClient string) []byte {
|
||||
b, err := exec.Command(birdClient, "show", "protocols", "all").Output()
|
||||
|
||||
if err != nil {
|
||||
b = make([]byte, 0)
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
return b
|
||||
}
|
||||
|
||||
func parseOutput(data []byte, ipVersion int) []*session {
|
||||
sessions := make([]*session, 0)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user