From 2d182775eac5a936fca6e9dbaf579635e2b491fa Mon Sep 17 00:00:00 2001 From: "jiaji.H" Date: Tue, 6 Jan 2026 13:49:23 +0800 Subject: [PATCH] =?UTF-8?q?Updata=EF=BC=9A=E5=A2=9E=E5=8A=A0=E5=81=A5?= =?UTF-8?q?=E5=BA=B7=E6=A3=80=E6=B5=8B=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/service/bundle/bundleBalance.go | 68 ----------- pkg/service/bundle/healthCheck.go | 181 ++++++++++++++++++++++++++++ 2 files changed, 181 insertions(+), 68 deletions(-) create mode 100644 pkg/service/bundle/healthCheck.go diff --git a/pkg/service/bundle/bundleBalance.go b/pkg/service/bundle/bundleBalance.go index 9316ea4..d191279 100644 --- a/pkg/service/bundle/bundleBalance.go +++ b/pkg/service/bundle/bundleBalance.go @@ -7,12 +7,10 @@ import ( "fmt" "fonchain-fiee/api/bundle" "fonchain-fiee/api/cast" - "fonchain-fiee/pkg/cache" "fonchain-fiee/pkg/service" "fonchain-fiee/pkg/service/bundle/common" "fonchain-fiee/pkg/utils" "io" - "net/http" "strconv" "strings" "time" @@ -579,69 +577,3 @@ func GetAccountBundleBalance(c *gin.Context) { }) service.Success(c, result) } - -// HealthCheck 健康检测接口 -func HealthCheck(c *gin.Context) { - healthStatus := gin.H{ - "status": "ok", - "timestamp": time.Now().Unix(), - "checks": make(map[string]interface{}), - } - - // 检查 Redis 连接 - redisStatus := "ok" - redisErr := "" - if cache.RedisClient != nil { - _, cancel := context.WithTimeout(context.Background(), 2*time.Second) - defer cancel() - _, err := cache.RedisClient.Ping().Result() - if err != nil { - redisStatus = "failed" - redisErr = err.Error() - } - } else { - redisStatus = "failed" - redisErr = "Redis client is nil" - } - healthStatus["checks"].(map[string]interface{})["redis"] = gin.H{ - "status": redisStatus, - "error": redisErr, - } - - // 检查微服务连接（调用一个简单的服务方法） - serviceStatus := "ok" - serviceErr := "" - ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) - defer cancel() - - // 调用 BundleProvider 的简单方法进行健康检测 - _, err := service.BundleProvider.GetBundleBalanceByUserId(ctx, &bundle.GetBundleBalanceByUserIdReq{ - UserId: 0, // 使用一个不存在的用户ID，只检测服务连通性 - }) - if err != nil { - // 检查是否是超时或连接错误 - if ctx.Err() == context.DeadlineExceeded { - serviceStatus = "timeout" - serviceErr = "Service timeout" - } else { - // 其他错误（如业务错误）认为服务是可用的 - serviceStatus = "ok" - serviceErr = "" - } - } - - healthStatus["checks"].(map[string]interface{})["microservice"] = gin.H{ - "status": serviceStatus, - "error": serviceErr, - } - - // 如果所有检查都通过，返回 200，否则返回 503 - httpStatus := http.StatusOK - if redisStatus != "ok" || serviceStatus != "ok" { - httpStatus = http.StatusServiceUnavailable - healthStatus["status"] = "degraded" - } - - c.JSON(httpStatus, healthStatus) - service.Success(c, healthStatus) -} diff --git a/pkg/service/bundle/healthCheck.go b/pkg/service/bundle/healthCheck.go new file mode 100644 index 0000000..cdbfa6c --- /dev/null +++ b/pkg/service/bundle/healthCheck.go @@ -0,0 +1,181 @@ +package bundle + +import ( + "context" + "fonchain-fiee/api/bundle" + "fonchain-fiee/api/cast" + "fonchain-fiee/api/files" + "fonchain-fiee/api/governance" + "fonchain-fiee/api/order" + "fonchain-fiee/api/payment" + "fonchain-fiee/pkg/cache" + "fonchain-fiee/pkg/service" + "net/http" + "time" + + "github.com/gin-gonic/gin" +) + +// HealthCheck 健康检测接口 +func HealthCheck(c *gin.Context) { + healthStatus := gin.H{ + "status": "ok", + "timestamp": time.Now().Unix(), + "checks": make(map[string]interface{}), + } + + // 检查 Redis 连接 + redisStatus := "ok" + redisErr := "" + if cache.RedisClient != nil { + _, cancel := context.WithTimeout(context.Background(), 2*time.Second) + defer cancel() + _, err := cache.RedisClient.Ping().Result() + if err != nil { + redisStatus = "failed" + redisErr = err.Error() + } + } else { + redisStatus = "failed" + redisErr = "Redis client is nil" + } + healthStatus["checks"].(map[string]interface{})["redis"] = gin.H{ + "status": redisStatus, + "error": redisErr, + } + + // 检查微服务连接（调用一个简单的服务方法） + serviceStatus := "ok" + serviceErr := "" + ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) + defer cancel() + + // 调用 BundleProvider 的简单方法进行健康检测 + _, err := service.BundleProvider.GetBundleBalanceByUserId(ctx, &bundle.GetBundleBalanceByUserIdReq{ + UserId: 0, // 使用一个不存在的用户ID，只检测服务连通性 + }) + if err != nil { + // 检查是否是超时或连接错误 + if ctx.Err() == context.DeadlineExceeded { + serviceStatus = "Bundle Service timeout" + serviceErr = "Service timeout" + } else { + // 其他错误（如业务错误）认为服务是可用的 + serviceStatus = "ok" + serviceErr = "" + } + } + healthStatus["checks"].(map[string]interface{})["bundleService"] = gin.H{ + "status": serviceStatus, + "error": serviceErr, + } + //调用 OrderProvider 的简单方法进行健康检测 + _, err = service.OrderProvider.GetOrder(ctx, &order.CommonRequest{ + ID: 0, + Domain: "", + SeriesUid: "", // 使用一个不存在的订单ID，只检测服务连通性 + }) + if err != nil { + if ctx.Err() == context.DeadlineExceeded { + serviceStatus = "Order Service timeout" + serviceErr = "Service timeout" + } else { + // 其他错误（如业务错误）认为服务是可用的 + serviceStatus = "ok" + serviceErr = "" + } + } + healthStatus["checks"].(map[string]interface{})["orderService"] = gin.H{ + "status": serviceStatus, + "error": serviceErr, + } + //调用 FilesProvider 的简单方法进行健康检测 + _, err = service.FilesProvider.List(ctx, &files.FileListReq{ + Path: "/", + UserSpacePath: "/fiee", + Page: 1, + PageSize: 50, + }) + if err != nil { + if ctx.Err() == context.DeadlineExceeded { + serviceStatus = "Filesbrowser Service timeout" + serviceErr = "Service timeout" + } else { + // 其他错误（如业务错误）认为服务是可用的 + serviceStatus = "ok" + serviceErr = "" + } + } + healthStatus["checks"].(map[string]interface{})["filesbrowserService"] = gin.H{ + "status": serviceStatus, + "error": serviceErr, + } + //调用 PaymentProvider 的简单方法进行健康检测 + _, err = service.PaymentProvider.QueryPayByOutTradeNo(ctx, &payment.PayQueryRequest{ + PayType: "2", + OutTradeNo: "1234567890", + }) + if err != nil { + if ctx.Err() == context.DeadlineExceeded { + serviceStatus = "Payment Service timeout" + serviceErr = "Service timeout" + } else { + // 其他错误（如业务错误）认为服务是可用的 + serviceStatus = "ok" + serviceErr = "" + } + } + healthStatus["checks"].(map[string]interface{})["paymentService"] = gin.H{ + "status": serviceStatus, + "error": serviceErr, + } + + //调用 CastProvider 的简单方法进行健康检测 + _, err = service.CastProvider.WorkList(ctx, &cast.WorkListReq{ + Page: 1, + PageSize: 10, + }) + if err != nil { + if ctx.Err() == context.DeadlineExceeded { + serviceStatus = "Cast Service timeout" + serviceErr = "Service timeout" + } else { + // 其他错误（如业务错误）认为服务是可用的 + serviceStatus = "ok" + serviceErr = "" + } + } + healthStatus["checks"].(map[string]interface{})["castService"] = gin.H{ + "status": serviceStatus, + "error": serviceErr, + } + //调用 GovernanceProvider 的简单方法进行健康检测 + _, err = service.GovernanceProvider.List(ctx, &governance.ListReq{ + Page: 1, + PageSize: 10, + }) + if err != nil { + if ctx.Err() == context.DeadlineExceeded { + serviceStatus = "Document Service timeout" + serviceErr = "Service timeout" + } else { + // 其他错误（如业务错误）认为服务是可用的 + serviceStatus = "ok" + serviceErr = "" + } + } + healthStatus["checks"].(map[string]interface{})["documentService"] = gin.H{ + "status": serviceStatus, + "error": serviceErr, + } + + // 如果所有检查都通过，返回 200，否则返回 503 + httpStatus := http.StatusOK + if redisStatus != "ok" || serviceStatus != "ok" { + httpStatus = http.StatusServiceUnavailable + healthStatus["status"] = "degraded" + } + + c.JSON(httpStatus, healthStatus) + service.Success(c, healthStatus) +}