mirror of
https://github.com/datarhei/core.git
synced 2025-10-07 00:43:39 +08:00
eb1cc37456
Commits (Ingo Oppermann): - Add experimental SRT connection stats and logs - Hide /config/reload endpoint in reade-only mode - Add SRT server - Create v16 in go.mod - Fix data races, tests, lint, and update dependencies - Add trailing slash for routed directories (datarhei/restreamer#340) - Allow relative URLs in content in static routes Co-Authored-By: Ingo Oppermann <57445+ioppermann@users.noreply.github.com>
115 lines
2.3 KiB
Go
115 lines
2.3 KiB
Go
package srt
|
|
|
|
import (
|
|
"runtime"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// Logger is for logging debug messages.
|
|
type Logger interface {
|
|
// HasTopic returns whether this Logger is logging messages of that topic.
|
|
HasTopic(topic string) bool
|
|
|
|
// Print adds a new message to the message queue. The message itself is
|
|
// a function that returns the string to be logges. It will only be
|
|
// executed if HasTopic returns true on the given topic.
|
|
Print(topic string, socketId uint32, skip int, message func() string)
|
|
|
|
// Listen returns a read channel for Log messages.
|
|
Listen() <-chan Log
|
|
|
|
// Close closes the logger. No more messages will be logged.
|
|
Close()
|
|
}
|
|
|
|
// logger implements a Logger
|
|
type logger struct {
|
|
logQueue chan Log
|
|
topics map[string]bool
|
|
}
|
|
|
|
// NewLogger returns a Logger that only listens on the given list of topics.
|
|
func NewLogger(topics []string) Logger {
|
|
l := &logger{
|
|
logQueue: make(chan Log, 1024),
|
|
topics: make(map[string]bool),
|
|
}
|
|
|
|
for _, topic := range topics {
|
|
l.topics[topic] = true
|
|
}
|
|
|
|
return l
|
|
}
|
|
|
|
func (l *logger) HasTopic(topic string) bool {
|
|
if len(l.topics) == 0 {
|
|
return false
|
|
}
|
|
|
|
if ok := l.topics[topic]; ok {
|
|
return true
|
|
}
|
|
|
|
len := len(topic)
|
|
|
|
for {
|
|
i := strings.LastIndexByte(topic[:len], ':')
|
|
if i == -1 {
|
|
break
|
|
}
|
|
|
|
len = i
|
|
|
|
if ok := l.topics[topic[:len]]; !ok {
|
|
continue
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (l *logger) Print(topic string, socketId uint32, skip int, message func() string) {
|
|
if !l.HasTopic(topic) {
|
|
return
|
|
}
|
|
|
|
_, file, line, _ := runtime.Caller(skip)
|
|
|
|
msg := Log{
|
|
Time: time.Now(),
|
|
SocketId: socketId,
|
|
Topic: topic,
|
|
Message: message(),
|
|
File: file,
|
|
Line: line,
|
|
}
|
|
|
|
// Write to log queue, but don't block if it's full
|
|
select {
|
|
case l.logQueue <- msg:
|
|
default:
|
|
}
|
|
}
|
|
|
|
func (l *logger) Listen() <-chan Log {
|
|
return l.logQueue
|
|
}
|
|
|
|
func (l *logger) Close() {
|
|
close(l.logQueue)
|
|
}
|
|
|
|
// Log represents a log message
|
|
type Log struct {
|
|
Time time.Time // Time of when this message has been logged
|
|
SocketId uint32 // The socketid if connection related, 0 otherwise
|
|
Topic string // The topic of this message
|
|
Message string // The message itself
|
|
File string // The file in which this message has been dispatched
|
|
Line int // The line number in the file in which this message has been dispatched
|
|
}
|