Add makefile & init scripts

Author: Novgorodov Igor, PMK-TV-OP

.gitignore

@@ -1,3 +1,6 @@
-nginx-prometheus*
+nginx-prometheus
+!deploy/*
 country.mmdb
 vendor*
+*.rpm
+*.deb

Makefile

@@ -0,0 +1,56 @@
+NAME := nginx-prometheus
+MAINTAINER:= Igor Novgorodov <[email protected]>
+DESCRIPTION := Service for exporting Nginx metrics to Prometheus
+LICENSE := MPLv2
+
+GO ?= go
+VERSION := $(shell cat VERSION)
+OUT := .out
+PACKAGE := github.com/blind-oracle/$(NAME)
+
+all: build
+
+build:
+	rm -rf $(OUT)
+	mkdir -p $(OUT)/root/etc/$(NAME)
+	go test ./...
+	GOARCH=amd64 GOOS=linux $(GO) build -o $(OUT) -ldflags "-s -w -X main.version=$(VERSION)"
+
+deb:
+	make build
+	make build-deb ARCH=amd64
+
+rpm:
+	make build
+	make build-rpm ARCH=amd64
+
+build-deb:
+	fpm -s dir -t deb -n $(NAME) -v $(VERSION) \
+		--deb-priority optional \
+		--category admin \
+		--force \
+		--url https://$(PACKAGE) \
+		--description "$(DESCRIPTION)" \
+		-m "$(MAINTAINER)" \
+		--license "$(LICENSE)" \
+		-a $(ARCH) \
+		$(OUT)/$(NAME)=/usr/bin/$(NAME) \
+		deploy/$(NAME).service=/usr/lib/systemd/system/$(NAME).service \
+		deploy/$(NAME)=/etc/default/$(NAME) \
+		$(OUT)/root/=/
+
+build-rpm:
+	fpm -s dir -t rpm -n $(NAME) -v $(VERSION) \
+		--force \
+		--rpm-compression bzip2 \
+		--rpm-os linux \
+		--url https://$(PACKAGE) \
+		--description "$(DESCRIPTION)" \
+		-m "$(MAINTAINER)" \
+		--license "$(LICENSE)" \
+		-a $(ARCH) \
+		--config-files /etc/$(NAME)/$(NAME).toml \
+		$(OUT)/$(NAME)=/usr/bin/$(NAME) \
+		deploy/$(NAME).service=/usr/lib/systemd/system/$(NAME).service \
+		deploy/$(NAME)=/etc/default/$(NAME) \
+		$(OUT)/root/=/

deploy/nginx-prometheus

@@ -0,0 +1,4 @@
+LISTEN_SYSLOG="0.0.0.0:1514"
+LISTEN_HTTP="0.0.0.0:11080"
+GEOIP_DB="-"
+URI_PREFIXES="-"

deploy/nginx-prometheus.service

@@ -0,0 +1,15 @@
+[Unit]
+Description=Service for exporting Nginx metrics to Prometheus
+Documentation=https://github.com/blind-oracle/nginx-prometheus
+After=network.target
+
+[Service]
+EnvironmentFile=/etc/default/nginx-prometheus
+Type=simple
+PermissionsStartOnly=true
+ExecStart=/usr/bin/nginx-prometheus -listenHTTP ${LISTEN_HTTP} -listenSyslog ${LISTEN_SYSLOG} -geoipCountryDB ${GEOIP_DB} -uriPrefixFile ${URI_PREFIXES}
+Restart=on-failure
+KillMode=control-group
+
+[Install]
+WantedBy=multi-user.target

main.go

@@ -69,20 +69,20 @@ func main() {
 
 	flag.StringVar(&listenSyslog, "listenSyslog", "0.0.0.0:1514", "ip:port to listen for syslog messages")
 	flag.StringVar(&listenHTTP, "listenHTTP", "0.0.0.0:11080", "ip:port to listen for http requests")
-	flag.StringVar(&uriPrefixFile, "uriPrefixFile", "", "file with allowed URI prefixes - one per line. If not specified - no per-URI stats gathered")
-	flag.StringVar(&geoipCountryDB, "geoipCountryDB", "", "path to MaxMind GeoIP country DB. If not specified - no lookups performed")
+	flag.StringVar(&uriPrefixFile, "uriPrefixFile", "-", "file with allowed URI prefixes - one per line. If not specified - no per-URI stats gathered")
+	flag.StringVar(&geoipCountryDB, "geoipCountryDB", "-", "path to MaxMind GeoIP country DB. If not specified - no lookups performed")
 	flag.BoolVar(&debug, "debug", false, "Enable debug")
 	flag.Parse()
 
-	if uriPrefixFile != "" {
+	if uriPrefixFile != "-" {
 		if uriTree, err = uriLoad(uriPrefixFile); err != nil {
 			log.Fatalf("Unable to load URIs: %s", err)
 		}
 
 		log.Printf("URI prefixes loaded: %d", uriTree.Len())
 	}
 
-	if geoipCountryDB != "" {
+	if geoipCountryDB != "-" {
 		if geoipDB, err = geoip.New(geoipCountryDB); err != nil {
 			log.Fatalf("Unable to load GeoIP database: %s", err)
 		}