feat : set the time zone for the service (#682)

* set the time zone for log output

* feat: change to global time zone setting

* feat: change to global time zone setting

Author: shootercheng

.env.example

@@ -1,6 +1,8 @@
 SERVER_HOST=0.0.0.0
 SERVER_PORT=5002
 SERVER_KEY=lYkiYYT6owG+71oLerGzA7GXCgOT++6ovaezWAjpCjf+Sjc3ZtU+qUEi
+# default server tz UTC
+SERVER_TZ=
 GIN_MODE=release
 PLATFORM=local
 

cmd/server/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"time"
 
 	"github.com/kelseyhightower/envconfig"
 	"github.com/langgenius/dify-plugin-daemon/internal/server"
@@ -19,6 +20,12 @@ func main() {
 
 	config.SetDefault()
 
+	loc, err := time.LoadLocation(config.ServerTimeZone)
+	if err != nil {
+		log.Panic("load location error", "error", err)
+	}
+	time.Local = loc
+
 	logCloser, err := log.Init(config.LogOutputFormat == "json", config.LogFile)
 	if err != nil {
 		log.Panic("failed to init logger", "error", err)

internal/types/app/config.go

@@ -24,9 +24,10 @@ const (
 
 type Config struct {
 	// server
-	ServerHost string `envconfig:"SERVER_HOST"`
-	ServerPort uint16 `envconfig:"SERVER_PORT" validate:"required"`
-	ServerKey  string `envconfig:"SERVER_KEY" validate:"required"`
+	ServerHost     string `envconfig:"SERVER_HOST"`
+	ServerPort     uint16 `envconfig:"SERVER_PORT" validate:"required"`
+	ServerKey      string `envconfig:"SERVER_KEY" validate:"required"`
+	ServerTimeZone string `envconfig:"SERVER_TZ" default:"UTC" validate:"timezone"`
 
 	// admin api enable
 	AdminApiEnabled bool   `envconfig:"ADMIN_API_ENABLED" default:"false"`

internal/types/app/default.go

@@ -12,6 +12,7 @@ func (config *Config) SetDefault() {
 	}
 	setDefaultString(&config.ServerHost, "0.0.0.0")
 	setDefaultInt(&config.ServerPort, 5002)
+	setDefaultString(&config.ServerTimeZone, "UTC")
 	setDefaultInt(&config.RoutinePoolSize, 10000)
 	setDefaultInt(&config.LifetimeCollectionGCInterval, 60)
 	setDefaultInt(&config.LifetimeCollectionHeartbeatInterval, 5)

pkg/utils/log/handler.go

@@ -44,7 +44,7 @@ func (h *Handler) Enabled(_ context.Context, level slog.Level) bool {
 func (h *Handler) Handle(ctx context.Context, r slog.Record) error {
 	fields := make(map[string]any)
 
-	fields["ts"] = r.Time.UTC().Format(time.RFC3339Nano)
+	fields["ts"] = r.Time.Local().Format(time.RFC3339Nano)
 	fields["severity"] = levelToSeverity(r.Level)
 	fields["service"] = h.opts.Service
 	fields["caller"] = getCaller(r.PC)