listen: add default param

This commit is contained in:
Motiejus Jakštys 2024-09-15 07:46:16 +03:00
parent 2000985e2f
commit 49d5f1200c
1 changed files with 9 additions and 5 deletions

View File

@ -3,6 +3,7 @@ package main
import ( import (
"bufio" "bufio"
"encoding/json" "encoding/json"
"flag"
"fmt" "fmt"
"io" "io"
"log" "log"
@ -12,7 +13,6 @@ import (
) )
const ( const (
_listen = ":9011"
_urlTemplate = "https://api.meteo.lt/v1/stations/%s/observations/%s" _urlTemplate = "https://api.meteo.lt/v1/stations/%s/observations/%s"
_station = "vilniaus-ams" _station = "vilniaus-ams"
_promTemplate = `weather_station_air_temperature_celsius{station="{{ .Station }}"} {{ .AirTemperature }} {{ .TS }} _promTemplate = `weather_station_air_temperature_celsius{station="{{ .Station }}"} {{ .AirTemperature }} {{ .TS }}
@ -28,12 +28,16 @@ weather_station_condition{station="{{ .Station }}",code="{{ .ConditionCode }}"}
` `
) )
var tpl = template.Must(template.New("prom").Parse(_promTemplate)) var (
_tpl = template.Must(template.New("prom").Parse(_promTemplate))
_listen = flag.String("l", "127.0.0.1:9011", "listen on")
)
func main() { func main() {
log.Printf("Listening on %s\n", _listen) flag.Parse()
log.Printf("Listening on %s\n", *_listen)
log.Fatal((&http.Server{ log.Fatal((&http.Server{
Addr: _listen, Addr: *_listen,
Handler: http.HandlerFunc(handler), Handler: http.HandlerFunc(handler),
}).ListenAndServe()) }).ListenAndServe())
} }
@ -82,7 +86,7 @@ func handler(w http.ResponseWriter, r *http.Request) {
observation.TS = ts.UnixMilli() observation.TS = ts.UnixMilli()
observation.Station = _station observation.Station = _station
if err := tpl.Execute(bw, observation); err != nil { if err := _tpl.Execute(bw, observation); err != nil {
log.Printf("error executing template: %v", err) log.Printf("error executing template: %v", err)
return return
} }