123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156 |
- // Copyright 2020 The Gitea Authors. All rights reserved.
- // Use of this source code is governed by a MIT-style
- // license that can be found in the LICENSE file.
-
- package private
-
- import (
- "encoding/json"
- "fmt"
- "net/http"
-
- "code.gitea.io/gitea/modules/graceful"
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/private"
- "code.gitea.io/gitea/modules/queue"
- "code.gitea.io/gitea/modules/setting"
-
- "gitea.com/macaron/macaron"
- )
-
- // FlushQueues flushes all the Queues
- func FlushQueues(ctx *macaron.Context, opts private.FlushOptions) {
- if opts.NonBlocking {
- // Save the hammer ctx here - as a new one is created each time you call this.
- baseCtx := graceful.GetManager().HammerContext()
- go func() {
- err := queue.GetManager().FlushAll(baseCtx, opts.Timeout)
- if err != nil {
- log.Error("Flushing request timed-out with error: %v", err)
- }
- }()
- ctx.JSON(http.StatusAccepted, map[string]interface{}{
- "err": "Flushing",
- })
- return
- }
- err := queue.GetManager().FlushAll(ctx.Req.Request.Context(), opts.Timeout)
- if err != nil {
- ctx.JSON(http.StatusRequestTimeout, map[string]interface{}{
- "err": fmt.Sprintf("%v", err),
- })
- }
- ctx.PlainText(http.StatusOK, []byte("success"))
- }
-
- // PauseLogging pauses logging
- func PauseLogging(ctx *macaron.Context) {
- log.Pause()
- ctx.PlainText(http.StatusOK, []byte("success"))
- }
-
- // ResumeLogging resumes logging
- func ResumeLogging(ctx *macaron.Context) {
- log.Resume()
- ctx.PlainText(http.StatusOK, []byte("success"))
- }
-
- // ReleaseReopenLogging releases and reopens logging files
- func ReleaseReopenLogging(ctx *macaron.Context) {
- if err := log.ReleaseReopen(); err != nil {
- ctx.JSON(http.StatusInternalServerError, map[string]interface{}{
- "err": fmt.Sprintf("Error during release and reopen: %v", err),
- })
- return
- }
- ctx.PlainText(http.StatusOK, []byte("success"))
- }
-
- // RemoveLogger removes a logger
- func RemoveLogger(ctx *macaron.Context) {
- group := ctx.Params("group")
- name := ctx.Params("name")
- ok, err := log.GetLogger(group).DelLogger(name)
- if err != nil {
- ctx.JSON(http.StatusInternalServerError, map[string]interface{}{
- "err": fmt.Sprintf("Failed to remove logger: %s %s %v", group, name, err),
- })
- return
- }
- if ok {
- setting.RemoveSubLogDescription(group, name)
- }
- ctx.PlainText(http.StatusOK, []byte(fmt.Sprintf("Removed %s %s", group, name)))
- }
-
- // AddLogger adds a logger
- func AddLogger(ctx *macaron.Context, opts private.LoggerOptions) {
- if len(opts.Group) == 0 {
- opts.Group = log.DEFAULT
- }
- if _, ok := opts.Config["flags"]; !ok {
- switch opts.Group {
- case "access":
- opts.Config["flags"] = log.FlagsFromString("")
- case "router":
- opts.Config["flags"] = log.FlagsFromString("date,time")
- default:
- opts.Config["flags"] = log.FlagsFromString("stdflags")
- }
- }
-
- if _, ok := opts.Config["colorize"]; !ok && opts.Mode == "console" {
- if _, ok := opts.Config["stderr"]; ok {
- opts.Config["colorize"] = log.CanColorStderr
- } else {
- opts.Config["colorize"] = log.CanColorStdout
- }
- }
-
- if _, ok := opts.Config["level"]; !ok {
- opts.Config["level"] = setting.LogLevel
- }
-
- if _, ok := opts.Config["stacktraceLevel"]; !ok {
- opts.Config["stacktraceLevel"] = setting.StacktraceLogLevel
- }
-
- if opts.Mode == "file" {
- if _, ok := opts.Config["maxsize"]; !ok {
- opts.Config["maxsize"] = 1 << 28
- }
- if _, ok := opts.Config["maxdays"]; !ok {
- opts.Config["maxdays"] = 7
- }
- if _, ok := opts.Config["compressionLevel"]; !ok {
- opts.Config["compressionLevel"] = -1
- }
- }
-
- bufferLen := setting.Cfg.Section("log").Key("BUFFER_LEN").MustInt64(10000)
- byteConfig, err := json.Marshal(opts.Config)
- if err != nil {
- log.Error("Failed to marshal log configuration: %v %v", opts.Config, err)
- ctx.JSON(http.StatusInternalServerError, map[string]interface{}{
- "err": fmt.Sprintf("Failed to marshal log configuration: %v %v", opts.Config, err),
- })
- return
- }
- config := string(byteConfig)
-
- if err := log.NewNamedLogger(opts.Group, bufferLen, opts.Name, opts.Mode, config); err != nil {
- log.Error("Failed to create new named logger: %s %v", config, err)
- ctx.JSON(http.StatusInternalServerError, map[string]interface{}{
- "err": fmt.Sprintf("Failed to create new named logger: %s %v", config, err),
- })
- return
- }
-
- setting.AddSubLogDescription(opts.Group, setting.SubLogDescription{
- Name: opts.Name,
- Provider: opts.Mode,
- Config: config,
- })
-
- ctx.PlainText(http.StatusOK, []byte("success"))
- }
|