870691e3d4
- Added functionality to archive traces by compressing and marking them as read-only, enhancing data management capabilities. - Introduced methods for checking if a trace is archived and for unarchiving traces, allowing for better control over trace data. - Updated the trace manager and driver interfaces to support archiving operations, ensuring consistency across components. - Enhanced error handling for operations on archived traces, preventing modifications to read-only data. - Improved trace metadata structure to include archived status and timestamp, facilitating better trace lifecycle management.
92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
package trace_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/yaoapp/yao/trace"
|
|
"github.com/yaoapp/yao/trace/types"
|
|
)
|
|
|
|
func TestArchive(t *testing.T) {
|
|
drivers := trace.GetTestDrivers()
|
|
|
|
for _, d := range drivers {
|
|
t.Run(d.Name, func(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
// Create a trace with AutoArchive disabled
|
|
traceID, manager, err := trace.New(ctx, d.DriverType, &types.TraceOption{
|
|
AutoArchive: false,
|
|
}, d.DriverOptions...)
|
|
assert.NoError(t, err)
|
|
defer trace.Release(traceID)
|
|
defer trace.Remove(ctx, d.DriverType, traceID, d.DriverOptions...)
|
|
|
|
// Create some test data
|
|
_, err = manager.Add("test input", types.TraceNodeOption{
|
|
Label: "Test Node",
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
manager.Info("Test log message", map[string]any{
|
|
"key": "value",
|
|
})
|
|
|
|
err = manager.Complete()
|
|
assert.NoError(t, err)
|
|
|
|
err = manager.MarkComplete()
|
|
assert.NoError(t, err)
|
|
|
|
// Wait a bit for completion
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
// Note: Archive functionality is tested at the driver level
|
|
// Here we just verify that traces can be created and completed
|
|
// without AutoArchive enabled
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestAutoArchive(t *testing.T) {
|
|
drivers := trace.GetTestDrivers()
|
|
|
|
for _, d := range drivers {
|
|
t.Run(d.Name, func(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
// Create a trace with AutoArchive enabled
|
|
traceID, manager, err := trace.New(ctx, d.DriverType, &types.TraceOption{
|
|
AutoArchive: true,
|
|
}, d.DriverOptions...)
|
|
assert.NoError(t, err)
|
|
defer trace.Release(traceID)
|
|
defer trace.Remove(ctx, d.DriverType, traceID, d.DriverOptions...)
|
|
|
|
// Create some test data
|
|
_, err = manager.Add("test input", types.TraceNodeOption{
|
|
Label: "Test Node",
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
manager.Info("Test log message", map[string]any{
|
|
"key": "value",
|
|
})
|
|
|
|
err = manager.Complete()
|
|
assert.NoError(t, err)
|
|
|
|
err = manager.MarkComplete()
|
|
assert.NoError(t, err)
|
|
|
|
// Wait for auto-archive to complete
|
|
time.Sleep(200 * time.Millisecond)
|
|
|
|
// Trace should still be accessible after auto-archive
|
|
assert.True(t, trace.IsLoaded(traceID))
|
|
})
|
|
}
|
|
}
|