正在显示
4 个修改的文件
包含
103 行增加
和
1 行删除
| @@ -7,6 +7,7 @@ import ( | @@ -7,6 +7,7 @@ import ( | ||
| 7 | "github.com/zeromicro/go-zero/rest/httpx" | 7 | "github.com/zeromicro/go-zero/rest/httpx" |
| 8 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/cmd/ep/chat/internal/pkg/db" | 8 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/cmd/ep/chat/internal/pkg/db" |
| 9 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/cmd/ep/chat/internal/pkg/domain" | 9 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/cmd/ep/chat/internal/pkg/domain" |
| 10 | + "gitlab.fjmaimaimai.com/allied-creation/su-micro/pkg/tool" | ||
| 10 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/pkg/xerr" | 11 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/pkg/xerr" |
| 11 | "net/http" | 12 | "net/http" |
| 12 | 13 | ||
| @@ -49,6 +50,7 @@ func main() { | @@ -49,6 +50,7 @@ func main() { | ||
| 49 | 50 | ||
| 50 | ctx := svc.NewServiceContext(c) | 51 | ctx := svc.NewServiceContext(c) |
| 51 | handler.RegisterHandlers(server, ctx) | 52 | handler.RegisterHandlers(server, ctx) |
| 53 | + server.AddRoutes(RoutersOpenapi(ctx)) | ||
| 52 | 54 | ||
| 53 | // 数据迁移 | 55 | // 数据迁移 |
| 54 | if c.Migrate { | 56 | if c.Migrate { |
| @@ -69,3 +71,18 @@ func systemSetup(c config.Config) { | @@ -69,3 +71,18 @@ func systemSetup(c config.Config) { | ||
| 69 | // 系统成功应答包装 | 71 | // 系统成功应答包装 |
| 70 | httpx.SetOkHandler(xerr.OkHandlerCtx) | 72 | httpx.SetOkHandler(xerr.OkHandlerCtx) |
| 71 | } | 73 | } |
| 74 | + | ||
| 75 | +func RoutersOpenapi(svc *svc.ServiceContext) []rest.Route { | ||
| 76 | + return []rest.Route{ | ||
| 77 | + { | ||
| 78 | + Method: http.MethodGet, | ||
| 79 | + Path: "/log/:module", | ||
| 80 | + Handler: tool.LogHandler(svc.Config.Log), | ||
| 81 | + }, | ||
| 82 | + { | ||
| 83 | + Method: http.MethodGet, | ||
| 84 | + Path: "/cache/clear/:app", | ||
| 85 | + Handler: tool.ClearCacheHandler(svc.Redis), | ||
| 86 | + }, | ||
| 87 | + } | ||
| 88 | +} |
| @@ -11,6 +11,7 @@ import ( | @@ -11,6 +11,7 @@ import ( | ||
| 11 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/cmd/ep/system/internal/pkg/db" | 11 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/cmd/ep/system/internal/pkg/db" |
| 12 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/cmd/ep/system/internal/pkg/domain" | 12 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/cmd/ep/system/internal/pkg/domain" |
| 13 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/pkg/gateway/smslib" | 13 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/pkg/gateway/smslib" |
| 14 | + "gitlab.fjmaimaimai.com/allied-creation/su-micro/pkg/tool" | ||
| 14 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/pkg/xerr" | 15 | "gitlab.fjmaimaimai.com/allied-creation/su-micro/pkg/xerr" |
| 15 | "net/http" | 16 | "net/http" |
| 16 | 17 | ||
| @@ -78,5 +79,15 @@ func RoutersOpenapi(svc *svc.ServiceContext) []rest.Route { | @@ -78,5 +79,15 @@ func RoutersOpenapi(svc *svc.ServiceContext) []rest.Route { | ||
| 78 | Path: "/openapi/sms/send-sms-code", | 79 | Path: "/openapi/sms/send-sms-code", |
| 79 | Handler: smslib.SendSmsCodeHandler(svc.SmsService), | 80 | Handler: smslib.SendSmsCodeHandler(svc.SmsService), |
| 80 | }, | 81 | }, |
| 82 | + { | ||
| 83 | + Method: http.MethodGet, | ||
| 84 | + Path: "/log/:module", | ||
| 85 | + Handler: tool.LogHandler(svc.Config.Log), | ||
| 86 | + }, | ||
| 87 | + { | ||
| 88 | + Method: http.MethodGet, | ||
| 89 | + Path: "/cache/clear/:app", | ||
| 90 | + Handler: tool.ClearCacheHandler(svc.Redis), | ||
| 91 | + }, | ||
| 81 | } | 92 | } |
| 82 | } | 93 | } |
pkg/tool/handlers.go
0 → 100644
| 1 | +package tool | ||
| 2 | + | ||
| 3 | +import ( | ||
| 4 | + "fmt" | ||
| 5 | + "github.com/zeromicro/go-zero/core/logx" | ||
| 6 | + "github.com/zeromicro/go-zero/core/stores/redis" | ||
| 7 | + "github.com/zeromicro/go-zero/rest/httpx" | ||
| 8 | + "net/http" | ||
| 9 | + "path/filepath" | ||
| 10 | + "strings" | ||
| 11 | +) | ||
| 12 | + | ||
| 13 | +func LogHandler(conf logx.LogConf) http.HandlerFunc { | ||
| 14 | + return func(w http.ResponseWriter, r *http.Request) { | ||
| 15 | + var req struct { | ||
| 16 | + Module string `path:"module"` | ||
| 17 | + } | ||
| 18 | + if err := httpx.Parse(r, &req); err != nil { | ||
| 19 | + httpx.ErrorCtx(r.Context(), w, err) | ||
| 20 | + return | ||
| 21 | + } | ||
| 22 | + path := conf.Path | ||
| 23 | + if conf.Mode != "file" { | ||
| 24 | + return | ||
| 25 | + } | ||
| 26 | + if path == "" { | ||
| 27 | + path = "logs" | ||
| 28 | + } | ||
| 29 | + if !strings.HasSuffix(req.Module, ".log") { | ||
| 30 | + req.Module += ".log" | ||
| 31 | + } | ||
| 32 | + handler := http.FileServer(http.Dir(path)) | ||
| 33 | + r.URL.Path = filepath.Join(req.Module) | ||
| 34 | + handler.ServeHTTP(w, r) | ||
| 35 | + } | ||
| 36 | +} | ||
| 37 | + | ||
| 38 | +func ClearCacheHandler(redis *redis.Redis) http.HandlerFunc { | ||
| 39 | + return func(w http.ResponseWriter, r *http.Request) { | ||
| 40 | + var ( | ||
| 41 | + success int | ||
| 42 | + appName string | ||
| 43 | + ) | ||
| 44 | + var req struct { | ||
| 45 | + App string `path:"app"` | ||
| 46 | + } | ||
| 47 | + if err := httpx.Parse(r, &req); err != nil { | ||
| 48 | + httpx.ErrorCtx(r.Context(), w, err) | ||
| 49 | + return | ||
| 50 | + } | ||
| 51 | + appName = req.App | ||
| 52 | + if strings.TrimSpace(appName) == "" { | ||
| 53 | + httpx.Ok(w) | ||
| 54 | + return | ||
| 55 | + } | ||
| 56 | + keyPattern := fmt.Sprintf("%s*", appName) | ||
| 57 | + list, err := redis.Keys(keyPattern) | ||
| 58 | + if err != nil { | ||
| 59 | + httpx.ErrorCtx(r.Context(), w, err) | ||
| 60 | + } | ||
| 61 | + for _, key := range list { | ||
| 62 | + if _, err = redis.Del(key); err == nil { | ||
| 63 | + success++ | ||
| 64 | + } | ||
| 65 | + } | ||
| 66 | + logx.Infof("清理缓存:%d/%d", success, len(list)) | ||
| 67 | + | ||
| 68 | + if err != nil { | ||
| 69 | + httpx.ErrorCtx(r.Context(), w, err) | ||
| 70 | + } else { | ||
| 71 | + httpx.OkJson(w, fmt.Sprintf("应用:%v 清理缓存:%d/%d", appName, success, len(list))) | ||
| 72 | + } | ||
| 73 | + } | ||
| 74 | +} |
-
请 注册 或 登录 后发表评论