weather/cmd/poller/main.go

95 lines
2.1 KiB
Go
Raw Normal View History

package main
import (
"flag"
"go/weather/internal"
"go/weather/internal/poller"
"go/weather/internal/storage"
"go/weather/internal/web"
"go/weather/pkg/logger"
"os"
"os/signal"
"strings"
"time"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
const bucket = "weather"
func main() {
f := flagDefinition()
loggerLevel := zap.NewAtomicLevelAt(f.logLevel)
defaultLogger := logger.NewLogger("poller", f.logOutput, loggerLevel)
sLogger := defaultLogger.Sugar()
defer defaultLogger.Sync()
config := internal.ParseConfiguration(sLogger, f.filename)
addr := web.NewListenAddr("api.openweathermap.org", 443)
poller := poller.NewWeatherPoller(defaultLogger, addr, 45.75, 4.85, config.OpenweatherSecret)
s3storage := storage.NewS3Storage(defaultLogger, config.S3Storage, bucket)
go func() {
for {
data := poller.Poll()
defer data.Close()
s3storage.Store(data)
defaultLogger.Sugar().Debug("Wait next poll")
time.Sleep(f.interval)
}
}()
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
<-c
defaultLogger.Sugar().Info("shutting down")
os.Exit(0)
}
//flagParameter contain all application customizable parameter
type flagParameter struct {
interval time.Duration
filename string
logLevel zapcore.Level
logOutput string
}
func flagDefinition() flagParameter {
var logLevel string
f := flagParameter{}
flag.DurationVar(&f.interval, "check-interval", time.Second*30, "the sleep duration between multiple call to remote weather api")
flag.StringVar(&f.filename, "filename", "config.hcl", "configuration filename")
flag.StringVar(&logLevel, "logLevel", "info", "Log level")
flag.StringVar(&f.logOutput, "logOutput", "weather.log", "Output log path")
flag.Parse()
f.logLevel = parseLogLevel(logLevel)
return f
}
func parseLogLevel(logLevel string) zapcore.Level {
zapLevel := zap.InfoLevel
switch strings.ToUpper(logLevel) {
case "DEBUG":
zapLevel = zapcore.DebugLevel
case "INFO":
zapLevel = zapcore.InfoLevel
case "WARN":
zapLevel = zapcore.WarnLevel
case "ERROR":
zapLevel = zapcore.ErrorLevel
case "PANIC":
zapLevel = zapcore.PanicLevel
case "FATAL":
zapLevel = zapcore.FatalLevel
}
return zapLevel
}