652035311d
- Introduced a global trace registry to manage active traces with thread-safe access. - Added support for two storage drivers: Local (file-based) and Store (Gou store). - Enhanced trace ID generation to include a date prefix and improved uniqueness. - Implemented functions for creating, loading, and releasing traces, along with checking their existence in both the registry and persistent storage. - Added functionality to retrieve trace metadata and list active traces, improving trace management capabilities.
64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package trace
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/yaoapp/yao/agent/trace/types"
|
|
)
|
|
|
|
// space implements the Space interface for custom space operations
|
|
type space struct {
|
|
ctx context.Context
|
|
traceID string
|
|
data *types.TraceSpace
|
|
driver types.Driver
|
|
}
|
|
|
|
// NewSpace creates a new space instance
|
|
func NewSpace(ctx context.Context, traceID string, data *types.TraceSpace, driver types.Driver) types.Space {
|
|
return &space{
|
|
ctx: ctx,
|
|
traceID: traceID,
|
|
data: data,
|
|
driver: driver,
|
|
}
|
|
}
|
|
|
|
// ID returns the space identifier
|
|
func (s *space) ID() string {
|
|
return s.data.ID
|
|
}
|
|
|
|
// Set stores a value by key
|
|
func (s *space) Set(key string, value any) error {
|
|
return s.driver.SetSpaceKey(s.ctx, s.traceID, s.data.ID, key, value)
|
|
}
|
|
|
|
// Get retrieves a value by key
|
|
func (s *space) Get(key string) (any, error) {
|
|
return s.driver.GetSpaceKey(s.ctx, s.traceID, s.data.ID, key)
|
|
}
|
|
|
|
// Has checks if a key exists
|
|
func (s *space) Has(key string) bool {
|
|
return s.driver.HasSpaceKey(s.ctx, s.traceID, s.data.ID, key)
|
|
}
|
|
|
|
// Delete removes a key-value pair
|
|
func (s *space) Delete(key string) error {
|
|
return s.driver.DeleteSpaceKey(s.ctx, s.traceID, s.data.ID, key)
|
|
}
|
|
|
|
// Clear removes all key-value pairs
|
|
func (s *space) Clear() error {
|
|
return s.driver.ClearSpaceKeys(s.ctx, s.traceID, s.data.ID)
|
|
}
|
|
|
|
// Keys returns all keys in the space
|
|
func (s *space) Keys() []string {
|
|
keys, err := s.driver.ListSpaceKeys(s.ctx, s.traceID, s.data.ID)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
return keys
|
|
}
|