From 8eb9adda68d41316232db0cf329f87b632bf3b5a Mon Sep 17 00:00:00 2001
From: vipg <vipg@vipg.com>
Date: Tue, 10 Feb 2026 10:57:11 +0800
Subject: [PATCH] add

---
 trading_assistant_api/common/logger/logger.go     | 13 +++++++++++++
 trading_assistant_api/services/country/main.go    |  3 +++
 trading_assistant_api/services/docker-compose.yml |  1 +
 trading_assistant_api/services/user/main.go       |  3 +++
 4 files changed, 20 insertions(+)

diff --git a/trading_assistant_api/common/logger/logger.go b/trading_assistant_api/common/logger/logger.go
index eec3301..52e4bfa 100644
--- a/trading_assistant_api/common/logger/logger.go
+++ b/trading_assistant_api/common/logger/logger.go
@@ -5,6 +5,7 @@ import (
 	"os"
 	"fmt"
 	"path/filepath"
+	"time"
 )
 
 type Logger interface {
@@ -64,3 +65,15 @@ func SetupFile(path string) error {
 	}
 	return nil
 }
+
+func SetupTimezone(tz string) error {
+	if tz == "" {
+		return fmt.Errorf("timezone empty")
+	}
+	loc, err := time.LoadLocation(tz)
+	if err != nil {
+		return err
+	}
+	time.Local = loc
+	return nil
+}
diff --git a/trading_assistant_api/services/country/main.go b/trading_assistant_api/services/country/main.go
index 35bc7d0..f678c36 100644
--- a/trading_assistant_api/services/country/main.go
+++ b/trading_assistant_api/services/country/main.go
@@ -15,6 +15,9 @@ import (
 
 func main() {
 	port := utils.GetEnv("PORT", "8081")
+	if err := logger.SetupTimezone("Asia/Shanghai"); err != nil {
+		logger.L().Printf("setup timezone error: %v", err)
+	}
 	if err := logger.SetupFile(utils.GetEnv("LOG_FILE", "/var/log/app/country.log")); err != nil {
 		logger.L().Printf("setup file logger error: %v", err)
 	}
diff --git a/trading_assistant_api/services/docker-compose.yml b/trading_assistant_api/services/docker-compose.yml
index 8b519d2..6294d26 100644
--- a/trading_assistant_api/services/docker-compose.yml
+++ b/trading_assistant_api/services/docker-compose.yml
@@ -26,6 +26,7 @@ services:
     image: user-service:latest
     container_name: user_service
     environment:
+      TZ: "Asia/Shanghai"
       PG_HOST: postgres
       PG_PORT: "5432"
       PG_USER: postgres
diff --git a/trading_assistant_api/services/user/main.go b/trading_assistant_api/services/user/main.go
index 8596f0e..1b5000c 100644
--- a/trading_assistant_api/services/user/main.go
+++ b/trading_assistant_api/services/user/main.go
@@ -23,6 +23,9 @@ var pg *sql.DB
 
 func main() {
 	port := utils.GetEnv("PORT", "8080")
+	if err := logger.SetupTimezone("Asia/Shanghai"); err != nil {
+		logger.L().Printf("setup timezone error: %v", err)
+	}
 	if err := logger.SetupFile(utils.GetEnv("LOG_FILE", "/var/log/app/user.log")); err != nil {
 		logger.L().Printf("setup file logger error: %v", err)
 	}