添加 otel 相关环境变量，优化部署配置

docker-compose.yaml

@@ -2,7 +2,7 @@ name: lanhu
 
 services:
   postgres:
-    image: postgres:17
+    image: postgres:17.7
     environment:
       POSTGRES_USER: ${DB_USERNAME}
       POSTGRES_PASSWORD: ${DB_PASSWORD}

pkg/env/env.go

@@ -36,6 +36,9 @@ var (
 	RedisPort     = "6379"
 	RedisPassword = ""
 
+	OtelHost string
+	OtelPort string
+
 	BaiyinCloudUrl string
 	BaiyinTokenUrl string
 
@@ -115,6 +118,9 @@ func Init() {
 	errs = append(errs, parse(&RedisPort, "REDIS_PORT", true, nil))
 	errs = append(errs, parse(&RedisPassword, "REDIS_PASS", true, nil))
 
+	errs = append(errs, parse(&OtelHost, "OTEL_HOST", true, nil))
+	errs = append(errs, parse(&OtelPort, "OTEL_PORT", true, nil))
+
 	errs = append(errs, parse(&BaiyinCloudUrl, "BAIYIN_CLOUD_URL", false, nil))
 	errs = append(errs, parse(&BaiyinTokenUrl, "BAIYIN_TOKEN_URL", false, nil))
 

web/globals/otel.go

@@ -3,6 +3,7 @@ package globals
 import (
 	"context"
 	"fmt"
+	"platform/pkg/env"
 
 	"go.opentelemetry.io/otel"
 	"go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc"
@@ -16,8 +17,13 @@ import (
 var tp *trace.TracerProvider
 
 func initOtel(ctx context.Context) error {
+	if env.OtelHost == "" || env.OtelPort == "" {
+		return nil
+	}
+
+	addr := env.OtelHost + ":" + env.OtelPort
 	exporter, err := otlptracegrpc.New(ctx,
-		otlptracegrpc.WithEndpoint("localhost:4317"),
+		otlptracegrpc.WithEndpoint(addr),
 		otlptracegrpc.WithInsecure(),
 	)
 	if err != nil {

web/handlers/resource.go

@@ -353,21 +353,10 @@ func StatisticResourceUsage(c *fiber.Ctx) error {
 	}
 
 	// 统计套餐提取数量
-	do := q.LogsUserUsage.Where(q.LogsUserUsage.UserID.Eq(authCtx.User.ID))
-	if req.ResourceNo != nil && *req.ResourceNo != "" {
-		var resourceID int32
-		err := q.Resource.
-			Where(
-				q.Resource.UserID.Eq(authCtx.User.ID),
-				q.Resource.ResourceNo.Eq(*req.ResourceNo),
-			).
-			Select(q.Resource.ID).
-			Scan(&resourceID)
-		if err != nil {
-			return err
-		}
-		do.Where(q.LogsUserUsage.ResourceID.Eq(resourceID))
-	}
+	do := q.LogsUserUsage.Where(
+		q.LogsUserUsage.UserID.Eq(authCtx.User.ID),
+	)
+
 	if req.TimeAfter != nil {
 		do.Where(q.LogsUserUsage.Time.Gte(*req.TimeAfter))
 	}
@@ -383,7 +372,7 @@ func StatisticResourceUsage(c *fiber.Ctx) error {
 		).
 		Where(do).
 		Group(
-			field.NewUnsafeFieldRaw("date_trunc('day', time)"),
+			field.NewField("", "date"),
 		).
 		Order(
 			field.NewField("", "date").Desc(),
@@ -397,9 +386,8 @@ func StatisticResourceUsage(c *fiber.Ctx) error {
 }
 
 type StatisticResourceUsageReq struct {
-	ResourceNo *string    `json:"resource_no"`
-	TimeAfter  *time.Time `json:"time_after"`
-	TimeBefore *time.Time `json:"time_before"`
+	TimeAfter  *time.Time `json:"time_start"`
+	TimeBefore *time.Time `json:"time_end"`
 }
 
 type StatisticResourceUsageResp []struct {

web/routes.go

@@ -1,6 +1,7 @@
 package web
 
 import (
+	"platform/pkg/env"
 	auth2 "platform/web/auth"
 	"platform/web/handlers"
 
@@ -86,7 +87,9 @@ func ApplyRouters(app *fiber.App) {
 	callbacks.Get("/identify", handlers.IdentifyCallbackNew)
 
 	// 临时
-	debug := app.Group("/debug")
-	debug.Get("/sms/:phone", handlers.DebugGetSmsCode)
-	debug.Get("/proxy/register", handlers.DebugRegisterProxyBaiYin)
+	if env.RunMode == env.RunModeDev {
+		debug := app.Group("/debug")
+		debug.Get("/sms/:phone", handlers.DebugGetSmsCode)
+		debug.Get("/proxy/register", handlers.DebugRegisterProxyBaiYin)
+	}
 }