Add logging admin APIs for dashboard #114

This commit is contained in:
Gabe Kangas 2020-10-29 18:17:04 -07:00
parent 3963568951
commit 5c6f5fc697
5 changed files with 138 additions and 2 deletions

50
controllers/admin/logs.go Normal file
View File

@ -0,0 +1,50 @@
package admin
import (
"encoding/json"
"net/http"
"time"
"github.com/owncast/owncast/logging"
"github.com/sirupsen/logrus"
)
// GetLogs will return all logs
func GetLogs(w http.ResponseWriter, r *http.Request) {
logs := logging.Logger.AllEntries()
response := make([]logsResponse, 0)
for i := 0; i < len(logs); i++ {
response = append(response, fromEntry(logs[i]))
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(response)
}
// GetWarnings will return only warning and error logs
func GetWarnings(w http.ResponseWriter, r *http.Request) {
logs := logging.Logger.WarningEntries()
response := make([]logsResponse, 0)
for i := 0; i < len(logs); i++ {
response = append(response, fromEntry(logs[i]))
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(response)
}
type logsResponse struct {
Message string `json:"message"`
Level string `json:"level"`
Time time.Time `json:"time"`
}
func fromEntry(e *logrus.Entry) logsResponse {
return logsResponse{
Message: e.Message,
Level: e.Level.String(),
Time: e.Time,
}
}

80
logging/logging.go Normal file
View File

@ -0,0 +1,80 @@
package logging
// Custom logging hooks for powering our logs API.
// Modeled after https://github.com/sirupsen/logrus/blob/master/hooks/test/test.go
import (
"os"
"sync"
"github.com/sirupsen/logrus"
logger "github.com/sirupsen/logrus"
)
const maxLogEntries = 500
type OCLogger struct {
Entries []logrus.Entry
mu sync.RWMutex
}
var Logger *OCLogger
// Setup configures our custom logging destinations
func Setup() {
logger.SetOutput(os.Stdout) // Send all logs to console
_logger := new(OCLogger)
logger.AddHook(_logger)
Logger = _logger
}
// Fire runs for every logging request
func (l *OCLogger) Fire(e *logger.Entry) error {
// Store all log messages to return back in the logging API
l.mu.Lock()
defer l.mu.Unlock()
if len(l.Entries) > maxLogEntries {
l.Entries = l.Entries[1:]
}
l.Entries = append(l.Entries, *e)
return nil
}
// Levels specifies what log levels we care about
func (l *OCLogger) Levels() []logrus.Level {
return logrus.AllLevels
}
// AllEntries returns all entries that were logged.
func (l *OCLogger) AllEntries() []*logrus.Entry {
l.mu.RLock()
defer l.mu.RUnlock()
// Make a copy so the returned value won't race with future log requests
entries := make([]*logrus.Entry, len(l.Entries))
for i := 0; i < len(l.Entries); i++ {
// Make a copy, for safety
entries[i] = &l.Entries[i]
}
return entries
}
// WarningEntries returns all warning or greater that were logged.
func (l *OCLogger) WarningEntries() []*logrus.Entry {
l.mu.RLock()
defer l.mu.RUnlock()
// Make a copy so the returned value won't race with future log requests
entries := make([]*logrus.Entry, 0)
for i := 0; i < len(l.Entries); i++ {
if l.Entries[i].Level <= logrus.WarnLevel {
// Make a copy, for safety
entries = append(entries, &l.Entries[i])
}
}
return entries
}

View File

@ -4,6 +4,7 @@ import (
"flag"
"fmt"
"github.com/owncast/owncast/logging"
"github.com/sirupsen/logrus"
log "github.com/sirupsen/logrus"
@ -80,6 +81,7 @@ func getVersion() string {
}
func configureLogging() {
logging.Setup()
log.SetFormatter(&log.TextFormatter{
FullTimestamp: true,
})

View File

@ -40,8 +40,6 @@ func RequireAdminAuth(handler http.HandlerFunc) http.HandlerFunc {
return
}
// Success
log.Traceln("Authenticated request OK for", r.URL.Path, "from", r.RemoteAddr, r.UserAgent())
handler(w, r)
}
}

View File

@ -71,6 +71,12 @@ func Start() error {
// Get a a detailed list of currently connected clients
http.HandleFunc("/api/admin/clients", middleware.RequireAdminAuth(controllers.GetConnectedClients))
// Get all logs
http.HandleFunc("/api/admin/logs", middleware.RequireAdminAuth(admin.GetLogs))
// Get warning/error logs
http.HandleFunc("/api/admin/logs/warnings", middleware.RequireAdminAuth(admin.GetWarnings))
port := config.Config.GetPublicWebServerPort()
log.Infof("Web server running on port: %d", port)