diff --git a/main.go b/main.go index 219ab4f..d36dea7 100644 --- a/main.go +++ b/main.go @@ -1,13 +1,13 @@ package main import ( - "fmt" "github.com/prometheus/client_golang/prometheus" "github.com/prometheus/client_golang/prometheus/promhttp" "github.com/sirupsen/logrus" "github.com/yitsushi/go-misskey" "net/http" "os" + "strings" ) //TIP
To run your code, right-click the code and select Run.
Alternatively, click @@ -20,6 +20,8 @@ var endpoint = os.Getenv("MISSKEY_ENDPOINT") var protocol, hasProtocolEnv = os.LookupEnv("MISSKEY_PROTOCOL") var path = os.Getenv("MISSKEY_PATH") var apiKey = os.Getenv("MISSKEY_API_TOKEN") +var logLevelEnv = os.Getenv("LOGGER_LEVEL") +var logFormatEnv = os.Getenv("LOGGER_FORMAT") var ( misskeyJobQueueJobsCount = prometheus.NewGaugeVec( @@ -177,19 +179,39 @@ func main() { options, err := misskey.NewClientWithOptions( misskey.WithAPIToken(apiKey), misskey.WithBaseURL(protocol, endpoint, path), - misskey.WithLogLevel(logrus.InfoLevel), ) + var logLevel logrus.Level + switch { + case strings.EqualFold("DEBUG", logLevelEnv): + logLevel = logrus.DebugLevel + break + case strings.EqualFold("INFO", logLevelEnv): + logLevel = logrus.InfoLevel + break + case strings.EqualFold("WARN", logLevelEnv): + logLevel = logrus.WarnLevel + break + case strings.EqualFold("ERROR", logLevelEnv): + logLevel = logrus.ErrorLevel + break + case strings.EqualFold("FATAL", logLevelEnv): + logLevel = logrus.FatalLevel + break + case strings.EqualFold("TRACE", logLevelEnv): + logLevel = logrus.TraceLevel + break + default: + logLevel = logrus.InfoLevel + } + if strings.EqualFold("json", logFormatEnv) { + logrus.SetFormatter(&logrus.JSONFormatter{}) + } + logrus.SetLevel(logLevel) client = options if err != nil { logrus.Error(err) return } - stats, err := client.Admin().Queue().Stats() - if err != nil { - logrus.Error(err) - return - } - fmt.Println(stats) prometheus.MustRegister(misskeyJobQueueJobsCount) prometheus.MustRegister(misskeyJobQueueDelayed)