mirror of
https://github.com/taigrr/log-socket
synced 2026-04-14 05:58:10 -07:00
feat(ws): add read pump for disconnect detection, GetContext, Level.String
- Add WebSocket read pump in LogSocketHandler so client disconnects are detected promptly instead of only on the next WriteMessage call - Add Client.GetContext(ctx) for context-aware blocking reads - Implement fmt.Stringer on Level type (Level.String()) - Add tests for GetContext, GetContext cancellation, and Level.String()
This commit is contained in:
17
log/log.go
17
log/log.go
@@ -1,6 +1,7 @@
|
||||
package log
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
@@ -153,6 +154,7 @@ func (c *Client) SetLogLevel(level Level) {
|
||||
c.LogLevel = level
|
||||
}
|
||||
|
||||
// Get blocks until a log entry is available and returns it.
|
||||
func (c *Client) Get() Entry {
|
||||
if !c.initialized {
|
||||
panic(errors.New("cannot get logs for uninitialized client, did you use CreateClient?"))
|
||||
@@ -160,6 +162,21 @@ func (c *Client) Get() Entry {
|
||||
return <-c.writer
|
||||
}
|
||||
|
||||
// GetContext blocks until a log entry is available or ctx is cancelled.
|
||||
// The second return value is false when the context was cancelled before
|
||||
// an entry arrived.
|
||||
func (c *Client) GetContext(ctx context.Context) (Entry, bool) {
|
||||
if !c.initialized {
|
||||
panic(errors.New("cannot get logs for uninitialized client, did you use CreateClient?"))
|
||||
}
|
||||
select {
|
||||
case e := <-c.writer:
|
||||
return e, true
|
||||
case <-ctx.Done():
|
||||
return Entry{}, false
|
||||
}
|
||||
}
|
||||
|
||||
// Trace prints out logs on trace level
|
||||
func Trace(args ...any) {
|
||||
output := fmt.Sprint(args...)
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
package log
|
||||
|
||||
import (
|
||||
"context"
|
||||
"strconv"
|
||||
"sync"
|
||||
"testing"
|
||||
@@ -378,8 +379,8 @@ func TestMultiNamespaceClient(t *testing.T) {
|
||||
authLogger := NewLogger("auth")
|
||||
dbLogger := NewLogger("database")
|
||||
|
||||
dbLogger.Info("db message") // filtered out
|
||||
apiLogger.Info("api message") // should arrive
|
||||
dbLogger.Info("db message") // filtered out
|
||||
apiLogger.Info("api message") // should arrive
|
||||
authLogger.Info("auth message") // should arrive
|
||||
|
||||
e1, ok := getEntry(c, time.Second)
|
||||
@@ -550,6 +551,68 @@ func TestMatchesNamespace(t *testing.T) {
|
||||
c2.Destroy()
|
||||
}
|
||||
|
||||
// TestGetContext verifies context cancellation stops blocking Get.
|
||||
func TestGetContext(t *testing.T) {
|
||||
c := CreateClient(DefaultNamespace)
|
||||
c.SetLogLevel(LTrace)
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
cancel() // cancel immediately
|
||||
|
||||
_, ok := c.GetContext(ctx)
|
||||
if ok {
|
||||
t.Error("expected GetContext to return false on cancelled context")
|
||||
}
|
||||
c.Destroy()
|
||||
}
|
||||
|
||||
// TestGetContextReceivesEntry verifies GetContext delivers entries normally.
|
||||
func TestGetContextReceivesEntry(t *testing.T) {
|
||||
c := CreateClient(DefaultNamespace)
|
||||
c.SetLogLevel(LTrace)
|
||||
|
||||
go func() {
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
Info("context entry")
|
||||
}()
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
||||
defer cancel()
|
||||
|
||||
e, ok := c.GetContext(ctx)
|
||||
if !ok {
|
||||
t.Fatal("expected GetContext to return entry")
|
||||
}
|
||||
if e.Output != "context entry" {
|
||||
t.Errorf("output = %q, want %q", e.Output, "context entry")
|
||||
}
|
||||
c.Destroy()
|
||||
}
|
||||
|
||||
// TestLevelString verifies the Level.String() method.
|
||||
func TestLevelString(t *testing.T) {
|
||||
tests := []struct {
|
||||
level Level
|
||||
want string
|
||||
}{
|
||||
{LTrace, "TRACE"},
|
||||
{LDebug, "DEBUG"},
|
||||
{LInfo, "INFO"},
|
||||
{LNotice, "NOTICE"},
|
||||
{LWarn, "WARN"},
|
||||
{LError, "ERROR"},
|
||||
{LPanic, "PANIC"},
|
||||
{LFatal, "FATAL"},
|
||||
{Level(99), "UNKNOWN"},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
got := tt.level.String()
|
||||
if got != tt.want {
|
||||
t.Errorf("Level(%d).String() = %q, want %q", tt.level, got, tt.want)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestFlush(t *testing.T) {
|
||||
defer Flush()
|
||||
}
|
||||
|
||||
25
log/types.go
25
log/types.go
@@ -15,6 +15,31 @@ const (
|
||||
|
||||
const DefaultNamespace = "default"
|
||||
|
||||
// String returns the human-readable name of the log level (e.g. "INFO").
|
||||
// It implements [fmt.Stringer].
|
||||
func (l Level) String() string {
|
||||
switch l {
|
||||
case LTrace:
|
||||
return "TRACE"
|
||||
case LDebug:
|
||||
return "DEBUG"
|
||||
case LInfo:
|
||||
return "INFO"
|
||||
case LNotice:
|
||||
return "NOTICE"
|
||||
case LWarn:
|
||||
return "WARN"
|
||||
case LError:
|
||||
return "ERROR"
|
||||
case LPanic:
|
||||
return "PANIC"
|
||||
case LFatal:
|
||||
return "FATAL"
|
||||
default:
|
||||
return "UNKNOWN"
|
||||
}
|
||||
}
|
||||
|
||||
type (
|
||||
LogWriter chan Entry
|
||||
Level int
|
||||
|
||||
Reference in New Issue
Block a user