stats

A Golang middleware to manage qps and latency stats and asynchronously write to influxdb

Installation

go get "github.com/guptachirag/stats"

Usage

package main

import (
	"log"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/guptachirag/stats"
	"github.com/urfave/negroni"
)

func main() {
	mux := http.NewServeMux()

	mux.HandleFunc("/", func(rw http.ResponseWriter, req *http.Request) {
		rw.Write([]byte("Home"))
	})
	mux.HandleFunc("/about", func(rw http.ResponseWriter, req *http.Request) {
		rw.Write([]byte("About"))
	})

	n := negroni.Classic()

	ss, err := stats.NewStats(
		&stats.StatsConfig{
			QueueSize:     1000,
			FlushDuration: time.Second,
		},
		&stats.InfluxDBConfig{
			Host:      "127.0.0.1",
			Port:      8086,
			User:      "root",
			Password:  "root",
			DB:        "server_stats",
			Precision: "ns",
		},
		map[string]string{
			"host": "example.com",
		})
	if err != nil {
		log.Fatal(err)
	}

	n.Use(ss)
	n.UseHandler(mux)

	s := http.Server{
		Addr:    ":8080",
		Handler: n,
	}

	sigs := make(chan os.Signal, 1)
	signal.Notify(sigs, syscall.SIGTERM)
	go func(sigs chan os.Signal) {
		for {
			sig := <-sigs
			switch sig {
			case syscall.SIGTERM:
				if err := s.Shutdown(nil); err != nil {
					if err := s.Close(); err != nil {
						ss.Close()
						log.Fatal(err)
					}
					ss.Close()
				}
				ss.Close()
			}
		}
	}(sigs)

	log.Fatal(s.ListenAndServe())
}